@astralyn/pi 0.85.3 → 0.86.0

package/dist/core/extensions/runner.js

@@ -3,6 +3,7 @@
  */
 import { theme } from "../../modes/interactive/theme/theme.js";
 import { BackgroundService } from "../background/service.js";
+import { buildSystemPrompt, normalizeBuildSystemPromptOptions, } from "../system-prompt.js";
 // Extension shortcuts compete with canonical keybinding ids from keybindings.json.
 // Only editor-global shortcuts are reserved here. Picker-specific bindings are not.
 const RESERVED_KEYBINDINGS_FOR_EXTENSION_CONFLICTS = [
@@ -48,6 +49,31 @@ const buildBuiltinKeybindings = (resolvedKeybindings) => {
     }
     return builtinKeybindings;
 };
+function isUserBashEventResult(value) {
+    if (typeof value !== "object" || value === null)
+        return false;
+    const candidate = value;
+    const hasOperations = candidate.operations !== undefined;
+    const hasResult = candidate.result !== undefined;
+    if (hasOperations === hasResult)
+        return false;
+    if (hasOperations) {
+        const operations = candidate.operations;
+        if (typeof operations !== "object" || operations === null)
+            return false;
+        return typeof operations.exec === "function";
+    }
+    const result = candidate.result;
+    if (typeof result !== "object" || result === null)
+        return false;
+    const resultRecord = result;
+    return (typeof resultRecord.output === "string" &&
+        "exitCode" in resultRecord &&
+        (resultRecord.exitCode === undefined || typeof resultRecord.exitCode === "number") &&
+        typeof resultRecord.cancelled === "boolean" &&
+        typeof resultRecord.truncated === "boolean" &&
+        (resultRecord.fullOutputPath === undefined || typeof resultRecord.fullOutputPath === "string"));
+}
 /**
  * Helper function to emit session_shutdown event to extensions.
  * Returns true if the event was emitted, false if there were no handlers.
@@ -59,14 +85,14 @@ export async function emitSessionShutdownEvent(extensionRunner, event) {
     }
     return false;
 }
+function snapshotEventHandlers(extensions, event) {
+    return extensions.map((ext) => ({ ext, handlers: ext.handlers.get(event)?.slice() ?? [] }));
+}
 export async function emitProjectTrustEvent(extensionsResult, event, ctx) {
     const errors = [];
-    for (const ext of extensionsResult.extensions) {
+    for (const { ext, handlers } of snapshotEventHandlers(extensionsResult.extensions, "project_trust")) {
         // A single extension may register multiple handlers for the same event.
         // The first project_trust handler that returns yes/no wins; undecided falls through.
-        const handlers = ext.handlers.get("project_trust");
-        if (!handlers || handlers.length === 0)
-            continue;
         for (const handler of handlers) {
             try {
                 const handlerResult = (await handler(event, ctx));
@@ -141,7 +167,7 @@ export class ExtensionRunner {
         };
         this.compactFn = () => { };
         this.getSystemPromptFn = () => "";
-        this.getSystemPromptOptionsFn = () => ({ cwd: this.cwd });
+        this.getSystemPromptOptionsFn = () => normalizeBuildSystemPromptOptions({ cwd: this.cwd });
         this.newSessionHandler = async () => ({ cancelled: false });
         this.forkHandler = async () => ({ cancelled: false });
         this.navigateTreeHandler = async () => ({ cancelled: false });
@@ -189,7 +215,8 @@ export class ExtensionRunner {
             this.getContextSnapshotFn = contextActions.getContextSnapshot;
         this.compactFn = contextActions.compact;
         this.getSystemPromptFn = contextActions.getSystemPrompt;
-        this.getSystemPromptOptionsFn = contextActions.getSystemPromptOptions ?? (() => ({ cwd: this.cwd }));
+        this.getSystemPromptOptionsFn =
+            contextActions.getSystemPromptOptions ?? (() => normalizeBuildSystemPromptOptions({ cwd: this.cwd }));
         // Flush provider registrations queued during extension loading
         for (const { name, config, extensionPath } of this.runtime.pendingProviderRegistrations) {
             try {
@@ -640,10 +667,7 @@ export class ExtensionRunner {
     async emit(event) {
         const ctx = this.createContext();
         let result;
-        for (const ext of this.extensions) {
-            const handlers = ext.handlers.get(event.type);
-            if (!handlers || handlers.length === 0)
-                continue;
+        for (const { ext, handlers } of snapshotEventHandlers(this.extensions, event.type)) {
             for (const handler of handlers) {
                 try {
                     const handlerResult = await handler(event, ctx);
@@ -668,14 +692,34 @@ export class ExtensionRunner {
         }
         return result;
     }
+    /** Returns the event's own action unless a handler overrides it; the last override wins. */
+    async emitCacheWarmingDecision(event) {
+        const ctx = this.createContext();
+        let action = event.action;
+        for (const { ext, handlers } of snapshotEventHandlers(this.extensions, event.type)) {
+            for (const handler of handlers) {
+                try {
+                    const result = (await handler(event, ctx));
+                    if (result?.action !== undefined)
+                        action = result.action;
+                }
+                catch (err) {
+                    this.emitError({
+                        extensionPath: ext.path,
+                        event: event.type,
+                        error: err instanceof Error ? err.message : String(err),
+                        stack: err instanceof Error ? err.stack : undefined,
+                    });
+                }
+            }
+        }
+        return action;
+    }
     async emitMessageEnd(event) {
         const ctx = this.createContext();
         let currentMessage = event.message;
         let modified = false;
-        for (const ext of this.extensions) {
-            const handlers = ext.handlers.get("message_end");
-            if (!handlers || handlers.length === 0)
-                continue;
+        for (const { ext, handlers } of snapshotEventHandlers(this.extensions, "message_end")) {
             for (const handler of handlers) {
                 try {
                     const currentEvent = { ...event, message: currentMessage };
@@ -711,10 +755,7 @@ export class ExtensionRunner {
         const ctx = this.createContext();
         const currentEvent = { ...event };
         let modified = false;
-        for (const ext of this.extensions) {
-            const handlers = ext.handlers.get("tool_result");
-            if (!handlers || handlers.length === 0)
-                continue;
+        for (const { ext, handlers } of snapshotEventHandlers(this.extensions, "tool_result")) {
             for (const handler of handlers) {
                 try {
                     const handlerResult = (await handler(currentEvent, ctx));
@@ -762,10 +803,7 @@ export class ExtensionRunner {
     async emitToolCall(event) {
         const ctx = this.createContext();
         let result;
-        for (const ext of this.extensions) {
-            const handlers = ext.handlers.get("tool_call");
-            if (!handlers || handlers.length === 0)
-                continue;
+        for (const { handlers } of snapshotEventHandlers(this.extensions, "tool_call")) {
             for (const handler of handlers) {
                 const handlerResult = await handler(event, ctx);
                 if (handlerResult) {
@@ -780,16 +818,16 @@ export class ExtensionRunner {
     }
     async emitUserBash(event) {
         const ctx = this.createContext();
-        for (const ext of this.extensions) {
-            const handlers = ext.handlers.get("user_bash");
-            if (!handlers || handlers.length === 0)
-                continue;
+        for (const { ext, handlers } of snapshotEventHandlers(this.extensions, "user_bash")) {
             for (const handler of handlers) {
                 try {
                     const handlerResult = await handler(event, ctx);
-                    if (handlerResult) {
-                        return handlerResult;
+                    if (handlerResult === undefined)
+                        continue;
+                    if (!isUserBashEventResult(handlerResult)) {
+                        throw new Error("Invalid user_bash handler result: return undefined for local execution or exactly one valid { operations } or { result } object");
                     }
+                    return handlerResult;
                 }
                 catch (err) {
                     const message = err instanceof Error ? err.message : String(err);
@@ -800,6 +838,7 @@ export class ExtensionRunner {
                         error: message,
                         stack,
                     });
+                    throw err;
                 }
             }
         }
@@ -808,10 +847,7 @@ export class ExtensionRunner {
     async emitContext(messages) {
         const ctx = this.createContext();
         let currentMessages = structuredClone(messages);
-        for (const ext of this.extensions) {
-            const handlers = ext.handlers.get("context");
-            if (!handlers || handlers.length === 0)
-                continue;
+        for (const { ext, handlers } of snapshotEventHandlers(this.extensions, "context")) {
             for (const handler of handlers) {
                 try {
                     const event = { type: "context", messages: currentMessages };
@@ -837,10 +873,7 @@ export class ExtensionRunner {
     async emitBeforeProviderRequest(payload) {
         const ctx = this.createContext();
         let currentPayload = payload;
-        for (const ext of this.extensions) {
-            const handlers = ext.handlers.get("before_provider_request");
-            if (!handlers || handlers.length === 0)
-                continue;
+        for (const { ext, handlers } of snapshotEventHandlers(this.extensions, "before_provider_request")) {
             for (const handler of handlers) {
                 try {
                     const event = {
@@ -868,10 +901,7 @@ export class ExtensionRunner {
     }
     async emitBeforeProviderHeaders(headers) {
         const ctx = this.createContext();
-        for (const ext of this.extensions) {
-            const handlers = ext.handlers.get("before_provider_headers");
-            if (!handlers || handlers.length === 0)
-                continue;
+        for (const { ext, handlers } of snapshotEventHandlers(this.extensions, "before_provider_headers")) {
             for (const handler of handlers) {
                 try {
                     // Handlers mutate `headers` in place; the return value is ignored.
@@ -895,37 +925,34 @@ export class ExtensionRunner {
         }
         return headers;
     }
-    async emitBeforeAgentStart(prompt, images, systemPrompt, systemPromptOptions) {
-        let currentSystemPrompt = systemPrompt;
+    async emitBeforeAgentStart(prompt, images, systemPromptOptions) {
+        const currentOptions = normalizeBuildSystemPromptOptions(systemPromptOptions);
+        const renderCurrentSystemPrompt = () => buildSystemPrompt(currentOptions);
         const ctx = Object.defineProperties({}, Object.getOwnPropertyDescriptors(this.createContext()));
         ctx.getSystemPrompt = () => {
             this.assertActive();
-            return currentSystemPrompt;
+            return renderCurrentSystemPrompt();
         };
         const messages = [];
-        let systemPromptModified = false;
-        for (const ext of this.extensions) {
-            const handlers = ext.handlers.get("before_agent_start");
-            if (!handlers || handlers.length === 0)
-                continue;
+        for (const { ext, handlers } of snapshotEventHandlers(this.extensions, "before_agent_start")) {
             for (const handler of handlers) {
                 try {
                     const event = {
                         type: "before_agent_start",
                         prompt,
                         images,
-                        systemPrompt: currentSystemPrompt,
-                        systemPromptOptions,
+                        get systemPrompt() {
+                            return renderCurrentSystemPrompt();
+                        },
+                        systemPromptOptions: currentOptions,
                     };
                     const handlerResult = await handler(event, ctx);
                     if (handlerResult) {
                         const result = handlerResult;
-                        if (result.message) {
+                        if (result.message)
                             messages.push(result.message);
-                        }
                         if (result.systemPrompt !== undefined) {
-                            currentSystemPrompt = result.systemPrompt;
-                            systemPromptModified = true;
+                            currentOptions.forceSystemPrompt = result.systemPrompt;
                         }
                     }
                 }
@@ -941,23 +968,14 @@ export class ExtensionRunner {
                 }
             }
         }
-        if (messages.length > 0 || systemPromptModified) {
-            return {
-                messages: messages.length > 0 ? messages : undefined,
-                systemPrompt: systemPromptModified ? currentSystemPrompt : undefined,
-            };
-        }
-        return undefined;
+        return { messages, systemPromptOptions: currentOptions };
     }
     async emitResourcesDiscover(cwd, reason) {
         const ctx = this.createContext();
         const skillPaths = [];
         const promptPaths = [];
         const themePaths = [];
-        for (const ext of this.extensions) {
-            const handlers = ext.handlers.get("resources_discover");
-            if (!handlers || handlers.length === 0)
-                continue;
+        for (const { ext, handlers } of snapshotEventHandlers(this.extensions, "resources_discover")) {
             for (const handler of handlers) {
                 try {
                     const event = { type: "resources_discover", cwd, reason };
@@ -992,8 +1010,8 @@ export class ExtensionRunner {
         const ctx = this.createContext();
         let currentText = text;
         let currentImages = images;
-        for (const ext of this.extensions) {
-            for (const handler of ext.handlers.get("input") ?? []) {
+        for (const { ext, handlers } of snapshotEventHandlers(this.extensions, "input")) {
+            for (const handler of handlers) {
                 try {
                     const event = {
                         type: "input",