`gj3-@dZddlmZddlZddlZddlZddlZddlZddlZ ddl m Z ddl m Z ddlZdZdZdZd Zd Zd Zd Zej,d Ze dGddZe dGddZd&d'dZd(dZd)dZd*dZddd+dZd,d-dZd&d.dZ eddf d/dZ!d0dZ"dedf d1dZ# d2 d3d Z$ d2 d4d!Z%eef d5d"Z&eef d6d#Z'dedef d7d$Z(dedef d8d%Z)y)9aTelegram Managed Bot onboarding client. Uses Telegram's Managed Bots feature to create a user-owned child bot without manual BotFather token copy-paste. Hermes talks only to the Nous onboarding service; the raw Telegram token is saved locally after one-time retrieval. ) annotationsN) dataclass)Optionalz+https://setup.hermes-agent.nousresearch.comTELEGRAM_ONBOARDING_URLHermesSetupBotz Hermes Agent abcdefghijklmnopqrstuvwxyz234567z^\d+:[A-Za-z0-9_-]{30,}$T)frozencReZdZUdZded<ded<ded<ded<ded<dZd ed <y) TelegramPairingz;Pairing record returned by the Telegram onboarding service.str pairing_id poll_tokensuggested_username deep_link qr_payloadN str | None expires_at)__name__ __module__ __qualname____doc____annotations__rR/root/.hermes/venv/lib/python3.12/site-packages/hermes_cli/telegram_managed_bot.pyr r &s)EOONO!J !rr c8eZdZUdZded<dZded<dZded<y) TelegramBotSetupResultzDSuccessful Telegram onboarding result returned by the setup service.rtokenNr bot_username int | None owner_user_id)rrrrrr!r#rrrrr2sN J#L*# $M:$rrc~|xs+tjjtxstj dS)z>Resolve the onboarding API URL, honoring the PoC env override./)osenvirongetTELEGRAM_ONBOARDING_URL_ENVDEFAULT_API_URLrstrip)api_urls r_api_urlr-;s- Q2::>>"=>Q/ fSkrcbt|txrttj |S)z8Return True when *token* has Telegram's bot-token shape.) isinstancerbool_TELEGRAM_BOT_TOKEN_REmatch)r s ris_valid_telegram_bot_tokenr3Bs% eS ! Od+A+G+G+N&OOrct|tryt|tr |dkDr|SdSt|tr$|j rt|}|dkDr|SdSy)Nr)r/r0intr isdecimal)valueparseds r_parse_owner_user_idr9GsZ%% u+t+%%//"3U!v-- rc0 ddl}ddl}|jd|jjdd}|j ||j d|j}|j|d|jS#t$rYywxYw) z>Render a URL as a QR code string suitable for terminal output.rN)versionerror_correctionbox_sizeborderT)fit)outinvert) ioqrcodeQRCode constantsERROR_CORRECT_Ladd_datamakeStringIO print_asciigetvalue ImportError)urlrDrEqrbufs rrender_qr_terminalrRRs ]]#--==   C Dkkm 3t,||~ sBB BB include_linkcnt|}|r t|n td|rtd|yy)zBPrint a QR code to stdout, with URL fallback if qrcode is missing.z@ (Install 'qrcode' for a scannable QR code: pip install qrcode) Link: N)rRprint)rOrTqr_texts r print_qr_coderYis6 %G g PQ rcDdjdt|DS)zGenerate a base32-ish slug for Telegram username correlation. Sixteen characters from a 32-symbol alphabet gives 80 bits of entropy while keeping ``hermes__bot`` under Telegram's 32-character username limit. rCc3NK|]}tjtywN)secretschoice_USERNAME_SLUG_ALPHABET).0_s r z)generate_username_slug..zsRq7>>"9:Rs#%)joinrange)lengths rgenerate_username_slugrfts 77RE&MR RRrc"|}dtdS)aGenerate a secure suggested bot username like ``hermes__bot``. ``profile_name`` is accepted for backward compatibility with the original PoC, but is intentionally not embedded in the username. The username has to carry enough entropy for backend correlation. hermes__bot)rf) profile_nameras rgenerate_bot_usernamerk}s A +-.d 33rc|jd}|xs t}dtjj |dtjj |}|r(tjj d|i}|d|S|S)z;Build a ``t.me/newbot`` deep link for managed bot creation.@zhttps://t.me/newbot/r%name?)lstriprkurllibparsequote urlencode) manager_botrsuggested_namemanagerusernamebase_urlparamss rgenerate_deep_linkr{s   %G!<%:%.s0TCz$((3--?$((3-?Ts,>+?"@A D  3 99;D xxW$ HHW E &u -88N+L ! lC (\" *488O+DE  rc>t|||}|r |jSdS)z?Poll the onboarding service once. Returns the token when ready.)rN)rr )r,rrresults rpoll_pairing_oncers# &gw HF!6<<+t+rctj|z}tj|kr? t||}|r|S tj |tj|kr?y#tjt f$rYIwxYw)zBPoll the pairing API until setup metadata is available or timeout.N)time monotonicrrrrsleep)r,rrintervaldeadliners rpoll_for_setup_resultrs~~')H .. X % -gw?F  8 .. X % ,   sA//B  B c@t||||}|r |jSdS)zAPoll the pairing API until the bot token is available or timeout.)rrN)rr )r,rrrrs rpoll_for_tokenr s%#7GWx XF!6<<+t+rc||f}t|}ttd|tjj t |}|stdtdytdtdtjj ttdtt |jd ttd |jttd td td }tj}||z} d} tj| kr|| t|z} ttj|z } tdt|| z } tjjd| d| dtjj | dz } t||}|r?tjjdtjj |S tj&t(tj| krtjjdtjj tdtdy#t j"t$f$rYwxYw)z2Run the full automatic Telegram bot creation flow.z1 Contacting Hermes Telegram onboarding service: u= ✗ Could not reach the Hermes Telegram onboarding service.z8 Try the manual setup instead, or check your network.Nu ✓ Pairing createdz Rendering QR code...z< Scan this QR code with your phone, or open the link below:FrSrVz3 When Telegram opens, tap 'Create Bot' to confirm.z7 (You can edit the bot display name before confirming)u⠋⠙⠹⠸⠼⠴⠦⠧⠇⠏rz z Waiting for bot creation... (z s remaining) r;u? ✓ Bot created successfully! u? ✗ Timed out waiting for bot creation. u4 The bot may still be created — check Telegram.z< You can paste the token manually below, or re-run setup.)r-rWsysstdoutflushrrYrrrrlenr5maxwriterrrrr POLL_INTERVAL)r,rurj poll_timeoutraresolved_api_urlr spinner_charsstartridxcharelapsed remainingrs rauto_setup_telegram_bot_resultrsJ \!A( G =>N=O PQJJ-.G  MN HI !" "#JJ G HI G'$$59 G HW&&' () G ?@ CD G4M NN E|#H C .. X %S3}#556dnn&./3|g567  4&6yk O   q -.>HF   W   "  =!) .. X %,JJXYJJ @A HI ,   s(A J<<KKc@t||||}|r |jSdS)zBRun automatic Telegram bot creation and return only the bot token.)r,rurjrN)rr )r,rurjrrs rauto_setup_telegram_botrYs-,!! F "6<<+t+rr\)r,rreturnr)r objectrr0)r7rrr")rOrrr)rOrrTr0rNone)r})rer5rr)rj Optional[str]rr)rurrrrvrrr)rr)r,rrrrfloatrzTelegramPairing | None)r)r,rrr rrrzTelegramBotSetupResult | None)r,rrr rrrr) r,rrr rrrrr Optional[TelegramBotSetupResult]) r,rrr rrrrrr) r,rrurrjrrrrr) r,rrurrjrrrrr)*r __future__rr&rer]rr urllib.parserq dataclassesrtypingrrr*r)DEFAULT_MANAGER_BOTDEFAULT_BOT_NAMEDEFAULT_POLL_TIMEOUTrr_compiler1r rr-r3r9rRrYrfrkr{rrrrrrrrrrrrs#  ! @7'! <#$?@ $""" $%%%P .59 S4+(,$(%" (!$' ''' 'Z  # D, , ,, ,*#     & ,*# , , ,, ,  ,*"&. ? ?? ? ? & ?F*"&. ,  , ,  , ,  ,r