߲j|dZddlZddlZddlZddlmZmZmZmZddl m Z ejdZ GddZ y)z=Running-process checkpoint persistence and PID-safe recovery.N)AnyDictListOptional)redact_sensitive_textztools.process_registryc<eZdZddeeeeeffdZde fdZ y)ProcessCheckpointMixinN extra_entriesc 4 ddlm}m} |j5g}|jj D]}|j r|j;|jdk(r,|jr |j|j|_d|ji|Dcic]}|t||c}}t|jd|d<|jxs |j |d <|j#||r9|Dchc]}|j%dc} |j' fd |Dddddd lm} | |ycc}wcc}w#1swY(xYw#t,$r"} t.j1d | d Yd} ~ yd} ~ wwxYw)zAWrite running process metadata to the checkpoint file atomically.r)_checkpoint_path_CHECKPOINT_FIELDSNhost session_idT) code_filecommand owner_task_idc3JK|]}|jdvs|yw)rN)get).0item tracked_idss F/home/ubuntu/.hermes/hermes-agent/tools/process_registry_checkpoint.py z;ProcessCheckpointMixin._write_checkpoint..(s""oDTXXlE[cnEn4"os##)atomic_json_writez#Failed to write checkpoint file: %s)exc_info)tools.process_registryr r _lock_runningvaluesexitedhost_start_time pid_scopepid_safe_host_start_timeidgetattrrrrtask_idappendrextendutilsr Exceptionloggerdebug) selfr r r entriessfentryrrers @r_write_checkpointz(ProcessCheckpointMixin._write_checkpointsjO R p--/*Axx ((0Q[[F5Jquu,0,F,Fquu,M))144bN`3aAwq!}4D3abE (=QYYRV'WE)$-.__-I E/*NN5)*!FM"Nd488L#9"NKNN"oM"oo' p( 0 .0' :4b#O% p p, R LL>DL Q Q RsN E,BE E 0AE EE 9E, E  E)%E,, F5FFreturnc ddlm}m}m}m}m}m}|}|jsy tj|jd}d} g} |D]} | jd| jdd} } | s*|j5| jd|jv}d d d r_| dk7r,tj!d | jd d d d | | |j#| | jds~|j%| r'tj!d| jdd| | jdd}|r0||s(tj'd|| | j)| /|Dcic]}|| j|||}}|j+| jd d | jddxs| jdd| jdt-j,|d | ddd|}|j5||j|j.<d d d | dz } tj!d|j0d d | |j2dkDs2|j4j)|j.|j2|j6d|Dcic]}|t9|d|c}|j:|j<d|j?| | S#t$rYywxYw#1swYzxYwcc}w#1swYxYwcc}w)!zvOn gateway startup, probe PIDs from the checkpoint file; returns how many were recovered as detached sessions.r)ProcessSessionr r _CHECKPOINT_DEFAULTS_WATCHER_ROUTE_KEYS_stop_systemd_unitzutf-8)encodingr#r"rrNz=Skipping recovery for non-host process: %s (pid=%s, scope=%s)runknown<r!uNot recovering session %s: pid %d is alive but its start time no longer matches — PID was recycled onto an unrelated process; refusing to adopt it.? systemd_unitzjCould not reap persisted scope %s for dead wrapper pid %s; retaining checkpoint entry for the next startuprr' started_at)rrrAT)r%detachedz'Recovered detached process: %s (pid=%d))rcheck_interval session_keywatcher_)notify_on_completeparent_session_id)r ) rr7r r r8r9r:existsjsonloads read_textr+rrrr,info_host_pid_is_ours_is_host_pid_alivewarningr(updatetimer%rwatcher_intervalpending_watchersrEr&rGrHr4)r.r7r r r8r9r:checkpoint_pathr/ recoveredunresolved_scope_entriesr2r#r"already_trackedr?r1fieldssessionkeys rrecover_from_checkpointz.ProcessCheckpointMixin.recover_from_checkpoint.s:  +,%%' jj!:!:G!:!LMG 9; 5 E"YYu-uyyf/MC K"'))L"9T]]"J KF" SIIi3CR8#yJ))#uyy9J/KL**3/KKF ,4c ; %yy< (:<(HNNJ$c+-33E:HZ[1a1&:1&=>>[F[ MM )Y7#ii<X )UW@X 99\499;?  A %Ul(;dUfUG 4,3 gjj) 4 NI KKA7??SVTVCWY\ ]''!+%%,,")**&-&>&>#*#6#6.K^^3sGGxu-=>>^ . +2*D*D)0)B)B .]5 l -EFw   K K6\ 4 4_s5%L L!L.+L3L?  LL!L+ 3L< )N) __name__ __module__ __qualname__rrrstrrr4intr]rIrr r s0RxT#s(^8L/MRris9C  ,,.   3 4iirc