#j JdZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z e rddl mZejeZdZdZe ejdee jd z d z Zd ZGd d ZddZddZdZefddZdS)zc Daemon mode support for Hindsight API. Provides idle timeout for running as a background daemon. ) annotationsN)Path) TYPE_CHECKING)IOi"HINDSIGHT_API_DAEMON_LOGz .hindsightz daemon.log_HINDSIGHT_DAEMON_CHILDc0eZdZdZefd dZdZdZdZdS) IdleTimeoutMiddlewarezBASGI middleware that tracks activity and exits after idle timeout. idle_timeoutintc`||_||_tj|_d|_dSN)appr time last_activity _checker_task)selfrr s \/home/rurouni/.hermes/hermes-agent/venv/lib/python3.11/site-packages/hindsight_api/daemon.py__init__zIdleTimeoutMiddleware.__init__)s-(!Y[[!ctKtj|_||||d{VdSr)rrr)rscopereceivesends r__call__zIdleTimeoutMiddleware.__call__/sB!Y[[hhugt,,,,,,,,,,,rc\tj||_dS)z7Start the background task that checks for idle timeout.N)asyncio create_task _check_idler)rs rstart_idle_checkerz(IdleTimeoutMiddleware.start_idle_checker4s'$01A1A1C1CDDrc|K|jdkrdS tjdd{Vtj|jz }||jkrlt d|jdtjdd{Vddl}tj tj |j )z:Background task that exits the process after idle timeout.rNTzIdle timeout reached (zs), shutting down daemon) r rsleeprrloggerinfosignaloskillgetpidSIGTERM)r idle_timer's rrz!IdleTimeoutMiddleware._check_idle8s   ! ! F 5-## # # # # # # # d&88I4,,, `T5F```aaamA&&&&&&&&&  V^444 5rN)r r ) __name__ __module__ __qualname____doc__DEFAULT_IDLE_TIMEOUTrrr rrrr r &sfLL0D""""" --- EEE55555rr log_handle 'IO[bytes]'returndictctjdkrKttdd}ttdd}||ztj|tjddSdtj||dS)anCross-platform kwargs to spawn a subprocess detached from the caller. On POSIX, ``start_new_session=True`` calls ``setsid(2)`` so the child survives the parent's terminal. On Windows we use ``DETACHED_PROCESS | CREATE_NEW_PROCESS_GROUP``. ``log_handle`` receives the child's stdout/stderr so output never leaks into the parent's terminal. WindowsDETACHED_PROCESSrCREATE_NEW_PROCESS_GROUPT) creationflagsstdinstdoutstderr close_fds)start_new_sessionr<r=r>)platformsystemgetattr subprocessDEVNULLSTDOUT)r3detached_processcreate_new_process_groups r_detach_popen_kwargsrIKsI%%":/A1EE#*:7QST#U#U -0HH'  '    "#   rNonectjddtjtjttj d5}tj | tj dddn #1swxYwYttd}tj | tj tj | tj dS)zpRedirect stdin/stdout/stderr to the daemon log file. Called in the daemon child process after re-exec. Tparentsexist_okrNa) DAEMON_LOG_PATHparentmkdirsysr=flushr>openr(devnulldup2filenor<)rWlog_fds r_redirect_stdio_to_logr[gs=    ===JJ bj#  6'   #)"2"2"4"4555666666666666666/3 ' 'FGFMMOOSZ..00111GFMMOOSZ..0011111s8ACC C ctjdkr#tjdddSt jtdkrtdSdtj ddD}tj dd g|z}t j }d|t<tt|d <tjddttd 5}tj|fd |it#|dddn #1swxYwYtjd dS)a`Detach the current process into a background daemon. Uses ``subprocess.Popen`` (which maps to ``posix_spawn`` on macOS) to re-exec the current command in a detached session. This replaces the traditional double-fork pattern because ``os.fork()`` without ``exec()`` corrupts Apple framework state (XPC, Metal/MPS, ObjC runtime) on macOS, causing SIGBUS crashes when PyTorch uses the MPS backend. The function has two code paths controlled by the ``_HINDSIGHT_DAEMON_CHILD`` environment variable: * **Parent** (env var not set): re-exec the same command via Popen with ``start_new_session=True``, stripping ``--daemon`` from argv and setting ``_HINDSIGHT_DAEMON_CHILD=1``. Then ``sys.exit(0)``. * **Child** (env var set): redirect stdio to the daemon log file and return. No fork, no re-exec. On Windows there is no fork model: the spawning parent is expected to detach us via ``CREATE_NEW_PROCESS_GROUP | DETACHED_PROCESS`` and to redirect stdout/stderr to ``HINDSIGHT_API_DAEMON_LOG`` before exec. We still ensure the log directory exists. win32TrLN1cg|] }|dk| S)z--daemonr2).0rPs r zdaemonize..s===Q*__!___rr#z-mzhindsight_api.mainrabenvr)rTrArQrRrSr(environgetENV_DAEMON_CHILDr[argv executablecopystrrVrDPopenrIexit) child_argscmdrcr3s r daemonizeroys. |w$$TD$AAA  z~~&''3..    >=SXabb\===J >4!5 6 CC *//  CC&)/&:&:C"#   === ot $ $K JJ#J)=j)I)IJJJKKKKKKKKKKKKKKKHQKKKKKs"D??EEportr boolcddl} ||j|j}|d|d|f}||dkS#t $rYdSwxYw)z>Check if a daemon is running and responsive on the given port.rNr#z 127.0.0.1F)socketAF_INET SOCK_STREAM settimeout connect_exclose Exception)rprssockresults rcheck_daemon_runningr|sMMM}}V^V-?@@ +t!455 { uusA%A,, A:9A:)r3r4r5r6)r5rJ)rpr r5rq)r0 __future__rrloggingr(rArDrTrpathlibrtypingrr getLoggerr-r%DEFAULT_DAEMON_PORTr1getenvrjhomerQrfr rIr[ror|r2rrrs #"""""     8 $ $$yry!;SS|A[^jAj=k=kllmm -"5"5"5"5"5"5"5"5J82222$222j&9       r