@codex-infinity/pi-infinity 0.64.2 → 0.65.1

package/dist/core/agent-session-runtime.d.ts

@@ -1,134 +1,83 @@
-import type { ThinkingLevel } from "@mariozechner/pi-agent-core";
-import type { Model } from "@mariozechner/pi-ai";
-import { AuthStorage } from "./auth-storage.js";
-import type { SessionStartEvent, ToolDefinition } from "./extensions/index.js";
-import { ModelRegistry } from "./model-registry.js";
-import { type DefaultResourceLoaderOptions, type ResourceLoader } from "./resource-loader.js";
-import { type CreateAgentSessionResult } from "./sdk.js";
+import type { AgentSession } from "./agent-session.js";
+import type { AgentSessionRuntimeDiagnostic, AgentSessionServices } from "./agent-session-services.js";
+import type { SessionStartEvent } from "./extensions/index.js";
+import type { CreateAgentSessionResult } from "./sdk.js";
 import { SessionManager } from "./session-manager.js";
-import { SettingsManager } from "./settings-manager.js";
-import type { Tool } from "./tools/index.js";
 /**
- * Stable bootstrap inputs reused whenever the active runtime is replaced.
+ * Result returned by runtime creation.
  *
- * Use this for process-level wiring that should survive `/new`, `/resume`,
- * `/fork`, and import flows. Session-local state belongs in the session file
- * or in settings, not here.
+ * The caller gets the created session, its cwd-bound services, and all
+ * diagnostics collected during setup.
  */
-export interface AgentSessionRuntimeBootstrap {
-    /** Agent directory used for auth, models, settings, sessions, and resource discovery. */
-    agentDir?: string;
-    /** Optional shared auth storage. If omitted, file-backed storage under agentDir is used. */
-    authStorage?: AuthStorage;
-    /** Initial model for the first created session runtime. */
-    model?: Model<any>;
-    /** Initial thinking level for the first created session runtime. */
-    thinkingLevel?: ThinkingLevel;
-    /** Optional scoped model list for model cycling and selection. */
-    scopedModels?: Array<{
-        model: Model<any>;
-        thinkingLevel?: ThinkingLevel;
-    }>;
-    /** Built-in tool set override. */
-    tools?: Tool[];
-    /** Additional custom tools registered directly through the SDK. */
-    customTools?: ToolDefinition[];
-    /**
-     * Resource loader input used for each created runtime.
-     *
-     * Pass either a factory that creates a fully custom ResourceLoader for the
-     * target cwd, or DefaultResourceLoader options without cwd/agentDir/
-     * settingsManager, which are supplied by the runtime.
-     */
-    resourceLoader?: ((cwd: string, agentDir: string) => Promise<ResourceLoader>) | Omit<DefaultResourceLoaderOptions, "cwd" | "agentDir" | "settingsManager">;
+export interface CreateAgentSessionRuntimeResult extends CreateAgentSessionResult {
+    services: AgentSessionServices;
+    diagnostics: AgentSessionRuntimeDiagnostic[];
 }
-/** Options for creating one concrete runtime instance. */
-export interface CreateAgentSessionRuntimeOptions {
-    /** Working directory for this runtime instance. */
-    cwd: string;
-    /** Optional preselected session manager. If omitted, normal session resolution applies. */
-    sessionManager?: SessionManager;
-    /** Optional session_start metadata to emit when the runtime binds extensions. */
-    sessionStartEvent?: SessionStartEvent;
-}
-type AgentSessionRuntime = CreateAgentSessionResult & {
-    cwd: string;
-    agentDir: string;
-    authStorage: AuthStorage;
-    modelRegistry: ModelRegistry;
-    settingsManager: SettingsManager;
-    resourceLoader: ResourceLoader;
-    sessionManager: SessionManager;
-};
 /**
- * Create one runtime instance containing an AgentSession plus the cwd-bound
- * services it depends on.
+ * Creates a full runtime for a target cwd and session manager.
  *
- * Most SDK callers should keep the returned value wrapped in an
- * AgentSessionRuntimeHost instead of holding it directly. The host owns
- * replacing the runtime when switching sessions across files or working
- * directories.
+ * The factory closes over process-global fixed inputs, recreates cwd-bound
+ * services for the effective cwd, resolves session options against those
+ * services, and finally creates the AgentSession.
  */
-export declare function createAgentSessionRuntime(bootstrap: AgentSessionRuntimeBootstrap, options: CreateAgentSessionRuntimeOptions): Promise<AgentSessionRuntime>;
+export type CreateAgentSessionRuntimeFactory = (options: {
+    cwd: string;
+    agentDir: string;
+    sessionManager: SessionManager;
+    sessionStartEvent?: SessionStartEvent;
+}) => Promise<CreateAgentSessionRuntimeResult>;
 /**
- * Stable wrapper around a replaceable AgentSession runtime.
+ * Owns the current AgentSession plus its cwd-bound services.
  *
- * Use this when your application needs `/new`, `/resume`, `/fork`, or import
- * behavior. After replacement, read `session` again and rebind any
- * session-local subscriptions or extension bindings.
+ * Session replacement methods tear down the current runtime first, then create
+ * and apply the next runtime. If creation fails, the error is propagated to the
+ * caller. The caller is responsible for user-facing error handling.
  */
-export declare class AgentSessionRuntimeHost {
-    private readonly bootstrap;
-    private runtime;
-    constructor(bootstrap: AgentSessionRuntimeBootstrap, runtime: AgentSessionRuntime);
-    /** The currently active session instance. Re-read this after runtime replacement. */
-    get session(): import("./agent-session.js").AgentSession;
+export declare class AgentSessionRuntime {
+    private _session;
+    private _services;
+    private readonly createRuntime;
+    private _diagnostics;
+    private _modelFallbackMessage?;
+    constructor(_session: AgentSession, _services: AgentSessionServices, createRuntime: CreateAgentSessionRuntimeFactory, _diagnostics?: AgentSessionRuntimeDiagnostic[], _modelFallbackMessage?: string | undefined);
+    get services(): AgentSessionServices;
+    get session(): AgentSession;
+    get cwd(): string;
+    get diagnostics(): readonly AgentSessionRuntimeDiagnostic[];
+    get modelFallbackMessage(): string | undefined;
     private emitBeforeSwitch;
     private emitBeforeFork;
-    private replace;
-    /**
-     * Replace the active runtime with one opened from an existing session file.
-     *
-     * Emits `session_before_switch` before replacement and returns
-     * `{ cancelled: true }` if an extension vetoes the switch.
-     */
+    private teardownCurrent;
+    private apply;
     switchSession(sessionPath: string): Promise<{
         cancelled: boolean;
     }>;
-    /**
-     * Replace the active runtime with a fresh session in the current cwd.
-     *
-     * `setup` runs after replacement against the new session manager, which lets
-     * callers seed entries before normal use begins.
-     */
     newSession(options?: {
-        /** Optional parent session path recorded in the new session header. */
         parentSession?: string;
-        /** Optional callback for seeding the new session manager after replacement. */
         setup?: (sessionManager: SessionManager) => Promise<void>;
     }): Promise<{
         cancelled: boolean;
     }>;
-    /**
-     * Replace the active runtime with a fork rooted at the given user-message
-     * entry.
-     *
-     * Returns the selected user text so UIs can restore it into the editor after
-     * the fork completes.
-     */
     fork(entryId: string): Promise<{
         cancelled: boolean;
         selectedText?: string;
     }>;
-    /**
-     * Import a JSONL session file into the current session directory and replace
-     * the active runtime with the imported session.
-     */
     importFromJsonl(inputPath: string): Promise<{
         cancelled: boolean;
     }>;
-    /** Emit session shutdown for the active runtime and dispose it permanently. */
     dispose(): Promise<void>;
 }
-export {};
+/**
+ * Create the initial runtime from a runtime factory and initial session target.
+ *
+ * The same factory is stored on the returned AgentSessionRuntime and reused for
+ * later /new, /resume, /fork, and import flows.
+ */
+export declare function createAgentSessionRuntime(createRuntime: CreateAgentSessionRuntimeFactory, options: {
+    cwd: string;
+    agentDir: string;
+    sessionManager: SessionManager;
+    sessionStartEvent?: SessionStartEvent;
+}): Promise<AgentSessionRuntime>;
+export { type AgentSessionRuntimeDiagnostic, type AgentSessionServices, type CreateAgentSessionFromServicesOptions, type CreateAgentSessionServicesOptions, createAgentSessionFromServices, createAgentSessionServices, } from "./agent-session-services.js";
 //# sourceMappingURL=agent-session-runtime.d.ts.map

package/dist/core/agent-session-runtime.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"agent-session-runtime.d.ts","sourceRoot":"","sources":["../../src/core/agent-session-runtime.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAC;AACjE,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,qBAAqB,CAAC;AAEjD,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,KAAK,EAAE,iBAAiB,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAE/E,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAyB,KAAK,4BAA4B,EAAE,KAAK,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACrH,OAAO,EAAE,KAAK,wBAAwB,EAAsB,MAAM,UAAU,CAAC;AAC7E,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AAE7C;;;;;;GAMG;AACH,MAAM,WAAW,4BAA4B;IAC5C,yFAAyF;IACzF,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,4FAA4F;IAC5F,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,2DAA2D;IAC3D,KAAK,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;IACnB,oEAAoE;IACpE,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,kEAAkE;IAClE,YAAY,CAAC,EAAE,KAAK,CAAC;QAAE,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;QAAC,aAAa,CAAC,EAAE,aAAa,CAAA;KAAE,CAAC,CAAC;IAC3E,kCAAkC;IAClC,KAAK,CAAC,EAAE,IAAI,EAAE,CAAC;IACf,mEAAmE;IACnE,WAAW,CAAC,EAAE,cAAc,EAAE,CAAC;IAC/B;;;;;;OAMG;IACH,cAAc,CAAC,EACZ,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC,cAAc,CAAC,CAAC,GAC5D,IAAI,CAAC,4BAA4B,EAAE,KAAK,GAAG,UAAU,GAAG,iBAAiB,CAAC,CAAC;CAC9E;AAED,0DAA0D;AAC1D,MAAM,WAAW,gCAAgC;IAChD,mDAAmD;IACnD,GAAG,EAAE,MAAM,CAAC;IACZ,2FAA2F;IAC3F,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC,iFAAiF;IACjF,iBAAiB,CAAC,EAAE,iBAAiB,CAAC;CACtC;AAED,KAAK,mBAAmB,GAAG,wBAAwB,GAAG;IACrD,GAAG,EAAE,MAAM,CAAC;IACZ,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,WAAW,CAAC;IACzB,aAAa,EAAE,aAAa,CAAC;IAC7B,eAAe,EAAE,eAAe,CAAC;IACjC,cAAc,EAAE,cAAc,CAAC;IAC/B,cAAc,EAAE,cAAc,CAAC;CAC/B,CAAC;AAEF;;;;;;;;GAQG;AACH,wBAAsB,yBAAyB,CAC9C,SAAS,EAAE,4BAA4B,EACvC,OAAO,EAAE,gCAAgC,GACvC,OAAO,CAAC,mBAAmB,CAAC,CAiD9B;AAaD;;;;;;GAMG;AACH,qBAAa,uBAAuB;IAElC,OAAO,CAAC,QAAQ,CAAC,SAAS;IAC1B,OAAO,CAAC,OAAO;IAFhB,YACkB,SAAS,EAAE,4BAA4B,EAChD,OAAO,EAAE,mBAAmB,EACjC;IAEJ,qFAAqF;IACrF,IAAI,OAAO,8CAEV;YAEa,gBAAgB;YAiBhB,cAAc;YAad,OAAO;IAUrB;;;;;OAKG;IACG,aAAa,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,SAAS,EAAE,OAAO,CAAA;KAAE,CAAC,CAcxE;IAED;;;;;OAKG;IACG,UAAU,CAAC,OAAO,CAAC,EAAE;QAC1B,uEAAuE;QACvE,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,+EAA+E;QAC/E,KAAK,CAAC,EAAE,CAAC,cAAc,EAAE,cAAc,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;KAC1D,GAAG,OAAO,CAAC;QAAE,SAAS,EAAE,OAAO,CAAA;KAAE,CAAC,CAsBlC;IAED;;;;;;OAMG;IACG,IAAI,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,SAAS,EAAE,OAAO,CAAC;QAAC,YAAY,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAkDlF;IAED;;;OAGG;IACG,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,SAAS,EAAE,OAAO,CAAA;KAAE,CAAC,CA6BxE;IAED,+EAA+E;IACzE,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,CAG7B;CACD","sourcesContent":["import { copyFileSync, existsSync, mkdirSync } from \"node:fs\";\nimport { basename, join, resolve } from \"node:path\";\nimport type { ThinkingLevel } from \"@mariozechner/pi-agent-core\";\nimport type { Model } from \"@mariozechner/pi-ai\";\nimport { getAgentDir } from \"../config.js\";\nimport { AuthStorage } from \"./auth-storage.js\";\nimport type { SessionStartEvent, ToolDefinition } from \"./extensions/index.js\";\nimport { emitSessionShutdownEvent } from \"./extensions/runner.js\";\nimport { ModelRegistry } from \"./model-registry.js\";\nimport { DefaultResourceLoader, type DefaultResourceLoaderOptions, type ResourceLoader } from \"./resource-loader.js\";\nimport { type CreateAgentSessionResult, createAgentSession } from \"./sdk.js\";\nimport { SessionManager } from \"./session-manager.js\";\nimport { SettingsManager } from \"./settings-manager.js\";\nimport type { Tool } from \"./tools/index.js\";\n\n/**\n * Stable bootstrap inputs reused whenever the active runtime is replaced.\n *\n * Use this for process-level wiring that should survive `/new`, `/resume`,\n * `/fork`, and import flows. Session-local state belongs in the session file\n * or in settings, not here.\n */\nexport interface AgentSessionRuntimeBootstrap {\n\t/** Agent directory used for auth, models, settings, sessions, and resource discovery. */\n\tagentDir?: string;\n\t/** Optional shared auth storage. If omitted, file-backed storage under agentDir is used. */\n\tauthStorage?: AuthStorage;\n\t/** Initial model for the first created session runtime. */\n\tmodel?: Model<any>;\n\t/** Initial thinking level for the first created session runtime. */\n\tthinkingLevel?: ThinkingLevel;\n\t/** Optional scoped model list for model cycling and selection. */\n\tscopedModels?: Array<{ model: Model<any>; thinkingLevel?: ThinkingLevel }>;\n\t/** Built-in tool set override. */\n\ttools?: Tool[];\n\t/** Additional custom tools registered directly through the SDK. */\n\tcustomTools?: ToolDefinition[];\n\t/**\n\t * Resource loader input used for each created runtime.\n\t *\n\t * Pass either a factory that creates a fully custom ResourceLoader for the\n\t * target cwd, or DefaultResourceLoader options without cwd/agentDir/\n\t * settingsManager, which are supplied by the runtime.\n\t */\n\tresourceLoader?:\n\t\t| ((cwd: string, agentDir: string) => Promise<ResourceLoader>)\n\t\t| Omit<DefaultResourceLoaderOptions, \"cwd\" | \"agentDir\" | \"settingsManager\">;\n}\n\n/** Options for creating one concrete runtime instance. */\nexport interface CreateAgentSessionRuntimeOptions {\n\t/** Working directory for this runtime instance. */\n\tcwd: string;\n\t/** Optional preselected session manager. If omitted, normal session resolution applies. */\n\tsessionManager?: SessionManager;\n\t/** Optional session_start metadata to emit when the runtime binds extensions. */\n\tsessionStartEvent?: SessionStartEvent;\n}\n\ntype AgentSessionRuntime = CreateAgentSessionResult & {\n\tcwd: string;\n\tagentDir: string;\n\tauthStorage: AuthStorage;\n\tmodelRegistry: ModelRegistry;\n\tsettingsManager: SettingsManager;\n\tresourceLoader: ResourceLoader;\n\tsessionManager: SessionManager;\n};\n\n/**\n * Create one runtime instance containing an AgentSession plus the cwd-bound\n * services it depends on.\n *\n * Most SDK callers should keep the returned value wrapped in an\n * AgentSessionRuntimeHost instead of holding it directly. The host owns\n * replacing the runtime when switching sessions across files or working\n * directories.\n */\nexport async function createAgentSessionRuntime(\n\tbootstrap: AgentSessionRuntimeBootstrap,\n\toptions: CreateAgentSessionRuntimeOptions,\n): Promise<AgentSessionRuntime> {\n\tconst cwd = options.cwd;\n\tconst agentDir = bootstrap.agentDir ?? getAgentDir();\n\tconst authStorage = bootstrap.authStorage ?? AuthStorage.create(join(agentDir, \"auth.json\"));\n\tconst settingsManager = SettingsManager.create(cwd, agentDir);\n\tconst modelRegistry = ModelRegistry.create(authStorage, join(agentDir, \"models.json\"));\n\tconst resourceLoader =\n\t\ttypeof bootstrap.resourceLoader === \"function\"\n\t\t\t? await bootstrap.resourceLoader(cwd, agentDir)\n\t\t\t: new DefaultResourceLoader({\n\t\t\t\t\t...(bootstrap.resourceLoader ?? {}),\n\t\t\t\t\tcwd,\n\t\t\t\t\tagentDir,\n\t\t\t\t\tsettingsManager,\n\t\t\t\t});\n\tawait resourceLoader.reload();\n\n\tconst extensionsResult = resourceLoader.getExtensions();\n\tfor (const { name, config } of extensionsResult.runtime.pendingProviderRegistrations) {\n\t\tmodelRegistry.registerProvider(name, config);\n\t}\n\textensionsResult.runtime.pendingProviderRegistrations = [];\n\n\tconst created = await createAgentSession({\n\t\tcwd,\n\t\tagentDir,\n\t\tauthStorage,\n\t\tmodelRegistry,\n\t\tsettingsManager,\n\t\tresourceLoader,\n\t\tsessionManager: options.sessionManager,\n\t\tmodel: bootstrap.model,\n\t\tthinkingLevel: bootstrap.thinkingLevel,\n\t\tscopedModels: bootstrap.scopedModels,\n\t\ttools: bootstrap.tools,\n\t\tcustomTools: bootstrap.customTools,\n\t\tsessionStartEvent: options.sessionStartEvent,\n\t});\n\n\treturn {\n\t\t...created,\n\t\tcwd,\n\t\tagentDir,\n\t\tauthStorage,\n\t\tmodelRegistry,\n\t\tsettingsManager,\n\t\tresourceLoader,\n\t\tsessionManager: created.session.sessionManager,\n\t};\n}\n\nfunction extractUserMessageText(content: string | Array<{ type: string; text?: string }>): string {\n\tif (typeof content === \"string\") {\n\t\treturn content;\n\t}\n\n\treturn content\n\t\t.filter((part): part is { type: \"text\"; text: string } => part.type === \"text\" && typeof part.text === \"string\")\n\t\t.map((part) => part.text)\n\t\t.join(\"\");\n}\n\n/**\n * Stable wrapper around a replaceable AgentSession runtime.\n *\n * Use this when your application needs `/new`, `/resume`, `/fork`, or import\n * behavior. After replacement, read `session` again and rebind any\n * session-local subscriptions or extension bindings.\n */\nexport class AgentSessionRuntimeHost {\n\tconstructor(\n\t\tprivate readonly bootstrap: AgentSessionRuntimeBootstrap,\n\t\tprivate runtime: AgentSessionRuntime,\n\t) {}\n\n\t/** The currently active session instance. Re-read this after runtime replacement. */\n\tget session() {\n\t\treturn this.runtime.session;\n\t}\n\n\tprivate async emitBeforeSwitch(\n\t\treason: \"new\" | \"resume\",\n\t\ttargetSessionFile?: string,\n\t): Promise<{ cancelled: boolean }> {\n\t\tconst runner = this.runtime.session.extensionRunner;\n\t\tif (!runner?.hasHandlers(\"session_before_switch\")) {\n\t\t\treturn { cancelled: false };\n\t\t}\n\n\t\tconst result = await runner.emit({\n\t\t\ttype: \"session_before_switch\",\n\t\t\treason,\n\t\t\ttargetSessionFile,\n\t\t});\n\t\treturn { cancelled: result?.cancel === true };\n\t}\n\n\tprivate async emitBeforeFork(entryId: string): Promise<{ cancelled: boolean }> {\n\t\tconst runner = this.runtime.session.extensionRunner;\n\t\tif (!runner?.hasHandlers(\"session_before_fork\")) {\n\t\t\treturn { cancelled: false };\n\t\t}\n\n\t\tconst result = await runner.emit({\n\t\t\ttype: \"session_before_fork\",\n\t\t\tentryId,\n\t\t});\n\t\treturn { cancelled: result?.cancel === true };\n\t}\n\n\tprivate async replace(options: CreateAgentSessionRuntimeOptions): Promise<void> {\n\t\tconst nextRuntime = await createAgentSessionRuntime(this.bootstrap, options);\n\t\tawait emitSessionShutdownEvent(this.runtime.session.extensionRunner);\n\t\tthis.runtime.session.dispose();\n\t\tif (process.cwd() !== nextRuntime.cwd) {\n\t\t\tprocess.chdir(nextRuntime.cwd);\n\t\t}\n\t\tthis.runtime = nextRuntime;\n\t}\n\n\t/**\n\t * Replace the active runtime with one opened from an existing session file.\n\t *\n\t * Emits `session_before_switch` before replacement and returns\n\t * `{ cancelled: true }` if an extension vetoes the switch.\n\t */\n\tasync switchSession(sessionPath: string): Promise<{ cancelled: boolean }> {\n\t\tconst beforeResult = await this.emitBeforeSwitch(\"resume\", sessionPath);\n\t\tif (beforeResult.cancelled) {\n\t\t\treturn beforeResult;\n\t\t}\n\n\t\tconst previousSessionFile = this.runtime.session.sessionFile;\n\t\tconst sessionManager = SessionManager.open(sessionPath);\n\t\tawait this.replace({\n\t\t\tcwd: sessionManager.getCwd(),\n\t\t\tsessionManager,\n\t\t\tsessionStartEvent: { type: \"session_start\", reason: \"resume\", previousSessionFile },\n\t\t});\n\t\treturn { cancelled: false };\n\t}\n\n\t/**\n\t * Replace the active runtime with a fresh session in the current cwd.\n\t *\n\t * `setup` runs after replacement against the new session manager, which lets\n\t * callers seed entries before normal use begins.\n\t */\n\tasync newSession(options?: {\n\t\t/** Optional parent session path recorded in the new session header. */\n\t\tparentSession?: string;\n\t\t/** Optional callback for seeding the new session manager after replacement. */\n\t\tsetup?: (sessionManager: SessionManager) => Promise<void>;\n\t}): Promise<{ cancelled: boolean }> {\n\t\tconst beforeResult = await this.emitBeforeSwitch(\"new\");\n\t\tif (beforeResult.cancelled) {\n\t\t\treturn beforeResult;\n\t\t}\n\n\t\tconst previousSessionFile = this.runtime.session.sessionFile;\n\t\tconst sessionDir = this.runtime.sessionManager.getSessionDir();\n\t\tconst sessionManager = SessionManager.create(this.runtime.cwd, sessionDir);\n\t\tif (options?.parentSession) {\n\t\t\tsessionManager.newSession({ parentSession: options.parentSession });\n\t\t}\n\t\tawait this.replace({\n\t\t\tcwd: this.runtime.cwd,\n\t\t\tsessionManager,\n\t\t\tsessionStartEvent: { type: \"session_start\", reason: \"new\", previousSessionFile },\n\t\t});\n\t\tif (options?.setup) {\n\t\t\tawait options.setup(this.runtime.sessionManager);\n\t\t\tthis.runtime.session.agent.state.messages = this.runtime.sessionManager.buildSessionContext().messages;\n\t\t}\n\t\treturn { cancelled: false };\n\t}\n\n\t/**\n\t * Replace the active runtime with a fork rooted at the given user-message\n\t * entry.\n\t *\n\t * Returns the selected user text so UIs can restore it into the editor after\n\t * the fork completes.\n\t */\n\tasync fork(entryId: string): Promise<{ cancelled: boolean; selectedText?: string }> {\n\t\tconst beforeResult = await this.emitBeforeFork(entryId);\n\t\tif (beforeResult.cancelled) {\n\t\t\treturn { cancelled: true };\n\t\t}\n\n\t\tconst selectedEntry = this.runtime.sessionManager.getEntry(entryId);\n\t\tif (!selectedEntry || selectedEntry.type !== \"message\" || selectedEntry.message.role !== \"user\") {\n\t\t\tthrow new Error(\"Invalid entry ID for forking\");\n\t\t}\n\n\t\tconst previousSessionFile = this.runtime.session.sessionFile;\n\t\tconst selectedText = extractUserMessageText(selectedEntry.message.content);\n\t\tif (this.runtime.sessionManager.isPersisted()) {\n\t\t\tconst currentSessionFile = this.runtime.session.sessionFile!;\n\t\t\tconst sessionDir = this.runtime.sessionManager.getSessionDir();\n\t\t\tif (!selectedEntry.parentId) {\n\t\t\t\tconst sessionManager = SessionManager.create(this.runtime.cwd, sessionDir);\n\t\t\t\tsessionManager.newSession({ parentSession: currentSessionFile });\n\t\t\t\tawait this.replace({\n\t\t\t\t\tcwd: this.runtime.cwd,\n\t\t\t\t\tsessionManager,\n\t\t\t\t\tsessionStartEvent: { type: \"session_start\", reason: \"fork\", previousSessionFile },\n\t\t\t\t});\n\t\t\t\treturn { cancelled: false, selectedText };\n\t\t\t}\n\n\t\t\tconst sourceManager = SessionManager.open(currentSessionFile, sessionDir);\n\t\t\tconst forkedSessionPath = sourceManager.createBranchedSession(selectedEntry.parentId)!;\n\t\t\tconst sessionManager = SessionManager.open(forkedSessionPath, sessionDir);\n\t\t\tawait this.replace({\n\t\t\t\tcwd: sessionManager.getCwd(),\n\t\t\t\tsessionManager,\n\t\t\t\tsessionStartEvent: { type: \"session_start\", reason: \"fork\", previousSessionFile },\n\t\t\t});\n\t\t\treturn { cancelled: false, selectedText };\n\t\t}\n\n\t\tconst sessionManager = this.runtime.sessionManager;\n\t\tif (!selectedEntry.parentId) {\n\t\t\tsessionManager.newSession({ parentSession: this.runtime.session.sessionFile });\n\t\t} else {\n\t\t\tsessionManager.createBranchedSession(selectedEntry.parentId);\n\t\t}\n\t\tawait this.replace({\n\t\t\tcwd: this.runtime.cwd,\n\t\t\tsessionManager,\n\t\t\tsessionStartEvent: { type: \"session_start\", reason: \"fork\", previousSessionFile },\n\t\t});\n\t\treturn { cancelled: false, selectedText };\n\t}\n\n\t/**\n\t * Import a JSONL session file into the current session directory and replace\n\t * the active runtime with the imported session.\n\t */\n\tasync importFromJsonl(inputPath: string): Promise<{ cancelled: boolean }> {\n\t\tconst resolvedPath = resolve(inputPath);\n\t\tif (!existsSync(resolvedPath)) {\n\t\t\tthrow new Error(`File not found: ${resolvedPath}`);\n\t\t}\n\n\t\tconst sessionDir = this.runtime.sessionManager.getSessionDir();\n\t\tif (!existsSync(sessionDir)) {\n\t\t\tmkdirSync(sessionDir, { recursive: true });\n\t\t}\n\n\t\tconst destinationPath = join(sessionDir, basename(resolvedPath));\n\t\tconst beforeResult = await this.emitBeforeSwitch(\"resume\", destinationPath);\n\t\tif (beforeResult.cancelled) {\n\t\t\treturn beforeResult;\n\t\t}\n\n\t\tconst previousSessionFile = this.runtime.session.sessionFile;\n\t\tif (resolve(destinationPath) !== resolvedPath) {\n\t\t\tcopyFileSync(resolvedPath, destinationPath);\n\t\t}\n\n\t\tconst sessionManager = SessionManager.open(destinationPath, sessionDir);\n\t\tawait this.replace({\n\t\t\tcwd: sessionManager.getCwd(),\n\t\t\tsessionManager,\n\t\t\tsessionStartEvent: { type: \"session_start\", reason: \"resume\", previousSessionFile },\n\t\t});\n\t\treturn { cancelled: false };\n\t}\n\n\t/** Emit session shutdown for the active runtime and dispose it permanently. */\n\tasync dispose(): Promise<void> {\n\t\tawait emitSessionShutdownEvent(this.runtime.session.extensionRunner);\n\t\tthis.runtime.session.dispose();\n\t}\n}\n"]}
+{"version":3,"file":"agent-session-runtime.d.ts","sourceRoot":"","sources":["../../src/core/agent-session-runtime.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,KAAK,EAAE,6BAA6B,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAC;AACvG,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAE/D,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,UAAU,CAAC;AACzD,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAEtD;;;;;GAKG;AACH,MAAM,WAAW,+BAAgC,SAAQ,wBAAwB;IAChF,QAAQ,EAAE,oBAAoB,CAAC;IAC/B,WAAW,EAAE,6BAA6B,EAAE,CAAC;CAC7C;AAED;;;;;;GAMG;AACH,MAAM,MAAM,gCAAgC,GAAG,CAAC,OAAO,EAAE;IACxD,GAAG,EAAE,MAAM,CAAC;IACZ,QAAQ,EAAE,MAAM,CAAC;IACjB,cAAc,EAAE,cAAc,CAAC;IAC/B,iBAAiB,CAAC,EAAE,iBAAiB,CAAC;CACtC,KAAK,OAAO,CAAC,+BAA+B,CAAC,CAAC;AAa/C;;;;;;GAMG;AACH,qBAAa,mBAAmB;IAE9B,OAAO,CAAC,QAAQ;IAChB,OAAO,CAAC,SAAS;IACjB,OAAO,CAAC,QAAQ,CAAC,aAAa;IAC9B,OAAO,CAAC,YAAY;IACpB,OAAO,CAAC,qBAAqB,CAAC;IAL/B,YACS,QAAQ,EAAE,YAAY,EACtB,SAAS,EAAE,oBAAoB,EACtB,aAAa,EAAE,gCAAgC,EACxD,YAAY,GAAE,6BAA6B,EAAO,EAClD,qBAAqB,CAAC,oBAAQ,EACnC;IAEJ,IAAI,QAAQ,IAAI,oBAAoB,CAEnC;IAED,IAAI,OAAO,IAAI,YAAY,CAE1B;IAED,IAAI,GAAG,IAAI,MAAM,CAEhB;IAED,IAAI,WAAW,IAAI,SAAS,6BAA6B,EAAE,CAE1D;IAED,IAAI,oBAAoB,IAAI,MAAM,GAAG,SAAS,CAE7C;YAEa,gBAAgB;YAiBhB,cAAc;YAad,eAAe;IAK7B,OAAO,CAAC,KAAK;IAUP,aAAa,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,SAAS,EAAE,OAAO,CAAA;KAAE,CAAC,CAkBxE;IAEK,UAAU,CAAC,OAAO,CAAC,EAAE;QAC1B,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,KAAK,CAAC,EAAE,CAAC,cAAc,EAAE,cAAc,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;KAC1D,GAAG,OAAO,CAAC;QAAE,SAAS,EAAE,OAAO,CAAA;KAAE,CAAC,CA2BlC;IAEK,IAAI,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,SAAS,EAAE,OAAO,CAAC;QAAC,YAAY,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAoElF;IAEK,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,SAAS,EAAE,OAAO,CAAA;KAAE,CAAC,CAiCxE;IAEK,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,CAG7B;CACD;AAED;;;;;GAKG;AACH,wBAAsB,yBAAyB,CAC9C,aAAa,EAAE,gCAAgC,EAC/C,OAAO,EAAE;IACR,GAAG,EAAE,MAAM,CAAC;IACZ,QAAQ,EAAE,MAAM,CAAC;IACjB,cAAc,EAAE,cAAc,CAAC;IAC/B,iBAAiB,CAAC,EAAE,iBAAiB,CAAC;CACtC,GACC,OAAO,CAAC,mBAAmB,CAAC,CAY9B;AAED,OAAO,EACN,KAAK,6BAA6B,EAClC,KAAK,oBAAoB,EACzB,KAAK,qCAAqC,EAC1C,KAAK,iCAAiC,EACtC,8BAA8B,EAC9B,0BAA0B,GAC1B,MAAM,6BAA6B,CAAC","sourcesContent":["import { copyFileSync, existsSync, mkdirSync } from \"node:fs\";\nimport { basename, join, resolve } from \"node:path\";\nimport type { AgentSession } from \"./agent-session.js\";\nimport type { AgentSessionRuntimeDiagnostic, AgentSessionServices } from \"./agent-session-services.js\";\nimport type { SessionStartEvent } from \"./extensions/index.js\";\nimport { emitSessionShutdownEvent } from \"./extensions/runner.js\";\nimport type { CreateAgentSessionResult } from \"./sdk.js\";\nimport { SessionManager } from \"./session-manager.js\";\n\n/**\n * Result returned by runtime creation.\n *\n * The caller gets the created session, its cwd-bound services, and all\n * diagnostics collected during setup.\n */\nexport interface CreateAgentSessionRuntimeResult extends CreateAgentSessionResult {\n\tservices: AgentSessionServices;\n\tdiagnostics: AgentSessionRuntimeDiagnostic[];\n}\n\n/**\n * Creates a full runtime for a target cwd and session manager.\n *\n * The factory closes over process-global fixed inputs, recreates cwd-bound\n * services for the effective cwd, resolves session options against those\n * services, and finally creates the AgentSession.\n */\nexport type CreateAgentSessionRuntimeFactory = (options: {\n\tcwd: string;\n\tagentDir: string;\n\tsessionManager: SessionManager;\n\tsessionStartEvent?: SessionStartEvent;\n}) => Promise<CreateAgentSessionRuntimeResult>;\n\nfunction extractUserMessageText(content: string | Array<{ type: string; text?: string }>): string {\n\tif (typeof content === \"string\") {\n\t\treturn content;\n\t}\n\n\treturn content\n\t\t.filter((part): part is { type: \"text\"; text: string } => part.type === \"text\" && typeof part.text === \"string\")\n\t\t.map((part) => part.text)\n\t\t.join(\"\");\n}\n\n/**\n * Owns the current AgentSession plus its cwd-bound services.\n *\n * Session replacement methods tear down the current runtime first, then create\n * and apply the next runtime. If creation fails, the error is propagated to the\n * caller. The caller is responsible for user-facing error handling.\n */\nexport class AgentSessionRuntime {\n\tconstructor(\n\t\tprivate _session: AgentSession,\n\t\tprivate _services: AgentSessionServices,\n\t\tprivate readonly createRuntime: CreateAgentSessionRuntimeFactory,\n\t\tprivate _diagnostics: AgentSessionRuntimeDiagnostic[] = [],\n\t\tprivate _modelFallbackMessage?: string,\n\t) {}\n\n\tget services(): AgentSessionServices {\n\t\treturn this._services;\n\t}\n\n\tget session(): AgentSession {\n\t\treturn this._session;\n\t}\n\n\tget cwd(): string {\n\t\treturn this._services.cwd;\n\t}\n\n\tget diagnostics(): readonly AgentSessionRuntimeDiagnostic[] {\n\t\treturn this._diagnostics;\n\t}\n\n\tget modelFallbackMessage(): string | undefined {\n\t\treturn this._modelFallbackMessage;\n\t}\n\n\tprivate async emitBeforeSwitch(\n\t\treason: \"new\" | \"resume\",\n\t\ttargetSessionFile?: string,\n\t): Promise<{ cancelled: boolean }> {\n\t\tconst runner = this.session.extensionRunner;\n\t\tif (!runner?.hasHandlers(\"session_before_switch\")) {\n\t\t\treturn { cancelled: false };\n\t\t}\n\n\t\tconst result = await runner.emit({\n\t\t\ttype: \"session_before_switch\",\n\t\t\treason,\n\t\t\ttargetSessionFile,\n\t\t});\n\t\treturn { cancelled: result?.cancel === true };\n\t}\n\n\tprivate async emitBeforeFork(entryId: string): Promise<{ cancelled: boolean }> {\n\t\tconst runner = this.session.extensionRunner;\n\t\tif (!runner?.hasHandlers(\"session_before_fork\")) {\n\t\t\treturn { cancelled: false };\n\t\t}\n\n\t\tconst result = await runner.emit({\n\t\t\ttype: \"session_before_fork\",\n\t\t\tentryId,\n\t\t});\n\t\treturn { cancelled: result?.cancel === true };\n\t}\n\n\tprivate async teardownCurrent(): Promise<void> {\n\t\tawait emitSessionShutdownEvent(this.session.extensionRunner);\n\t\tthis.session.dispose();\n\t}\n\n\tprivate apply(result: CreateAgentSessionRuntimeResult): void {\n\t\tif (process.cwd() !== result.services.cwd) {\n\t\t\tprocess.chdir(result.services.cwd);\n\t\t}\n\t\tthis._session = result.session;\n\t\tthis._services = result.services;\n\t\tthis._diagnostics = result.diagnostics;\n\t\tthis._modelFallbackMessage = result.modelFallbackMessage;\n\t}\n\n\tasync switchSession(sessionPath: string): Promise<{ cancelled: boolean }> {\n\t\tconst beforeResult = await this.emitBeforeSwitch(\"resume\", sessionPath);\n\t\tif (beforeResult.cancelled) {\n\t\t\treturn beforeResult;\n\t\t}\n\n\t\tconst previousSessionFile = this.session.sessionFile;\n\t\tconst sessionManager = SessionManager.open(sessionPath);\n\t\tawait this.teardownCurrent();\n\t\tthis.apply(\n\t\t\tawait this.createRuntime({\n\t\t\t\tcwd: sessionManager.getCwd(),\n\t\t\t\tagentDir: this.services.agentDir,\n\t\t\t\tsessionManager,\n\t\t\t\tsessionStartEvent: { type: \"session_start\", reason: \"resume\", previousSessionFile },\n\t\t\t}),\n\t\t);\n\t\treturn { cancelled: false };\n\t}\n\n\tasync newSession(options?: {\n\t\tparentSession?: string;\n\t\tsetup?: (sessionManager: SessionManager) => Promise<void>;\n\t}): Promise<{ cancelled: boolean }> {\n\t\tconst beforeResult = await this.emitBeforeSwitch(\"new\");\n\t\tif (beforeResult.cancelled) {\n\t\t\treturn beforeResult;\n\t\t}\n\n\t\tconst previousSessionFile = this.session.sessionFile;\n\t\tconst sessionDir = this.session.sessionManager.getSessionDir();\n\t\tconst sessionManager = SessionManager.create(this.cwd, sessionDir);\n\t\tif (options?.parentSession) {\n\t\t\tsessionManager.newSession({ parentSession: options.parentSession });\n\t\t}\n\n\t\tawait this.teardownCurrent();\n\t\tthis.apply(\n\t\t\tawait this.createRuntime({\n\t\t\t\tcwd: this.cwd,\n\t\t\t\tagentDir: this.services.agentDir,\n\t\t\t\tsessionManager,\n\t\t\t\tsessionStartEvent: { type: \"session_start\", reason: \"new\", previousSessionFile },\n\t\t\t}),\n\t\t);\n\t\tif (options?.setup) {\n\t\t\tawait options.setup(this.session.sessionManager);\n\t\t\tthis.session.agent.state.messages = this.session.sessionManager.buildSessionContext().messages;\n\t\t}\n\t\treturn { cancelled: false };\n\t}\n\n\tasync fork(entryId: string): Promise<{ cancelled: boolean; selectedText?: string }> {\n\t\tconst beforeResult = await this.emitBeforeFork(entryId);\n\t\tif (beforeResult.cancelled) {\n\t\t\treturn { cancelled: true };\n\t\t}\n\n\t\tconst selectedEntry = this.session.sessionManager.getEntry(entryId);\n\t\tif (!selectedEntry || selectedEntry.type !== \"message\" || selectedEntry.message.role !== \"user\") {\n\t\t\tthrow new Error(\"Invalid entry ID for forking\");\n\t\t}\n\n\t\tconst previousSessionFile = this.session.sessionFile;\n\t\tconst selectedText = extractUserMessageText(selectedEntry.message.content);\n\t\tif (this.session.sessionManager.isPersisted()) {\n\t\t\tconst currentSessionFile = this.session.sessionFile;\n\t\t\tif (!currentSessionFile) {\n\t\t\t\tthrow new Error(\"Persisted session is missing a session file\");\n\t\t\t}\n\t\t\tconst sessionDir = this.session.sessionManager.getSessionDir();\n\t\t\tif (!selectedEntry.parentId) {\n\t\t\t\tconst sessionManager = SessionManager.create(this.cwd, sessionDir);\n\t\t\t\tsessionManager.newSession({ parentSession: currentSessionFile });\n\t\t\t\tawait this.teardownCurrent();\n\t\t\t\tthis.apply(\n\t\t\t\t\tawait this.createRuntime({\n\t\t\t\t\t\tcwd: this.cwd,\n\t\t\t\t\t\tagentDir: this.services.agentDir,\n\t\t\t\t\t\tsessionManager,\n\t\t\t\t\t\tsessionStartEvent: { type: \"session_start\", reason: \"fork\", previousSessionFile },\n\t\t\t\t\t}),\n\t\t\t\t);\n\t\t\t\treturn { cancelled: false, selectedText };\n\t\t\t}\n\n\t\t\tconst sourceManager = SessionManager.open(currentSessionFile, sessionDir);\n\t\t\tconst forkedSessionPath = sourceManager.createBranchedSession(selectedEntry.parentId);\n\t\t\tif (!forkedSessionPath) {\n\t\t\t\tthrow new Error(\"Failed to create forked session\");\n\t\t\t}\n\t\t\tconst sessionManager = SessionManager.open(forkedSessionPath, sessionDir);\n\t\t\tawait this.teardownCurrent();\n\t\t\tthis.apply(\n\t\t\t\tawait this.createRuntime({\n\t\t\t\t\tcwd: sessionManager.getCwd(),\n\t\t\t\t\tagentDir: this.services.agentDir,\n\t\t\t\t\tsessionManager,\n\t\t\t\t\tsessionStartEvent: { type: \"session_start\", reason: \"fork\", previousSessionFile },\n\t\t\t\t}),\n\t\t\t);\n\t\t\treturn { cancelled: false, selectedText };\n\t\t}\n\n\t\tconst sessionManager = this.session.sessionManager;\n\t\tif (!selectedEntry.parentId) {\n\t\t\tsessionManager.newSession({ parentSession: this.session.sessionFile });\n\t\t} else {\n\t\t\tsessionManager.createBranchedSession(selectedEntry.parentId);\n\t\t}\n\t\tawait this.teardownCurrent();\n\t\tthis.apply(\n\t\t\tawait this.createRuntime({\n\t\t\t\tcwd: this.cwd,\n\t\t\t\tagentDir: this.services.agentDir,\n\t\t\t\tsessionManager,\n\t\t\t\tsessionStartEvent: { type: \"session_start\", reason: \"fork\", previousSessionFile },\n\t\t\t}),\n\t\t);\n\t\treturn { cancelled: false, selectedText };\n\t}\n\n\tasync importFromJsonl(inputPath: string): Promise<{ cancelled: boolean }> {\n\t\tconst resolvedPath = resolve(inputPath);\n\t\tif (!existsSync(resolvedPath)) {\n\t\t\tthrow new Error(`File not found: ${resolvedPath}`);\n\t\t}\n\n\t\tconst sessionDir = this.session.sessionManager.getSessionDir();\n\t\tif (!existsSync(sessionDir)) {\n\t\t\tmkdirSync(sessionDir, { recursive: true });\n\t\t}\n\n\t\tconst destinationPath = join(sessionDir, basename(resolvedPath));\n\t\tconst beforeResult = await this.emitBeforeSwitch(\"resume\", destinationPath);\n\t\tif (beforeResult.cancelled) {\n\t\t\treturn beforeResult;\n\t\t}\n\n\t\tconst previousSessionFile = this.session.sessionFile;\n\t\tif (resolve(destinationPath) !== resolvedPath) {\n\t\t\tcopyFileSync(resolvedPath, destinationPath);\n\t\t}\n\n\t\tconst sessionManager = SessionManager.open(destinationPath, sessionDir);\n\t\tawait this.teardownCurrent();\n\t\tthis.apply(\n\t\t\tawait this.createRuntime({\n\t\t\t\tcwd: sessionManager.getCwd(),\n\t\t\t\tagentDir: this.services.agentDir,\n\t\t\t\tsessionManager,\n\t\t\t\tsessionStartEvent: { type: \"session_start\", reason: \"resume\", previousSessionFile },\n\t\t\t}),\n\t\t);\n\t\treturn { cancelled: false };\n\t}\n\n\tasync dispose(): Promise<void> {\n\t\tawait emitSessionShutdownEvent(this.session.extensionRunner);\n\t\tthis.session.dispose();\n\t}\n}\n\n/**\n * Create the initial runtime from a runtime factory and initial session target.\n *\n * The same factory is stored on the returned AgentSessionRuntime and reused for\n * later /new, /resume, /fork, and import flows.\n */\nexport async function createAgentSessionRuntime(\n\tcreateRuntime: CreateAgentSessionRuntimeFactory,\n\toptions: {\n\t\tcwd: string;\n\t\tagentDir: string;\n\t\tsessionManager: SessionManager;\n\t\tsessionStartEvent?: SessionStartEvent;\n\t},\n): Promise<AgentSessionRuntime> {\n\tconst result = await createRuntime(options);\n\tif (process.cwd() !== result.services.cwd) {\n\t\tprocess.chdir(result.services.cwd);\n\t}\n\treturn new AgentSessionRuntime(\n\t\tresult.session,\n\t\tresult.services,\n\t\tcreateRuntime,\n\t\tresult.diagnostics,\n\t\tresult.modelFallbackMessage,\n\t);\n}\n\nexport {\n\ttype AgentSessionRuntimeDiagnostic,\n\ttype AgentSessionServices,\n\ttype CreateAgentSessionFromServicesOptions,\n\ttype CreateAgentSessionServicesOptions,\n\tcreateAgentSessionFromServices,\n\tcreateAgentSessionServices,\n} from \"./agent-session-services.js\";\n"]}

package/dist/core/agent-session-runtime.js

@@ -1,68 +1,7 @@
 import { copyFileSync, existsSync, mkdirSync } from "node:fs";
 import { basename, join, resolve } from "node:path";
-import { getAgentDir } from "../config.js";
-import { AuthStorage } from "./auth-storage.js";
 import { emitSessionShutdownEvent } from "./extensions/runner.js";
-import { ModelRegistry } from "./model-registry.js";
-import { DefaultResourceLoader } from "./resource-loader.js";
-import { createAgentSession } from "./sdk.js";
 import { SessionManager } from "./session-manager.js";
-import { SettingsManager } from "./settings-manager.js";
-/**
- * Create one runtime instance containing an AgentSession plus the cwd-bound
- * services it depends on.
- *
- * Most SDK callers should keep the returned value wrapped in an
- * AgentSessionRuntimeHost instead of holding it directly. The host owns
- * replacing the runtime when switching sessions across files or working
- * directories.
- */
-export async function createAgentSessionRuntime(bootstrap, options) {
-    const cwd = options.cwd;
-    const agentDir = bootstrap.agentDir ?? getAgentDir();
-    const authStorage = bootstrap.authStorage ?? AuthStorage.create(join(agentDir, "auth.json"));
-    const settingsManager = SettingsManager.create(cwd, agentDir);
-    const modelRegistry = ModelRegistry.create(authStorage, join(agentDir, "models.json"));
-    const resourceLoader = typeof bootstrap.resourceLoader === "function"
-        ? await bootstrap.resourceLoader(cwd, agentDir)
-        : new DefaultResourceLoader({
-            ...(bootstrap.resourceLoader ?? {}),
-            cwd,
-            agentDir,
-            settingsManager,
-        });
-    await resourceLoader.reload();
-    const extensionsResult = resourceLoader.getExtensions();
-    for (const { name, config } of extensionsResult.runtime.pendingProviderRegistrations) {
-        modelRegistry.registerProvider(name, config);
-    }
-    extensionsResult.runtime.pendingProviderRegistrations = [];
-    const created = await createAgentSession({
-        cwd,
-        agentDir,
-        authStorage,
-        modelRegistry,
-        settingsManager,
-        resourceLoader,
-        sessionManager: options.sessionManager,
-        model: bootstrap.model,
-        thinkingLevel: bootstrap.thinkingLevel,
-        scopedModels: bootstrap.scopedModels,
-        tools: bootstrap.tools,
-        customTools: bootstrap.customTools,
-        sessionStartEvent: options.sessionStartEvent,
-    });
-    return {
-        ...created,
-        cwd,
-        agentDir,
-        authStorage,
-        modelRegistry,
-        settingsManager,
-        resourceLoader,
-        sessionManager: created.session.sessionManager,
-    };
-}
 function extractUserMessageText(content) {
     if (typeof content === "string") {
         return content;
@@ -73,23 +12,37 @@ function extractUserMessageText(content) {
         .join("");
 }
 /**
- * Stable wrapper around a replaceable AgentSession runtime.
+ * Owns the current AgentSession plus its cwd-bound services.
  *
- * Use this when your application needs `/new`, `/resume`, `/fork`, or import
- * behavior. After replacement, read `session` again and rebind any
- * session-local subscriptions or extension bindings.
+ * Session replacement methods tear down the current runtime first, then create
+ * and apply the next runtime. If creation fails, the error is propagated to the
+ * caller. The caller is responsible for user-facing error handling.
  */
-export class AgentSessionRuntimeHost {
-    constructor(bootstrap, runtime) {
-        this.bootstrap = bootstrap;
-        this.runtime = runtime;
+export class AgentSessionRuntime {
+    constructor(_session, _services, createRuntime, _diagnostics = [], _modelFallbackMessage) {
+        this._session = _session;
+        this._services = _services;
+        this.createRuntime = createRuntime;
+        this._diagnostics = _diagnostics;
+        this._modelFallbackMessage = _modelFallbackMessage;
+    }
+    get services() {
+        return this._services;
     }
-    /** The currently active session instance. Re-read this after runtime replacement. */
     get session() {
-        return this.runtime.session;
+        return this._session;
+    }
+    get cwd() {
+        return this._services.cwd;
+    }
+    get diagnostics() {
+        return this._diagnostics;
+    }
+    get modelFallbackMessage() {
+        return this._modelFallbackMessage;
     }
     async emitBeforeSwitch(reason, targetSessionFile) {
-        const runner = this.runtime.session.extensionRunner;
+        const runner = this.session.extensionRunner;
         if (!runner?.hasHandlers("session_before_switch")) {
             return { cancelled: false };
         }
@@ -101,7 +54,7 @@ export class AgentSessionRuntimeHost {
         return { cancelled: result?.cancel === true };
     }
     async emitBeforeFork(entryId) {
-        const runner = this.runtime.session.extensionRunner;
+        const runner = this.session.extensionRunner;
         if (!runner?.hasHandlers("session_before_fork")) {
             return { cancelled: false };
         }
@@ -111,128 +64,125 @@ export class AgentSessionRuntimeHost {
         });
         return { cancelled: result?.cancel === true };
     }
-    async replace(options) {
-        const nextRuntime = await createAgentSessionRuntime(this.bootstrap, options);
-        await emitSessionShutdownEvent(this.runtime.session.extensionRunner);
-        this.runtime.session.dispose();
-        if (process.cwd() !== nextRuntime.cwd) {
-            process.chdir(nextRuntime.cwd);
-        }
-        this.runtime = nextRuntime;
+    async teardownCurrent() {
+        await emitSessionShutdownEvent(this.session.extensionRunner);
+        this.session.dispose();
+    }
+    apply(result) {
+        if (process.cwd() !== result.services.cwd) {
+            process.chdir(result.services.cwd);
+        }
+        this._session = result.session;
+        this._services = result.services;
+        this._diagnostics = result.diagnostics;
+        this._modelFallbackMessage = result.modelFallbackMessage;
     }
-    /**
-     * Replace the active runtime with one opened from an existing session file.
-     *
-     * Emits `session_before_switch` before replacement and returns
-     * `{ cancelled: true }` if an extension vetoes the switch.
-     */
     async switchSession(sessionPath) {
         const beforeResult = await this.emitBeforeSwitch("resume", sessionPath);
         if (beforeResult.cancelled) {
             return beforeResult;
         }
-        const previousSessionFile = this.runtime.session.sessionFile;
+        const previousSessionFile = this.session.sessionFile;
         const sessionManager = SessionManager.open(sessionPath);
-        await this.replace({
+        await this.teardownCurrent();
+        this.apply(await this.createRuntime({
             cwd: sessionManager.getCwd(),
+            agentDir: this.services.agentDir,
             sessionManager,
             sessionStartEvent: { type: "session_start", reason: "resume", previousSessionFile },
-        });
+        }));
         return { cancelled: false };
     }
-    /**
-     * Replace the active runtime with a fresh session in the current cwd.
-     *
-     * `setup` runs after replacement against the new session manager, which lets
-     * callers seed entries before normal use begins.
-     */
     async newSession(options) {
         const beforeResult = await this.emitBeforeSwitch("new");
         if (beforeResult.cancelled) {
             return beforeResult;
         }
-        const previousSessionFile = this.runtime.session.sessionFile;
-        const sessionDir = this.runtime.sessionManager.getSessionDir();
-        const sessionManager = SessionManager.create(this.runtime.cwd, sessionDir);
+        const previousSessionFile = this.session.sessionFile;
+        const sessionDir = this.session.sessionManager.getSessionDir();
+        const sessionManager = SessionManager.create(this.cwd, sessionDir);
         if (options?.parentSession) {
             sessionManager.newSession({ parentSession: options.parentSession });
         }
-        await this.replace({
-            cwd: this.runtime.cwd,
+        await this.teardownCurrent();
+        this.apply(await this.createRuntime({
+            cwd: this.cwd,
+            agentDir: this.services.agentDir,
             sessionManager,
             sessionStartEvent: { type: "session_start", reason: "new", previousSessionFile },
-        });
+        }));
         if (options?.setup) {
-            await options.setup(this.runtime.sessionManager);
-            this.runtime.session.agent.state.messages = this.runtime.sessionManager.buildSessionContext().messages;
+            await options.setup(this.session.sessionManager);
+            this.session.agent.state.messages = this.session.sessionManager.buildSessionContext().messages;
         }
         return { cancelled: false };
     }
-    /**
-     * Replace the active runtime with a fork rooted at the given user-message
-     * entry.
-     *
-     * Returns the selected user text so UIs can restore it into the editor after
-     * the fork completes.
-     */
     async fork(entryId) {
         const beforeResult = await this.emitBeforeFork(entryId);
         if (beforeResult.cancelled) {
             return { cancelled: true };
         }
-        const selectedEntry = this.runtime.sessionManager.getEntry(entryId);
+        const selectedEntry = this.session.sessionManager.getEntry(entryId);
         if (!selectedEntry || selectedEntry.type !== "message" || selectedEntry.message.role !== "user") {
             throw new Error("Invalid entry ID for forking");
         }
-        const previousSessionFile = this.runtime.session.sessionFile;
+        const previousSessionFile = this.session.sessionFile;
         const selectedText = extractUserMessageText(selectedEntry.message.content);
-        if (this.runtime.sessionManager.isPersisted()) {
-            const currentSessionFile = this.runtime.session.sessionFile;
-            const sessionDir = this.runtime.sessionManager.getSessionDir();
+        if (this.session.sessionManager.isPersisted()) {
+            const currentSessionFile = this.session.sessionFile;
+            if (!currentSessionFile) {
+                throw new Error("Persisted session is missing a session file");
+            }
+            const sessionDir = this.session.sessionManager.getSessionDir();
             if (!selectedEntry.parentId) {
-                const sessionManager = SessionManager.create(this.runtime.cwd, sessionDir);
+                const sessionManager = SessionManager.create(this.cwd, sessionDir);
                 sessionManager.newSession({ parentSession: currentSessionFile });
-                await this.replace({
-                    cwd: this.runtime.cwd,
+                await this.teardownCurrent();
+                this.apply(await this.createRuntime({
+                    cwd: this.cwd,
+                    agentDir: this.services.agentDir,
                     sessionManager,
                     sessionStartEvent: { type: "session_start", reason: "fork", previousSessionFile },
-                });
+                }));
                 return { cancelled: false, selectedText };
             }
             const sourceManager = SessionManager.open(currentSessionFile, sessionDir);
             const forkedSessionPath = sourceManager.createBranchedSession(selectedEntry.parentId);
+            if (!forkedSessionPath) {
+                throw new Error("Failed to create forked session");
+            }
             const sessionManager = SessionManager.open(forkedSessionPath, sessionDir);
-            await this.replace({
+            await this.teardownCurrent();
+            this.apply(await this.createRuntime({
                 cwd: sessionManager.getCwd(),
+                agentDir: this.services.agentDir,
                 sessionManager,
                 sessionStartEvent: { type: "session_start", reason: "fork", previousSessionFile },
-            });
+            }));
             return { cancelled: false, selectedText };
         }
-        const sessionManager = this.runtime.sessionManager;
+        const sessionManager = this.session.sessionManager;
         if (!selectedEntry.parentId) {
-            sessionManager.newSession({ parentSession: this.runtime.session.sessionFile });
+            sessionManager.newSession({ parentSession: this.session.sessionFile });
         }
         else {
             sessionManager.createBranchedSession(selectedEntry.parentId);
         }
-        await this.replace({
-            cwd: this.runtime.cwd,
+        await this.teardownCurrent();
+        this.apply(await this.createRuntime({
+            cwd: this.cwd,
+            agentDir: this.services.agentDir,
             sessionManager,
             sessionStartEvent: { type: "session_start", reason: "fork", previousSessionFile },
-        });
+        }));
         return { cancelled: false, selectedText };
     }
-    /**
-     * Import a JSONL session file into the current session directory and replace
-     * the active runtime with the imported session.
-     */
     async importFromJsonl(inputPath) {
         const resolvedPath = resolve(inputPath);
         if (!existsSync(resolvedPath)) {
             throw new Error(`File not found: ${resolvedPath}`);
         }
-        const sessionDir = this.runtime.sessionManager.getSessionDir();
+        const sessionDir = this.session.sessionManager.getSessionDir();
         if (!existsSync(sessionDir)) {
             mkdirSync(sessionDir, { recursive: true });
         }
@@ -241,22 +191,37 @@ export class AgentSessionRuntimeHost {
         if (beforeResult.cancelled) {
             return beforeResult;
         }
-        const previousSessionFile = this.runtime.session.sessionFile;
+        const previousSessionFile = this.session.sessionFile;
         if (resolve(destinationPath) !== resolvedPath) {
             copyFileSync(resolvedPath, destinationPath);
         }
         const sessionManager = SessionManager.open(destinationPath, sessionDir);
-        await this.replace({
+        await this.teardownCurrent();
+        this.apply(await this.createRuntime({
             cwd: sessionManager.getCwd(),
+            agentDir: this.services.agentDir,
             sessionManager,
             sessionStartEvent: { type: "session_start", reason: "resume", previousSessionFile },
-        });
+        }));
         return { cancelled: false };
     }
-    /** Emit session shutdown for the active runtime and dispose it permanently. */
     async dispose() {
-        await emitSessionShutdownEvent(this.runtime.session.extensionRunner);
-        this.runtime.session.dispose();
+        await emitSessionShutdownEvent(this.session.extensionRunner);
+        this.session.dispose();
+    }
+}
+/**
+ * Create the initial runtime from a runtime factory and initial session target.
+ *
+ * The same factory is stored on the returned AgentSessionRuntime and reused for
+ * later /new, /resume, /fork, and import flows.
+ */
+export async function createAgentSessionRuntime(createRuntime, options) {
+    const result = await createRuntime(options);
+    if (process.cwd() !== result.services.cwd) {
+        process.chdir(result.services.cwd);
     }
+    return new AgentSessionRuntime(result.session, result.services, createRuntime, result.diagnostics, result.modelFallbackMessage);
 }
+export { createAgentSessionFromServices, createAgentSessionServices, } from "./agent-session-services.js";
 //# sourceMappingURL=agent-session-runtime.js.map