i&dZddlmZddlZddlZddlmZddlmZm Z ddl m Z ddl m Z mZmZmZmZmZdd lmZmZdd lmZejeZGd d eZ ddlZn#eef$rdZYnwxYwd dZ e fd!dZ!e fd"dZ"d#dZ#dZ$d$d%dZ%dS)&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"eZdZdZdZdZdZdZdS) BindStatusz/Status codes returned by ``_poll_bind_result``.rrN)__name__ __module__ __qualname____doc__NONEPENDING COMPLETEDEXPIREDD/home/ubuntu/.hermes/hermes-agent/gateway/platforms/qqbot/onboard.pyrr+s(99 DGIGGGrrurlstrreturnboolctdS tjtjjd}|||d|ddS#t$rYdSwxYw)zDTry to render a QR code in the terminal. Returns True if successful.NFr)error_correctionborderT)fit)invert) _qrcode_modQRCode constantsERROR_CORRECT_Madd_datamake print_ascii Exception)r!qrs r _render_qrr3>su   (2B    C D d###t uusA&A33 BBtimeoutfloatTuple[str, str]clddl}dtt}t}||d5}||d|it }||}dddn #1swxYwY| ddkr#t| d d | d i d }|std t d|||fS)zCreate a bind task and return *(task_id, aes_key_base64)*. Raises: RuntimeError: If the API returns a non-zero ``retcode``. rNhttps://Tr4follow_redirectskeyjsonheadersretcodemsgzcreate_bind_task faileddatatask_idz-create_bind_task: missing task_id in responsezcreate_bind_task ok: task_id=%s) httpxr r rClientpostrraise_for_statusr=get RuntimeErrorloggerdebug)r4rCr!r;clientresprArBs r _create_bind_taskrMTsa LLL 7[ 7"5 7 7C   C g = ={{3eS\?;L;L{MM yy{{  xx a488E+DEEFFFhhvr""&&y11G LJKKK LL2G<<< C<sABBBrB Tuple[BindStatus, str, str, str]cddl}dtt}||d5}||d|it }||}dddn #1swxYwY|ddkr#t|d d |d i}t|d dt|d d|dd|ddfS)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``. rNr8Tr9rBr<r?r@zpoll_bind_result failedrAstatus bot_appidbot_encrypt_secret user_openid) rCr r rDrErrFr=rGrHrr")rBr4rCr!rKrLrAds r _poll_bind_resultrVosqLLL 5[ 5"3 5 5C g = ={{3i%9?CTCT{UU yy{{  xx a488E+DEEFFF A1551%%&& AEE+r " "## "B'' mR   sABB B cFtjt|S)z3Build the QR-code target URL for a given *task_id*.rB)rformatrrXs r build_connect_urlrZs  !%.. 9 9 99rrXtimeout_secondsintOptional[dict]ctj|z}ttdzD]} t \}}n4#t $r'}t d|Yd}~dSd}~wwxYwt|}tt|rtd|n!td|tdttj|kr* t|\}}} } n)#t $rtj tYSwxYw|tjkrMt!| |} ttd|d| rtd | || | d cS|tjkrM|tkr#t d tdStd |dzd tdnOtj ttj|k*t d|dSdS)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_secretrTu6[QQBot onboard] QR code expired %d times — giving upz# QR code expired, refreshing... (/z([QQBot onboard] Poll timed out after %ds)time monotonicrange_MAX_REFRESHESrMr1rIwarningrZprintr3rVsleepr rrrr) r\deadline refresh_countrBaes_keyexcr!rPraencrypted_secretrTrbs r qr_registerrps~/1H~12222  022 GWW    NNKS Q Q Q444444  ((  c?? T RSRR S S S S BSBB C C C R S S S n)) @QRY@Z@Z= 0++    0111 --- ./? I I ?f???@@@@>>>???$%2#. +++ N22NN#[]klll44b]Q=NbbQ_bbbccc J, - - -5n)): NNE W W W44  4s)A A5A00A5?D#D:9D:)r!r"r#r$)r4r5r#r6)rBr"r4r5r#rN)rBr"r#r")r[)r\r]r#r^)&r __future__rloggingrdenumrtypingrr urllib.parserr,r r r r r rcryptorrutilsr getLoggerrrIrqrcoder* ImportError TypeErrorr3rMrVrZrgrprrr r|s  #""""" """"""""65555555""""""  8 $ $     YKKK,(;:)B::::@@@@@@@s"A'' A32A3