`gj*dZddlmZddlZddlZddlmZddlmZm Z ddl m Z ddl m Z mZmZmZmZmZdd lmZmZdd lmZej0eZGd d eZ ddlZdd Z e fddZ!e f ddZ"ddZ#dZ$dddZ%y#eef$rdZY,wxYw)u\ QQBot scan-to-configure (QR code onboard) module. Mirrors the Feishu onboarding pattern: synchronous HTTP + a single public entry-point ``qr_register()`` that handles the full flow (create task → display QR code → poll → decrypt credentials). Calls the ``q.qq.com`` ``create_bind_task`` / ``poll_bind_result`` APIs to generate a QR-code URL and poll for scan completion. On success the caller receives the bot's *app_id*, *client_secret* (decrypted locally), and the scanner's *user_openid* — enough to fully configure the QQBot gateway. Reference: https://bot.q.qq.com/wiki/develop/api-v2/ ) annotationsN)IntEnum)OptionalTuple)quote)ONBOARD_API_TIMEOUTONBOARD_CREATE_PATHONBOARD_POLL_INTERVALONBOARD_POLL_PATH PORTAL_HOSTQR_URL_TEMPLATE)decrypt_secretgenerate_bind_key)get_api_headersc eZdZdZdZdZdZdZy) BindStatusz/Status codes returned by ``_poll_bind_result``.rrN)__name__ __module__ __qualname____doc__NONEPENDING COMPLETEDEXPIREDR/root/.hermes/venv/lib/python3.12/site-packages/gateway/platforms/qqbot/onboard.pyrr+s9 DGIGrrcty tjtjjd}|j ||j d|j dy#t$rYywxYw)zDTry to render a QR code in the terminal. Returns True if successful.Fr)error_correctionborderT)fit)invert) _qrcode_modQRCode constantsERROR_CORRECT_Madd_datamake print_ascii Exception)urlqrs r _render_qrr0>sq    (22BB  C D d# sA$A.. A:9A:cddl}dtt}t}|j |d5}|j |d|it }|j|j}dddjddk7rt|jd d |jd ijd }|s td tjd|||fS#1swYxYw)zCreate a bind task and return *(task_id, aes_key_base64)*. Raises: RuntimeError: If the API returns a non-zero ``retcode``. rNhttps://Ttimeoutfollow_redirectskeyjsonheadersretcodemsgzcreate_bind_task faileddatatask_idz-create_bind_task: missing task_id in responsezcreate_bind_task ok: task_id=%s) httpxr r rClientpostrraise_for_statusr8get RuntimeErrorloggerdebug)r4r>r.r6clientrespr<r=s r _create_bind_taskrHTs  [M"5!6 7C  C g ={{3eS\?;L{M yy{  xx a488E+DEFFhhvr"&&y1G JKK LL2G< C<s ?C//C8c&ddl}dtt}|j|d5}|j |d|it }|j |j}dddjddk7rt|jd d |jd i}t|jd dt|jd d|jdd|jddfS#1swYxYw)zPoll the bind result for *task_id*. Returns: A 4-tuple of ``(status, bot_appid, bot_encrypt_secret, user_openid)``. Raises: RuntimeError: If the API returns a non-zero ``retcode``. rNr2Tr3r=r7r:r;zpoll_bind_result failedr<status bot_appidbot_encrypt_secret user_openid) r>r r r?r@rrAr8rBrCrstr)r=r4r>r.rFrGr<ds r _poll_bind_resultrQos [M"3!4 5C g ={{3i%9?CT{U yy{  xx a488E+DEFF A1551%& AEE+r "# "B' mR  s ?DDc@tjt|S)z3Build the QR-code target URL for a given *task_id*.r=)rformatrrSs r build_connect_urlrUs  ! !%. 99rrctj|z}ttdzD]u} t \}}t|}tt|rtd|ntd|tdttj|kr t|\}}} } |tjk(r=t!| |} ttd|d| rtd | || | d cS|tj"k(r@|tk\rt jd tytd |dzd td/tjttj|krt jd|yy#t $r!}t jd|Yd}~yd}~wwxYw#t $rtjtYhwxYw)ugRun the QQBot scan-to-configure QR registration flow. Mirrors ``feishu.qr_register()``: handles create → display → poll → decrypt in one call. Unexpected errors propagate to the caller. :returns: ``{"app_id": ..., "client_secret": ..., "user_openid": ...}`` on success, or ``None`` on failure / expiry / cancellation. rz.[QQBot onboard] Failed to create bind task: %sNz7 Scan the QR code above, or open this URL directly: z' Open this URL in QQ on your phone: z> Tip: pip install qrcode to display a scannable QR code herez QR scan complete! (App ID: )z Scanner's OpenID: )app_id client_secretrNu6[QQBot onboard] QR code expired %d times — giving upz# QR code expired, refreshing... (/z([QQBot onboard] Poll timed out after %ds)time monotonicrange_MAX_REFRESHESrHr-rDwarningrUprintr0rQsleepr rrrr) timeout_secondsdeadline refresh_countr=aes_keyexcr.rJrXencrypted_secretrNrYs r qr_registerrhs~~/1H~122  02 GW  (  c? LSER S ?$%2#. +++ N2NN#[]kl<]Q=Nrxs # "6"   8 $ ,(;:)  &B:@E YKs!B BB