catui-agent 1.0.0-beta.0

package/dist/core/extensions-host/loader.d.ts

@@ -0,0 +1,21 @@
+import { type EventBus } from "../runtime/event-bus.js";
+import type { Extension, ExtensionFactory, ExtensionRuntime, LoadExtensionsResult } from "./types.js";
+export declare function __getExtensionAliasesForTest(): Record<string, string>;
+export declare function __getExtensionVirtualModulesForTest(): Promise<Record<string, unknown>>;
+/**
+ * Create a runtime with throwing stubs for action methods.
+ * Runner.bindCore() replaces these with real implementations.
+ */
+export declare function createExtensionRuntime(): ExtensionRuntime;
+/**
+ * Create an Extension from an inline factory function.
+ */
+export declare function loadExtensionFromFactory(factory: ExtensionFactory, cwd: string, agentDir: string, eventBus: EventBus, runtime: ExtensionRuntime, extensionPath?: string): Promise<Extension>;
+/**
+ * Load extensions from paths.
+ */
+export declare function loadExtensions(paths: string[], cwd: string, agentDir: string, eventBus?: EventBus): Promise<LoadExtensionsResult>;
+/**
+ * Discover and load extensions from standard locations.
+ */
+export declare function discoverAndLoadExtensions(configuredPaths: string[], cwd: string, agentDir?: string, eventBus?: EventBus): Promise<LoadExtensionsResult>;

package/dist/core/extensions-host/loader.js

@@ -0,0 +1 @@
+var _=Object.defineProperty;var a=(t,n)=>_(t,"name",{value:n,configurable:!0});import*as d from"node:fs";import{createRequire as T}from"node:module";import*as x from"node:os";import*as l from"node:path";import{fileURLToPath as w}from"node:url";import{createJiti as M}from"@mariozechner/jiti";import*as F from"@catui/agent-core";import*as k from"@catui/ai";import*as L from"@catui/ai/env";import*as C from"@catui/ai/events";import*as N from"@catui/ai/json";import*as U from"@catui/ai/models";import*as I from"@catui/ai/oauth";import*as R from"@catui/ai/overflow";import*as D from"@catui/ai/registry";import*as B from"@catui/ai/schema";import*as O from"@catui/ai/stream";import*as z from"@catui/ai/types";import*as J from"@catui/tui";import*as V from"@sinclair/typebox";import{CONFIG_DIR_NAME as W,getAgentDir as $,isBunBinary as Q}from"../../config.js";import{createEventBus as q}from"../runtime/event-bus.js";import{execCommand as H}from"../platform/exec/exec.js";const G=["types","schema","events","models","registry","stream","env","overflow","json","oauth"],K={"@catui/ai/types":z,"@catui/ai/schema":B,"@catui/ai/events":C,"@catui/ai/models":U,"@catui/ai/registry":D,"@catui/ai/stream":O,"@catui/ai/env":L,"@catui/ai/overflow":R,"@catui/ai/json":N,"@catui/ai/oauth":I};async function v(){return{"@sinclair/typebox":V,"@catui/agent-core":F,"@catui/tui":J,"@catui/ai":k,...K,"@catui/agent":await import("../../index.js"),"@pencil-agent/nano-pencil":await import("../../index.js")}}a(v,"getVirtualModules");const g=T(import.meta.url);let h=null;function E(){if(h)return h;const t=l.dirname(w(import.meta.url)),n=l.resolve(t,"../..","index.js"),c=g.resolve("@sinclair/typebox").replace(/[\\/]build[\\/]cjs[\\/]index\.js$/,""),i=Object.fromEntries(G.map(r=>[`@catui/ai/${r}`,g.resolve(`@catui/ai/${r}`)]));return h={"@catui/agent":n,"@pencil-agent/nano-pencil":n,"@catui/agent-core":g.resolve("@catui/agent-core"),"@catui/tui":g.resolve("@catui/tui"),...i,"@catui/ai":g.resolve("@catui/ai"),"@sinclair/typebox":c},h}a(E,"getAliases");function pe(){return E()}a(pe,"__getExtensionAliasesForTest");async function me(){return v()}a(me,"__getExtensionVirtualModulesForTest");const X=/[\u00A0\u2000-\u200A\u202F\u205F\u3000]/g;function Y(t){return t.replace(X," ")}a(Y,"normalizeUnicodeSpaces");function Z(t){const n=Y(t);return n.startsWith("~/")?l.join(x.homedir(),n.slice(2)):n.startsWith("~")?l.join(x.homedir(),n.slice(1)):n}a(Z,"expandPath");function S(t,n){const o=Z(t);return l.isAbsolute(o)?o:l.resolve(n,o)}a(S,"resolvePath");function ee(){const t=a(()=>{throw new Error("Extension runtime not initialized. Action methods cannot be called during extension loading.")},"notInitialized");return{sendMessage:t,sendUserMessage:t,executeCommand:a(()=>Promise.reject(new Error("Extension runtime not initialized")),"executeCommand"),isIdle:t,clearFollowUpQueue:t,appendEntry:t,setSessionName:t,getSessionName:t,setLabel:t,getActiveTools:t,getAllTools:t,setActiveTools:t,getCommands:t,setModel:a(()=>Promise.reject(new Error("Extension runtime not initialized")),"setModel"),getThinkingLevel:t,setThinkingLevel:t,flagValues:new Map,pendingProviderRegistrations:[]}}a(ee,"createExtensionRuntime");function A(t,n,o,c,i){return{cwd:o,agentDir:c,on(e,s){const u=t.handlers.get(e)??[];u.push(s),t.handlers.set(e,u)},registerTool(e){t.tools.set(e.name,{definition:e,extensionPath:t.path})},registerCommand(e,s){t.commands.set(e,{name:e,...s})},registerShortcut(e,s){t.shortcuts.set(e,{shortcut:e,extensionPath:t.path,...s})},registerFlag(e,s){t.flags.set(e,{name:e,extensionPath:t.path,...s}),s.default!==void 0&&!n.flagValues.has(e)&&n.flagValues.set(e,s.default)},registerMessageRenderer(e,s){t.messageRenderers.set(e,s)},getFlag(e){if(t.flags.has(e))return n.flagValues.get(e)},sendMessage(e,s){n.sendMessage(e,s)},sendUserMessage(e,s){n.sendUserMessage(e,s)},executeCommand(e){return n.executeCommand(e)},isIdle(){return n.isIdle()},clearFollowUpQueue(){n.clearFollowUpQueue()},appendEntry(e,s){n.appendEntry(e,s)},setSessionName(e){n.setSessionName(e)},getSessionName(){return n.getSessionName()},setLabel(e,s){n.setLabel(e,s)},exec(e,s,u){return H(e,s,u?.cwd??o,u)},getActiveTools(){return n.getActiveTools()},getAllTools(){return n.getAllTools()},setActiveTools(e){n.setActiveTools(e)},getCommands(){return n.getCommands()},setModel(e){return n.setModel(e)},getThinkingLevel(){return n.getThinkingLevel()},setThinkingLevel(e){n.setThinkingLevel(e)},registerProvider(e,s){n.pendingProviderRegistrations.push({name:e,config:s})},events:i}}a(A,"createExtensionAPI");async function te(t){const c=await M(import.meta.url,{moduleCache:!1,...Q?{virtualModules:await v(),tryNative:!1}:{alias:E()}}).import(t,{default:!0});return typeof c!="function"?void 0:c}a(te,"loadExtensionModule");function b(t,n){return{path:t,resolvedPath:n,handlers:new Map,tools:new Map,messageRenderers:new Map,commands:new Map,flags:new Map,shortcuts:new Map}}a(b,"createExtension");async function ne(t,n,o,c,i){const r=S(t,n);try{const e=await te(r);if(!e)return{extension:null,error:`Extension does not export a valid factory function: ${t}`};const s=b(t,r),u=A(s,i,n,o,c);return await e(u),{extension:s,error:null}}catch(e){return{extension:null,error:`Failed to load extension: ${e instanceof Error?e.message:String(e)}`}}}a(ne,"loadExtension");async function ge(t,n,o,c,i,r="<inline>"){const e=b(r,r),s=A(e,i,n,o,c);return await t(s),e}a(ge,"loadExtensionFromFactory");async function se(t,n,o,c){const i=[],r=[],e=c??q(),s=ee(),u=await Promise.all(t.map(f=>ne(f,n,o,e,s).then(p=>({...p,path:f}))));for(const{extension:f,error:p,path:m}of u){if(p){r.push({path:m,error:p});continue}f&&i.push(f)}return{extensions:i,errors:r,runtime:s}}a(se,"loadExtensions");function P(t){try{const n=d.readFileSync(t,"utf-8"),o=JSON.parse(n);return o.catui&&typeof o.catui=="object"?o.catui:o.nanopencil&&typeof o.nanopencil=="object"?o.nanopencil:null}catch{return null}}a(P,"readCatuiManifest");function oe(t){return t.endsWith(".ts")||t.endsWith(".js")}a(oe,"isExtensionFile");function j(t){const n=l.join(t,"package.json");if(d.existsSync(n)){const i=P(n);if(i?.extensions?.length){const r=[];for(const e of i.extensions){const s=l.resolve(t,e);d.existsSync(s)&&r.push(s)}if(r.length>0)return r}}const o=l.join(t,"index.ts"),c=l.join(t,"index.js");return d.existsSync(o)?[o]:d.existsSync(c)?[c]:null}a(j,"resolveExtensionEntries");function y(t){if(!d.existsSync(t))return[];const n=[];try{const o=d.readdirSync(t,{withFileTypes:!0});for(const c of o){const i=l.join(t,c.name);if((c.isFile()||c.isSymbolicLink())&&oe(c.name)){n.push(i);continue}if(c.isDirectory()||c.isSymbolicLink()){const r=j(i);r&&n.push(...r)}}}catch{return[]}return n}a(y,"discoverExtensionsInDir");function ie(t){const n=[],o=a(i=>{const r=l.join(i,"package.json");if(!d.existsSync(r))return;const e=P(r);if(e?.extensions?.length)for(const s of e.extensions){const u=l.resolve(i,s);d.existsSync(u)&&n.push(u)}},"scanPackage"),c=a(i=>{if(!d.existsSync(i))return;let r;try{r=d.readdirSync(i,{withFileTypes:!0})}catch{return}for(const e of r)if(!(!e.isDirectory()&&!e.isSymbolicLink())&&!(e.name===".bin"||e.name===".cache")&&!(e.name==="@catui"||e.name==="@pencil-agent")){if(e.name.startsWith("@")){const s=l.join(i,e.name);let u;try{u=d.readdirSync(s,{withFileTypes:!0})}catch{continue}for(const f of u)(f.isDirectory()||f.isSymbolicLink())&&o(l.join(s,f.name));continue}o(l.join(i,e.name))}},"scanNodeModules");for(const i of t)c(l.join(i,"node_modules"));return n}a(ie,"discoverNpmExtensions");async function he(t,n,o=$(),c){const i=[],r=new Set,e=a(f=>{for(const p of f){const m=l.resolve(p);r.has(m)||(r.add(m),i.push(p))}},"addPaths"),s=l.join(n,W,"extensions");e(y(s));const u=l.join(o,"extensions");e(y(u));for(const f of t){const p=S(f,n);if(d.existsSync(p)&&d.statSync(p).isDirectory()){const m=j(p);if(m){e(m);continue}e(y(p));continue}e([p])}return e(ie([n,o])),se(i,n,o,c)}a(he,"discoverAndLoadExtensions");export{pe as __getExtensionAliasesForTest,me as __getExtensionVirtualModulesForTest,ee as createExtensionRuntime,he as discoverAndLoadExtensions,ge as loadExtensionFromFactory,se as loadExtensions};

package/dist/core/extensions-host/runner.d.ts

@@ -0,0 +1,233 @@
+/**
+ * [WHO]: ExtensionRunner class, lifecycle management, event emission, slash-command dispatch chokepoint (invokeCommand), telemetry sink wiring (setTelemetrySink)
+ * [FROM]: Depends on agent-core, ai, tui, modes/theme, session-manager, types.ts, core/platform/telemetry (ExtensionTelemetrySink + classifyArgsSignature for the P1 ext_command_events writer)
+ * [TO]: Consumed by core/extensions-host/index.ts, core/extensions-host/wrapper.ts, core/runtime/agent-session.ts (delegates command dispatch via invokeCommand)
+ * [HERE]: core/extensions-host/runner.ts - extension execution and lifecycle management; owns the single try/catch around command.handler so telemetry can wrap every invocation regardless of caller mode
+ */
+import type { AgentMessage } from "@catui/agent-core";
+import type { ImageContent } from "@catui/ai/types";
+import type { KeyId } from "@catui/tui";
+import type { ResourceDiagnostic } from "../platform/config/diagnostics.js";
+import type { KeybindingsConfig } from "../platform/keybindings.js";
+import type { ModelRegistry } from "../model-registry.js";
+import type { SessionManager } from "../session/session-manager.js";
+import { type ExtensionTelemetrySink, type LlmCallEventInput } from "../platform/telemetry/index.js";
+import type { BeforeAgentStartEvent, BeforeAgentStartEventResult, ContextEvent, Extension, ExtensionActions, ExtensionCommandContext, ExtensionCommandContextActions, ExtensionContext, ExtensionContextActions, ExtensionError, ExtensionEvent, ExtensionFlag, ExtensionRuntime, ExtensionShortcut, ExtensionUIContext, InputEvent, InputEventResult, InputSource, MessageRenderer, RegisteredCommand, RegisteredTool, ResourcesDiscoverEvent, SessionBeforeCompactResult, SessionBeforeForkResult, SessionBeforeSwitchResult, SessionBeforeTreeResult, ToolCallEvent, ToolCallEventResult, ToolResultEvent, ToolResultEventResult, UserBashEvent, UserBashEventResult } from "./types.js";
+/** Combined result from all before_agent_start handlers */
+interface BeforeAgentStartCombinedResult {
+    messages?: NonNullable<BeforeAgentStartEventResult["message"]>[];
+    systemPrompt?: string;
+}
+/**
+ * Events handled by the generic emit() method.
+ * Events with dedicated emitXxx() methods are excluded for stronger type safety.
+ */
+type RunnerEmitEvent = Exclude<ExtensionEvent, ToolCallEvent | ToolResultEvent | UserBashEvent | ContextEvent | BeforeAgentStartEvent | ResourcesDiscoverEvent | InputEvent>;
+type RunnerEmitResult<TEvent extends RunnerEmitEvent> = TEvent extends {
+    type: "session_before_switch";
+} ? SessionBeforeSwitchResult | undefined : TEvent extends {
+    type: "session_before_fork";
+} ? SessionBeforeForkResult | undefined : TEvent extends {
+    type: "session_before_compact";
+} ? SessionBeforeCompactResult | undefined : TEvent extends {
+    type: "session_before_tree";
+} ? SessionBeforeTreeResult | undefined : undefined;
+export type ExtensionErrorListener = (error: ExtensionError) => void;
+export type NewSessionHandler = (options?: {
+    parentSession?: string;
+    setup?: (sessionManager: SessionManager) => Promise<void>;
+}) => Promise<{
+    cancelled: boolean;
+}>;
+export type ForkHandler = (entryId: string) => Promise<{
+    cancelled: boolean;
+}>;
+export type NavigateTreeHandler = (targetId: string, options?: {
+    summarize?: boolean;
+    customInstructions?: string;
+    replaceInstructions?: boolean;
+    label?: string;
+}) => Promise<{
+    cancelled: boolean;
+}>;
+export type SwitchSessionHandler = (sessionPath: string) => Promise<{
+    cancelled: boolean;
+}>;
+export type ReloadHandler = () => Promise<void>;
+export type ShutdownHandler = () => void;
+/**
+ * Helper function to emit session_shutdown event to extensions.
+ * Returns true if the event was emitted, false if there were no handlers.
+ */
+export declare function emitSessionShutdownEvent(extensionRunner: ExtensionRunner | undefined): Promise<boolean>;
+export declare class ExtensionRunner {
+    private extensions;
+    private runtime;
+    private uiContext;
+    private cwd;
+    private agentDir;
+    private sessionManager;
+    private modelRegistry;
+    private errorListeners;
+    private getModel;
+    private completeSimpleFn;
+    private completeSimpleWithUsageFn;
+    private completeJsonFn;
+    private isIdleFn;
+    private waitForIdleFn;
+    private abortFn;
+    private clearFollowUpQueueFn;
+    private hasPendingMessagesFn;
+    private getContextUsageFn;
+    private compactFn;
+    private getSystemPromptFn;
+    private getSoulManagerFn;
+    private getSettingsFn;
+    private getSkillsFn;
+    private newSessionHandler;
+    private forkHandler;
+    private navigateTreeHandler;
+    private switchSessionHandler;
+    private reloadHandler;
+    private shutdownHandler;
+    private shortcutDiagnostics;
+    private commandDiagnostics;
+    private telemetrySink?;
+    private _beforeAgentStartTimeoutMs;
+    private get beforeAgentStartTimeoutMs();
+    private readonly beforeAgentStartTimeoutSentinel;
+    private beforeAgentStartTimeoutLastReported;
+    constructor(extensions: Extension[], runtime: ExtensionRuntime, cwd: string, agentDir: string, sessionManager: SessionManager, modelRegistry: ModelRegistry);
+    private withTimeout;
+    private reportBeforeAgentStartTimeout;
+    bindCore(actions: ExtensionActions, contextActions: ExtensionContextActions): void;
+    bindCommandContext(actions?: ExtensionCommandContextActions): void;
+    setUIContext(uiContext?: ExtensionUIContext): void;
+    getUIContext(): ExtensionUIContext;
+    hasUI(): boolean;
+    getExtensionPaths(): string[];
+    /** Get all registered tools from all extensions (first registration per name wins). */
+    getAllRegisteredTools(): RegisteredTool[];
+    /** Get a tool definition by name. Returns undefined if not found. */
+    getToolDefinition(toolName: string): RegisteredTool["definition"] | undefined;
+    getFlags(): Map<string, ExtensionFlag>;
+    setFlagValue(name: string, value: boolean | string): void;
+    getFlagValues(): Map<string, boolean | string>;
+    getShortcuts(effectiveKeybindings: Required<KeybindingsConfig>): Map<KeyId, ExtensionShortcut>;
+    getShortcutDiagnostics(): ResourceDiagnostic[];
+    onError(listener: ExtensionErrorListener): () => void;
+    emitError(error: ExtensionError): void;
+    /** Set the timeout for before_agent_start extension hooks (default: 1500ms). */
+    setBeforeAgentStartTimeout(ms: number): void;
+    /** Clean up all handlers, listeners, and diagnostics. Call on session shutdown. */
+    dispose(): void;
+    hasHandlers(eventType: string): boolean;
+    getMessageRenderer(customType: string): MessageRenderer | undefined;
+    getRegisteredCommands(reserved?: Set<string>): RegisteredCommand[];
+    getCommandDiagnostics(): ResourceDiagnostic[];
+    getRegisteredCommandsWithPaths(): Array<{
+        command: RegisteredCommand;
+        extensionPath: string;
+    }>;
+    getCommand(name: string): RegisteredCommand | undefined;
+    /**
+     * Returns the owning Extension for a given slash command name. Used by the
+     * telemetry middleware to stamp `extension_name` on each ext_command_events
+     * row. Falls back to undefined when no extension claims the command.
+     */
+    private findCommandOwner;
+    /**
+     * Derive a short, stable extension name from an Extension record. For
+     * built-ins (extensions/builtin/<name>) and most user extensions
+     * (packages/<name>) the directory basename is the right answer.
+     */
+    private deriveExtensionName;
+    /**
+     * Attach (or replace) the extension telemetry sink. The runner owns sink
+     * lifecycle from this point — invokeCommand() will fire-and-forget one
+     * `ext_command_events` row per invocation, and writeLlmCallEvent() (called
+     * from extension-core-bindings) writes one `ext_llm_calls` row per
+     * extension-initiated LLM call. Passing the noop sink (the factory's
+     * default when no insforge credentials exist) is the safe way to disable
+     * telemetry without scattering null-checks at the call sites.
+     */
+    setTelemetrySink(sink: ExtensionTelemetrySink): void;
+    /**
+     * Passthrough used by core/runtime/extension-core-bindings.ts after each
+     * extension-initiated LLM call. The runner owns the sink; the binding
+     * doesn't import it directly to keep its concerns scoped to LLM plumbing.
+     */
+    writeLlmCallEvent(input: LlmCallEventInput): void;
+    /**
+     * Wrap an extension hook handler invocation in three layers:
+     *
+     *   1. AsyncLocalStorage frame so any LLM call placed by the handler gets
+     *      attributed to this extension + hook + isUserInitiated=false in
+     *      ext_llm_calls.
+     *   2. Wall-clock timing measurement (only when the sample roll passes).
+     *   3. One fire-and-forget ext_hook_events row per sampled invocation,
+     *      capturing duration_ms + ok + error_code + sample_rate.
+     *
+     * High-frequency hooks (tool_*) are sampled at 10% so the table doesn't
+     * drown in tool-execution rows; the sample_rate column on each row lets
+     * dashboards extrapolate with `count(*) * 1/sample_rate`. Sampling decision
+     * sits inside the AsyncLocalStorage frame so even skipped-emit hooks still
+     * attribute their LLM calls.
+     */
+    private invokeHookHandler;
+    /**
+     * Single chokepoint for slash command dispatch. All modes (interactive /
+     * print / rpc / acp) funnel through agent-session._tryExecuteExtensionCommand,
+     * which calls this method. The wrapper measures wall-clock duration,
+     * captures outcome (ok / error / cancelled), and emits one telemetry row
+     * per invocation. Errors are still routed through emitError() so existing
+     * UI surfaces (toasts, logs) keep working unchanged.
+     *
+     * Returns `{ found: false }` when no extension owns the command, letting
+     * the caller fall through to built-in command handling without emitting a
+     * telemetry row for an unknown command.
+     */
+    invokeCommand(commandName: string, args: string, ctx: ExtensionCommandContext, metadata?: {
+        sessionId?: string | null;
+        runId?: string | null;
+        variant?: string | null;
+    }): Promise<{
+        found: boolean;
+    }>;
+    /**
+     * Request a graceful shutdown. Called by extension tools and event handlers.
+     * The actual shutdown behavior is provided by the mode via bindExtensions().
+     */
+    shutdown(): void;
+    /**
+     * Create an ExtensionContext for use in event handlers and tool execution.
+     * Context values are resolved at call time, so changes via bindCore/bindUI are reflected.
+     */
+    createContext(): ExtensionContext;
+    createCommandContext(): ExtensionCommandContext;
+    private isSessionBeforeEvent;
+    emit<TEvent extends RunnerEmitEvent>(event: TEvent): Promise<RunnerEmitResult<TEvent>>;
+    emitToolResult(event: ToolResultEvent): Promise<ToolResultEventResult | undefined>;
+    emitToolCall(event: ToolCallEvent): Promise<ToolCallEventResult | undefined>;
+    emitUserBash(event: UserBashEvent): Promise<UserBashEventResult | undefined>;
+    emitContext(messages: AgentMessage[]): Promise<AgentMessage[]>;
+    emitBeforeAgentStart(prompt: string, images: ImageContent[] | undefined, systemPrompt: string): Promise<BeforeAgentStartCombinedResult | undefined>;
+    private appendSystemPrompt;
+    emitResourcesDiscover(cwd: string, reason: ResourcesDiscoverEvent["reason"]): Promise<{
+        skillPaths: Array<{
+            path: string;
+            extensionPath: string;
+        }>;
+        promptPaths: Array<{
+            path: string;
+            extensionPath: string;
+        }>;
+        themePaths: Array<{
+            path: string;
+            extensionPath: string;
+        }>;
+    }>;
+    /** Emit input event. Transforms chain, "handled" short-circuits. */
+    emitInput(text: string, images: ImageContent[] | undefined, source: InputSource): Promise<InputEventResult>;
+}
+export {};

package/dist/core/extensions-host/runner.js

@@ -0,0 +1,3 @@
+var k=Object.defineProperty;var s=(f,e)=>k(f,"name",{value:e,configurable:!0});import{theme as E}from"../../modes/interactive/theme/theme.js";import{classifyArgsSignature as F,HOOK_SAMPLE_RATES as v,runWithExtCallerContext as x}from"../platform/telemetry/index.js";const C=["interrupt","clear","exit","suspend","cycleThinkingLevel","cycleModelForward","cycleModelBackward","selectModel","selectProviderThenModel","expandTools","toggleThinking","externalEditor","followUp","submit","selectConfirm","selectCancel","copy","deleteToLineEnd"],T=s(f=>{const e={};for(const[n,t]of Object.entries(f)){const o=n,i=Array.isArray(t)?t:[t],a=C.includes(o);for(const r of i){const l=r.toLowerCase();e[l]={action:o,restrictOverride:a}}}return e},"buildBuiltinKeybindings");async function _(f){return f?.hasHandlers("session_shutdown")?(await f.emit({type:"session_shutdown"}),!0):!1}s(_,"emitSessionShutdownEvent");const p={select:s(async()=>{},"select"),confirm:s(async()=>!1,"confirm"),input:s(async()=>{},"input"),notify:s(()=>{},"notify"),onTerminalInput:s(()=>()=>{},"onTerminalInput"),setStatus:s(()=>{},"setStatus"),setWorkingMessage:s(()=>{},"setWorkingMessage"),setWidget:s(()=>{},"setWidget"),setFooter:s(()=>{},"setFooter"),setHeader:s(()=>{},"setHeader"),setTitle:s(()=>{},"setTitle"),custom:s(async()=>{},"custom"),pasteToEditor:s(()=>{},"pasteToEditor"),setEditorText:s(()=>{},"setEditorText"),getEditorText:s(()=>"","getEditorText"),editor:s(async()=>{},"editor"),openExternalEditor:s(async()=>!1,"openExternalEditor"),setEditorComponent:s(()=>{},"setEditorComponent"),get theme(){return E},getAllThemes:s(()=>[],"getAllThemes"),getTheme:s(()=>{},"getTheme"),setTheme:s(f=>({success:!1,error:"UI not available"}),"setTheme"),getToolsExpanded:s(()=>!1,"getToolsExpanded"),setToolsExpanded:s(()=>{},"setToolsExpanded")};class H{static{s(this,"ExtensionRunner")}extensions;runtime;uiContext;cwd;agentDir;sessionManager;modelRegistry;errorListeners=new Set;getModel=s(()=>{},"getModel");completeSimpleFn=s(async()=>{},"completeSimpleFn");completeSimpleWithUsageFn=s(async()=>{},"completeSimpleWithUsageFn");completeJsonFn;isIdleFn=s(()=>!0,"isIdleFn");waitForIdleFn=s(async()=>{},"waitForIdleFn");abortFn=s(()=>{},"abortFn");clearFollowUpQueueFn=s(()=>{},"clearFollowUpQueueFn");hasPendingMessagesFn=s(()=>!1,"hasPendingMessagesFn");getContextUsageFn=s(()=>{},"getContextUsageFn");compactFn=s(()=>{},"compactFn");getSystemPromptFn=s(()=>"","getSystemPromptFn");getSoulManagerFn=s(()=>{},"getSoulManagerFn");getSettingsFn=s(()=>({}),"getSettingsFn");getSkillsFn=s(()=>[],"getSkillsFn");newSessionHandler=s(async()=>({cancelled:!1}),"newSessionHandler");forkHandler=s(async()=>({cancelled:!1}),"forkHandler");navigateTreeHandler=s(async()=>({cancelled:!1}),"navigateTreeHandler");switchSessionHandler=s(async()=>({cancelled:!1}),"switchSessionHandler");reloadHandler=s(async()=>{},"reloadHandler");shutdownHandler=s(()=>{},"shutdownHandler");shortcutDiagnostics=[];commandDiagnostics=[];telemetrySink;_beforeAgentStartTimeoutMs=1500;get beforeAgentStartTimeoutMs(){return this._beforeAgentStartTimeoutMs??1500}beforeAgentStartTimeoutSentinel=Symbol("before_agent_start_timeout");beforeAgentStartTimeoutLastReported=new Map;constructor(e,n,t,o,i,a){this.extensions=e,this.runtime=n,this.uiContext=p,this.cwd=t,this.agentDir=o,this.sessionManager=i,this.modelRegistry=a}async withTimeout(e,n){return new Promise(t=>{const o=setTimeout(()=>t(this.beforeAgentStartTimeoutSentinel),n);Promise.resolve(e).then(i=>{clearTimeout(o),t(i)}).catch(()=>{clearTimeout(o),t(this.beforeAgentStartTimeoutSentinel)})})}reportBeforeAgentStartTimeout(e){const n=Date.now(),t=this.beforeAgentStartTimeoutLastReported.get(e)??0;n-t<6e4||(this.beforeAgentStartTimeoutLastReported.set(e,n),console.warn(`Extension before_agent_start timed out (${this.beforeAgentStartTimeoutMs}ms): ${e}`))}bindCore(e,n){this.runtime.sendMessage=e.sendMessage,this.runtime.sendUserMessage=e.sendUserMessage,this.runtime.executeCommand=e.executeCommand,this.runtime.appendEntry=e.appendEntry,this.runtime.setSessionName=e.setSessionName,this.runtime.getSessionName=e.getSessionName,this.runtime.setLabel=e.setLabel,this.runtime.getActiveTools=e.getActiveTools,this.runtime.getAllTools=e.getAllTools,this.runtime.setActiveTools=e.setActiveTools,this.runtime.getCommands=e.getCommands,this.runtime.setModel=e.setModel,this.runtime.getThinkingLevel=e.getThinkingLevel,this.runtime.setThinkingLevel=e.setThinkingLevel,this.getModel=n.getModel,this.completeSimpleFn=n.completeSimple,this.completeSimpleWithUsageFn=n.completeSimpleWithUsage,this.completeJsonFn=n.completeJson,this.isIdleFn=n.isIdle,this.runtime.isIdle=()=>this.isIdleFn(),this.abortFn=n.abort,this.clearFollowUpQueueFn=n.clearFollowUpQueue,this.runtime.clearFollowUpQueue=()=>this.clearFollowUpQueueFn(),this.hasPendingMessagesFn=n.hasPendingMessages,this.shutdownHandler=n.shutdown,this.getContextUsageFn=n.getContextUsage,this.compactFn=n.compact,this.getSystemPromptFn=n.getSystemPrompt,this.getSoulManagerFn=n.getSoulManager,this.getSettingsFn=n.getSettings,this.getSkillsFn=n.getSkills;for(const{name:t,config:o}of this.runtime.pendingProviderRegistrations)this.modelRegistry.registerProvider(t,o);this.runtime.pendingProviderRegistrations=[]}bindCommandContext(e){if(e){this.waitForIdleFn=e.waitForIdle,this.newSessionHandler=e.newSession,this.forkHandler=e.fork,this.navigateTreeHandler=e.navigateTree,this.switchSessionHandler=e.switchSession,this.reloadHandler=e.reload;return}this.waitForIdleFn=async()=>{},this.newSessionHandler=async()=>({cancelled:!1}),this.forkHandler=async()=>({cancelled:!1}),this.navigateTreeHandler=async()=>({cancelled:!1}),this.switchSessionHandler=async()=>({cancelled:!1}),this.reloadHandler=async()=>{}}setUIContext(e){this.uiContext=e??p}getUIContext(){return this.uiContext}hasUI(){return this.uiContext===p?!1:this.uiContext.__nonInteractive!==!0}getExtensionPaths(){return this.extensions.map(e=>e.path)}getAllRegisteredTools(){const e=new Map;for(const n of this.extensions)for(const t of n.tools.values())e.has(t.definition.name)||e.set(t.definition.name,t);return Array.from(e.values())}getToolDefinition(e){for(const n of this.extensions){const t=n.tools.get(e);if(t)return t.definition}}getFlags(){const e=new Map;for(const n of this.extensions)for(const[t,o]of n.flags)e.has(t)||e.set(t,o);return e}setFlagValue(e,n){this.runtime.flagValues.set(e,n)}getFlagValues(){return new Map(this.runtime.flagValues)}getShortcuts(e){this.shortcutDiagnostics=[];const n=T(e),t=new Map,o=s((i,a)=>{this.shortcutDiagnostics.push({type:"warning",message:i,path:a}),this.hasUI()||console.warn(i)},"addDiagnostic");for(const i of this.extensions)for(const[a,r]of i.shortcuts){const l=a.toLowerCase(),c=n[l];if(c?.restrictOverride===!0){o(`Extension shortcut '${a}' from ${r.extensionPath} conflicts with built-in shortcut. Skipping.`,r.extensionPath);continue}c?.restrictOverride===!1&&o(`Extension shortcut conflict: '${a}' is built-in shortcut for ${c.action} and ${r.extensionPath}. Using ${r.extensionPath}.`,r.extensionPath);const h=t.get(l);h&&o(`Extension shortcut conflict: '${a}' registered by both ${h.extensionPath} and ${r.extensionPath}. Using ${r.extensionPath}.`,r.extensionPath),t.set(l,r)}return t}getShortcutDiagnostics(){return this.shortcutDiagnostics}onError(e){return this.errorListeners.add(e),()=>this.errorListeners.delete(e)}emitError(e){for(const n of this.errorListeners)n(e)}setBeforeAgentStartTimeout(e){this._beforeAgentStartTimeoutMs=e}dispose(){this.errorListeners.clear(),this.beforeAgentStartTimeoutLastReported.clear(),this.shortcutDiagnostics=[],this.commandDiagnostics=[];for(const e of this.extensions)e.handlers.clear()}hasHandlers(e){for(const n of this.extensions){const t=n.handlers.get(e);if(t&&t.length>0)return!0}return!1}getMessageRenderer(e){for(const n of this.extensions){const t=n.messageRenderers.get(e);if(t)return t}}getRegisteredCommands(e){this.commandDiagnostics=[];const n=[],t=new Map;for(const o of this.extensions)for(const i of o.commands.values()){if(e?.has(i.name)){const r=`Extension command '${i.name}' from ${o.path} conflicts with built-in commands. Skipping.`;this.commandDiagnostics.push({type:"warning",message:r,path:o.path}),this.hasUI()||console.warn(r);continue}const a=t.get(i.name);if(a){const r=`Extension command '${i.name}' from ${o.path} conflicts with ${a}. Skipping.`;this.commandDiagnostics.push({type:"warning",message:r,path:o.path}),this.hasUI()||console.warn(r);continue}t.set(i.name,o.path),n.push(i)}return n}getCommandDiagnostics(){return this.commandDiagnostics}getRegisteredCommandsWithPaths(){const e=[];for(const n of this.extensions)for(const t of n.commands.values())e.push({command:t,extensionPath:n.path});return e}getCommand(e){for(const n of this.extensions){const t=n.commands.get(e);if(t)return t}}findCommandOwner(e){for(const n of this.extensions)if(n.commands.has(e))return n}deriveExtensionName(e){const t=(e.path||e.resolvedPath||"").replace(/\/+$/,"").split(/[\\/]/);return t[t.length-1]||"unknown"}setTelemetrySink(e){this.telemetrySink=e}writeLlmCallEvent(e){try{this.telemetrySink?.writeLlmCallEvent(e)}catch{}}invokeHookHandler(e,n,t){const o=this.deriveExtensionName(e),i={extensionName:o,callerContext:`hook:${n}`,isUserInitiated:!1};return x(i,async()=>{const a=v[n]??1;if(a<1&&Math.random()>=a)return await t();const r=new Date,l=performance.now();let c=!0,h=null;try{return await t()}catch(u){throw c=!1,h=u instanceof Error?u.constructor.name:"unknown",u}finally{try{this.telemetrySink?.writeHookEvent({extensionName:o,hookName:n,durationMs:Math.round(performance.now()-l),ok:c,errorCode:h,sampleRate:a,recordedAt:r})}catch{}}})}async invokeCommand(e,n,t,o){const i=this.getCommand(e);if(!i)return{found:!1};const a=this.findCommandOwner(e),r=a?this.deriveExtensionName(a):"unknown",l=new Date,c=performance.now();let h="ok",u=null,m;const d=F(n),y={extensionName:r,callerContext:d==="no-args"?`command:/${e}`:`command:/${e} ${d}`,isUserInitiated:!0,sessionId:o?.sessionId??null,runId:o?.runId??null,variant:o?.variant??null};try{await x(y,()=>i.handler(n,t))}catch(g){h="error",u=g instanceof Error?g.constructor.name:"unknown",m=g,this.emitError({extensionPath:`command:${e}`,event:"command",error:g instanceof Error?g.message:String(g)})}const S=Math.round(performance.now()-c),w=new Date;try{this.telemetrySink?.writeCommandEvent({extensionName:r,commandName:e,argsSignature:d,argsLength:n.length,outcome:h,errorCode:u,durationMs:S,startedAt:l,endedAt:w,sessionId:o?.sessionId??null,runId:o?.runId??null,variant:o?.variant??null})}catch{}return{found:!0}}shutdown(){this.shutdownHandler()}createContext(){const e=this.getModel;return{ui:this.uiContext,hasUI:this.hasUI(),cwd:this.cwd,agentDir:this.agentDir,sessionManager:this.sessionManager,modelRegistry:this.modelRegistry,get model(){return e()},completeSimple:s((n,t)=>this.completeSimpleFn(n,t),"completeSimple"),completeSimpleWithUsage:s((n,t)=>this.completeSimpleWithUsageFn(n,t),"completeSimpleWithUsage"),completeJson:this.completeJsonFn?(n,t,o,i)=>this.completeJsonFn(n,t,o,i):void 0,isIdle:s(()=>this.isIdleFn(),"isIdle"),abort:s(()=>this.abortFn(),"abort"),clearFollowUpQueue:s(()=>this.clearFollowUpQueueFn(),"clearFollowUpQueue"),hasPendingMessages:s(()=>this.hasPendingMessagesFn(),"hasPendingMessages"),shutdown:s(()=>this.shutdownHandler(),"shutdown"),getContextUsage:s(()=>this.getContextUsageFn(),"getContextUsage"),compact:s(n=>this.compactFn(n),"compact"),getSystemPrompt:s(()=>this.getSystemPromptFn(),"getSystemPrompt"),getSoulManager:s(()=>this.getSoulManagerFn(),"getSoulManager"),getSettings:s(()=>this.getSettingsFn(),"getSettings"),getSkills:s(()=>this.getSkillsFn(),"getSkills")}}createCommandContext(){return{...this.createContext(),waitForIdle:s(()=>this.waitForIdleFn(),"waitForIdle"),newSession:s(e=>this.newSessionHandler(e),"newSession"),fork:s(e=>this.forkHandler(e),"fork"),navigateTree:s((e,n)=>this.navigateTreeHandler(e,n),"navigateTree"),switchSession:s(e=>this.switchSessionHandler(e),"switchSession"),reload:s(()=>this.reloadHandler(),"reload")}}isSessionBeforeEvent(e){return e.type==="session_before_switch"||e.type==="session_before_fork"||e.type==="session_before_compact"||e.type==="session_before_tree"}async emit(e){const n=this.createContext();let t;for(const o of this.extensions){const i=o.handlers.get(e.type);if(!(!i||i.length===0))for(const a of i)try{const r=await this.invokeHookHandler(o,e.type,()=>a(e,n));if(this.isSessionBeforeEvent(e)&&r&&(t=r,t.cancel))return t}catch(r){const l=r instanceof Error?r.message:String(r),c=r instanceof Error?r.stack:void 0;this.emitError({extensionPath:o.path,event:e.type,error:l,stack:c})}}return t}async emitToolResult(e){const n=this.createContext(),t={...e};let o=!1;for(const i of this.extensions){const a=i.handlers.get("tool_result");if(!(!a||a.length===0))for(const r of a)try{const l=await this.invokeHookHandler(i,"tool_result",()=>r(t,n));if(!l)continue;l.content!==void 0&&(t.content=l.content,o=!0),l.details!==void 0&&(t.details=l.details,o=!0),l.isError!==void 0&&(t.isError=l.isError,o=!0)}catch(l){const c=l instanceof Error?l.message:String(l),h=l instanceof Error?l.stack:void 0;this.emitError({extensionPath:i.path,event:"tool_result",error:c,stack:h})}}if(o)return{content:t.content,details:t.details,isError:t.isError}}async emitToolCall(e){const n=this.createContext();let t;for(const o of this.extensions){const i=o.handlers.get("tool_call");if(!(!i||i.length===0))for(const a of i){const r=await this.invokeHookHandler(o,"tool_call",()=>a(e,n));if(r&&(t=r,t.block))return t}}return t}async emitUserBash(e){const n=this.createContext();for(const t of this.extensions){const o=t.handlers.get("user_bash");if(!(!o||o.length===0))for(const i of o)try{const a=await this.invokeHookHandler(t,"user_bash",()=>i(e,n));if(a)return a}catch(a){const r=a instanceof Error?a.message:String(a),l=a instanceof Error?a.stack:void 0;this.emitError({extensionPath:t.path,event:"user_bash",error:r,stack:l})}}}async emitContext(e){const n=this.createContext();let t=structuredClone(e);for(const o of this.extensions){const i=o.handlers.get("context");if(!(!i||i.length===0))for(const a of i)try{const r={type:"context",messages:t},l=await this.invokeHookHandler(o,"context",()=>a(r,n));l&&l.messages&&(t=l.messages)}catch(r){const l=r instanceof Error?r.message:String(r),c=r instanceof Error?r.stack:void 0;this.emitError({extensionPath:o.path,event:"context",error:l,stack:c})}}return t}async emitBeforeAgentStart(e,n,t){const o=this.createContext(),i=[];let a=t,r=!1;for(const l of this.extensions){const c=l.handlers.get("before_agent_start");if(!(!c||c.length===0))for(const h of c)try{const u={type:"before_agent_start",prompt:e,images:n,systemPrompt:a},m=await this.withTimeout(this.invokeHookHandler(l,"before_agent_start",()=>h(u,o)),this.beforeAgentStartTimeoutMs);if(m===this.beforeAgentStartTimeoutSentinel){this.reportBeforeAgentStartTimeout(l.path);continue}if(m){const d=m;d.message&&i.push(d.message),d.systemPrompt!==void 0&&(a=d.systemPrompt,r=!0),d.appendSystemPrompt!==void 0&&(a=this.appendSystemPrompt(a,d.appendSystemPrompt),r=!0)}}catch(u){const m=u instanceof Error?u.message:String(u),d=u instanceof Error?u.stack:void 0;this.emitError({extensionPath:l.path,event:"before_agent_start",error:m,stack:d})}}if(i.length>0||r)return{messages:i.length>0?i:void 0,systemPrompt:r?a:void 0}}appendSystemPrompt(e,n){const t=n.trim();return t?`${e}
+
+${t}`:e}async emitResourcesDiscover(e,n){const t=this.createContext(),o=[],i=[],a=[];for(const r of this.extensions){const l=r.handlers.get("resources_discover");if(!(!l||l.length===0))for(const c of l)try{const h={type:"resources_discover",cwd:e,reason:n},m=await this.invokeHookHandler(r,"resources_discover",()=>c(h,t));m?.skillPaths?.length&&o.push(...m.skillPaths.map(d=>({path:d,extensionPath:r.path}))),m?.promptPaths?.length&&i.push(...m.promptPaths.map(d=>({path:d,extensionPath:r.path}))),m?.themePaths?.length&&a.push(...m.themePaths.map(d=>({path:d,extensionPath:r.path})))}catch(h){const u=h instanceof Error?h.message:String(h),m=h instanceof Error?h.stack:void 0;this.emitError({extensionPath:r.path,event:"resources_discover",error:u,stack:m})}}return{skillPaths:o,promptPaths:i,themePaths:a}}async emitInput(e,n,t){const o=this.createContext();let i=e,a=n;for(const r of this.extensions)for(const l of r.handlers.get("input")??[])try{const c={type:"input",text:i,images:a,source:t},h=await this.invokeHookHandler(r,"input",()=>l(c,o));if(h?.action==="handled")return h;h?.action==="transform"&&(i=h.text,a=h.images??a)}catch(c){this.emitError({extensionPath:r.path,event:"input",error:c instanceof Error?c.message:String(c),stack:c instanceof Error?c.stack:void 0})}return i!==e||a!==n?{action:"transform",text:i,images:a}:{action:"continue"}}}export{H as ExtensionRunner,_ as emitSessionShutdownEvent};