agent-relay 1.2.0 → 1.3.0

package/dist/wrapper/pty-wrapper.d.ts

@@ -4,49 +4,31 @@
  * Unlike TmuxWrapper which provides interactive terminal access,
  * PtyWrapper runs agents headlessly with output capture for logs.
  * Used for spawned workers that don't need direct user interaction.
+ *
+ * Extends BaseWrapper for shared message handling, spawn/release,
+ * and continuity integration.
  */
-import { EventEmitter } from 'node:events';
-import type { ParsedSummary, SessionEndMarker } from './parser.js';
-import type { SpeakOnTrigger } from '../protocol/types.js';
+import { BaseWrapper, type BaseWrapperConfig } from './base-wrapper.js';
+import type { ParsedCommand, ParsedSummary, SessionEndMarker } from './parser.js';
+import type { SendPayload, SendMeta } from '../protocol/types.js';
 import { type LifecycleHooks } from '../hooks/index.js';
-import { type InjectionMetrics, type CliType } from './shared.js';
-export interface PtyWrapperConfig {
-    name: string;
-    command: string;
-    args?: string[];
-    socketPath?: string;
-    cwd?: string;
-    env?: Record<string, string>;
-    /** Relay prefix pattern (default: '->relay:') */
-    relayPrefix?: string;
-    /** CLI type for special handling (auto-detected from command if not set) */
-    cliType?: CliType;
+/**
+ * PtyWrapper-specific configuration.
+ * Extends BaseWrapperConfig with PTY-specific options.
+ */
+export interface PtyWrapperConfig extends BaseWrapperConfig {
     /** Directory to write log files (optional) */
     logsDir?: string;
-    /** Dashboard port for spawn/release API calls (enables nested spawning from spawned agents) */
-    dashboardPort?: number;
     /** Allow this agent to spawn other agents (default: true for Lead, false for spawned workers) */
     allowSpawn?: boolean;
-    /** Callback for spawn commands (fallback if dashboardPort not set) */
-    onSpawn?: (name: string, cli: string, task: string) => Promise<void>;
-    /** Callback for release commands (fallback if dashboardPort not set) */
-    onRelease?: (name: string) => Promise<void>;
     /** Callback when agent exits */
     onExit?: (code: number) => void;
-    /** Primary agent to shadow (if this agent is a shadow) */
-    shadowOf?: string;
-    /** When the shadow should speak (default: ['EXPLICIT_ASK']) */
-    shadowSpeakOn?: SpeakOnTrigger[];
-    /** Stream output to daemon for dashboard log viewing (default: true) */
-    streamLogs?: boolean;
-    /** Task/role description for trajectory tracking */
-    task?: string;
     /** Custom lifecycle hooks */
     hooks?: LifecycleHooks;
     /** Enable trajectory tracking hooks (default: true if task provided) */
     trajectoryTracking?: boolean;
-    /** Resume from a previous agent ID (for crash recovery) */
-    resumeAgentId?: string;
+    /** Interactive mode - disables auto-accept of permission prompts (for auth setup flows) */
+    interactive?: boolean;
     /**
      * Summary reminder configuration. Set to false to disable.
      * Default: { intervalMinutes: 15, minOutputs: 50 }
@@ -71,6 +53,12 @@ export interface SessionEndEvent {
     agentName: string;
     marker: SessionEndMarker;
 }
+export interface AuthRevokedEvent {
+    agentName: string;
+    provider: string;
+    message?: string;
+    confidence: 'high' | 'medium' | 'low';
+}
 export interface PtyWrapperEvents {
     output: (data: string) => void;
     exit: (code: number) => void;
@@ -80,41 +68,25 @@ export interface PtyWrapperEvents {
     'summary': (event: SummaryEvent) => void;
     /** Emitted when agent outputs a [[SESSION_END]] block. Cloud services can handle session closure. */
     'session-end': (event: SessionEndEvent) => void;
+    /** Emitted when auth revocation is detected. Cloud services can handle re-auth flow. */
+    'auth_revoked': (event: AuthRevokedEvent) => void;
 }
-export declare class PtyWrapper extends EventEmitter {
-    private config;
+export declare class PtyWrapper extends BaseWrapper {
+    protected config: PtyWrapperConfig;
     private ptyProcess?;
-    private client;
-    private running;
     private outputBuffer;
     private rawBuffer;
-    private relayPrefix;
-    private cliType;
-    private sentMessageHashes;
-    private receivedMessageIds;
-    private processedSpawnCommands;
-    private processedReleaseCommands;
-    private pendingFencedSpawn;
-    private messageQueue;
-    private isInjecting;
     private readyForMessages;
     private lastOutputTime;
-    private injectionMetrics;
     private logFilePath?;
     private logStream?;
     private acceptedPrompts;
     private hookRegistry;
     private sessionStartTime;
-    private continuity?;
-    private agentId?;
-    private processedContinuityCommands;
-    private lastSummaryRawContent;
-    private sessionEndProcessed;
     private inThinkingBlock;
     private lastSummaryTime;
     private outputsSinceSummary;
     private detectedTask?;
-    private sessionEndData?;
     private instructionsInjected;
     private continuityInjected;
     private recentLogChunks;
@@ -122,19 +94,24 @@ export declare class PtyWrapper extends EventEmitter {
     private static readonly LOG_DEDUP_MAX_SIZE;
     private lastParsedLength;
     private lastContinuityParsedLength;
+    private authRevoked;
+    private lastAuthCheck;
+    private readonly AUTH_CHECK_INTERVAL;
     constructor(config: PtyWrapperConfig);
     /**
-     * Start the agent process
+     * Inject content into the PTY process.
+     * Used by BaseWrapper for message injection.
      */
-    start(): Promise<void>;
+    protected performInjection(content: string): Promise<void>;
     /**
-     * Initialize agent ID for continuity/resume functionality
+     * Get cleaned output buffer for command parsing.
+     * Strips ANSI codes and returns raw buffer content.
      */
-    private initializeAgentId;
+    protected getCleanOutput(): string;
     /**
-     * Get the current agent ID
+     * Start the agent process
      */
-    getAgentId(): string | undefined;
+    start(): Promise<void>;
     /**
      * Inject continuity context from previous session.
      * Called after agent ID initialization to restore state from ledger.
@@ -150,7 +127,11 @@ export declare class PtyWrapper extends EventEmitter {
      *   ->continuity:uncertain "..."  - Mark item as uncertain
      *   ->continuity:handoff <<<...>>> - Create explicit handoff
      */
-    private parseContinuityCommands;
+    /**
+     * Parse continuity commands from output.
+     * Overrides BaseWrapper to use client.sendMessage instead of queuing.
+     */
+    protected parseContinuityCommands(content: string): Promise<void>;
     /**
      * Handle output from the process
      */
@@ -214,9 +195,10 @@ export declare class PtyWrapper extends EventEmitter {
      */
     private parseSingleLineMessages;
     /**
-     * Send relay command to daemon
+     * Send relay command to daemon.
+     * Extends BaseWrapper to add PTY-specific logging and hook dispatch.
      */
-    private sendRelayCommand;
+    protected sendRelayCommand(cmd: ParsedCommand): void;
     /** Valid CLI types for spawn commands */
     private static readonly VALID_CLI_TYPES;
     /** Validate agent name format (PascalCase, alphanumeric, 2-30 chars) */
@@ -237,21 +219,25 @@ export declare class PtyWrapper extends EventEmitter {
      * - Command must be at start of line (after whitespace)
      * - Agent name must be PascalCase (e.g., Backend, Frontend, Worker1)
      * - CLI must be a known type (claude, codex, gemini, etc.)
+     *
+     * Overrides BaseWrapper to add PTY-specific validation and logging.
      */
-    private parseSpawnReleaseCommands;
+    protected parseSpawnReleaseCommands(content: string): void;
     /**
-     * Execute spawn via API or callback
+     * Execute spawn via API or callback.
+     * Overrides BaseWrapper to add PTY-specific logging and API path.
      */
-    private executeSpawn;
+    protected executeSpawn(name: string, cli: string, task: string): Promise<void>;
     /**
-     * Execute release via API or callback
+     * Execute release via API or callback.
+     * Overrides BaseWrapper to add PTY-specific logging and API path.
      */
-    private executeRelease;
+    protected executeRelease(name: string): Promise<void>;
     /**
-     * Handle incoming message from relay
-     * @param originalTo - The original 'to' field from sender. '*' indicates this was a broadcast message.
+     * Handle incoming message from relay.
+     * Extends BaseWrapper to add PTY-specific behavior.
      */
-    private handleIncomingMessage;
+    protected handleIncomingMessage(from: string, payload: SendPayload, messageId: string, meta?: SendMeta, originalTo?: string): void;
     /**
      * Wait for output to stabilize before injection.
      * Returns true if output has been stable for the required duration.
@@ -303,8 +289,6 @@ export declare class PtyWrapper extends EventEmitter {
      * Close the log file stream
      */
     private closeLogStream;
-    get isRunning(): boolean;
-    get name(): string;
     get pid(): number | undefined;
     get logPath(): string | undefined;
     /**
@@ -318,11 +302,6 @@ export declare class PtyWrapper extends EventEmitter {
      * Also updates the local continuity ledger for session recovery.
      */
     private checkForSummaryAndEmit;
-    /**
-     * Save a parsed summary to the continuity ledger.
-     * Maps summary fields to ledger fields for session recovery.
-     */
-    private saveSummaryToLedger;
     /**
      * Check for [[SESSION_END]] blocks and emit 'session-end' event.
      * Allows cloud services to handle session closure without hardcoding storage.
@@ -330,19 +309,18 @@ export declare class PtyWrapper extends EventEmitter {
      */
     private checkForSessionEndAndEmit;
     /**
-     * Reset session-specific state for wrapper reuse.
-     * Call this when starting a new session with the same wrapper instance.
+     * Check for auth revocation patterns in output.
+     * Detects when the CLI's OAuth session has been revoked (e.g., user logged in elsewhere).
+     * Emits 'auth_revoked' event and sends notification to relay daemon.
      */
-    resetSessionState(): void;
+    private checkAuthRevocation;
     /**
-     * Get injection reliability metrics
+     * Reset auth state (e.g., after re-authentication)
      */
-    getInjectionMetrics(): InjectionMetrics & {
-        successRate: number;
-    };
+    resetAuthState(): void;
     /**
-     * Get count of pending messages in queue
+     * Check if auth has been revoked
      */
-    get pendingMessageCount(): number;
+    isAuthRevoked(): boolean;
 }
 //# sourceMappingURL=pty-wrapper.d.ts.map