߲j(dZddlmZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z mZmZmZmZeeeefZdZdZdd Zdd Zdd Zdd Zdd ZGddZeZddZddddZd dZ d!dZ!d"dZ"d#dZ#d$dZ$gdZ%y)%aCross-agent file state coordination. Prevents mangled edits when concurrent subagents (same process, same filesystem) touch the same file: B writes a file A already read, so A's next write would clobber B's changes. Complements the single-agent path-overlap check in ``agent.tool_dispatch_helpers._should_parallelize_tool_batch``. A process-wide ``FileStateRegistry`` tracks per-agent read stamps, the global last writer and a per-path lock; every method is a no-op under ``HERMES_DISABLE_FILE_STATE_GUARD=1``. ) annotationsN) defaultdict)contextmanager)Path)DictIterableListOptionalTupleicdtjjddjdk(S)NHERMES_DISABLE_FILE_STATE_GUARD1)osenvirongetstrip5/home/ubuntu/.hermes/hermes-agent/tools/file_state.py _disabledrs& ::>>;R @ F F HC OOrctS)zTrue when the user switched the read-before-write guard off; the file tools then warn instead of refusing stale/unread write_file overwrites.)rrrrguard_disabledr#s  ;rc` tjj|S#t$rYywxYwN)rpathgetmtimeOSError)resolveds r_mtime_or_noner )s.ww)) s ! --cTtjdtj|S)Nz%H:%M:%S)timestrftime localtime)tss r_fmt_tsr&0s ==T^^B%7 88rctt||z D]G} t|tr|j n#|j t t |Iy#ttf$rYywxYw)zPop entries until *container* is within *cap* (sets: arbitrary; dicts: oldest by insertion order). An eviction only costs one redundant re-send or staleness check.N) rangelen isinstancesetpopnextiter StopIterationKeyError) containercap_s r _evict_oldestr45sh3y>C' ( )S)  d4 ?34  x(   sAA##A65A6ceZdZdZddZeddZddZddd ddZdd dd Z dd Z dd Z dd Z ddZ ddZy)FileStateRegistryz4Process-wide coordinator for cross-agent file edits.ctt|_i|_i|_i|_t j|_t j|_ yr) rdict_reads _last_writer _path_locks_path_lock_users threadingLock _meta_lock _state_lockselfs r__init__zFileStateRegistry.__init__EsC7B47H :<6802#..*$>>+rc#jK|j5|jj|tj}|j j |ddz|j |<dddj d|j|j5|j |dz }|r||j |<n8|j j|d|jj|ddddy#1swYxYw#1swYyxYw#|j|j5|j |dz }|r||j |<n8|j j|d|jj|ddddw#1swYwxYwxYww)zPer-path lock: threads on the same path serialize, different paths proceed. The lock entry is dropped once the last holder/waiter exits.rN) r?r; setdefaultr=r>r<racquirereleaser,)rBrlockuserss r lock_pathzFileStateRegistry.lock_pathMs|__ Y##..x9IJD.2.C.C.G.GRS.TWX.XD ! !( + Y  9  LLN 9--h7!;6;D))(3))--h=$$((48  9 9 Y Y 9 9 LLN 9--h7!;6;D))(3))--h=$$((48  9 9 9sg F3AD *F3D!F3#AD F3 DF3DF3!F0>AF$ F0$F- )F00F3cv|j|}t||t|f||<t|ty)zCaller holds ``_state_lock``.N)r9floatboolr4_MAX_PATHS_PER_AGENT)rBtask_idrmtimenowpartial agent_readss r_stampzFileStateRegistry._stampas4kk'* !&usDM B Hk#78rFN)rSrQctry| t|n|}|y|j5|j|||t j|dddy#1swYyxYwr)rr r@rUr")rBrPrrSrQs r record_readzFileStateRegistry.record_readgs^ ; ,1Mx(u =     H KK5$))+w G H H Hs (AA$)rQc(try| t|n|}|ytj}|j5||f|j|<t |jt |j||||ddddy#1swYyxYw)zRecord a successful write: global last-writer AND this agent's own read stamp (a write is an implicit read of the current content).NF)rr r"r@r:r4_MAX_GLOBAL_WRITERSrU)rBrPrrQrRs r note_writezFileStateRegistry.note_writeqs ; ,1Mx(u = iik    >+2C.D  h ' $++-@ A KK5#u = > > >s ABBc try|j5|jj|ij|}|jj|}dddyt |}|y>|\}}||k7r4||d|dS|d}||kDr |d|dt |dt |dS||\} } } || k7r|dS| r|d Sy|d S#1swY{xYw) zModel-facing warning if this write would be stale, else ``None``. Severity order: sibling wrote after our read > mtime drift / partial read > never read.Nz" was modified by sibling subagent zg but this agent never read it. Read the file before writing to avoid overwriting the sibling's changes.rEz at u% — after this agent's last read at z". Re-read the file before writing.zs was modified since you last read it on disk (external edit or unrecorded writer). Re-read the file before writing.zz was last read with offset/limit pagination (partial view). Read the remaining pages, or use patch, before overwriting it.zS was not read by this agent. Read the file first so you can write an informed edit.)rr@r9rr:r r&) rBrPrstamp last_writer current_mtime writer_tid writer_tsread_ts read_mtime_read_tsrSs r check_stalezFileStateRegistry.check_staleso ;    :KKOOGR044X>E++//9K : =[0&x0    "$/ !J W$=#*$F%.)112  (w&#*$F%.WY-?,@A55'>'@ .##J0R8^Y O**1- . .Cy  .Cys)BB#B(BB%ctrgS|j5t|jj |ij cdddS#1swYyxYw)z#Resolved paths this agent has read.N)rr@rfr9rkeys)rBrPs r known_readszFileStateRegistry.known_readssJ ;I    = 499;< = = =s 3AAc"|j5|jj|d|jj Dcgc]\}\}}||k(s|c}}}D]}|j|= dddycc}}}w#1swYyxYw)acRelease read stamps and writer claims owned by a task after its lifecycle ends. A finished task is not a concurrent sibling: leaving its writer claims behind makes the next run of the same job (a fresh ``cron::`` id) refuse to write the same scratch path as "modified by sibling subagent" hours after the writer exited.N)r@r9r,r:rg)rBrPrnr__tss r forget_taskzFileStateRegistry.forget_tasks    ) KKOOGT *484E4E4K4K4Mgg0A0 CQ[_fQfag )%%a( ) ) )g ) )s#;BA> A> B>BBcX|j5|jj|jjddd|j5|j j|j jdddy#1swYSxYw#1swYyxYw)z)Reset all state. Intended for tests only.N)r@r9clearr:r?r;r<rAs rrwzFileStateRegistry.clears    & KK       # # % &__ *    " " $  ! ! ' ' ) * * & & * *s5B5B B B))returnNone)rstr) rPrzrrzrQrMrRrMrSrNrxry) rPrzrrzrSrNrQOptional[float]rxry)rPrzrrzrQr{rxry)rPrzrrzrx Optional[str])rirzrjrMrkz Iterable[str]rxDict[str, List[str]]rPrzrxz List[str])rPrzrxry)__name__ __module__ __qualname____doc__rCrrKrUrWrZrdrorrrurwrrrr6r6Bs|>,99&9 KP-1H*H6:HSW >0Fd ) .B = )*rr6ctSr) _registryrrr get_registryrs rFrScFtj|t||y)Nr)rrWrz)rPresolved_or_pathrSs rrWrWs '3'7#8'JrcBtj|t|yr)rrZrzrPrs rrZrZs #&6"78rc@tj|t|Sr)rrdrzrs rrdrds  #.>*? @@rc>tjt|Sr)rrKrz)rs rrKrKs   s#34 55rc htj|||Dcgc] }t|c}Scc}wr)rrorz)rirjrkrns rroros)  ! !/8e=Tc!f=T UU=Ts/ c,tj|Sr)rrr)rPs rrrrrs   ))r)r6rrWrZrdrKrorr)rxrN)rrzrxr{)r%rMrxrz)r2intrxry)rxr6)rPrzr str | PathrSrNrxry)rPrzrrrxry)rPrzrrrxr|)rr)rirzrjrMrkzIterable[str | Path]rxr}r~)&r __future__rrr=r" collectionsr contextlibrpathlibrtypingrrr r r rMrN ReadStamprOrYrrr r&r4r6rrrWrZrdrKrorr__all__rrrrs#  #%88 %$ % P  9 W*W*t    PUK9A6V* r