@poncho-ai/harness 0.35.0 → 0.36.1

package/src/storage/engine.ts

@@ -0,0 +1,106 @@
+import type {
+  Conversation,
+  ConversationSummary,
+  PendingSubagentResult,
+} from "../state.js";
+import type { MainMemory } from "../memory.js";
+import type { TodoItem } from "../todo-tools.js";
+import type { Reminder } from "../reminder-store.js";
+
+// ---------------------------------------------------------------------------
+// VFS types
+// ---------------------------------------------------------------------------
+
+export interface VfsStat {
+  type: "file" | "directory" | "symlink";
+  size: number;
+  mode: number;
+  mimeType?: string;
+  symlinkTarget?: string;
+  createdAt: number;
+  updatedAt: number;
+}
+
+export interface VfsDirEntry {
+  name: string;
+  type: "file" | "directory" | "symlink";
+}
+
+// ---------------------------------------------------------------------------
+// StorageEngine – single interface replacing all KV stores
+// ---------------------------------------------------------------------------
+
+export interface StorageEngine {
+  /** Run migrations and prepare the storage backend. */
+  initialize(): Promise<void>;
+  /** Gracefully release resources. */
+  close(): Promise<void>;
+
+  // --- Conversations (replaces ConversationStore) ---
+  conversations: {
+    list(ownerId?: string, tenantId?: string | null): Promise<ConversationSummary[]>;
+    get(conversationId: string): Promise<Conversation | undefined>;
+    create(ownerId?: string, title?: string, tenantId?: string | null): Promise<Conversation>;
+    update(conversation: Conversation): Promise<void>;
+    rename(conversationId: string, title: string): Promise<Conversation | undefined>;
+    delete(conversationId: string): Promise<boolean>;
+    search(query: string, tenantId?: string | null): Promise<ConversationSummary[]>;
+    appendSubagentResult(
+      conversationId: string,
+      result: PendingSubagentResult,
+    ): Promise<void>;
+    clearCallbackLock(conversationId: string): Promise<Conversation | undefined>;
+  };
+
+  // --- Memory (replaces MemoryStore) ---
+  memory: {
+    get(tenantId?: string | null): Promise<MainMemory>;
+    update(content: string, tenantId?: string | null): Promise<MainMemory>;
+  };
+
+  // --- Todos (replaces TodoStore) ---
+  todos: {
+    get(conversationId: string): Promise<TodoItem[]>;
+    set(conversationId: string, todos: TodoItem[]): Promise<void>;
+  };
+
+  // --- Reminders (replaces ReminderStore) ---
+  reminders: {
+    list(tenantId?: string | null): Promise<Reminder[]>;
+    create(input: {
+      task: string;
+      scheduledAt: number;
+      timezone?: string;
+      conversationId: string;
+      ownerId?: string;
+      tenantId?: string | null;
+    }): Promise<Reminder>;
+    cancel(id: string): Promise<Reminder>;
+    delete(id: string): Promise<void>;
+  };
+
+  // --- Virtual Filesystem (replaces UploadStore + new VFS) ---
+  vfs: {
+    readFile(tenantId: string, path: string): Promise<Uint8Array>;
+    writeFile(
+      tenantId: string,
+      path: string,
+      content: Uint8Array,
+      mimeType?: string,
+    ): Promise<void>;
+    appendFile(tenantId: string, path: string, content: Uint8Array): Promise<void>;
+    deleteFile(tenantId: string, path: string): Promise<void>;
+    deleteDir(tenantId: string, path: string, recursive?: boolean): Promise<void>;
+    stat(tenantId: string, path: string): Promise<VfsStat | undefined>;
+    readdir(tenantId: string, path: string): Promise<VfsDirEntry[]>;
+    mkdir(tenantId: string, path: string, recursive?: boolean): Promise<void>;
+    rename(tenantId: string, oldPath: string, newPath: string): Promise<void>;
+    chmod(tenantId: string, path: string, mode: number): Promise<void>;
+    utimes(tenantId: string, path: string, mtime: Date): Promise<void>;
+    symlink(tenantId: string, target: string, linkPath: string): Promise<void>;
+    readlink(tenantId: string, path: string): Promise<string>;
+    lstat(tenantId: string, path: string): Promise<VfsStat | undefined>;
+    listAllPaths(tenantId: string): string[];
+    getUsage(tenantId: string): Promise<{ fileCount: number; totalBytes: number }>;
+  };
+}

package/src/storage/index.ts

@@ -0,0 +1,59 @@
+// ---------------------------------------------------------------------------
+// Storage engine factory + re-exports
+// ---------------------------------------------------------------------------
+
+export type { StorageEngine, VfsDirEntry, VfsStat } from "./engine.js";
+export { InMemoryEngine } from "./memory-engine.js";
+export { SqliteEngine } from "./sqlite-engine.js";
+export { PostgresEngine } from "./postgres-engine.js";
+
+import type { StorageEngine } from "./engine.js";
+import { InMemoryEngine } from "./memory-engine.js";
+import { SqliteEngine } from "./sqlite-engine.js";
+import { PostgresEngine } from "./postgres-engine.js";
+
+export type StorageProvider = "memory" | "sqlite" | "postgresql" | "local";
+
+export interface StorageFactoryOptions {
+  provider?: StorageProvider;
+  workingDir: string;
+  agentId: string;
+  /** Env var name for the PostgreSQL connection URL (default: DATABASE_URL). */
+  urlEnv?: string;
+  /** Override the SQLite database file path. */
+  dbPath?: string;
+}
+
+export function createStorageEngine(options: StorageFactoryOptions): StorageEngine {
+  const provider = options.provider ?? "sqlite";
+
+  switch (provider) {
+    case "memory":
+      return new InMemoryEngine(options.agentId);
+
+    case "local":
+    case "sqlite":
+      return new SqliteEngine({
+        workingDir: options.workingDir,
+        agentId: options.agentId,
+        dbPath: options.dbPath,
+      });
+
+    case "postgresql":
+      return new PostgresEngine({
+        agentId: options.agentId,
+        urlEnv: options.urlEnv,
+      });
+
+    default: {
+      const deprecated = ["redis", "upstash", "dynamodb"];
+      if (deprecated.includes(provider)) {
+        throw new Error(
+          `Storage provider '${provider}' is no longer supported. ` +
+            `Please migrate to 'sqlite' or 'postgresql'.`,
+        );
+      }
+      throw new Error(`Unknown storage provider: ${provider}`);
+    }
+  }
+}