@downcity/city 1.1.4 → 1.1.5

package/src/store/StoreModelRepository.ts

@@ -1,324 +0,0 @@
-/**
- * ConsoleStore 模型与 Provider 仓储。
- *
- * 关键点（中文）
- * - 只负责 model/provider 相关读写，不处理 env、channel account、secure settings。
- * - 对外暴露纯函数，`ConsoleStore` 作为门面调用。
- */
-
-import { eq } from "drizzle-orm";
-import type {
-  StoredModel,
-  StoredModelProvider,
-  StoredProviderMeta,
-  UpsertModelInput,
-  UpsertModelProviderInput,
-} from "@/types/store/Store.js";
-import { decryptText, encryptText } from "./crypto.js";
-import { modelProvidersTable, modelsTable } from "./schema.js";
-import type { ConsoleStoreContext } from "./StoreShared.js";
-import { nowIso } from "./StoreShared.js";
-
-/**
- * 列出 providers。
- */
-export async function listStoredProviders(
-  context: ConsoleStoreContext,
-): Promise<StoredModelProvider[]> {
-  const rows = context.db.select().from(modelProvidersTable).all();
-  const result: StoredModelProvider[] = [];
-  for (const row of rows) {
-    let apiKey: string | undefined;
-    if (row.apiKeyEncrypted) {
-      apiKey = await decryptText(row.apiKeyEncrypted);
-    }
-    result.push({
-      id: row.id,
-      type: row.type as StoredModelProvider["type"],
-      baseUrl: row.baseUrl || undefined,
-      apiKey,
-      createdAt: row.createdAt,
-      updatedAt: row.updatedAt,
-    });
-  }
-  return result;
-}
-
-/**
- * 同步列出 provider 元信息（不含 API Key）。
- */
-export function listStoredProviderMetas(
-  context: ConsoleStoreContext,
-): StoredProviderMeta[] {
-  const rows = context.db.select().from(modelProvidersTable).all();
-  return rows.map((row) => ({
-    id: row.id,
-    type: row.type,
-    baseUrl: row.baseUrl || undefined,
-  }));
-}
-
-/**
- * 获取单个 provider。
- */
-export async function getStoredProvider(
-  context: ConsoleStoreContext,
-  providerId: string,
-): Promise<StoredModelProvider | null> {
-  const row = context.db
-    .select()
-    .from(modelProvidersTable)
-    .where(eq(modelProvidersTable.id, providerId))
-    .get();
-  if (!row) return null;
-  let apiKey: string | undefined;
-  if (row.apiKeyEncrypted) {
-    apiKey = await decryptText(row.apiKeyEncrypted);
-  }
-  return {
-    id: row.id,
-    type: row.type as StoredModelProvider["type"],
-    baseUrl: row.baseUrl || undefined,
-    apiKey,
-    createdAt: row.createdAt,
-    updatedAt: row.updatedAt,
-  };
-}
-
-/**
- * 新增或更新 provider。
- */
-export async function upsertStoredProvider(
-  context: ConsoleStoreContext,
-  input: UpsertModelProviderInput,
-): Promise<void> {
-  const id = String(input.id || "").trim();
-  if (!id) throw new Error("providerId cannot be empty");
-
-  const existing = context.db
-    .select()
-    .from(modelProvidersTable)
-    .where(eq(modelProvidersTable.id, id))
-    .get();
-  const createdAt = existing?.createdAt || nowIso();
-  const updatedAt = nowIso();
-  const hasApiKeyField = Object.prototype.hasOwnProperty.call(input, "apiKey");
-  let apiKeyEncrypted: string | null = existing?.apiKeyEncrypted || null;
-  if (hasApiKeyField) {
-    if (typeof input.apiKey === "string" && input.apiKey.length > 0) {
-      apiKeyEncrypted = await encryptText(input.apiKey);
-    } else {
-      apiKeyEncrypted = null;
-    }
-  }
-
-  context.db
-    .insert(modelProvidersTable)
-    .values({
-      id,
-      type: input.type,
-      baseUrl: input.baseUrl || null,
-      apiKeyEncrypted,
-      createdAt,
-      updatedAt,
-    })
-    .onConflictDoUpdate({
-      target: modelProvidersTable.id,
-      set: {
-        type: input.type,
-        baseUrl: input.baseUrl || null,
-        apiKeyEncrypted,
-        updatedAt,
-      },
-    })
-    .run();
-}
-
-/**
- * 删除 provider。
- */
-export function removeStoredProvider(
-  context: ConsoleStoreContext,
-  providerId: string,
-): void {
-  const refs = context.db
-    .select()
-    .from(modelsTable)
-    .where(eq(modelsTable.providerId, providerId))
-    .all();
-  if (refs.length > 0) {
-    throw new Error(
-      `Provider "${providerId}" is referenced by models: ${refs.map((item) => item.id).join(", ")}`,
-    );
-  }
-  context.db
-    .delete(modelProvidersTable)
-    .where(eq(modelProvidersTable.id, providerId))
-    .run();
-}
-
-/**
- * 列出 models。
- */
-export function listStoredModels(context: ConsoleStoreContext): StoredModel[] {
-  const rows = context.db.select().from(modelsTable).all();
-  return rows.map((row) => ({
-    id: row.id,
-    providerId: row.providerId,
-    name: row.name,
-    temperature: row.temperature ?? undefined,
-    maxTokens: row.maxTokens ?? undefined,
-    topP: row.topP ?? undefined,
-    frequencyPenalty: row.frequencyPenalty ?? undefined,
-    presencePenalty: row.presencePenalty ?? undefined,
-    anthropicVersion: row.anthropicVersion ?? undefined,
-    isPaused: Number(row.isPaused || 0) === 1,
-    createdAt: row.createdAt,
-    updatedAt: row.updatedAt,
-  }));
-}
-
-/**
- * 获取单个 model。
- */
-export function getStoredModel(
-  context: ConsoleStoreContext,
-  modelId: string,
-): StoredModel | null {
-  const row = context.db
-    .select()
-    .from(modelsTable)
-    .where(eq(modelsTable.id, modelId))
-    .get();
-  if (!row) return null;
-  return {
-    id: row.id,
-    providerId: row.providerId,
-    name: row.name,
-    temperature: row.temperature ?? undefined,
-    maxTokens: row.maxTokens ?? undefined,
-    topP: row.topP ?? undefined,
-    frequencyPenalty: row.frequencyPenalty ?? undefined,
-    presencePenalty: row.presencePenalty ?? undefined,
-    anthropicVersion: row.anthropicVersion ?? undefined,
-    isPaused: Number(row.isPaused || 0) === 1,
-    createdAt: row.createdAt,
-    updatedAt: row.updatedAt,
-  };
-}
-
-/**
- * 新增或更新 model。
- */
-export function upsertStoredModel(
-  context: ConsoleStoreContext,
-  input: UpsertModelInput,
-): void {
-  const id = String(input.id || "").trim();
-  if (!id) throw new Error("modelId cannot be empty");
-  const providerId = String(input.providerId || "").trim();
-  if (!providerId) throw new Error("providerId cannot be empty");
-  const provider = context.db
-    .select()
-    .from(modelProvidersTable)
-    .where(eq(modelProvidersTable.id, providerId))
-    .get();
-  if (!provider) throw new Error(`Provider not found: ${providerId}`);
-
-  const existing = context.db
-    .select()
-    .from(modelsTable)
-    .where(eq(modelsTable.id, id))
-    .get();
-  const createdAt = existing?.createdAt || nowIso();
-  const updatedAt = nowIso();
-
-  context.db
-    .insert(modelsTable)
-    .values({
-      id,
-      providerId,
-      name: input.name,
-      temperature: input.temperature ?? null,
-      maxTokens: input.maxTokens ?? null,
-      topP: input.topP ?? null,
-      frequencyPenalty: input.frequencyPenalty ?? null,
-      presencePenalty: input.presencePenalty ?? null,
-      anthropicVersion: input.anthropicVersion ?? null,
-      isPaused: input.isPaused === true ? 1 : 0,
-      createdAt,
-      updatedAt,
-    })
-    .onConflictDoUpdate({
-      target: modelsTable.id,
-      set: {
-        providerId,
-        name: input.name,
-        temperature: input.temperature ?? null,
-        maxTokens: input.maxTokens ?? null,
-        topP: input.topP ?? null,
-        frequencyPenalty: input.frequencyPenalty ?? null,
-        presencePenalty: input.presencePenalty ?? null,
-        anthropicVersion: input.anthropicVersion ?? null,
-        isPaused: input.isPaused === true ? 1 : 0,
-        updatedAt,
-      },
-    })
-    .run();
-}
-
-/**
- * 切换 model 暂停状态。
- */
-export function setStoredModelPaused(
-  context: ConsoleStoreContext,
-  modelId: string,
-  paused: boolean,
-): void {
-  const id = String(modelId || "").trim();
-  if (!id) throw new Error("modelId cannot be empty");
-  const current = getStoredModel(context, id);
-  if (!current) throw new Error(`Model not found: ${id}`);
-  context.db
-    .update(modelsTable)
-    .set({
-      isPaused: paused ? 1 : 0,
-      updatedAt: nowIso(),
-    })
-    .where(eq(modelsTable.id, id))
-    .run();
-}
-
-/**
- * 删除 model。
- */
-export function removeStoredModel(
-  context: ConsoleStoreContext,
-  modelId: string,
-): void {
-  context.db.delete(modelsTable).where(eq(modelsTable.id, modelId)).run();
-}
-
-/**
- * 获取“model + provider”聚合信息。
- */
-export async function getResolvedStoredModel(
-  context: ConsoleStoreContext,
-  modelId: string,
-): Promise<{ model: StoredModel; provider: StoredModelProvider } | null> {
-  const model = getStoredModel(context, modelId);
-  if (!model) return null;
-  const provider = await getStoredProvider(context, model.providerId);
-  if (!provider) return null;
-  return { model, provider };
-}
-
-/**
- * 清空模型相关表。
- */
-export function clearStoredModelsAndProviders(
-  context: ConsoleStoreContext,
-): void {
-  context.sqlite.exec("DELETE FROM models;");
-  context.sqlite.exec("DELETE FROM model_providers;");
-}

package/src/store/StoreSchema.ts

@@ -1,344 +0,0 @@
-/**
- * ConsoleStore Schema 管理。
- *
- * 关键点（中文）
- * - 负责 `ConsoleStore` 的建表与轻量迁移。
- * - 启动时执行，不承担任何查询写入业务逻辑。
- */
-
-import type { ConsoleStoreContext } from "./StoreShared.js";
-
-/**
- * 初始化 ConsoleStore 所需表结构。
- */
-export function ensureConsoleStoreSchema(context: ConsoleStoreContext): void {
-  context.sqlite.exec(`
-    CREATE TABLE IF NOT EXISTS model_providers (
-      id TEXT PRIMARY KEY NOT NULL,
-      type TEXT NOT NULL,
-      base_url TEXT,
-      api_key_encrypted TEXT,
-      created_at TEXT NOT NULL,
-      updated_at TEXT NOT NULL
-    );
-  `);
-  context.sqlite.exec(`
-    CREATE TABLE IF NOT EXISTS models (
-      id TEXT PRIMARY KEY NOT NULL,
-      provider_id TEXT NOT NULL,
-      name TEXT NOT NULL,
-      temperature REAL,
-      max_tokens INTEGER,
-      top_p REAL,
-      frequency_penalty REAL,
-      presence_penalty REAL,
-      anthropic_version TEXT,
-      is_paused INTEGER NOT NULL DEFAULT 0,
-      created_at TEXT NOT NULL,
-      updated_at TEXT NOT NULL
-    );
-  `);
-  ensureModelsTableColumns(context);
-  context.sqlite.exec(`
-    CREATE INDEX IF NOT EXISTS models_provider_id_idx
-    ON models(provider_id);
-  `);
-  context.sqlite.exec(`
-    CREATE TABLE IF NOT EXISTS console_secure_settings (
-      key TEXT PRIMARY KEY NOT NULL,
-      value_encrypted TEXT NOT NULL,
-      created_at TEXT NOT NULL,
-      updated_at TEXT NOT NULL
-    );
-  `);
-  context.sqlite.exec(`
-    CREATE TABLE IF NOT EXISTS env_entries (
-      scope TEXT NOT NULL,
-      agent_id TEXT NOT NULL DEFAULT '',
-      key TEXT NOT NULL,
-      description TEXT,
-      value_encrypted TEXT NOT NULL,
-      created_at TEXT NOT NULL,
-      updated_at TEXT NOT NULL,
-      PRIMARY KEY (scope, agent_id, key)
-    );
-  `);
-  context.sqlite.exec(`
-    CREATE INDEX IF NOT EXISTS env_entries_scope_idx
-    ON env_entries(scope);
-  `);
-  context.sqlite.exec(`
-    CREATE INDEX IF NOT EXISTS env_entries_agent_id_idx
-    ON env_entries(agent_id);
-  `);
-  ensureEnvEntriesMigration(context);
-  context.sqlite.exec(`
-    CREATE TABLE IF NOT EXISTS channel_accounts (
-      id TEXT PRIMARY KEY NOT NULL,
-      channel TEXT NOT NULL,
-      name TEXT NOT NULL,
-      identity TEXT,
-      owner TEXT,
-      creator TEXT,
-      bot_token_encrypted TEXT,
-      app_id_encrypted TEXT,
-      app_secret_encrypted TEXT,
-      domain TEXT,
-      sandbox INTEGER,
-      created_at TEXT NOT NULL,
-      updated_at TEXT NOT NULL
-    );
-  `);
-  context.sqlite.exec(`
-    CREATE INDEX IF NOT EXISTS channel_accounts_channel_idx
-    ON channel_accounts(channel);
-  `);
-  ensureChannelAccountsTableColumns(context);
-  ensureAuthSchema(context);
-}
-
-/**
- * 初始化 Console 认证与授权表结构。
- *
- * 关键点（中文）
- * - 该 schema 属于 console 级全局能力，不依赖任何单个 agent 项目。
- * - V1 只建表与索引，不在这里写入默认数据，默认数据由 auth bootstrap 负责。
- */
-function ensureAuthSchema(context: ConsoleStoreContext): void {
-  context.sqlite.exec(`
-    CREATE TABLE IF NOT EXISTS auth_users (
-      id TEXT PRIMARY KEY NOT NULL,
-      username TEXT NOT NULL,
-      password_hash TEXT NOT NULL,
-      display_name TEXT,
-      status TEXT NOT NULL,
-      created_at TEXT NOT NULL,
-      updated_at TEXT NOT NULL
-    );
-  `);
-  context.sqlite.exec(`
-    CREATE UNIQUE INDEX IF NOT EXISTS auth_users_username_uq
-    ON auth_users(username);
-  `);
-
-  context.sqlite.exec(`
-    CREATE TABLE IF NOT EXISTS auth_roles (
-      id TEXT PRIMARY KEY NOT NULL,
-      name TEXT NOT NULL,
-      description TEXT,
-      created_at TEXT NOT NULL,
-      updated_at TEXT NOT NULL
-    );
-  `);
-  context.sqlite.exec(`
-    CREATE UNIQUE INDEX IF NOT EXISTS auth_roles_name_uq
-    ON auth_roles(name);
-  `);
-
-  context.sqlite.exec(`
-    CREATE TABLE IF NOT EXISTS auth_permissions (
-      id TEXT PRIMARY KEY NOT NULL,
-      key TEXT NOT NULL,
-      description TEXT,
-      created_at TEXT NOT NULL,
-      updated_at TEXT NOT NULL
-    );
-  `);
-  context.sqlite.exec(`
-    CREATE UNIQUE INDEX IF NOT EXISTS auth_permissions_key_uq
-    ON auth_permissions(key);
-  `);
-
-  context.sqlite.exec(`
-    CREATE TABLE IF NOT EXISTS auth_user_roles (
-      id TEXT PRIMARY KEY NOT NULL,
-      user_id TEXT NOT NULL,
-      role_id TEXT NOT NULL,
-      created_at TEXT NOT NULL
-    );
-  `);
-  context.sqlite.exec(`
-    CREATE UNIQUE INDEX IF NOT EXISTS auth_user_roles_user_role_uq
-    ON auth_user_roles(user_id, role_id);
-  `);
-  context.sqlite.exec(`
-    CREATE INDEX IF NOT EXISTS auth_user_roles_user_id_idx
-    ON auth_user_roles(user_id);
-  `);
-  context.sqlite.exec(`
-    CREATE INDEX IF NOT EXISTS auth_user_roles_role_id_idx
-    ON auth_user_roles(role_id);
-  `);
-
-  context.sqlite.exec(`
-    CREATE TABLE IF NOT EXISTS auth_role_permissions (
-      id TEXT PRIMARY KEY NOT NULL,
-      role_id TEXT NOT NULL,
-      permission_id TEXT NOT NULL,
-      created_at TEXT NOT NULL
-    );
-  `);
-  context.sqlite.exec(`
-    CREATE UNIQUE INDEX IF NOT EXISTS auth_role_permissions_role_permission_uq
-    ON auth_role_permissions(role_id, permission_id);
-  `);
-  context.sqlite.exec(`
-    CREATE INDEX IF NOT EXISTS auth_role_permissions_role_id_idx
-    ON auth_role_permissions(role_id);
-  `);
-  context.sqlite.exec(`
-    CREATE INDEX IF NOT EXISTS auth_role_permissions_permission_id_idx
-    ON auth_role_permissions(permission_id);
-  `);
-
-  context.sqlite.exec(`
-    CREATE TABLE IF NOT EXISTS auth_tokens (
-      id TEXT PRIMARY KEY NOT NULL,
-      user_id TEXT NOT NULL,
-      name TEXT NOT NULL,
-      token_hash TEXT NOT NULL,
-      expires_at TEXT,
-      revoked_at TEXT,
-      last_used_at TEXT,
-      created_at TEXT NOT NULL,
-      updated_at TEXT NOT NULL
-    );
-  `);
-  context.sqlite.exec(`
-    CREATE UNIQUE INDEX IF NOT EXISTS auth_tokens_token_hash_uq
-    ON auth_tokens(token_hash);
-  `);
-  context.sqlite.exec(`
-    CREATE INDEX IF NOT EXISTS auth_tokens_user_id_idx
-    ON auth_tokens(user_id);
-  `);
-  context.sqlite.exec(`
-    CREATE INDEX IF NOT EXISTS auth_tokens_expires_at_idx
-    ON auth_tokens(expires_at);
-  `);
-
-  context.sqlite.exec(`
-    CREATE TABLE IF NOT EXISTS auth_audit_logs (
-      id TEXT PRIMARY KEY NOT NULL,
-      actor_user_id TEXT,
-      actor_token_id TEXT,
-      resource_type TEXT NOT NULL,
-      resource_id TEXT,
-      action TEXT NOT NULL,
-      result TEXT NOT NULL,
-      request_id TEXT,
-      ip TEXT,
-      user_agent TEXT,
-      meta_json TEXT,
-      created_at TEXT NOT NULL
-    );
-  `);
-  context.sqlite.exec(`
-    CREATE INDEX IF NOT EXISTS auth_audit_logs_actor_created_idx
-    ON auth_audit_logs(actor_user_id, created_at);
-  `);
-  context.sqlite.exec(`
-    CREATE INDEX IF NOT EXISTS auth_audit_logs_action_created_idx
-    ON auth_audit_logs(action, created_at);
-  `);
-  context.sqlite.exec(`
-    CREATE INDEX IF NOT EXISTS auth_audit_logs_resource_idx
-    ON auth_audit_logs(resource_type, resource_id);
-  `);
-}
-
-/**
- * 补齐 models 表的增量列。
- */
-function ensureModelsTableColumns(context: ConsoleStoreContext): void {
-  const rows = context.sqlite
-    .prepare("PRAGMA table_info(models)")
-    .all() as Array<{ name?: unknown }>;
-  const columns = new Set(
-    rows.map((row) => String(row.name || "").trim()).filter(Boolean),
-  );
-  if (!columns.has("is_paused")) {
-    context.sqlite.exec(
-      "ALTER TABLE models ADD COLUMN is_paused INTEGER NOT NULL DEFAULT 0;",
-    );
-  }
-}
-
-/**
- * 补齐 channel_accounts 表的增量列。
- */
-function ensureChannelAccountsTableColumns(context: ConsoleStoreContext): void {
-  const rows = context.sqlite
-    .prepare("PRAGMA table_info(channel_accounts)")
-    .all() as Array<{ name?: unknown }>;
-  const columns = new Set(
-    rows.map((row) => String(row.name || "").trim()).filter(Boolean),
-  );
-  if (!columns.has("owner")) {
-    context.sqlite.exec("ALTER TABLE channel_accounts ADD COLUMN owner TEXT;");
-  }
-  if (!columns.has("creator")) {
-    context.sqlite.exec("ALTER TABLE channel_accounts ADD COLUMN creator TEXT;");
-  }
-}
-
-/**
- * 迁移历史 env 双表到统一单表。
- */
-function ensureEnvEntriesMigration(context: ConsoleStoreContext): void {
-  const envEntryColumns = context.sqlite
-    .prepare("PRAGMA table_info(env_entries)")
-    .all() as Array<{ name?: unknown }>;
-  const envEntryColumnNames = new Set(
-    envEntryColumns.map((row) => String(row.name || "").trim()).filter(Boolean),
-  );
-  if (!envEntryColumnNames.has("description")) {
-    context.sqlite.exec("ALTER TABLE env_entries ADD COLUMN description TEXT;");
-  }
-
-  const tableRows = context.sqlite
-    .prepare(
-      "SELECT name FROM sqlite_master WHERE type = 'table' AND name IN ('global_env', 'agent_env');",
-    )
-    .all() as Array<{ name?: unknown }>;
-  const tableNames = new Set(
-    tableRows.map((row) => String(row.name || "").trim()).filter(Boolean),
-  );
-  if (tableNames.has("global_env")) {
-    context.sqlite.exec(`
-      INSERT OR IGNORE INTO env_entries (
-        scope, agent_id, key, description, value_encrypted, created_at, updated_at
-      )
-      SELECT
-        'global',
-        '',
-        key,
-        NULL,
-        value_encrypted,
-        created_at,
-        updated_at
-      FROM global_env;
-    `);
-  }
-  if (tableNames.has("agent_env")) {
-    context.sqlite.exec(`
-      INSERT OR IGNORE INTO env_entries (
-        scope, agent_id, key, description, value_encrypted, created_at, updated_at
-      )
-      SELECT
-        'agent',
-        agent_id,
-        key,
-        NULL,
-        value_encrypted,
-        created_at,
-        updated_at
-      FROM agent_env;
-    `);
-    context.sqlite.exec(`
-      UPDATE env_entries
-      SET agent_id = ''
-      WHERE scope = 'global' AND agent_id IS NULL;
-    `);
-  }
-}