鎒joXdZddlZddlZddlmZddlmZddlm Z GddZ dS)aManages the lifecycle of registered triggers. The TriggerRunner starts all triggers as concurrent asyncio tasks at session start and cancels them at session end. Unhandled exceptions in a trigger are logged but do not crash the session or restart the trigger. N)Sequence connection)triggersceZdZdZdeejdejddfdZ ddZ ddZ dd Z dd Z edefd Zed ejd ejddfdZdS) TriggerRunnerz8Manages registration, startup, and shutdown of triggers.rrreturnNcJt||_||_g|_dS)zInitializes the TriggerRunner. Args: triggers: Sequence of trigger functions to manage. connection: The live connection to the agent. N)list _triggers _connection_tasks)selfrrs XC:\Projects\DSCons\.venv\Lib\site-packages\google\antigravity\triggers\trigger_runner.py__init__zTriggerRunner.__init__!s$(^^DN!D,.DKKKc *K|jrtd|jD]r}tj|j}t j|||dt|dd}|j |sdS)uStart all triggers as concurrent asyncio tasks. Each trigger receives its own TriggerContext. If a trigger raises an unhandled exception, it is logged and the task stops — no auto-restart, no impact on other triggers or the session. Triggers run as independent asyncio tasks with no ordering guarantees — they may execute in any order. Raises: RuntimeError: If the runner is already started. z!TriggerRunner is already started.rztrigger-__name__unknown)nameN) r RuntimeErrorr triggers_moduleTriggerContextr asyncio create_task _run_triggergetattrappend)rtriggerctxtasks rstartzTriggerRunner.start0s {> < = ==>  *%   c   GS ) )C'':yAACCd krcK|jsdS|jD]*}|s|+tj|jddid{V|jdS)zCancel all trigger tasks and wait for them to finish. Safe to call multiple times. After stop(), the runner can be started again with start(). Nreturn_exceptionsT)rdonecancelrgatherclear)rr!s rstopzTriggerRunner.stopJs ;  f  YY[[  .$+ > > >>>>>>>>Krc>K|d{V|S)z\Start triggers on entering the context. Returns: The TriggerRunner instance. N)r"rs r __aenter__zTriggerRunner.__aenter__[s- **,, Krc>K|d{VdS)zStop triggers on exiting the context. Args: exc_type: The exception type, if any. exc_val: The exception value, if any. exc_tb: The traceback, if any. N)r))rexc_typeexc_valexc_tbs r __aexit__zTriggerRunner.__aexit__ds. ))++rc>td|jDS)z%True if any trigger tasks are active.c3@K|]}| VdS)N)r%).0r!s r z+TriggerRunner.is_running..qs,774499;;777777r)anyrr+s r is_runningzTriggerRunner.is_runningns# 774;777 7 77rrr cKt|dt|} ||d{VdS#tj$rt jd|t $rt jd|YdSwxYw)a(Wraps a trigger call with error handling. CancelledError is allowed to propagate so that trigger-internal cleanup code works correctly. Non-cancellation exceptions are logged and swallowed. Args: trigger: The trigger function to run. ctx: The trigger's context. rNzTrigger '%s' cancelled.z-Trigger '%s' failed with unhandled exception.)rreprrCancelledErrorlogginginfo Exception exception)rr trigger_names rrzTriggerRunner._run_triggerss7JW >>L GCLL  !    l,l;;;   9<s5AA<;A<)r N)r r)r __module__ __qualname____doc__rrTriggerconnection_module Connectionrr"r)r,r1propertyboolr7 staticmethodrrrrrrs @@ /01 /$. / / / / /4" 8$888 88&  ) <rr) rBrr;typingrgoogle.antigravity.connectionsrrDgoogle.antigravity.triggersrrrrIrrrMsJJJJJJCCCCCCmmmmmmmmmmr