@oh-my-pi/pi-coding-agent 12.18.1 → 12.19.0

package/src/session/session-manager.ts

@@ -4,8 +4,9 @@ import * as path from "node:path";
 import type { AgentMessage } from "@oh-my-pi/pi-agent-core";
 import type { ImageContent, Message, TextContent, Usage } from "@oh-my-pi/pi-ai";
 import { getTerminalId } from "@oh-my-pi/pi-tui";
-import { isEnoent, logger, parseJsonlLenient, Snowflake } from "@oh-my-pi/pi-utils";
+import { isEnoent, logger, parseJsonlLenient, Snowflake, toError } from "@oh-my-pi/pi-utils";
 import { getBlobsDir, getAgentDir as getDefaultAgentDir, getProjectDir } from "@oh-my-pi/pi-utils/dirs";
+import { ArtifactManager } from "./artifacts";
 import { type BlobPutResult, BlobStore, externalizeImageData, isBlobRef, resolveImageData } from "./blob-store";
 import {
 	type BashExecutionMessage,
@@ -216,6 +217,10 @@ export type ReadonlySessionManager = Pick<
 	| "getSessionDir"
 	| "getSessionId"
 	| "getSessionFile"
+	| "getArtifactsDir"
+	| "allocateArtifactPath"
+	| "saveArtifact"
+	| "getArtifactPath"
 	| "getLeafId"
 	| "getLeafEntry"
 	| "getEntry"
@@ -379,10 +384,6 @@ export function getLatestCompactionEntry(entries: SessionEntry[]): CompactionEnt
 	return null;
 }
 
-function toError(value: unknown): Error {
-	return value instanceof Error ? value : new Error(String(value));
-}
-
 /**
  * Build the session context from entries using tree traversal.
  * If leafId is provided, walks from that entry to root.
@@ -1133,6 +1134,8 @@ export class SessionManager {
 	#persistChain: Promise<void> = Promise.resolve();
 	#persistError: Error | undefined;
 	#persistErrorReported = false;
+	#artifactManager: ArtifactManager | null = null;
+	#artifactManagerSessionFile: string | null = null;
 	readonly #blobStore: BlobStore;
 
 	private constructor(
@@ -1534,6 +1537,67 @@ export class SessionManager {
 		return this.#sessionFile;
 	}
 
+	/**
+	 * Returns the session artifacts directory path (session file path without .jsonl).
+	 * Returns null when the session is not persisted to a file.
+	 */
+	getArtifactsDir(): string | null {
+		const sessionFile = this.#sessionFile;
+		return sessionFile ? sessionFile.slice(0, -6) : null;
+	}
+
+	/**
+	 * Returns an artifact manager bound to the current session file.
+	 * Recreates the manager when the active session file changes.
+	 */
+	#getOrCreateArtifactManager(): ArtifactManager | null {
+		const sessionFile = this.#sessionFile;
+		if (!sessionFile) {
+			this.#artifactManager = null;
+			this.#artifactManagerSessionFile = null;
+			return null;
+		}
+
+		if (this.#artifactManager && this.#artifactManagerSessionFile === sessionFile) {
+			return this.#artifactManager;
+		}
+
+		const manager = new ArtifactManager(sessionFile);
+		this.#artifactManager = manager;
+		this.#artifactManagerSessionFile = sessionFile;
+		return manager;
+	}
+
+	/**
+	 * Allocate a new artifact path and ID for the current session.
+	 * Returns an empty object when the session is not persisted.
+	 */
+	async allocateArtifactPath(toolType: string): Promise<{ id?: string; path?: string }> {
+		const manager = this.#getOrCreateArtifactManager();
+		if (!manager) return {};
+		return manager.allocatePath(toolType);
+	}
+
+	/**
+	 * Save artifact content under the current session and return artifact ID.
+	 * Returns undefined when the session is not persisted.
+	 */
+	async saveArtifact(content: string, toolType: string): Promise<string | undefined> {
+		const manager = this.#getOrCreateArtifactManager();
+		if (!manager) return undefined;
+		return manager.save(content, toolType);
+	}
+
+	/**
+	 * Resolve an artifact ID to an on-disk path for the current session.
+	 * Returns null when missing or when the session is not persisted.
+	 */
+	async getArtifactPath(id: string): Promise<string | null> {
+		const manager = this.#getOrCreateArtifactManager();
+		if (!manager) return null;
+		return manager.getPath(id);
+	}
+
 	getSessionName(): string | undefined {
 		return this.#sessionName;
 	}

package/src/session/session-storage.ts

@@ -1,6 +1,6 @@
 import * as fs from "node:fs";
 import * as path from "node:path";
-import { isEnoent } from "@oh-my-pi/pi-utils";
+import { isEnoent, toError } from "@oh-my-pi/pi-utils";
 
 export interface SessionStorageStat {
 	size: number;
@@ -32,10 +32,6 @@ export interface SessionStorage {
 	openWriter(path: string, options?: { flags?: "a" | "w"; onError?: (err: Error) => void }): SessionStorageWriter;
 }
 
-function toError(value: unknown): Error {
-	return value instanceof Error ? value : new Error(String(value));
-}
-
 // FinalizationRegistry to clean up leaked file descriptors
 const writerRegistry = new FinalizationRegistry<number>(fd => {
 	try {