agentboss 0.1.0

package/server/db/connection.js

@@ -0,0 +1,175 @@
+/**
+ * Database connection manager for Agent Boss.
+ *
+ * Opens (or creates) the boss.db SQLite database via sql.js and exposes a
+ * singleton Promise-based accessor.  The in-memory database is periodically
+ * flushed to disk and saved on close.
+ *
+ * Usage:
+ *   const { getDb, saveDb, closeDb } = require('./connection');
+ *   const db = await getDb();
+ *   db.run('SELECT 1');
+ *
+ * @author Felix
+ */
+
+const fs = require('fs');
+const path = require('path');
+const initSqlJs = require('sql.js');
+const { initDatabase, getDbPath } = require('./schema');
+
+// ---------------------------------------------------------------------------
+// Internal state
+// ---------------------------------------------------------------------------
+
+/** @type {Promise<import('sql.js').Database> | null} */
+let _dbPromise = null;
+
+/** @type {import('sql.js').Database | null} */
+let _dbInstance = null;
+
+/** @type {NodeJS.Timeout | null} */
+let _autoSaveTimer = null;
+
+/** Auto-save interval in milliseconds (default: 30 s). */
+const AUTO_SAVE_INTERVAL_MS = 30_000;
+
+// ---------------------------------------------------------------------------
+// Helpers
+// ---------------------------------------------------------------------------
+
+/**
+ * Ensure the directory for the database file exists.
+ *
+ * @param {string} dbPath Absolute path to the database file.
+ */
+function ensureDir(dbPath) {
+  const dir = path.dirname(dbPath);
+  if (!fs.existsSync(dir)) {
+    fs.mkdirSync(dir, { recursive: true });
+  }
+}
+
+/**
+ * Write the current in-memory database to disk.
+ *
+ * Safe to call at any time — if there is no active database instance the
+ * call is a no-op.
+ */
+function saveDb() {
+  if (!_dbInstance) {
+    return;
+  }
+  const dbPath = getDbPath();
+  ensureDir(dbPath);
+  const data = _dbInstance.export();
+  fs.writeFileSync(dbPath, Buffer.from(data));
+}
+
+/**
+ * Start the periodic auto-save timer.  If a timer is already running it is
+ * left untouched.
+ */
+function startAutoSave() {
+  if (_autoSaveTimer) {
+    return;
+  }
+  _autoSaveTimer = setInterval(() => {
+    try {
+      saveDb();
+    } catch (_err) {
+      // Swallow — auto-save is best-effort.  The caller can always invoke
+      // saveDb() explicitly for error handling.
+    }
+  }, AUTO_SAVE_INTERVAL_MS);
+
+  // Allow the Node process to exit even if the timer is still active.
+  if (_autoSaveTimer.unref) {
+    _autoSaveTimer.unref();
+  }
+}
+
+// ---------------------------------------------------------------------------
+// Public API
+// ---------------------------------------------------------------------------
+
+/**
+ * Return the singleton sql.js Database instance, initialising it on the
+ * first call.
+ *
+ * - If `~/.agent-boss/boss.db` exists the file is loaded into memory.
+ * - Otherwise a fresh in-memory database is created.
+ * - The full schema from `schema.js` is applied (CREATE IF NOT EXISTS).
+ * - An auto-save timer is started to periodically flush to disk.
+ *
+ * @returns {Promise<import('sql.js').Database>}  Resolves to the database.
+ */
+function getDb() {
+  if (_dbPromise) {
+    return _dbPromise;
+  }
+
+  _dbPromise = (async () => {
+    const SQL = await initSqlJs();
+    const dbPath = getDbPath();
+
+    let db;
+    if (fs.existsSync(dbPath)) {
+      const fileBuffer = fs.readFileSync(dbPath);
+      db = new SQL.Database(fileBuffer);
+    } else {
+      ensureDir(dbPath);
+      db = new SQL.Database();
+    }
+
+    // Apply / migrate schema (all statements are IF NOT EXISTS).
+    initDatabase(db);
+
+    _dbInstance = db;
+
+    // Persist the freshly-created / migrated database immediately so the
+    // file on disk is always up-to-date with the latest schema.
+    saveDb();
+
+    startAutoSave();
+
+    return db;
+  })();
+
+  // If initialisation fails, reset so the next call retries.
+  _dbPromise.catch(() => {
+    _dbPromise = null;
+    _dbInstance = null;
+  });
+
+  return _dbPromise;
+}
+
+/**
+ * Flush the database to disk and release all resources.
+ *
+ * After calling this the next `getDb()` call will re-open the database from
+ * the saved file.
+ */
+async function closeDb() {
+  if (_autoSaveTimer) {
+    clearInterval(_autoSaveTimer);
+    _autoSaveTimer = null;
+  }
+
+  if (_dbPromise) {
+    try {
+      const db = await _dbPromise;
+      // Final save before close.
+      saveDb();
+      db.close();
+    } catch (_err) {
+      // If getDb() itself had failed there is nothing to close.
+    }
+  }
+
+  _dbPromise = null;
+  _dbInstance = null;
+}
+
+module.exports = { getDb, saveDb, closeDb };