sj pdZddlZddlmZddlmZdZdZdZe edd Z e edd Z e edd Z d Z d!d eeefdzdefdZdefdZdede fdZdede fdZdede fdZde dde d e de de de de f dZddde de de de fd ZdS)"zBShared gateway restart constants and supervisor detection helpers.N)Mapping)DEFAULT_CONFIGKN"HERMES_GATEWAY_EXTERNAL_SUPERVISORagentrestart_drain_timeoutrestart_after_turn_timeoutcron_drain_timeoutg$@environreturncZ| tjn|}|drdS|drdS|dd}|r|dkrdSt|tddvS) z=Return whether this gateway process is owned by a supervisor.N INVOCATION_IDTHERMES_S6_SUPERVISED_CHILDXPC_SERVICE_NAME0>1onyestrue)osr getstrEXTERNAL_GATEWAY_SUPERVISOR_ENVstriplower)r env xpc_services 3/home/agent/.hermes/hermes-agent/gateway/restart.pyis_gateway_supervisor_processr!7s "**WC wwt ww+,,t'',b11K{c))t sww6;; < < B B D D J J L LQ c~tjdptjdS)uReturn whether the gateway is running inside a container for restart routing purposes (Docker/Podman ⇒ the detached setsid path dies with the cgroup; exit-75 service restart is the only viable path). Extracted from the inline probe in the /restart handler so tests can mock container detection hermetically — a real ``/.dockerenv`` on a containerized CI runner otherwise flips the routing under the test. z /.dockerenvz/run/.containerenv)rpathexistsr"r is_container_restart_contextr'Ks, 7>>- ( ( PBGNN;O,P,PPr"rawc t|pdrt|nt}n#tt f$r tcYSwxYwt d|S)zEParse a configured drain timeout, falling back to the shared default.r)rrfloat%DEFAULT_GATEWAY_RESTART_DRAIN_TIMEOUT TypeError ValueErrormaxr(values r parse_restart_drain_timeoutr2Wsn5!#)nn2244_c :_ z "55544445 sE??s9<AAc|tSt|tr|stS t |}n#t t f$r tcYSwxYwtd|S)uParse the after-turn wait cap for in-band restart, falling back to default. ``0`` is a deliberate disable (legacy immediate drain) and must not fall through to the default — unlike empty/missing input. Nr*)*DEFAULT_GATEWAY_RESTART_AFTER_TURN_TIMEOUT isinstancerrr+r-r.r/r0s r parse_restart_after_turn_timeoutr6`s  {99#s:CIIKK:99:c  z ":::9999: sE??A A&%A&c|tSt|tr|stS t |}n#t t f$r tcYSwxYwtd|S)uParse the cron-only drain floor, falling back to the shared default. ``0`` is a deliberate opt-out — cron work is then interrupted on the same budget as chat work, the pre-#82161 behaviour — and must not fall through to the default, unlike empty/missing input. Nr*)"DEFAULT_GATEWAY_CRON_DRAIN_TIMEOUTr5rrr+r-r.r/r0s r parse_cron_drain_timeoutr:qs {11#s2CIIKK2112c  z "22211112 sE??r7r*)elapsedcleanup_reserve_s drain_timeoutwatchdog_delayr;r<cddtdtdtfd}||}||}|dkr|S||||z ||tz }t|t ||S)uSeconds the shutdown drain may spend waiting on in-flight cron work. The configured floor is clamped to what this process can actually honour. The shutdown watchdog hard-exits at ``watchdog_delay`` and the service manager's ``TimeoutStopSec`` is sized from the same drain timeout, so waiting past that leash (minus ``cleanup_reserve_s`` for the teardown that follows the drain) would swap a cleanly-interrupted job for a SIGKILL that leaves it wedged mid-run — strictly worse than the bug being fixed. Never returns less than ``drain_timeout``: the cron floor only ever extends the wait, so an operator who deliberately configured a long ``restart_drain_timeout`` keeps it. r*r1fallbackr cp tt|dS#ttf$r|cYSwxYw)Nr*r/r+r-r.)r1r@s r _secondsz+resolve_cron_drain_budget.._secondssF uU||S)) ):&   OOO s 55)r*)objectr+CRON_DRAIN_CLEANUP_RESERVE_Sr/min) r=r r>r;r<rCdrainfloorceilings r resolve_cron_drain_budgetrJs,%% H] # #E H' ( (E ||   (7    ($&B C C D uc%)) * **r"g.@)headroomafter_turn_timeoutrKcb tt|d}n#ttf$rd}YnwxYw tt|d}n#ttf$rd}YnwxYw tt|d}n#ttf$rd}YnwxYw||z|zS)aTSeconds a CLI should wait for the gateway PID to exit after SIGUSR1. In-band restart may defer ``stop()`` until active turns finish (``after_turn_timeout``) and then spend up to ``drain_timeout`` inside ``stop()``. Callers that fall back to a hard kill on wait expiry must cover both phases or they reintroduce #77184. r*rB)r=rLrKrG after_turnmargins r resolve_restart_exit_wait_budgetrPsE-((#.. z "122C88 z " U8__c** z " :  &&s/ 66AA.-A.2BB&%B&)N)__doc__rcollections.abcrhermes_cli.configr!GATEWAY_SERVICE_RESTART_EXIT_CODEGATEWAY_FATAL_CONFIG_EXIT_CODErr+r,r4r9rErboolr!r'rDr2r6r:rJrPr&r"r rWs/HH ######,,,,,,%'! "$ #G(-734))%.3U789..*&+U701&&" $)- S#X  % ( Qd Q Q Q QV&U"&U.; %+%+%+%+%+ %+  %+  %+ %+%+%+%+X ''''' '  ''''''r"