߲j <RdZddlmZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z m Z mZddlmZmZej$dZddZdd Zed d hZdd Zd ZdZdZeeeehZeZdZdZdZ de dZ!dZ"e GddZ#e GddZ$GddZ%y)uTDialog capture + response half of the CDP supervisor. Two capture paths feed one ``PendingDialog`` queue: native ``Page.javascriptDialogOpening`` events (answered with ``Page.handleJavaScriptDialog``), and the injected *dialog bridge* — a page script rewriting alert/confirm/prompt into a sync XHR to a magic host we intercept via the CDP ``Fetch`` domain and answer with ``Fetch.fulfillRequest``. The bridge works on Browserbase (whose CDP proxy auto-dismisses native dialogs) because the native dialog never fires. ``DialogSupervisionMixin`` relies on state ``CDPSupervisor.__init__`` sets. ) annotationsN) dataclass)AnyCallableDictOptional)parse_qsurlparseztools.browser_supervisorc"ddlm}||dS)zARedact page-originated text before exposing supervisor snapshots.r)redact_sensitive_textT)force) agent.redactr )valuer s E/home/ubuntu/.hermes/hermes-agent/tools/browser_supervisor_dialogs.py_redact_supervisor_textrs2 d 33c4t||dzkDr|| dS|S)zTCap a ring buffer at ``keep`` entries once it overflows to 2x (slack reduces churn).N)len)eventskeeps r _trim_ringr!s$ [4!8364%&>??rmessagedefault_promptc ||Dcic],}||tvrtt||n t||.c}Scc}w)zDSnapshot dict of ``keys`` with page-originated text fields redacted.)_REDACTED_FIELDSrgetattr)objkeysks r _dialog_dictr!)s<os tjkA1@P;P&wsA7V]^acdVe e tt ts19 must_respond auto_dismiss auto_acceptgr@zhermes-dialog-bridge.invalidzhttp://z/*a (() => { if (window.__hermesDialogBridgeInstalled) return; window.__hermesDialogBridgeInstalled = true; const ENDPOINT = "http://hermes-dialog-bridge.invalid/"; function ask(kind, message, defaultPrompt) { try { const xhr = new XMLHttpRequest(); const params = new URLSearchParams({ kind: String(kind || ""), message: String(message == null ? "" : message), default_prompt: String(defaultPrompt == null ? "" : defaultPrompt), }); xhr.open("GET", ENDPOINT + "?" + params.toString(), false); // sync xhr.send(null); if (xhr.status !== 200) return null; let parsed; try { parsed = JSON.parse(xhr.responseText || ""); } catch (e) { return null; } if (kind === "alert") return undefined; if (kind === "confirm") return Boolean(parsed && parsed.accept); if (kind === "prompt") { if (!parsed || !parsed.accept) return null; return parsed.prompt_text == null ? "" : String(parsed.prompt_text); } return null; } catch (e) { return null; } } window.alert = function(message) { ask("alert", message, ""); }; window.confirm = function(message) { const r = ask("confirm", message, ""); return r === null ? false : Boolean(r); }; window.prompt = function(message, def) { const r = ask("prompt", message, def == null ? "" : def); return r === null ? null : String(r); }; })(); creZdZUdZded<ded<ded<ded<ded<ded <d Zd ed <d Zd ed <ddZy ) PendingDialogz3A JS dialog currently open on some frame's session.stridtyperrfloat opened_atcdp_session_idN Optional[str]frame_idbridge_request_idct|dS)N)r)r*rrr,r/r!selfs rto_dictzPendingDialog.to_dict|sD"fggrreturnDict[str, Any])__name__ __module__ __qualname____doc____annotations__r/r0r5rrr'r'nsA= G I L"Hm"'+}+hrr'cdeZdZUdZded<ded<ded<ded<ded<ded <d Zd ed <dd Zy ) DialogRecordzOA dialog that was opened and then handled (kept briefly in ``recent_dialogs``).r(r)r*rr+r, closed_at closed_byNr.r/ct|dS)Nr)r*rr,rArBr/r2r3s rr5zDialogRecord.to_dictsD"noorr6)r9r:r;r<r=r/r5r>rrr@r@s5Y G I LN"Hm"prr@ceZdZUdZ ddZddZddZddZdd ddZdd Z dd Z dd Z dd Z dd Z ddZddZeeedZded<y)DialogSupervisionMixinzJDialog event handling for ``CDPSupervisor`` (all methods run on its loop).cK |j||||d{y7#t$r"}tjd|||Yd}~yd}~wwxYww)zABest-effort CDP call: failures are logged at debug and swallowed. session_idtimeoutNz%s failed (%s): %s)_cdp Exceptionloggerdebug)r4methodparamsrIrJwhates r _cdp_quietz!DialogSupervisionMixin._cdp_quietsM @))FFz7)S S S @ LL-vtQ ? ? @s6A$"$A$ AA A AAcK|xsddd}dtdddd|fd td d gd d dd|fdtdddd|ff}|D]%\}}}}|j|||||d{'y7w)aJInstall the dialog-bridge init script + Fetch interceptor on a session. Idempotent at the CDP level (Chromium de-dupes identical add-script calls; Fetch.enable replaces prior patterns); the final Runtime.evaluate injects into the already-loaded document so existing pages pick up the override on reconnect.Nz%Page.addScriptToEvaluateOnNewDocumentT)sourcerunImmediately@zdialog bridge sid=z Fetch.enableRequest) urlPattern requestStageF)patternshandleAuthRequestszRuntime.evaluate) expression returnByValue@zdialog bridge inject sid=rIrJrQ)_DIALOG_BRIDGE_SCRIPTDIALOG_BRIDGE_URL_PATTERNrS)r4rIsidstepsrOrPrJrQs r_install_dialog_bridgez-DialogSupervisionMixin._install_dialog_bridges R"% 4AVjn6o &se, . :Sen+o*p49;!D!D!F GAQ!W G G v)'155B3GXYX]X]^nprXs!FJJy,AUXYcUd   Z Hs%A&D(D)4DD-A'DDN)r0c xjdz c_tdj|||tj|xsjxsd||tdt d|fij j}|Uj5jddddtjj|d |d yj5jj<dddtjj!j"fd j$j<y#1swYxYw#1swY_xYw) zCreate the dialog and apply the policy: auto-respond, or queue + arm the watchdog. Auto policies archive FIRST (tagged ``auto_policy``) so the ``closed`` event that follows our own response isn't re-archived as ``remote``.zd-rU)r)r*rrr,r-r/r0)FrUTN auto_policyraccept prompt_textc`tjjjSN)asyncio create_task_dialog_timeout_expiredr))dialogr4srz6DialogSupervisionMixin._admit_dialog..s!G''(D(DVYY(OPr) _dialog_seqr'time_page_session_idDIALOG_POLICY_AUTO_DISMISSDIALOG_POLICY_AUTO_ACCEPTrl dialog_policy _state_lock_archive_dialog_lockedrr_respond_quiet_pending_dialogsr)get_running_loop call_laterdialog_timeout_s_dialog_watchdogs) r4r*rrrIr/r0autors ` @rrkz$DialogSupervisionMixin._admit_dialogsT AD$$%&T7Saiik*2[@U@U2[Y[1B  +K9RUY[iTjkoo      !! C++FMB C    3 3F47X\]^X_ 3 ` a     6/5D ! !&)) , 6,3,D,D,F,Q,Q  ! ! P- vyy)  C C 6 6sE(E E E)c  K|jxsd}|jrtjt ||j |j dk(r|xsdnddj}|jd|jdddd d d d gtj|jd |d d|j d{yd|i}||j dk(r||d<|jd||d d{y7=7w)uBridge-fulfill for XHR-captured dialogs (swallows failures so the page unblocks), else native CDP — ``promptText`` only for prompt dialogs when given; raises on CDP failure.NpromptrU)r dialog_idrzFetch.fulfillRequestz Content-Typezapplication/json)namerzAccess-Control-Allow-Origin*)rq responseCoderesponseHeadersbodyrYzbridge fulfill rbr promptTextzPage.handleJavaScriptDialogrH) r-r0jsondumpsboolr)r*encoderSbase64 b64encodedecoderK)r4rrrrIrrPs r_respondzDialogSupervisionMixin._responds(**2d  # #::fFIIEK[[T\E\{/@bbdfggmgmgo //&$66.#.F< ii5v*^aibbb  cs$C D D  7DD D DcK |j|||d{y7#t$r+}tjd|j|Yd}~yd}~wwxYww)zWAuto-policy / watchdog response (already archived by the caller); failures logged only.rNzauto response failed for %s: %s)rrLrMrNr))r4rrrrRs rrz%DialogSupervisionMixin._respond_quietsM J--v;-O O O J LL:FIIq I I Js6A#!#A# A!A AAAcK |j|||d{|j|jdy7!#|j|jdwxYww)uAgent response path. The dialog is retired regardless of outcome — a CDP error usually means it already closed (browser auto-dismissed after navigation).rNagent)r_retire_dialogr))r4rrrs r_handle_dialog_cdpz)DialogSupervisionMixin._handle_dialog_cdpsR 4--v;-O O O    7 3 P    7 3s%A ?=?A ?AA cXK|j5|jj|}dddytj d|j ||j |j|j|d|j|ddd{y#1swYrxYw7w)NzBCDP supervisor %s: dialog %s (%s) auto-dismissed after %ss timeoutwatchdogFr) rrrlrMwarningtask_idr*rrr)r4rrs rrz.DialogSupervisionMixin._dialog_timeout_expireds    :**..y9F : > [||Y T=R=R T Iz2!!&D!III : : Js( B*BA+B*B(B*B%!B*c|j5|jj|d}||j||ddd|jj|d}||j yy#1swY9xYw)zRRemove a pending dialog (archiving it with ``closed_by``) and cancel its watchdog.N)rrpoprrcancel)r4rrBrhandles rrz%DialogSupervisionMixin._retire_dialogs|    ?**..y$?F!++FI> ?''++It<   MMO   ? ?s 1A66A?c t|j|j|j|jt j ||j }tg|j|t|_y)zNMove a pending dialog to the recent_dialogs ring buffer. Must hold state_lock.rDN) r@r)r*rr,rr/r_recent_dialogsRECENT_DIALOGS_MAX)r4rrBrecords rrz-DialogSupervisionMixin._archive_dialog_lockeds]fnn`f`p`p(, ySYSbSbd)*ID,@,@*I&*IK]^rcK|j5tfd|jjDd}dddr|j |dyy#1swYxYww)Nc3lK|]+}|jk(r|j|j-ywr)r-r0r)).0drIs r z;DialogSupervisionMixin._on_dialog_closed..)s7aq!"!1!1Z!?ADWDWD_ ddas14remote)rnextrvaluesr)r4rPrI candidates ` r_on_dialog_closedz(DialogSupervisionMixin._on_dialog_closed$so   haD,A,A,H,H,JabfhI h     8 4  h hs A(.AA(A%!A()zPage.javascriptDialogOpeningzPage.javascriptDialogClosedzFetch.requestPausedzDict[str, Callable[..., Any]]EVENT_HANDLERS) rOr(rPr8rIr.rJr+rQr(r7None)rIr(r7r)rPr8rIr.r7r)r*r(rr(rr(rIr.r/r.r0r.r7r)rr'rrrr.r7r)rr'rrrr(r7r)rr(r7r)rr(rBr(r7r)rr'rBr(r7r)r9r:r;r<rSrgrmr{rkrrrrrrrrr=r>rrrFrFsT@"'@/2@7;@e&  &SW - BO [_ 8c,J4 J_ 5);'8/5N1rrF)rr(r7r()rlistrintr7r)rrrtupler7r8)&r< __future__rrrrloggingr dataclassesrtypingrrrr urllib.parser r getLoggerrMrr frozensetrr!DIALOG_POLICY_MUST_RESPONDrr_VALID_POLICIESDEFAULT_DIALOG_POLICYDEFAULT_DIALOG_TIMEOUT_Srrurdrcr'r@rFr>rrrs #  !00+   5 64@ i)9:;u ,+)!;=VW3 4%&8%9< 'T hh h"  p p  pccr