@ai-setting/roy-agent-core 1.0.0

package/src/env/workflow/storage/sqlite.ts

@@ -0,0 +1,163 @@
+/**
+ * @fileoverview SQLite Storage for Workflow Component
+ * 
+ * Uses Bun's built-in bun:sqlite module for persistence.
+ * 
+ * 存储路径：~/.local/share/roy-agent/workflows.db
+ * 
+ * Note: After session-based refactoring, workflow runs are managed via SessionComponent.
+ * Only the workflows table is used here for storing workflow definitions.
+ */
+
+import { join } from "path";
+import { createLogger } from "../../log-trace/logger";
+
+const logger = createLogger("workflow:sqlite");
+
+// ============================================================================
+// Database Path
+// ============================================================================
+
+/**
+ * Get default data directory
+ */
+function getDefaultDataDir(): string {
+  const home = process.env.HOME || process.env.USERPROFILE || "/tmp";
+  return join(home, ".local", "share", "roy-agent");
+}
+
+/**
+ * Get the database file path
+ * Default: ~/.local/share/roy-agent/workflows.db
+ */
+export function getDatabasePath(): string {
+  return join(getDefaultDataDir(), "workflows.db");
+}
+
+// ============================================================================
+// Database Singleton
+// ============================================================================
+
+let dbInstance: any = null;
+
+/**
+ * Get the singleton database instance
+ * @param path - Optional custom database path. If not provided, uses default path.
+ *               Use ':memory:' for in-memory database (testing)
+ */
+export function getDatabase(path?: string): any {
+  if (!dbInstance) {
+    const dbPath = path ?? getDatabasePath();
+    
+    // 确保目录存在
+    const dbDir = dbPath.substring(0, dbPath.lastIndexOf('/'));
+    if (dbDir && dbDir !== '/') {
+      try {
+        const { mkdirSync } = require("fs");
+        mkdirSync(dbDir, { recursive: true });
+      } catch {
+        // 目录可能已存在，忽略错误
+      }
+    }
+    
+    // Import bun:sqlite using dynamic require
+    // eslint-disable-next-line @typescript-eslint/no-require-imports
+    const { Database } = require("bun:sqlite");
+    
+    dbInstance = new Database(dbPath);
+    
+    // Enable WAL mode for better performance (if available)
+    // Some SQLite implementations may not have pragma method
+    try {
+      if (typeof dbInstance.pragma === 'function') {
+        dbInstance.pragma("journal_mode = WAL");
+      }
+    } catch {
+      // pragma is optional, continue without it
+      logger?.debug("WAL mode not available, using default journal mode");
+    }
+  }
+  return dbInstance;
+}
+
+/**
+ * Close the database connection
+ */
+export function closeDatabase(): void {
+  if (dbInstance) {
+    dbInstance.close();
+    dbInstance = null;
+  }
+}
+
+// Re-export Database type for Bun
+// Note: bun:sqlite types may not be available in all environments
+// Use dynamic import at runtime and cast as needed
+export type BunDatabase = {
+  exec(sql: string): void;
+  query<T>(sql: string): { all(): T[]; get(...params: any[]): T | null };
+  run(sql: string, ...params: any[]): { lastInsertRowid: number; changes: number };
+  close(): void;
+  pragma?: (sql: string) => void;
+};
+
+// ============================================================================
+// Table Initialization
+// ============================================================================
+
+/**
+ * Initialize all required tables
+ * 
+ * Note: Only workflows table is created. Workflow runs are managed via SessionComponent
+ * using the session/message storage, following the session-based design.
+ */
+export function initializeTables(): void {
+  const db = getDatabase();
+  
+  // Create workflows table for storing workflow definitions
+  db.exec(`
+    CREATE TABLE IF NOT EXISTS workflows (
+      id TEXT PRIMARY KEY,
+      name TEXT NOT NULL UNIQUE,
+      version TEXT NOT NULL DEFAULT '1.0',
+      description TEXT,
+      definition TEXT NOT NULL,
+      config TEXT NOT NULL DEFAULT '{}',
+      metadata TEXT NOT NULL DEFAULT '{}',
+      tags TEXT NOT NULL DEFAULT '[]',
+      task_id INTEGER,
+      created_at TEXT NOT NULL,
+      updated_at TEXT NOT NULL
+    )
+  `);
+  
+  // Create indexes for better query performance
+  db.exec(`
+    CREATE INDEX IF NOT EXISTS idx_workflows_name ON workflows(name);
+    CREATE INDEX IF NOT EXISTS idx_workflows_tags ON workflows(tags);
+    CREATE INDEX IF NOT EXISTS idx_workflows_task_id ON workflows(task_id);
+  `);
+}
+
+// ============================================================================
+// Re-initialize (for testing)
+// ============================================================================
+
+/**
+ * Reset the database instance (for testing purposes)
+ * This is useful for tests that need a fresh database
+ */
+export function resetDatabase(): void {
+  closeDatabase();
+  dbInstance = null;
+}
+
+/**
+ * Initialize tables and return a ready database
+ * This is a convenience function that combines getDatabase and initializeTables
+ */
+export function initDatabase(path?: string): any {
+  const db = getDatabase(path);
+  initializeTables();
+  return db;
+}