@vladimirven/openswe 0.1.0

package/src/store/db.ts

@@ -0,0 +1,249 @@
+/**
+ * Database connection and initialization
+ *
+ * Manages the SQLite database connection using Bun's native bun:sqlite.
+ * Uses WAL mode for better concurrency.
+ */
+
+import { Database } from "bun:sqlite"
+import { dirname } from "path"
+import { getStateDatabasePath } from "../workspace/paths"
+
+// ============================================================================
+// Schema
+// ============================================================================
+
+// Import schema as raw text
+import schema from "./schema.sql" with { type: "text" }
+
+// ============================================================================
+// Manager Types
+// ============================================================================
+
+export interface DatabaseManager {
+  getDatabase: () => Database
+  initDatabase: (projectRoot: string) => Promise<void>
+  initDatabaseWithPath: (dbPath: string) => Promise<void>
+  closeDatabase: () => void
+  isDatabaseInitialized: () => boolean
+  getDatabasePath: () => string | null
+  getSchemaVersion: () => number
+  runMigrations: () => void
+  withTransaction: <T>(fn: () => T) => T
+}
+
+// ============================================================================
+// Database Manager Factory
+// ============================================================================
+
+export function createDatabaseManager(): DatabaseManager {
+  let db: Database | null = null
+  let currentDbPath: string | null = null
+
+  const getDatabase = (): Database => {
+    if (!db) {
+      throw new Error(
+        "Database not initialized. Call initDatabase() first."
+      )
+    }
+    return db
+  }
+
+  const applySchema = (): void => {
+    if (!db) {
+      throw new Error("Database not initialized")
+    }
+
+    // Execute schema as a transaction
+    db.exec(schema)
+  }
+
+  const initDatabase = async (projectRoot: string): Promise<void> => {
+    const dbPath = getStateDatabasePath(projectRoot)
+
+    // If already connected to this database, skip
+    if (db && currentDbPath === dbPath) {
+      return
+    }
+
+    // Close existing connection if any
+    if (db) {
+      closeDatabase()
+    }
+
+    // Ensure directory exists
+    const dir = dirname(dbPath)
+    await Bun.write(dir + "/.keep", "")
+
+    // Open database connection
+    db = new Database(dbPath, { create: true })
+    currentDbPath = dbPath
+
+    // Enable WAL mode for better concurrency
+    db.exec("PRAGMA journal_mode = WAL")
+
+    // Enable foreign keys
+    db.exec("PRAGMA foreign_keys = ON")
+
+    // Apply schema
+    applySchema()
+    runMigrations()
+  }
+
+  const initDatabaseWithPath = async (dbPath: string): Promise<void> => {
+    // Close existing connection if any
+    if (db) {
+      closeDatabase()
+    }
+
+    // For in-memory databases, skip directory creation
+    if (dbPath !== ":memory:") {
+      const dir = dirname(dbPath)
+      await Bun.write(dir + "/.keep", "")
+    }
+
+    // Open database connection
+    db = new Database(dbPath, { create: true })
+    currentDbPath = dbPath
+
+    // Enable WAL mode for better concurrency (skip for in-memory)
+    if (dbPath !== ":memory:") {
+      db.exec("PRAGMA journal_mode = WAL")
+    }
+
+    // Enable foreign keys
+    db.exec("PRAGMA foreign_keys = ON")
+
+    // Apply schema
+    applySchema()
+    runMigrations()
+  }
+
+  const closeDatabase = (): void => {
+    if (db) {
+      db.close()
+      db = null
+      currentDbPath = null
+    }
+  }
+
+  const isDatabaseInitialized = (): boolean => {
+    return db !== null
+  }
+
+  const getDatabasePath = (): string | null => {
+    return currentDbPath
+  }
+
+  const getSchemaVersion = (): number => {
+    const database = getDatabase()
+    const result = database
+      .query<{ version: number }, []>(
+        "SELECT MAX(version) as version FROM schema_version"
+      )
+      .get()
+    return result?.version ?? 0
+  }
+
+  const runMigrations = (): void => {
+    const version = getSchemaVersion()
+    const database = getDatabase()
+
+    // Migration to version 2: Add ai_session_data column
+    if (version < 2) {
+      // Check if column exists (for databases created with new schema)
+      const columns = database
+        .query<{ name: string }, []>("PRAGMA table_info(sessions)")
+        .all()
+      const hasColumn = columns.some((col) => col.name === "ai_session_data")
+
+      if (!hasColumn) {
+        database.exec("ALTER TABLE sessions ADD COLUMN ai_session_data TEXT")
+      }
+
+      // Update version
+      database.exec("INSERT OR REPLACE INTO schema_version (version) VALUES (2)")
+    }
+
+    // Add future migrations here:
+    // if (version < 3) { migrateTo3() }
+  }
+
+  const withTransaction = <T>(fn: () => T): T => {
+    const database = getDatabase()
+    return database.transaction(fn)()
+  }
+
+  return {
+    getDatabase,
+    initDatabase,
+    initDatabaseWithPath,
+    closeDatabase,
+    isDatabaseInitialized,
+    getDatabasePath,
+    getSchemaVersion,
+    runMigrations,
+    withTransaction,
+  }
+}
+
+// ============================================================================
+// Active Database Manager
+// ============================================================================
+
+let activeManager = createDatabaseManager()
+
+export function setDatabaseManager(manager: DatabaseManager): void {
+  activeManager = manager
+}
+
+// ============================================================================
+// Default Manager Exports
+// ============================================================================
+
+export function getDatabase(): Database {
+  return activeManager.getDatabase()
+}
+
+export async function initDatabase(projectRoot: string): Promise<void> {
+  return activeManager.initDatabase(projectRoot)
+}
+
+export async function initDatabaseWithPath(dbPath: string): Promise<void> {
+  return activeManager.initDatabaseWithPath(dbPath)
+}
+
+export function closeDatabase(): void {
+  return activeManager.closeDatabase()
+}
+
+export function isDatabaseInitialized(): boolean {
+  return activeManager.isDatabaseInitialized()
+}
+
+export function getDatabasePath(): string | null {
+  return activeManager.getDatabasePath()
+}
+
+export function getSchemaVersion(): number {
+  return activeManager.getSchemaVersion()
+}
+
+export function runMigrations(): void {
+  return activeManager.runMigrations()
+}
+
+export function withTransaction<T>(fn: () => T): T {
+  return activeManager.withTransaction(fn)
+}
+
+// ============================================================================
+// Utility Functions
+// ============================================================================
+
+/**
+ * Generate an ISO timestamp for the current time
+ */
+export function nowISO(): string {
+  return new Date().toISOString()
+}

package/src/store/index.ts

@@ -0,0 +1,101 @@
+/**
+ * Store module - SQLite database layer for OpenSWE
+ *
+ * Provides persistent storage for sessions, human tasks, project state,
+ * and output buffers using Bun's native bun:sqlite.
+ */
+
+// ============================================================================
+// Types
+// ============================================================================
+
+export type {
+  Phase,
+  Status,
+  AISessionData,
+  Session,
+  CreateSessionInput,
+  UpdateSessionInput,
+  ProjectState,
+  CreateProjectInput,
+  OutputBuffer,
+} from "./types"
+
+export {
+  isValidPhase,
+  isValidStatus,
+  isValidAISessionData,
+} from "./types"
+
+// ============================================================================
+// Database Connection
+// ============================================================================
+
+export {
+  createDatabaseManager,
+  setDatabaseManager,
+  getDatabase,
+  initDatabase,
+  initDatabaseWithPath,
+  closeDatabase,
+  isDatabaseInitialized,
+  getDatabasePath,
+  getSchemaVersion,
+  runMigrations,
+  withTransaction,
+  nowISO,
+} from "./db"
+
+// ============================================================================
+// Project Operations
+// ============================================================================
+
+export {
+  getProject,
+  createProject,
+  updateLastOpened,
+  projectExists,
+  deleteProject,
+} from "./project"
+
+// ============================================================================
+// Session Operations
+// ============================================================================
+
+export {
+  getAllSessions,
+  getSession,
+  getSessionsByStatus,
+  getSessionsByPhase,
+  getActiveSessionCount,
+  getSessionCount,
+  createSession,
+  updateSession,
+  updateSessionPhase,
+  updateSessionStatus,
+  incrementRetryCount,
+  updateTokensUsed,
+  setPid,
+  setAISessionData,
+  getAISessionData,
+  deleteSession,
+  deleteAllSessions,
+} from "./sessions"
+
+// ============================================================================
+// Buffer Operations
+// ============================================================================
+
+export {
+  MAX_BUFFER_LINES,
+  getBuffer,
+  getRecentLines,
+  getAllLines,
+  createBuffer,
+  ensureBuffer,
+  appendLines,
+  appendLine,
+  setLines,
+  clearBuffer,
+  deleteBuffer,
+} from "./buffers"

package/src/store/project.ts

@@ -0,0 +1,119 @@
+/**
+ * Project state database operations
+ *
+ * Manages the project singleton in the database.
+ * This mirrors essential fields from ProjectConfig for querying.
+ */
+
+import { getDatabase, nowISO } from "./db"
+import type { ProjectState, CreateProjectInput } from "./types"
+
+// ============================================================================
+// Database Row Type
+// ============================================================================
+
+interface ProjectRow {
+  id: number
+  repo_full_name: string
+  repo_url: string
+  created_at: string
+  last_opened_at: string
+}
+
+// ============================================================================
+// Row Mapping
+// ============================================================================
+
+/**
+ * Convert database row to ProjectState
+ */
+function rowToProject(row: ProjectRow): ProjectState {
+  return {
+    id: 1,
+    repoFullName: row.repo_full_name,
+    repoUrl: row.repo_url,
+    createdAt: row.created_at,
+    lastOpenedAt: row.last_opened_at,
+  }
+}
+
+// ============================================================================
+// CRUD Operations
+// ============================================================================
+
+/**
+ * Get the project state from the database
+ *
+ * @returns ProjectState or null if not created
+ */
+export function getProject(): ProjectState | null {
+  const db = getDatabase()
+  const row = db
+    .query<ProjectRow, []>("SELECT * FROM project WHERE id = 1")
+    .get()
+
+  return row ? rowToProject(row) : null
+}
+
+/**
+ * Create the project state (singleton)
+ *
+ * @param data - Project creation data
+ * @returns The created ProjectState
+ * @throws Error if project already exists
+ */
+export function createProject(data: CreateProjectInput): ProjectState {
+  const db = getDatabase()
+  const now = nowISO()
+
+  // Check if project already exists
+  const existing = getProject()
+  if (existing) {
+    throw new Error("Project already exists. Use updateProject instead.")
+  }
+
+  db.query(
+    `INSERT INTO project (id, repo_full_name, repo_url, created_at, last_opened_at)
+     VALUES (1, ?, ?, ?, ?)`
+  ).run(
+    data.repoFullName,
+    data.repoUrl,
+    now,
+    now
+  )
+
+  return {
+    id: 1,
+    repoFullName: data.repoFullName,
+    repoUrl: data.repoUrl,
+    createdAt: now,
+    lastOpenedAt: now,
+  }
+}
+
+/**
+ * Update the lastOpenedAt timestamp
+ */
+export function updateLastOpened(): void {
+  const db = getDatabase()
+  const now = nowISO()
+
+  db.query("UPDATE project SET last_opened_at = ? WHERE id = 1").run(now)
+}
+
+/**
+ * Check if project exists in the database
+ */
+export function projectExists(): boolean {
+  return getProject() !== null
+}
+
+/**
+ * Delete the project state
+ *
+ * Warning: This also cascades to delete all sessions and tasks.
+ */
+export function deleteProject(): void {
+  const db = getDatabase()
+  db.query("DELETE FROM project WHERE id = 1").run()
+}

package/src/store/schema.sql

@@ -0,0 +1,71 @@
+-- OpenSWE Database Schema
+-- SQLite database for persistent storage of sessions, tasks, and project state
+
+-- ============================================================================
+-- Schema Version
+-- ============================================================================
+
+-- Track schema version for migrations
+CREATE TABLE IF NOT EXISTS schema_version (
+  version INTEGER PRIMARY KEY,
+  applied_at TEXT NOT NULL DEFAULT (datetime('now'))
+);
+
+-- Insert initial version if not exists
+INSERT OR IGNORE INTO schema_version (version) VALUES (1);
+
+-- ============================================================================
+-- Project Table (Singleton)
+-- ============================================================================
+
+-- Project metadata - only one row (id = 1)
+CREATE TABLE IF NOT EXISTS project (
+  id INTEGER PRIMARY KEY CHECK (id = 1),
+  repo_full_name TEXT NOT NULL,
+  repo_url TEXT NOT NULL,
+  created_at TEXT NOT NULL DEFAULT (datetime('now')),
+  last_opened_at TEXT NOT NULL DEFAULT (datetime('now'))
+);
+
+-- ============================================================================
+-- Sessions Table
+-- ============================================================================
+
+-- Sessions represent work on specific issues or tasks
+CREATE TABLE IF NOT EXISTS sessions (
+  id TEXT PRIMARY KEY,
+  name TEXT NOT NULL,
+  issue_number INTEGER,
+  issue_title TEXT,
+  issue_body TEXT,
+  issue_url TEXT,
+  worktree_path TEXT NOT NULL,
+  branch_name TEXT NOT NULL,
+  phase TEXT NOT NULL DEFAULT 'pending',
+  status TEXT NOT NULL DEFAULT 'queued',
+  attention_reason TEXT,
+  retry_count INTEGER NOT NULL DEFAULT 0,
+  tokens_used INTEGER NOT NULL DEFAULT 0,
+  pid INTEGER,
+  ai_session_data TEXT,                   -- JSON: {backend, sessionId, ...}
+  created_at TEXT NOT NULL DEFAULT (datetime('now')),
+  updated_at TEXT NOT NULL DEFAULT (datetime('now'))
+);
+
+-- Index for querying sessions by status
+CREATE INDEX IF NOT EXISTS idx_sessions_status ON sessions(status);
+
+-- Index for querying sessions by phase
+CREATE INDEX IF NOT EXISTS idx_sessions_phase ON sessions(phase);
+
+-- ============================================================================
+-- Output Buffers Table
+-- ============================================================================
+
+-- Circular buffers for session output (stored as JSON array)
+CREATE TABLE IF NOT EXISTS output_buffers (
+  session_id TEXT PRIMARY KEY,
+  lines TEXT NOT NULL DEFAULT '[]',
+  last_updated TEXT NOT NULL DEFAULT (datetime('now')),
+  FOREIGN KEY (session_id) REFERENCES sessions(id) ON DELETE CASCADE
+);