鎒jrHdZddlmZddlZddlmZmZGddZdS)zDThread-safe hierarchical key-value store for extensibility contexts.) annotationsN)AnyCallablecReZdZdZdddZddd ZddZ dddZddZddZ d dZ dS)! StateStorea@Thread-safe hierarchical key-value store for extensibility contexts. Provides atomic state operations (``get_state``, ``set_state``, ``update_state``) and explicit lock/context manager support across both flat and hierarchical (parent-child) extensibility handles such as ``ToolContext`` and ``HookContext``. NparentStateStore | NonereturnNonecR||_i|_tj|_dS)zInitializes the StateStore. Args: parent: Optional parent state store for fallback lookups in hierarchical contexts. N)r_store threadingRLock_lock)selfrs LC:\Projects\DSCons\.venv\Lib\site-packages\google\antigravity\utils\state.py__init__zStateStore.__init__ s%DK"$DK""DJJJkeystrdefaultrc|j5||jvr|j|cdddS dddn #1swxYwY|j|j||S|S)aRetrieves a value from the local state store or its parents. Args: key: The state key. default: Value returned when key is absent in this store and any ancestors. Returns: The stored value, or ``default`` if the key is not found. N)rr r get_state)rrrs rrzStateStore.get_state+s     {3                        { [ " "3 0 00 Ns 8<<valuecZ|j5||j|<ddddS#1swxYwYdS)z{Stores a value in the local state store. Args: key: The state key. value: The value to store locally. N)rr )rrrs r set_statezStateStore.set_state=s| dk#s $$ updater_fnCallable[[Any], Any]c|j5|||}||}||j|<|cdddS#1swxYwYdS)a\Atomically updates a value in the local state store. Note: ``updater_fn`` executes while holding the global reentrant lock (``self._lock``). To avoid blocking concurrent execution across the session, ``updater_fn`` should be a fast, non-blocking callback without synchronous I/O or long sleeps. If the key exists only in a parent store, the current parent value is passed to ``updater_fn``, and the updated result is stored in the local store so that local turn updates do not mutate parent session scope. Args: key: The state key. updater_fn: Function taking current value (or default if missing) and returning updated value to store. default: Value passed to ``updater_fn`` if key is not found across the hierarchy. Returns: The updated value after applying ``updater_fn``. N)rrr )rrrrvalnew_vals r update_statezStateStore.update_stateGs6  NN3 ( (c 3g dk#  s-AA Athreading.RLockc|jS)zReturns the underlying reentrant lock for custom critical sections. Returns: The `threading.RLock` protecting this state store. )rrs rlockzStateStore.lockhs :rc8|j|S)z6Acquires the store lock for use in a `with` statement.)racquirer%s r __enter__zStateStore.__enter__psJ Krexc_typeexc_valexc_tbc8|jdS)z8Releases the store lock when exiting a `with` statement.N)rrelease)rr*r+r,s r__exit__zStateStore.__exit__usJr)N)rr r r )rrrrr r)rrrrr r )rrrrrrr r)r r#)r r)r*rr+rr,rr r ) __name__ __module__ __qualname____doc__rrrr"r&r)r/rrrrs # # # # #$ B rr)r3 __future__rrtypingrrrr4rrr7sKJ"""""" ``````````r