AgentCompany/src/main/database/connectors-crud.ts

import type { DatabaseSync } from "node:sqlite";
import { normalizeCommandValue } from "@agentcompany/adapter-utils/server-utils";
import type { ConnectorId, ConnectorRecord } from "@shared/types";
import type { ReadAll } from "./helpers";
import { nowIso } from "./helpers";
import { getConnectorDefinition, buildDefaultConnectorRows } from "../connectors";


export function buildConnectorMethods(
  getDb: () => DatabaseSync,
  readAll: ReadAll,
) {
  function readConnectors(): ConnectorRecord[] {
    const rows = readAll<{
      id: string; label: string; description: string; status: ConnectorRecord["status"];
      command: string; version: string | null; auth_state: ConnectorRecord["authState"];
      last_checked_at: string | null; last_error: string | null; model: string | null;
      env_binding_text: string; notes: string;
    }>(
      "select id, label, description, status, command, version, auth_state, last_checked_at, last_error, model, env_binding_text, notes from connectors order by label asc",
    );
    const result: ConnectorRecord[] = [];
    for (const row of rows) {
      let definition;
      try {
        definition = getConnectorDefinition(row.id as ConnectorId);
      } catch {
        continue;
      }
      result.push({
        id: row.id as ConnectorId,
        label: row.label,
        description: row.description,
        status: row.status,
        command: normalizeCommandValue(row.command),
        version: row.version,
        authState: row.auth_state,
        lastCheckedAt: row.last_checked_at,
        lastError: row.last_error,
        model: row.model,
        envBindingText: row.env_binding_text,
        notes: row.notes,
        configurationDoc: definition.configurationDoc,
        capabilityMatrix: definition.capabilityMatrix,
      });
    }
    return result;
  }

  return {
    readConnectors,

    saveConnector(input: {
      id: ConnectorId;
      command: string;
      model: string | null;
      envBindingText: string;
      notes: string;
    }) {
      const normalizedCommand = normalizeCommandValue(input.command);
      getDb()
        .prepare(
          "update connectors set command = ?, model = ?, env_binding_text = ?, notes = ?, updated_at = ? where id = ?",
        )
        .run(normalizedCommand, input.model, input.envBindingText, input.notes, nowIso(), input.id);
    },

    updateConnectorHealth(input: {
      id: ConnectorId;
      status: string;
      authState: string;
      version: string | null;
      lastError: string | null;
    }) {
      getDb()
        .prepare(
          "update connectors set status = ?, auth_state = ?, version = ?, last_checked_at = ?, last_error = ?, updated_at = ? where id = ?",
        )
        .run(input.status, input.authState, input.version, nowIso(), input.lastError, nowIso(), input.id);
    },

    getConnector(id: ConnectorId): ConnectorRecord {
      const record = readConnectors().find((entry) => entry.id === id);
      if (!record) throw new Error(`Connector not found: ${id}`);
      return record;
    },

    seedConnectors() {
      getDb().prepare("delete from connectors where id not in ('codex_local', 'claude_local', 'gemini_local') and id not like 'claude_compat%'").run();
      const currentRows = readAll<Partial<ConnectorRecord>>(
        "select id, label, description, status, command, version, auth_state as authState, last_checked_at as lastCheckedAt, last_error as lastError, model, env_binding_text as envBindingText, notes from connectors",
      );
      // Migrate first compat slot if its definition defaults have changed (edition upgrade).
      // When the definition ships with preset notes that differ from the DB, relocate the
      // old user config to the next empty compat slot so definition defaults take effect.
      const firstCompat = currentRows.find((r) => r.id === "claude_compatible");
      if (firstCompat?.envBindingText && firstCompat.notes) {
        let firstDef: { notes: string } | undefined;
        try { firstDef = getConnectorDefinition("claude_compatible"); } catch { /* skip */ }
        if (firstDef?.notes && firstCompat.notes !== firstDef.notes) {
          const migrationTargets: ConnectorId[] = [
            "claude_compat_2", "claude_compat_3", "claude_compat_4", "claude_compat_5",
            "claude_compat_6", "claude_compat_7", "claude_compat_8", "claude_compat_9", "claude_compat_10",
          ];
          const occupied = new Set(
            currentRows
              .filter((r) => r.id !== "claude_compatible" && r.id?.startsWith("claude_compat") && r.envBindingText)
              .map((r) => r.id),
          );
          const targetId = migrationTargets.find((id) => !occupied.has(id));
          if (targetId) {
            const migrated = {
              envBindingText: firstCompat.envBindingText,
              notes: firstCompat.notes,
              model: firstCompat.model,
              status: firstCompat.status,
              authState: firstCompat.authState,
              version: firstCompat.version,
              lastCheckedAt: firstCompat.lastCheckedAt,
              lastError: firstCompat.lastError,
            };
            const targetRow = currentRows.find((r) => r.id === targetId);
            if (targetRow) {
              Object.assign(targetRow, migrated);
            } else {
              currentRows.push({ id: targetId as ConnectorId, ...migrated });
            }
            firstCompat.envBindingText = undefined;
            firstCompat.notes = undefined;
            firstCompat.model = undefined;
            firstCompat.status = undefined;
            firstCompat.authState = undefined;
            firstCompat.version = undefined;
            firstCompat.lastCheckedAt = undefined;
            firstCompat.lastError = undefined;
          }
        }
      }

      const merged = buildDefaultConnectorRows(currentRows);
      const statement = getDb().prepare(
        "insert or replace into connectors (id, label, description, status, command, version, auth_state, last_checked_at, last_error, model, env_binding_text, notes, updated_at) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)",
      );
      const updatedAt = nowIso();
      for (const row of merged) {
        statement.run(
          row.id, row.label, row.description, row.status,
          row.command, row.version, row.authState, row.lastCheckedAt,
          row.lastError, row.model, row.envBindingText, row.notes, updatedAt,
        );
      }

      // One-time migration: mark native connectors that are actively used by
      // agents as model-pool-managed so they remain visible to the CEO after
      // the opt-in policy change. Connectors not used by any agent stay
      // unmanaged and will not appear in CEO prompts until the user explicitly
      // enables models for them in Settings > Model Management.
      const migrationKey = "migration:model_pool_opt_in_v1";
      const migrated = readAll<{ value: string }>("SELECT value FROM settings WHERE key = ?", migrationKey);
      if (migrated.length === 0) {
        const usedIds = new Set(
          readAll<{ connector_id: string }>(
            "SELECT DISTINCT connector_id FROM agents WHERE status NOT IN ('terminated')",
          ).map((r) => r.connector_id),
        );
        const nativeIds = ["codex_local", "claude_local", "gemini_local"];
        const upsert = getDb().prepare(
          "INSERT INTO settings (key, value) VALUES (?, ?) ON CONFLICT(key) DO UPDATE SET value = excluded.value",
        );
        for (const id of nativeIds) {
          const hasSettings = readAll<{ key: string }>(
            "SELECT key FROM settings WHERE key = ?",
            `disabled_models:${id}`,
          );
          if (hasSettings.length === 0 && usedIds.has(id)) {
            upsert.run(`disabled_models:${id}`, "[]");
          }
        }
        upsert.run(migrationKey, "1");
      }
    },
  };
}