@gotgenes/pi-subagents 5.5.0 → 5.7.0

package/src/runtime.ts

@@ -6,7 +6,7 @@
  * Follows the same pattern as pi-permission-system's ExtensionRuntime.
  */
 
-import type { AgentActivity, AgentWidget } from "./ui/agent-widget.js";
+import type { AgentActivity, AgentWidget, UICtx } from "./ui/agent-widget.js";
 
 /**
  * Narrow config subset read by AgentManager when constructing RunOptions.
@@ -23,27 +23,65 @@ export interface RunConfig {
  * Created once inside `piSubagentsExtension()` via `createSubagentRuntime()`.
  * Tests construct a fresh runtime per test for full isolation.
  */
-export interface SubagentRuntime {
+export class SubagentRuntime {
   // ── Execution config (was module-scope in agent-runner.ts) ──────────────
   /** Default max turns for all agents. undefined = unlimited. */
-  defaultMaxTurns: number | undefined;
+  defaultMaxTurns: number | undefined = undefined;
   /** Additional turns allowed after the soft-limit steer message. */
-  graceTurns: number;
+  graceTurns: number = 5;
 
   // ── Session state (was closure-scoped in index.ts) ───────────────────────
   /** Active Pi session context — set on session_start, cleared on session_shutdown. */
-  currentCtx: { pi: unknown; ctx: unknown } | undefined;
+  currentCtx: { pi: unknown; ctx: unknown } | undefined = undefined;
   /**
    * Per-agent live activity state shared across the notification system,
    * widget, and tool handlers. The Map itself is never replaced.
    */
-  readonly agentActivity: Map<string, AgentActivity>;
+  readonly agentActivity: Map<string, AgentActivity> = new Map();
   /**
    * Persistent widget reference. Null until constructed after AgentManager.
-   * Notification closures use `runtime.widget!` — safe because agents always
-   * complete after widget construction.
+   * Delegation methods use optional chaining so callers never need `widget!`.
    */
-  widget: AgentWidget | null;
+  widget: AgentWidget | null = null;
+
+  // ── Session-context methods ──────────────────────────────────────────────
+
+  /** Store the active Pi session context (called from session_start). */
+  setSessionContext(pi: unknown, ctx: unknown): void {
+    this.currentCtx = { pi, ctx };
+  }
+
+  /** Clear the session context (called from session_shutdown). */
+  clearSessionContext(): void {
+    this.currentCtx = undefined;
+  }
+
+  // ── Widget delegation methods ─────────────────────────────────────────────
+
+  /** Delegate to widget.setUICtx — no-op when widget is null. */
+  setUICtx(ctx: UICtx): void {
+    this.widget?.setUICtx(ctx);
+  }
+
+  /** Delegate to widget.onTurnStart — no-op when widget is null. */
+  onTurnStart(): void {
+    this.widget?.onTurnStart();
+  }
+
+  /** Delegate to widget.markFinished — no-op when widget is null. */
+  markFinished(id: string): void {
+    this.widget?.markFinished(id);
+  }
+
+  /** Delegate to widget.update — no-op when widget is null. */
+  updateWidget(): void {
+    this.widget?.update();
+  }
+
+  /** Delegate to widget.ensureTimer — no-op when widget is null. */
+  ensureTimer(): void {
+    this.widget?.ensureTimer();
+  }
 }
 
 /**
@@ -52,11 +90,5 @@ export interface SubagentRuntime {
  * Call once at extension startup; pass the result to factories and handlers.
  */
 export function createSubagentRuntime(): SubagentRuntime {
-  return {
-    defaultMaxTurns: undefined,
-    graceTurns: 5,
-    currentCtx: undefined,
-    agentActivity: new Map(),
-    widget: null,
-  };
+  return new SubagentRuntime();
 }