Struct ToolHandleRegistry 

pub struct ToolHandleRegistry { /* private fields */ }

Registry of live (and recently-terminal) detached tool invocations. One per crate::executor::Runtime; cheap to share via Arc.

Implementations

impl ToolHandleRegistry

pub fn new() -> Self

pub fn subscribe(&self) -> Receiver<ToolStreamEvent>

Subscribe to the live event fanout (every chunk from every handle, wrapped in a ToolStreamEvent). Used by the WS layer to forward tools.stream.event notifications.

pub async fn register( &self, tool: &str, action_id: &str, ) -> (ToolHandle, CancellationToken)

Register a new detached invocation. Returns the opaque handle and the cancel token the drain task must select on.

pub async fn push_chunk(&self, handle_id: &str, chunk: ToolStreamChunk)

Append a chunk from the executor’s stream. A terminal chunk seals the status (done → succeeded + result, error → failed + message). Chunks arriving after a terminal state (e.g. a racing executor after cancel) are dropped. Every accepted chunk is also broadcast as a ToolStreamEvent.

pub async fn mark_stream_closed(&self, handle_id: &str)

The executor’s stream ended without a terminal chunk — a dropped sender (panicked task, closed pipe). Treated as failure, unless the entry was already sealed (normal after a done/error/cancel).

pub async fn cancel(&self, handle_id: &str) -> bool

Request cancellation: fires the cancel token (the drain task and a cooperative executor observe it) and seals the entry as cancelled unless already terminal. Returns false for an unknown handle.

pub async fn cancel_all(&self) -> usize

Cancel every live invocation (linus review D3): wired into the daemon’s session teardown so a dropped WS connection doesn’t orphan running detached tools — each session’s runtime (and thus registry) is unreachable after disconnect, so anything still running would be uncancellable until daemon restart. Returns how many live invocations were cancelled.

pub async fn poll(&self, handle_id: &str) -> Option<ToolPollResult>

Drain buffered chunks + report status. None for an unknown (or already fully-consumed) handle. The FIRST poll that observes the entry in a terminal state marks it consumed (it also drains any remaining chunks); the next poll removes it — so a caller always gets to see the terminal status at least once.

pub async fn status(&self, handle_id: &str) -> Option<ToolStatus>

Current status without draining (used by tests and liveness checks).

pub async fn live_count(&self) -> usize

Number of live (non-terminal) invocations.

Trait Implementations

impl Default for ToolHandleRegistry

fn default() -> Self

Returns the “default value” for a type.