@build-astron-co/nimbus 0.2.0 → 0.4.0

package/dist/src/state/schema.js

@@ -0,0 +1,233 @@
+/**
+ * Unified schema with all 16 tables for the Nimbus persistence layer.
+ *
+ * Tables cover: operations, config, templates, conversations, artifacts,
+ * projects, audit_logs, safety_checks, checkpoints, device_codes, tokens,
+ * teams, team_members, users, subscriptions, and usage.
+ */
+/**
+ * Run all CREATE TABLE / CREATE INDEX migrations against the given database.
+ * Every statement uses IF NOT EXISTS so the function is safe to call on an
+ * already-initialised database.
+ */
+export function runMigrations(db) {
+    db.exec(`
+    -- Operations / history
+    CREATE TABLE IF NOT EXISTS operations (
+      id TEXT PRIMARY KEY,
+      timestamp TEXT NOT NULL DEFAULT (datetime('now')),
+      type TEXT NOT NULL,
+      command TEXT NOT NULL,
+      input TEXT,
+      output TEXT,
+      status TEXT NOT NULL DEFAULT 'pending',
+      duration_ms INTEGER,
+      model TEXT,
+      tokens_used INTEGER,
+      cost_usd REAL,
+      metadata TEXT
+    );
+
+    -- Config key-value store
+    CREATE TABLE IF NOT EXISTS config (
+      key TEXT PRIMARY KEY,
+      value TEXT NOT NULL,
+      updated_at TEXT NOT NULL DEFAULT (datetime('now'))
+    );
+
+    -- Templates
+    CREATE TABLE IF NOT EXISTS templates (
+      id TEXT PRIMARY KEY,
+      name TEXT NOT NULL,
+      type TEXT NOT NULL,
+      content TEXT NOT NULL,
+      variables TEXT,
+      created_at TEXT NOT NULL DEFAULT (datetime('now')),
+      updated_at TEXT NOT NULL DEFAULT (datetime('now'))
+    );
+
+    -- Conversations
+    CREATE TABLE IF NOT EXISTS conversations (
+      id TEXT PRIMARY KEY,
+      title TEXT NOT NULL,
+      messages TEXT NOT NULL,
+      model TEXT,
+      created_at TEXT NOT NULL DEFAULT (datetime('now')),
+      updated_at TEXT NOT NULL DEFAULT (datetime('now')),
+      metadata TEXT
+    );
+
+    -- Artifacts
+    CREATE TABLE IF NOT EXISTS artifacts (
+      id TEXT PRIMARY KEY,
+      conversation_id TEXT,
+      name TEXT NOT NULL,
+      type TEXT NOT NULL,
+      content TEXT NOT NULL,
+      language TEXT,
+      created_at TEXT NOT NULL DEFAULT (datetime('now')),
+      updated_at TEXT NOT NULL DEFAULT (datetime('now')),
+      metadata TEXT
+    );
+
+    -- Projects
+    CREATE TABLE IF NOT EXISTS projects (
+      id TEXT PRIMARY KEY,
+      name TEXT NOT NULL,
+      path TEXT NOT NULL UNIQUE,
+      config TEXT NOT NULL DEFAULT '{}',
+      last_scanned TEXT,
+      created_at TEXT NOT NULL DEFAULT (datetime('now')),
+      updated_at TEXT NOT NULL DEFAULT (datetime('now'))
+    );
+
+    -- Audit logs
+    CREATE TABLE IF NOT EXISTS audit_logs (
+      id TEXT PRIMARY KEY,
+      timestamp TEXT NOT NULL DEFAULT (datetime('now')),
+      user_id TEXT,
+      action TEXT NOT NULL,
+      resource_type TEXT,
+      resource_id TEXT,
+      input TEXT,
+      output TEXT,
+      status TEXT NOT NULL,
+      duration_ms INTEGER,
+      metadata TEXT
+    );
+
+    -- Safety checks
+    CREATE TABLE IF NOT EXISTS safety_checks (
+      id TEXT PRIMARY KEY,
+      operation_id TEXT,
+      check_type TEXT NOT NULL,
+      check_name TEXT NOT NULL,
+      passed INTEGER NOT NULL DEFAULT 0,
+      severity TEXT,
+      message TEXT,
+      requires_approval INTEGER NOT NULL DEFAULT 0,
+      approved_by TEXT,
+      approved_at TEXT,
+      created_at TEXT NOT NULL DEFAULT (datetime('now'))
+    );
+
+    -- Checkpoints (resumable operations)
+    CREATE TABLE IF NOT EXISTS checkpoints (
+      id TEXT PRIMARY KEY,
+      operation_id TEXT NOT NULL,
+      step INTEGER NOT NULL,
+      state TEXT NOT NULL,
+      created_at TEXT NOT NULL DEFAULT (datetime('now'))
+    );
+
+    -- Device codes (auth)
+    CREATE TABLE IF NOT EXISTS device_codes (
+      device_code TEXT PRIMARY KEY,
+      user_code TEXT NOT NULL UNIQUE,
+      client_id TEXT,
+      scope TEXT,
+      status TEXT NOT NULL DEFAULT 'pending',
+      token TEXT,
+      expires_at TEXT NOT NULL,
+      created_at TEXT NOT NULL DEFAULT (datetime('now'))
+    );
+
+    -- Auth tokens
+    CREATE TABLE IF NOT EXISTS tokens (
+      id TEXT PRIMARY KEY,
+      user_id TEXT,
+      token TEXT NOT NULL UNIQUE,
+      type TEXT NOT NULL DEFAULT 'access',
+      expires_at TEXT,
+      created_at TEXT NOT NULL DEFAULT (datetime('now'))
+    );
+
+    -- Teams
+    CREATE TABLE IF NOT EXISTS teams (
+      id TEXT PRIMARY KEY,
+      name TEXT NOT NULL,
+      owner_id TEXT NOT NULL,
+      plan TEXT NOT NULL DEFAULT 'free',
+      created_at TEXT NOT NULL DEFAULT (datetime('now')),
+      updated_at TEXT NOT NULL DEFAULT (datetime('now'))
+    );
+
+    -- Team members
+    CREATE TABLE IF NOT EXISTS team_members (
+      id TEXT PRIMARY KEY,
+      team_id TEXT NOT NULL,
+      user_id TEXT NOT NULL,
+      role TEXT NOT NULL DEFAULT 'member',
+      joined_at TEXT NOT NULL DEFAULT (datetime('now')),
+      FOREIGN KEY (team_id) REFERENCES teams(id) ON DELETE CASCADE
+    );
+
+    -- Users
+    CREATE TABLE IF NOT EXISTS users (
+      id TEXT PRIMARY KEY,
+      email TEXT UNIQUE,
+      name TEXT,
+      avatar_url TEXT,
+      created_at TEXT NOT NULL DEFAULT (datetime('now')),
+      updated_at TEXT NOT NULL DEFAULT (datetime('now'))
+    );
+
+    -- Subscriptions (billing)
+    CREATE TABLE IF NOT EXISTS subscriptions (
+      id TEXT PRIMARY KEY,
+      team_id TEXT NOT NULL,
+      plan TEXT NOT NULL DEFAULT 'free',
+      status TEXT NOT NULL DEFAULT 'active',
+      current_period_start TEXT,
+      current_period_end TEXT,
+      created_at TEXT NOT NULL DEFAULT (datetime('now')),
+      updated_at TEXT NOT NULL DEFAULT (datetime('now')),
+      FOREIGN KEY (team_id) REFERENCES teams(id) ON DELETE CASCADE
+    );
+
+    -- Usage tracking (billing)
+    CREATE TABLE IF NOT EXISTS usage (
+      id TEXT PRIMARY KEY,
+      team_id TEXT,
+      user_id TEXT,
+      type TEXT NOT NULL,
+      quantity REAL NOT NULL DEFAULT 0,
+      unit TEXT NOT NULL DEFAULT 'tokens',
+      cost_usd REAL DEFAULT 0,
+      metadata TEXT,
+      created_at TEXT NOT NULL DEFAULT (datetime('now'))
+    );
+
+    -- Shared sessions
+    CREATE TABLE IF NOT EXISTS shares (
+      id TEXT PRIMARY KEY,
+      session_id TEXT NOT NULL,
+      name TEXT NOT NULL,
+      messages TEXT NOT NULL,
+      model TEXT,
+      mode TEXT,
+      cost_usd REAL DEFAULT 0,
+      token_count INTEGER DEFAULT 0,
+      is_live INTEGER NOT NULL DEFAULT 0,
+      write_token TEXT,
+      created_at TEXT NOT NULL DEFAULT (datetime('now')),
+      expires_at TEXT NOT NULL
+    );
+
+    -- Indexes
+    CREATE INDEX IF NOT EXISTS idx_operations_timestamp ON operations(timestamp);
+    CREATE INDEX IF NOT EXISTS idx_operations_type ON operations(type);
+    CREATE INDEX IF NOT EXISTS idx_conversations_updated ON conversations(updated_at);
+    CREATE INDEX IF NOT EXISTS idx_artifacts_type ON artifacts(type);
+    CREATE INDEX IF NOT EXISTS idx_artifacts_conversation ON artifacts(conversation_id);
+    CREATE INDEX IF NOT EXISTS idx_audit_logs_timestamp ON audit_logs(timestamp);
+    CREATE INDEX IF NOT EXISTS idx_audit_logs_user ON audit_logs(user_id);
+    CREATE INDEX IF NOT EXISTS idx_audit_logs_action ON audit_logs(action);
+    CREATE INDEX IF NOT EXISTS idx_checkpoints_operation ON checkpoints(operation_id);
+    CREATE INDEX IF NOT EXISTS idx_team_members_team ON team_members(team_id);
+    CREATE INDEX IF NOT EXISTS idx_usage_team ON usage(team_id);
+    CREATE INDEX IF NOT EXISTS idx_usage_created ON usage(created_at);
+    CREATE INDEX IF NOT EXISTS idx_shares_session ON shares(session_id);
+    CREATE INDEX IF NOT EXISTS idx_shares_expires ON shares(expires_at);
+  `);
+}

package/dist/src/state/sessions.js

@@ -0,0 +1,79 @@
+/**
+ * Session / operation CRUD helpers.
+ *
+ * Refactored from SQLiteAdapter.saveOperation, getOperation, listOperations,
+ * and listOperationsByType.
+ */
+import { getDb } from './db';
+// ---------------------------------------------------------------------------
+// Internal helpers
+// ---------------------------------------------------------------------------
+function rowToOperation(row) {
+    return {
+        id: row.id,
+        timestamp: new Date(row.timestamp),
+        type: row.type,
+        command: row.command,
+        input: row.input,
+        output: row.output,
+        status: row.status,
+        durationMs: row.duration_ms,
+        model: row.model,
+        tokensUsed: row.tokens_used,
+        costUsd: row.cost_usd,
+        metadata: row.metadata ? JSON.parse(row.metadata) : undefined,
+    };
+}
+// ---------------------------------------------------------------------------
+// Public API
+// ---------------------------------------------------------------------------
+/**
+ * Persist a single operation record.
+ */
+export function saveOperation(operation, db) {
+    const d = db || getDb();
+    const stmt = d.prepare(`
+    INSERT INTO operations (id, timestamp, type, command, input, output, status, duration_ms, model, tokens_used, cost_usd, metadata)
+    VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+  `);
+    stmt.run(operation.id, operation.timestamp.toISOString(), operation.type, operation.command, operation.input || null, operation.output || null, operation.status, operation.durationMs || null, operation.model || null, operation.tokensUsed || null, operation.costUsd || null, operation.metadata ? JSON.stringify(operation.metadata) : null);
+}
+/**
+ * Retrieve a single operation by id.
+ */
+export function getOperation(id, db) {
+    const d = db || getDb();
+    const stmt = d.prepare('SELECT * FROM operations WHERE id = ?');
+    const row = stmt.get(id);
+    if (!row) {
+        return null;
+    }
+    return rowToOperation(row);
+}
+/**
+ * List operations ordered by timestamp descending.
+ */
+export function listOperations(limit = 50, offset = 0, db) {
+    const d = db || getDb();
+    const stmt = d.prepare(`
+    SELECT * FROM operations
+    ORDER BY timestamp DESC
+    LIMIT ? OFFSET ?
+  `);
+    const rows = stmt.all(limit, offset);
+    return rows.map(row => rowToOperation(row));
+}
+/**
+ * List operations filtered by type, ordered by timestamp descending.
+ */
+export function listOperationsByType(type, limit = 50, offset = 0, db) {
+    const d = db || getDb();
+    const stmt = d.prepare(`
+    SELECT * FROM operations
+    WHERE type = ?
+    ORDER BY timestamp DESC
+    LIMIT ? OFFSET ?
+  `);
+    const rows = stmt.all(type, limit, offset);
+    return rows.map(row => rowToOperation(row));
+}

package/dist/src/state/teams.js

@@ -0,0 +1,131 @@
+/**
+ * Team and user CRUD helpers.
+ *
+ * Refactored from the team-service database adapter
+ * (services/team-service/src/db/adapter.ts) into standalone functions
+ * that operate against the unified Nimbus database.
+ */
+import { getDb } from './db';
+// ---------------------------------------------------------------------------
+// Team helpers
+// ---------------------------------------------------------------------------
+/**
+ * Create a new team.
+ */
+export function createTeam(id, name, ownerId, plan = 'free', db) {
+    const d = db || getDb();
+    const stmt = d.prepare(`
+    INSERT INTO teams (id, name, owner_id, plan, created_at, updated_at)
+    VALUES (?, ?, ?, ?, CURRENT_TIMESTAMP, CURRENT_TIMESTAMP)
+  `);
+    stmt.run(id, name, ownerId, plan);
+}
+/**
+ * Retrieve a team by id.
+ */
+export function getTeam(id, db) {
+    const d = db || getDb();
+    const stmt = d.prepare('SELECT * FROM teams WHERE id = ?');
+    const row = stmt.get(id);
+    if (!row) {
+        return null;
+    }
+    return {
+        id: row.id,
+        name: row.name,
+        ownerId: row.owner_id,
+        plan: row.plan,
+        createdAt: row.created_at,
+        updatedAt: row.updated_at,
+    };
+}
+/**
+ * List all teams, ordered by most-recently updated first.
+ */
+export function listTeams(db) {
+    const d = db || getDb();
+    const stmt = d.prepare('SELECT * FROM teams ORDER BY updated_at DESC');
+    const rows = stmt.all();
+    return rows.map(row => ({
+        id: row.id,
+        name: row.name,
+        ownerId: row.owner_id,
+        plan: row.plan,
+        createdAt: row.created_at,
+        updatedAt: row.updated_at,
+    }));
+}
+// ---------------------------------------------------------------------------
+// Team member helpers
+// ---------------------------------------------------------------------------
+/**
+ * Add a member to a team.
+ */
+export function addTeamMember(id, teamId, userId, role = 'member', db) {
+    const d = db || getDb();
+    const stmt = d.prepare(`
+    INSERT INTO team_members (id, team_id, user_id, role, joined_at)
+    VALUES (?, ?, ?, ?, CURRENT_TIMESTAMP)
+  `);
+    stmt.run(id, teamId, userId, role);
+}
+/**
+ * Remove a member from a team by user id.
+ */
+export function removeTeamMember(teamId, userId, db) {
+    const d = db || getDb();
+    const stmt = d.prepare('DELETE FROM team_members WHERE team_id = ? AND user_id = ?');
+    stmt.run(teamId, userId);
+}
+/**
+ * List all members of a team.
+ */
+export function listTeamMembers(teamId, db) {
+    const d = db || getDb();
+    const stmt = d.prepare(`
+    SELECT * FROM team_members
+    WHERE team_id = ?
+    ORDER BY joined_at ASC
+  `);
+    const rows = stmt.all(teamId);
+    return rows.map(row => ({
+        id: row.id,
+        teamId: row.team_id,
+        userId: row.user_id,
+        role: row.role,
+        joinedAt: row.joined_at,
+    }));
+}
+// ---------------------------------------------------------------------------
+// User helpers
+// ---------------------------------------------------------------------------
+/**
+ * Create a new user.
+ */
+export function createUser(id, email, name, avatarUrl, db) {
+    const d = db || getDb();
+    const stmt = d.prepare(`
+    INSERT INTO users (id, email, name, avatar_url, created_at, updated_at)
+    VALUES (?, ?, ?, ?, CURRENT_TIMESTAMP, CURRENT_TIMESTAMP)
+  `);
+    stmt.run(id, email || null, name || null, avatarUrl || null);
+}
+/**
+ * Retrieve a user by id.
+ */
+export function getUser(id, db) {
+    const d = db || getDb();
+    const stmt = d.prepare('SELECT * FROM users WHERE id = ?');
+    const row = stmt.get(id);
+    if (!row) {
+        return null;
+    }
+    return {
+        id: row.id,
+        email: row.email,
+        name: row.name,
+        avatarUrl: row.avatar_url,
+        createdAt: row.created_at,
+        updatedAt: row.updated_at,
+    };
+}

package/dist/src/telemetry.js

@@ -0,0 +1,91 @@
+/**
+ * Telemetry Module
+ *
+ * Thin wrapper around the PostHog analytics client (src/utils/analytics.ts).
+ * Reads the user's opt-in preference from ~/.nimbus/config.json and only
+ * forwards events when telemetry is explicitly enabled.
+ *
+ * Every exported function is guaranteed to never throw -- callers do not
+ * need try-catch around telemetry calls (though several already have it).
+ */
+import { analytics } from './utils/analytics';
+/**
+ * Read the telemetry section from ~/.nimbus/config.json.
+ * Returns `{ enabled: false }` on any error so the default is always off.
+ */
+function readTelemetryConfig() {
+    try {
+        // eslint-disable-next-line @typescript-eslint/no-var-requires
+        const fs = require('node:fs');
+        // eslint-disable-next-line @typescript-eslint/no-var-requires
+        const path = require('node:path');
+        // eslint-disable-next-line @typescript-eslint/no-var-requires
+        const { homedir } = require('os');
+        const configPath = path.join(homedir(), '.nimbus', 'config.json');
+        if (!fs.existsSync(configPath)) {
+            return { enabled: false };
+        }
+        const raw = JSON.parse(fs.readFileSync(configPath, 'utf-8'));
+        return raw?.telemetry ?? { enabled: false };
+    }
+    catch {
+        return { enabled: false };
+    }
+}
+// Cache the config for the lifetime of the process so we only read the
+// filesystem once.  The user would need to restart nimbus after toggling
+// telemetry anyway, so a stale value is acceptable.
+let _cachedConfig = null;
+function getConfig() {
+    if (_cachedConfig === null) {
+        _cachedConfig = readTelemetryConfig();
+    }
+    return _cachedConfig;
+}
+// ---------------------------------------------------------------------------
+// Public API
+// ---------------------------------------------------------------------------
+/**
+ * Track an IaC generation event (terraform apply, k8s apply, helm install, etc.).
+ *
+ * @param type  - A short identifier for the generation kind, e.g. "terraform-apply"
+ * @param tags  - Freeform tags for categorisation, e.g. ["terraform"]
+ */
+export function trackGeneration(type, tags) {
+    try {
+        const config = getConfig();
+        if (!config.enabled) {
+            return;
+        }
+        // Fire-and-forget -- analytics.trackEvent already swallows errors.
+        analytics.trackEvent('generation', {
+            type,
+            tags,
+            userId: config.anonymousId,
+        });
+    }
+    catch {
+        // Absolutely never throw from telemetry.
+    }
+}
+/**
+ * Track a generic named event with optional properties.
+ *
+ * @param name        - The event name, e.g. "analysis_completed"
+ * @param properties  - Arbitrary key/value metadata
+ */
+export function trackEvent(name, properties) {
+    try {
+        const config = getConfig();
+        if (!config.enabled) {
+            return;
+        }
+        analytics.trackEvent(name, {
+            ...properties,
+            userId: config.anonymousId,
+        });
+    }
+    catch {
+        // Absolutely never throw from telemetry.
+    }
+}