@ugm/desiagent 0.1.21

package/dist/db/client.js

@@ -0,0 +1,178 @@
+import { drizzle } from 'drizzle-orm/bun-sqlite';
+import { Database } from 'bun:sqlite';
+import * as schema from './schema.js';
+import { getLogger } from '../util/logger.js';
+import { existsSync, mkdirSync } from 'fs';
+import { dirname } from 'path';
+import { DatabaseError, InitializationError } from '../errors/index.js';
+let dbInstance = null;
+/**
+ * SQL statements to create all tables based on the schema
+ */
+const CREATE_TABLES_SQL = `
+-- Agents table
+CREATE TABLE IF NOT EXISTS agents (
+  id TEXT PRIMARY KEY,
+  name TEXT NOT NULL,
+  version TEXT NOT NULL,
+  prompt_template TEXT NOT NULL,
+  provider TEXT,
+  model TEXT,
+  active INTEGER NOT NULL DEFAULT 0,
+  metadata TEXT,
+  created_at INTEGER NOT NULL DEFAULT (unixepoch()),
+  updated_at INTEGER NOT NULL DEFAULT (unixepoch())
+);
+
+-- Unique index for agent name+version
+CREATE UNIQUE INDEX IF NOT EXISTS idx_name_version ON agents(name, version);
+
+-- Partial unique index for active agent per name
+CREATE UNIQUE INDEX IF NOT EXISTS idx_active_agent ON agents(name) WHERE active = 1;
+
+-- DAGs table
+CREATE TABLE IF NOT EXISTS dags (
+  id TEXT PRIMARY KEY,
+  status TEXT NOT NULL,
+  result TEXT,
+  usage TEXT,
+  generation_stats TEXT,
+  attempts INTEGER NOT NULL DEFAULT 0,
+  params TEXT,
+  agent_name TEXT,
+  dag_title TEXT,
+  cron_schedule TEXT,
+  schedule_active INTEGER NOT NULL DEFAULT 0,
+  created_at INTEGER NOT NULL DEFAULT (unixepoch()),
+  updated_at INTEGER NOT NULL DEFAULT (unixepoch())
+);
+
+-- DAG Executions table
+CREATE TABLE IF NOT EXISTS dag_executions (
+  id TEXT PRIMARY KEY,
+  dag_id TEXT NOT NULL REFERENCES dags(id) ON DELETE CASCADE,
+  status TEXT NOT NULL,
+  execution_results TEXT,
+  failure_reason TEXT,
+  created_at INTEGER NOT NULL DEFAULT (unixepoch()),
+  updated_at INTEGER NOT NULL DEFAULT (unixepoch())
+);
+
+-- DAG Sub-Steps table
+CREATE TABLE IF NOT EXISTS dag_sub_steps (
+  id TEXT PRIMARY KEY,
+  execution_id TEXT NOT NULL REFERENCES dag_executions(id) ON DELETE CASCADE,
+  node_id TEXT NOT NULL,
+  "index" INTEGER NOT NULL,
+  type TEXT NOT NULL CHECK(type IN ('action', 'result', 'decision')),
+  content TEXT NOT NULL,
+  created_at INTEGER NOT NULL DEFAULT (unixepoch())
+);
+`;
+/**
+ * Initialize database tables if they don't exist
+ */
+function initializeTables(sqlite, logger) {
+    try {
+        sqlite.exec(CREATE_TABLES_SQL);
+        logger.info('Database tables initialized');
+    }
+    catch (error) {
+        throw new DatabaseError(`Failed to initialize database tables: ${error instanceof Error ? error.message : String(error)}`, 'initializeTables', error instanceof Error ? error : undefined);
+    }
+}
+/**
+ * Create and initialize database connection
+ */
+function createDatabase(dbPath) {
+    const logger = getLogger();
+    try {
+        // Ensure data directory exists
+        const dbDir = dirname(dbPath);
+        if (!existsSync(dbDir)) {
+            mkdirSync(dbDir, { recursive: true });
+            logger.info(`Created database directory: ${dbDir}`);
+        }
+        // Check if database file exists (to determine if we need to create tables)
+        const isNewDatabase = !existsSync(dbPath);
+        // Open SQLite database using bun:sqlite
+        const sqlite = new Database(dbPath);
+        // Enable foreign keys and WAL mode
+        sqlite.exec('PRAGMA journal_mode = WAL;');
+        sqlite.exec('PRAGMA foreign_keys = ON;');
+        // Initialize tables if this is a new database
+        if (isNewDatabase) {
+            logger.info('New database detected, creating tables...');
+            initializeTables(sqlite, logger);
+        }
+        else {
+            // For existing databases, ensure tables exist (handles partial initialization)
+            initializeTables(sqlite, logger);
+        }
+        const db = drizzle(sqlite, { schema });
+        logger.info(`Database initialized: ${dbPath}`);
+        return db;
+    }
+    catch (error) {
+        const logger = getLogger();
+        logger.error('Failed to initialize database');
+        throw new InitializationError(`Failed to initialize database: ${error instanceof Error ? error.message : String(error)}`, 'DatabaseClient', error instanceof Error ? error : undefined);
+    }
+}
+/**
+ * Get or create the global database instance
+ */
+export function getDatabase(dbPath) {
+    if (!dbInstance) {
+        dbInstance = createDatabase(dbPath);
+    }
+    return dbInstance;
+}
+/**
+ * Close database connection
+ */
+export function closeDatabase() {
+    if (dbInstance) {
+        try {
+            const logger = getLogger();
+            logger.info('Closing database connection');
+            // bun:sqlite closes automatically when garbage collected
+            // but we can mark it as null
+            dbInstance = null;
+        }
+        catch (error) {
+            const logger = getLogger();
+            logger.error('Error closing database');
+        }
+    }
+}
+/**
+ * Run database migrations
+ * Note: For Phase 2, this is a placeholder.
+ * In production, migrations should be managed via drizzle-kit
+ */
+export async function runMigrations(db) {
+    try {
+        void db; // Suppress unused variable warning
+        const logger = getLogger();
+        logger.info('Migrations check complete (no migrations yet)');
+        // TODO: Implement migration system in Phase 4
+    }
+    catch (error) {
+        throw new DatabaseError(`Migration failed: ${error instanceof Error ? error.message : String(error)}`, 'runMigrations', error instanceof Error ? error : undefined);
+    }
+}
+/**
+ * Execute a query with error handling
+ */
+export async function withDatabase(db, operation) {
+    try {
+        return await operation(db);
+    }
+    catch (error) {
+        const logger = getLogger();
+        logger.error('Database operation failed');
+        throw new DatabaseError(`Database operation failed: ${error instanceof Error ? error.message : String(error)}`, 'query', error instanceof Error ? error : undefined);
+    }
+}
+//# sourceMappingURL=client.js.map

package/dist/db/client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.js","sourceRoot":"","sources":["../../src/db/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,wBAAwB,CAAC;AACjD,OAAO,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AACtC,OAAO,KAAK,MAAM,MAAM,aAAa,CAAC;AACtC,OAAO,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AAC9C,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAC3C,OAAO,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AAE/B,OAAO,EAAE,aAAa,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AAIxE,IAAI,UAAU,GAAqB,IAAI,CAAC;AAExC;;GAEG;AACH,MAAM,iBAAiB,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA2DzB,CAAC;AAEF;;GAEG;AACH,SAAS,gBAAgB,CAAC,MAAW,EAAE,MAAW;IAChD,IAAI,CAAC;QACH,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC/B,MAAM,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;IAC7C,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,aAAa,CACrB,yCAAyC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,EACjG,kBAAkB,EAClB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAC3C,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAS,cAAc,CAAC,MAAc;IACpC,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAE3B,IAAI,CAAC;QACH,+BAA+B;QAC/B,MAAM,KAAK,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;QAC9B,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YACvB,SAAS,CAAC,KAAK,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACtC,MAAM,CAAC,IAAI,CAAC,+BAA+B,KAAK,EAAE,CAAC,CAAC;QACtD,CAAC;QAED,2EAA2E;QAC3E,MAAM,aAAa,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QAE1C,wCAAwC;QACxC,MAAM,MAAM,GAAG,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC;QAEpC,mCAAmC;QACnC,MAAM,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;QAC1C,MAAM,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;QAEzC,8CAA8C;QAC9C,IAAI,aAAa,EAAE,CAAC;YAClB,MAAM,CAAC,IAAI,CAAC,2CAA2C,CAAC,CAAC;YACzD,gBAAgB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACnC,CAAC;aAAM,CAAC;YACN,+EAA+E;YAC/E,gBAAgB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACnC,CAAC;QAED,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;QAEvC,MAAM,CAAC,IAAI,CAAC,yBAAyB,MAAM,EAAE,CAAC,CAAC;QAE/C,OAAO,EAAE,CAAC;IACZ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;QAC3B,MAAM,CAAC,KAAK,CAAC,+BAA+B,CAAC,CAAC;QAC9C,MAAM,IAAI,mBAAmB,CAC3B,kCAAkC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,EAC1F,gBAAgB,EAChB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAC3C,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,WAAW,CAAC,MAAc;IACxC,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,UAAU,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;IACtC,CAAC;IACD,OAAO,UAAU,CAAC;AACpB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,aAAa;IAC3B,IAAI,UAAU,EAAE,CAAC;QACf,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;YAC3B,MAAM,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;YAC3C,yDAAyD;YACzD,6BAA6B;YAC7B,UAAU,GAAG,IAAI,CAAC;QACpB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;YAC3B,MAAM,CAAC,KAAK,CAAC,wBAAwB,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;AACH,CAAC;AAED;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,aAAa,CAAC,EAAa;IAC/C,IAAI,CAAC;QACH,KAAK,EAAE,CAAC,CAAC,mCAAmC;QAC5C,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;QAC3B,MAAM,CAAC,IAAI,CAAC,+CAA+C,CAAC,CAAC;QAC7D,8CAA8C;IAChD,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,aAAa,CACrB,qBAAqB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,EAC7E,eAAe,EACf,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAC3C,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,EAAa,EACb,SAAwC;IAExC,IAAI,CAAC;QACH,OAAO,MAAM,SAAS,CAAC,EAAE,CAAC,CAAC;IAC7B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;QAC3B,MAAM,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC;QAC1C,MAAM,IAAI,aAAa,CACrB,8BAA8B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,EACtF,OAAO,EACP,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAC3C,CAAC;IACJ,CAAC;AACH,CAAC"}