PL jcUdZddlmZddlZddlZddlZddlZddlZddlZddl m Z m Z m Z ddl Z ddlmZmZddlmZddlmZejeZdZd Zdad ed <d adIdZdJdZdJdZdKdZ dLdZ!dJdZ"dMdZ#dNdZ$dOdZ%iZ&ded <ej'Z(dPd"Z)dQd#Z*dRdSd&Z+dTd(Z,dUdVd)Z-efdWd/Z.defdXd1Z/defdYd3Z0defdWd4Z1dUdZd5Z2 d[d\d8Z3dUd]d:Z4dUd^d<Z5dUd_d>Z6dUd`d?Z7dUdadAZ8dUd`dBZ9dUd`dCZ: dbdcdFZ;dbdddHZ y"B''--//u !! " ""rct}|sdS tj|dd}|jdkrts |}|d}t |tr2d|cxkrdkr%nn"d d lm }||}|j pd }d |d |a n#ttf$rYnwxYwda|jdkS#t$rYdSwxYw)z'Verify the Camofox server is reachable.Fz/healthtimeoutvncPortir)urlparse localhostzhttp://:T)rrequestsget status_code_vnc_url_checkedjson isinstanceint urllib.parser'hostnamer ValueErrorKeyError Exception)urlrespdatavnc_portr'parsedhosts rcheck_camofox_availabler<EsA   C u|sOOOQ777  s " "+; " yy{{88I..h,,;h1G1G1G1G%1G1G1G1G1G555555%Xc]]F!?9kD:::::H)    # 3&& uus6+CA0B10C1CCCC C#"C#c:tsttS)z>Return the VNC URL if the Camofox server exposes one, or None.)r-r<r rrr get_vnc_urlr>^s "!!! OrDict[str, Any]c  tdidi}n4#t$r'}td|icYd}~Sd}~wwxYwt |t r|niS)z>Return the ``browser.camofox`` config block, or an empty dict.browsercamofoxz7camofox config check failed, defaulting to disabled: %sN)rr+r5loggerwarningr/dict) camofox_cfgexcs r_get_camofox_configrHes!mm'' 266::9bII PRUVVV %[$77 ?;;R?s69 A*A%A*%A*c^ttdS)alReturn whether Hermes-managed persistence is enabled for Camofox. When enabled, sessions use a stable profile-scoped userId so the Camofox server can map it to a persistent browser profile directory. When disabled (default), each session gets a random userId (ephemeral). Controlled by ``browser.camofox.managed_persistence`` in config.yaml. managed_persistence)rrHr+rrr_managed_persistence_enabledrKos( #%%))*?@@ A AArtask_idrFOptional[Dict[str, str]]ctjddp5t|dpd}|sdStjddpDt|dpdpd|pddd }||d S) zReturn an externally configured Camofox identity, if one is set. Integrations that own the visible Camofox browser can set a shared user ID so Hermes operates in the same browser profile instead of creating a separate private session. CAMOFOX_USER_IDruser_idNCAMOFOX_SESSION_KEY session_keytask_default)rPrR)rrrrr+)rLrFrPrRs r_camofox_identity_overriderV{si)2..4466g#kooi>X>X>^\^:_:_:e:e:g:gG t ',,2244 1 {}--3 4 4 : : < < 1 0G(y#2#. 0 0 { ; ;;rnameOptional[bool]ctj|d}|sdS|dvrdS|dvrdStd||dS)Nr>1onyestrueT>0noofffalseFz"Ignoring invalid boolean env %s=%r)rrrlowerrCdebug)rWraws r _env_flagresv )D"   # # % % + + - -C t (((t )))u LL5tSAAA 4rcltd}||St|dS)z@Return whether Hermes should recover an existing Camofox tab ID.CAMOFOX_ADOPT_EXISTING_TABNadopt_existing_tab)rerr+)rF env_values r_adopt_existing_tab_enabledrjs7677I   455 6 66rzDict[str, Dict[str, Any]] _sessionssessionc |ds|ds|Sts|S tdd|diddg}nH#t$r;}td |d||cYd }~Sd }~wwxYwt |tr|s|S|d fd |D}|p d |D}|r|dnd }t |tr|dnd }t |tr6|r4||d<td||d|S)a0Attach process-local state to an already-open managed Camofox tab. Some integrations own the visible Camofox tab outside Hermes. Gateway restarts can leave this module's in-memory session cache empty even though Camofox still has that tab, so rehydrate tab_id before creating a new tab. tab_idrh/tabsuserIdrPr!paramsr#tabsz&Camofox tab adoption failed for %s: %sNrRcpg|]2}t|tr|dk0|3S) listItemId)r/rEr+).0tabrRs r z'_adopt_existing_tab..sN  c4 &)WW\%:%:k%I%I %I%I%Irc<g|]}t|t|Sr)r/rE)rvrws rrxz'_adopt_existing_tab..s'"P"P"P3*S$:O:O"P3"P"P"PrtabIdz&Adopted existing Camofox tab %s for %s) r+r_getr5rCrcr/listrEr)rlrsrG matching_tabs candidateslatestrnrRs @r_adopt_existing_tabrs{{8GKK0D$E$E   GXwy/A$BANNNRRSY[]^^  =w{{9?U?UWZ[[[ dD ! !++m,,KM P"P"P$"P"P"PJ) 3Z^^tF$.vt$<$< FVZZ $F&#_6_" =vw{{S\G]G]^^^ Ns.A.. B380B.(B3.B3cn|pd}t5|tvr&tt|cdddSt}t ||}|r"|dd|ddt |d}nt |dr1t|}|dd|ddt |d}n4dtj j dd dd |dd d d d}|t|<t|cdddS#1swxYwYdS) zGet or create a camofox session for the given task. When managed persistence is enabled, uses a deterministic userId derived from the Hermes profile so the Camofox server can map it to the same persistent browser profile across restarts. rTNrPrRT)rPrnrRmanagedrhrJhermes_ rSrUF) _sessions_lockrkrrHrVrjrr+r uuiduuid4hex)rLrFidentity_overriderlidentitys r _get_sessionrs"G  , , i  &y'9:: , , , , , , , ,*++ 6w LL  ,Y70?&A+&N&N GG+//"788 9 9 +G44H#I.' 6&A+&N&N GG=TZ\\%5crc%:<<5wss|55 &+ G% '"7++A , , , , , , , , , , , , , , , , , ,s#D*C!D**D.1D. about:blankr6c2t|}|dr|St}tj|d|d|d|dt}||}|d|d<|S)zr|tools.browser_toolrrlenr5r.dumpsr* HTTPErrorr ConnectionErrorrr) r6rLrlr8resultvnc snap_data snapshot_textrres rcamofox_navigaterPs71w''x !'3//Gs++DD5*555"9-c::D 88E3''XXgr**   mm   #F9 T :   5*555 ')"45I&MM*b99M        =!!$@@@ 2 2= A A !.F: &/mmK&C&CF? # #    D z&!!!  DDD333UCCCCCCCCC  #   z__5F5F___       111#a&&%0000000001s[BD?A=DD? D(%D?'D((D??GE'!G'9G" G+GGGfull user_taskc t|}|dstddStd|ddd|di }|d d }|d d }d dlm}m}m} t||kr|r |||}n | |}tj d||dS#t$r(} tt| dcYd} ~ Sd} ~ wwxYw)z-Get accessibility tree snapshot from Camofox.rn0No browser session. Call browser_navigate first.FrrrrprPrrrrr)r_extract_relevant_contentrT)rrrN) rr r|r+rrrrrr.rr5r) rrLrrlr8r refs_countrrrrs rcamofox_snapshotrsu 1w''x  aPZ_``` ` 1WX& 1 1 1gi01    88J++XXk1--           x==7 7 7 844XyII--h77z '     111#a&&%0000000001s#'CBC C:C5/C:5C:refc t|}|dstddS|d}td|dd|d|d }t jd ||d d d S#t$r(}tt|dcYd}~Sd}~wwxYw)z$Click an element by ref via Camofox.rnrFr@rz/clickrP)rprTr6r)rclickedr6N) rr lstriprr.rr+r5r)rrLrl clean_refr8rs r camofox_clickrs1w''x  aPZ_``` `JJsOO  .WX& . . .y)) < <  z 88E2&&     111#a&&%0000000001s#'BA#B CB;5C;Ctextc` t|}|dstddS|d}td|dd|d||d t jd ||d S#t $r(}tt|dcYd }~Sd }~wwxYw) z-Type text into an element by ref via Camofox.rnrFrrrz/typerP)rprrT)rtypedelementN)rr rrr.rr5r)rrrLrlrrs r camofox_typers1w''x  aPZ_``` `JJsOO  -WX& - - -y))T J J   z      111#a&&%0000000001s#'A;AA;; B-B("B-(B- directionc2 t|}|dstddStd|dd|d|dtjd |d S#t $r(}tt |dcYd }~Sd }~wwxYw) zScroll the page via Camofox.rnrFrrz/scrollrP)rprT)rscrolledNrr rr.rr5r)rrLrlrs rcamofox_scrollrs 1w''x  aPZ_``` ` /WX& / / /y) B B   zd BBCCC 111#a&&%0000000001"'A$9A$$ B.B BBcX t|}|dstddStd|ddd|di}tjd |d d d S#t $r(}tt|dcYd }~Sd }~wwxYw)zNavigate back via Camofox.rnrFrrz/backrprPTr6r)rr6N)rr rr.rr+r5r)rLrlr8rs r camofox_backrs 1w''x  aPZ_``` ` -WX& - - - wy) *  zd488E23F3FGGHHH 111#a&&%0000000001s#'A7A A77 B)B$B)$B)keyc2 t|}|dstddStd|dd|d|dtjd |d S#t $r(}tt |dcYd }~Sd }~wwxYw) z!Press a keyboard key via Camofox.rnrFrrz/pressrP)rprT)rpressedNr)rrLrlrs r camofox_pressrs 1w''x  aPZ_``` ` .WX& . . .y)# 6 6   zds;;<<< 111#a&&%0000000001rc* t|}|stjdddStd|dtjdddS#t$r/}tjddt |dcYd}~Sd}~wwxYw)z&Close the browser session via Camofox.T)rclosedz /sessions/rP)rrrDN)rr.rrr5r)rLrlrs r camofox_closers P(( A:$$??@@ @ -+ - -   zdd;;<<< PPPzdds1vvNNOOOOOOOOPs"'A.A B#$B B Bc t|}|dstddSddl}td|ddd |d i }|d d }g}|d}t |D]\}}|} | dr| d| } | r| dnd } d } |dzt|krH| d||dz} | r| d} | s| r| | | dtjd|t|dS#t$r(}tt!|dcYd}~Sd}~wwxYw)zGet images on the current page via Camofox. Extracts image information from the accessibility tree snapshot, since Camofox does not expose a dedicated /images endpoint. rnrFrrNrrrprPrrr )z- img zimg zimg\s+"([^"]*)"r&z /url:\s*(\S+))srcaltT)rimagescount)rr rer|r+split enumerater startswithsearchgrouprappendr.rr5r)rLrlrr8rrlinesilinestripped alt_matchrr url_matchrs rcamofox_get_imagesrs  &1w''x  aPZ_``` `  1WX& 1 1 1gi01   88J++ t$$ '' < F93F>9F>questionannotatec t|}|dstddStd|ddd|di }d d lm}|d z }|d d t |dtjj dddz }t|d5}| |j dddn #1swxYwYtj|j d} d} |rV t!d|ddd|di } d| dddd} n#t$$rYnwxYwd dlm} | | } d dlm} d|| } t/}t1|ddi}t3|d d!}t3|d"d#}n#t$$rd$}d#}YnwxYw| d%d&|d'd(d)d*| id+gd,gd||-}|jr+|jd jj pdnd}d dlm} | |}t;jd ||d.S#t$$r(}tt |dcYd}~Sd}~wwxYw)/zrHrKrVrerjrkLockrrrrrrrr|rrrrrrrrrrrr*r2rrrr@s,#"""""   &&&&&&&&&&22222222<<<<<<%%%%%%  8 $ $ 4444 # # # #2@@@@ B B B B<<<<&    7777(* ))))!!####L(,(,(,(,V*,,,,*1A$(8H(,