iR:dZddlZddlZddlZddlZddlZddlZddlZddlZddl m Z m Z ddl m Z ddlmZmZmZddlmZddlmZejeZejZdeegdfdzd dfd Zd eegdfdzfd Zd ed ed dfdZd e fdZ dej!ded dfdZ" d#de#ededzd ej!fdZ$de d efdZ%de ded dfdZ&ded e'e(e)fdzfdZ*GddeZ+GddZ,ded efd Z-Gd!d"e Z.dS)$aCBase class for all Hermes execution environment backends. Unified spawn-per-call model: every command spawns a fresh ``bash -c`` process. A session snapshot (env vars, functions, aliases) is captured once at init and re-sourced before each command. CWD persists via in-band stdout markers (remote) or a temp file (local). N)ABCabstractmethod)Path)IOCallableProtocol)get_hermes_home)is_interruptedcbreturnc|t_dS)z>Register a callback that _wait_for_process fires periodically.N)_activity_callback_localcallback)r s ?/home/agentuser/.hermes/hermes-agent/tools/environments/base.pyset_activity_callbackrs(*%%%c.ttddS)Nr)getattrrrr_get_activity_callbackr$s +Z > >>rstatelabelctj}|dd}||dz |krdS||d< t}|r+t ||dz }||d|ddSdS#t $rYdSwxYw)a_Fire the activity callback at most once every ``state['interval']`` seconds. *state* must contain ``last_touch`` (monotonic timestamp) and ``start`` (monotonic timestamp of the operation start). An optional ``interval`` key overrides the default 10 s cadence. Swallows all exceptions so callers don't need their own try/except. intervalg$@ last_touchNstartz (z s elapsed))time monotonicgetrint Exception)rrnowrr elapseds rtouch_activity_if_duer$(s .  CyyT**H U<  8++E, # % %  0#g.//G B%..7... / / / / / 0 0      s9A>> B  B ctjd}|rt|}ntdz }|dd|S)zReturn the host-side root for all sandbox storage (Docker workspaces, Singularity overlays/SIF cache, etc.). Configurable via TERMINAL_SANDBOX_DIR. Defaults to {HERMES_HOME}/sandboxes/. TERMINAL_SANDBOX_DIR sandboxesTparentsexist_ok)osgetenvrr mkdir)customps rget_sandbox_dirr0BsT Y- . .F , LL    +GGD4G((( Hrprocdatacffd}tj|ddS)zMWrite *data* to proc.stdin on a daemon thread to avoid pipe-buffer deadlocks.c jjdS#ttf$rYdSwxYwN)stdinwritecloseBrokenPipeErrorOSError)r2r1sr_writez_pipe_stdin.._writeYs_  J  T " " " J       )    DD s38A  A TtargetdaemonN) threadingThreadr)r1r2r;s`` r _pipe_stdinrAVsLF40006688888rcmd stdin_datac tj|ftjtj| tjn tjdd|}|t |||S)aSpawn a subprocess with standard stdout/stderr/stdin setup. If *stdin_data* is provided, writes it asynchronously via :func:`_pipe_stdin`. Backends with special Popen needs (e.g. local's ``preexec_fn``) can bypass this and call :func:`_pipe_stdin` directly. NT)stdoutstderrr6text) subprocessPopenPIPESTDOUTDEVNULLrA)rBrCkwargsr1s r _popen_bashrNcsi     !+!7jooZ=O        DD*%%% Krpathc|r7 tj|S#t$rYnwxYwiS)z:Load a JSON file as a dict, returning ``{}`` on any error.)existsjsonloads read_textr!)rOs r_load_json_storerUysV {{}} :dnn..// /    D  Is%< A A c|jdd|tj|ddS)z.Write *data* as pretty-printed JSON to *path*.Tr()indentN)parentr- write_textrRdumps)rOr2s r_save_json_storer\sDKdT222OODJtA.../////r host_pathc t|}|j|jfS#t$rYdSwxYw)zIReturn ``(mtime, size)`` for cache comparison, or ``None`` if unreadable.N)rstatst_mtimest_sizer:)r]sts r_file_mtime_keyrcsO )__ ! ! # # RZ(( tts .1 ??ceZdZdZdedzfdZd dZd dedzdefdZe de e dzfdZ e dedzfd Z dS) ProcessHandlezDuck type that every backend's _run_bash() must return. subprocess.Popen satisfies this natively. SDK backends (Modal, Daytona) return _ThreadedProcessHandle which adapts their blocking calls. r NcdSr5rselfs rpollzProcessHandle.pollrcdSr5rrgs rkillzProcessHandle.killrjrtimeoutcdSr5rrhrms rwaitzProcessHandle.waitrjrcdSr5rrgs rrEzProcessHandle.stdout(+rcdSr5rrgs r returncodezProcessHandle.returncoderrrr Nr5)__name__ __module__ __qualname____doc__r rirlfloatrppropertyrstrrErtrrrreres &cDj%%%%< (output_str, exit_code)`` in a background thread and exposes a ProcessHandle-compatible interface. An optional ``cancel_fn`` is invoked on ``kill()`` for backend-specific cancellation (e.g. Modal sandbox.terminate, Daytona sandbox.stop). Nexec_fn cancel_fnc8|_tj_d_d_t j\}}t j|ddd_ |_ fd}tj |d}| dS)Nrutf-8replace)encodingerrorsc \}}|_ tjj|ddn#t $rYnwxYwn%#t $r}|_d_Yd}~nd}~wwxYw tjjn#t $rYnwxYwj dS# tjjn#t $rYnwxYwj wxYw)Nrr)r) _returncoder+r7 _write_fdencoder:r!_errorr8_doneset)output exit_codeexcrrhs r_workerz0_ThreadedProcessHandle.__init__.._workersO !$+GII! #, HT^V]]79]-U-UVVVVD % % %! #$       %HT^,,,,D      HT^,,,,D     sA/AA AAAAC A;#A61C6A;;C?B B&%B&D C D  C-*D ,C--D Tr<) _cancel_fnr?Eventrrrr+pipefdopen_stdoutrr@r)rhrrread_fdwrite_fdrts`` r__init__z_ThreadedProcessHandle.__init__s $_&& '+(, GIIy# RRR ! ! ! ! ! ! !&  GD 9 9 9  rc|jSr5)rrgs rrEz_ThreadedProcessHandle.stdouts |rr c|jSr5)rrgs rrtz!_ThreadedProcessHandle.returncodes rcF|jr|jndSr5)ris_setrrgs rriz_ThreadedProcessHandle.polls##':#4#4#6#6@tD@rcd|jr( |dS#t$rYdSwxYwdSr5)rr!rgs rrlz_ThreadedProcessHandle.killsS ?  !!!!!      s  --rmcF|j||jS)Nrm)rrprros rrpz_ThreadedProcessHandle.waits! (((rr5)rvrwrxryrtupler|r rr{rErtrirlrzrprrrr~r~s04##"eCHo-.#BH%,####JX C$J   X AcDjAAAA  EDL C      rr~ session_idcd|dS)N __HERMES_CWD___r)rs r _cwd_markerrs ): ) ) ))rc zeZdZUdZdZeed<dZeed<defdZ d&d ed ed e fd Z d ddddede d ededzde f dZedZdZded edefdZedededefdZd'de d ede fdZde fdZde fdZde fdZd(dZ d)ddd!ded ed edzdedzde f d"Zd#Zd$Zdedeeedzffd%ZdS)*BaseEnvironmentaCommon interface and unified execution flow for all Hermes backends. Subclasses implement ``_run_bash()`` and ``cleanup()``. The base class provides ``execute()`` with session snapshot sourcing, CWD tracking, interrupt handling, and timeout enforcement. r _stdin_mode_snapshot_timeoutr cdS)a,Return the backend temp directory used for session artifacts. Most sandboxed backends use ``/tmp`` inside the target environment. LocalEnvironment overrides this on platforms like Termux where ``/tmp`` may be missing and ``TMPDIR`` is the portable writable location. z/tmprrgs r get_temp_dirzBaseEnvironment.get_temp_dir s vrNcwdrmenvcX||_||_|pi|_tjjdd|_|dpd}|d|jd|_ |d|jd|_ t|j|_ d|_ dS)N /z /hermes-snap-z.shz /hermes-cwd-z.txtF) rrmruuiduuid4hex _session_idrrstrip_snapshot_path _cwd_filer_snapshot_ready)rhrrmrtemp_dirs rrzBaseEnvironment.__init__s 9":<<+CRC0$$&&--c229c!)MM8HMMM$HH$2BHHH&t'788$rFxloginrmrC cmd_stringrrCcJtt|jd)zSpawn a bash process to run *cmd_string*. Returns a ProcessHandle (subprocess.Popen or _ThreadedProcessHandle). Must be overridden by every backend. z must implement _run_bash())NotImplementedErrortyperv)rhrrrmrCs r _run_bashzBaseEnvironment._run_bash#s$"T$ZZ%8"U"U"UVVVrcdS)z z# declare -f | grep -vE '^_[^_]' >> z alias -p >> z# echo 'shopt -s expand_aliases' >> z echo 'set +e' >> z echo 'set +u' >> z pwd -P > z 2>/dev/null || true printf '\n%sz\n' "$(pwd -P)" T)rrmrz-Session snapshot created (session=%s, cwd=%s)uLinit_session failed (session=%s): %s — falling back to bash -l per commandFN)rrrrr_wait_for_processr _update_cwdloggerinforrr!warning)rh bootstrapr1resultrs r init_sessionzBaseEnvironment.init_session;s U4. U U151D U U. U U261D U U!% 3  U U !% 3  U U   U U* U U/3.> U U U  )>>)4AW>XXD++D$:P+QQF#'D   V $ $ $ KK?        ) ) ) NN6      $)D  )s A;C D(DDcommandc|dd}g}|jr|d|jd|dkr)|dst j|n|}|d|d|d |d|d |jr|d |jd|d |jd|d |jd|jd|dd |S)zwBuild the full bash script that sources snapshot, cd's, runs command, re-dumps env vars, and emits CWD markers.'z'\''zsource z 2>/dev/null || true~z~/zcd z || exit 126zeval 'z__hermes_ec=$?rz pwd -P > z printf '\nrz\n' "$(pwd -P)"zexit $__hermes_ec ) rrappendr startswithshlexquoterrjoin)rhrrescapedparts quoted_cwds r _wrap_commandzBaseEnvironment._wrap_commandds{//#w//   N LLL4#6LLL M M M!$s 3>>$3G3G EK   S   3:333444  (g((())) %&&&   S LLQ(;QQQ R R R  EEEEFFF  R$* R Rd.> R R R     ()))yyrcbdtjjdd}|d|d|d|S)z;Append stdin_data as a shell heredoc to the command string. HERMES_STDIN_Nrz << 'z' r)rrr)rrC delimiters r_embed_stdin_heredocz$BaseEnvironment._embed_stdin_heredocsG._drains  K//D!((....//%   ##%%%$$L(    s$3A.A.-A.Tr<)rrNrWrz [Command interrupted])rrtz [Command timed out after zs]|zterminal command runningg?)r?r@rrrrir _kill_processrlstripr$sleeprEr8r!rt) rhr1rmr drain_threaddeadline_now_activity_statepartial timeout_msgrs ` @rrz!BaseEnvironment._wait_for_processs,$&       !'vdCCC >##g-~   iikk! ""4(((!!!!,,, ggm447PP"%~(**""4(((!!!!,,,''-00GGGGG .g 33$++--"%  "/3M N N N JsOOO+iikk!. !$$$  K          D ''-00PPPsF11 F>=F>cl |dS#tttf$rYdSwxYw)zDTerminate a process. Subclasses may override for process-group kill.N)rlProcessLookupErrorPermissionErrorr:)rhr1s rrzBaseEnvironment._kill_processsA  IIKKKKK"OW=    DD s 33rc0||dS)zDExtract CWD from command output. Override for local file-based read.N)_extract_cwd_from_output)rhrs rrzBaseEnvironment._update_cwds %%f-----rcB|dd}|j}||}|dkrdStd|dz }||||}|dks||krdS||t |z|}|r||_|dd|}|dkr|}|d|t |z} | dkr| dznt |} |d||| dz|d<dS) zParse the __HERMES_CWD_{session}__ marker from stdout output. Updates self.cwd and strips the marker from result["output"]. Used by remote backends (Docker, SSH, Modal, Daytona, Singularity). rrNrirr)rrrfindmaxlenstriprfind) rhrrmarkerlast search_startfirstcwd_path line_startline_ends rrz(BaseEnvironment._extract_cwd_from_outputs: Hb))!||F## 2:: F1dTk**  V\488 B;;%4-- F%#f++-45;;==  DH \\$511   J;;tTCKK%788#+r>>8a<Hook called before each command execution. Remote backends (SSH, Modal, Daytona) override this to trigger their FileSyncManager. Bind-mount backends (Docker, Singularity) and Local don't need file sync — the host filesystem is directly visible inside the container/process. Nrrgs r_before_executezBaseEnvironment._before_execute s  rr)rmrCc|||\}}|p|j}|p|j}||||z} n||} n|} | r#|jdkr||| }d} |||} |j } || | || } | | |} | | | S)z=Execute a command, return {"output": str, "returncode": int}.Nheredocrr) r_prepare_commandrmrrrrrrrr)rhrrrmrC exec_command sudo_stdineffective_timeout effective_cwdeffective_stdinwrappedrr1rs rexecutezBaseEnvironment.executes #'#8#8#A#A j#3t|tx   !j&<(:5OO  #(OO(O  #t/9<<44\?SSL"O$$\=AA((~~ 5*;  ''6G'HH     rc.|dS)z.Alias for cleanup (compat with older callers).N)rrgs rstopzBaseEnvironment.stopHs rcR |dS#t$rYdSwxYwr5)rr!rgs r__del__zBaseEnvironment.__del__Ls:  LLNNNNN    DD s  &&c$ddlm}||S)z6Transform sudo commands if SUDO_PASSWORD is available.r)_transform_sudo_command)tools.terminal_toolr)rhrrs rrz BaseEnvironment._prepare_commandRs%??????&&w///rr5)rru)r)rvrwrxryrr|__annotations__rr rdictrboolrerrrrr staticmethodrrrrrrrrrrrrrrrrsK sc % %C %# %D % % % %(!% W W W W W  W $J W  W W W W  ^ #)#)#)R$ S$ s$ s$ $ $ $ THcHsHsHHH\H>Q>Qm>Qc>QD>Q>Q>Q>Q@-.$.... Ct C C C CL    "' #!% ''''' t ' $J ' ''''Z 00c3:o0F000000rrr5)/ryrRloggingr+rrHr?rrabcrrpathlibrtypingrrrhermes_constantsr tools.interruptr getLoggerrvrlocalrr|rrr!r$r0rIrAlistrNrUr\rrzr rcrer~rrrrrr-s8   ########)))))))))),,,,,,******  8 $ $+9?,,+hud{3d:+t++++ ?# 5 <????        4       ( 9j& 9c 9d 9 9 9 9.2 c #d ,4D040t00000 suUCZ'84'?,,,,,H,,,$B B B B B B B B T*C*C****Z0Z0Z0Z0Z0cZ0Z0Z0Z0Z0r