wind-agent-cli 1.37.2

package/server/modules/database/repositories/provider-models.ts

@@ -0,0 +1,168 @@
+import { getConnection } from '@/modules/database/connection.js';
+import type {
+  CustomProviderModelInput,
+  CustomProviderModelRecord,
+  LLMProvider,
+} from '@/shared/types.js';
+
+type CustomProviderModelRow = {
+  id: number;
+  provider: LLMProvider;
+  model_id: string;
+  model_name: string;
+  sort_order: number;
+};
+
+const toCustomProviderModelRecord = (
+  row: CustomProviderModelRow,
+): CustomProviderModelRecord => ({
+  recordId: row.id,
+  provider: row.provider,
+  modelId: row.model_id,
+  model: row.model_name,
+  sortOrder: row.sort_order,
+});
+
+const readCustomProviderModelRow = (
+  provider: LLMProvider,
+  recordId: number,
+): CustomProviderModelRow | null => {
+  const row = getConnection().prepare(`
+    SELECT id, provider, model_id, model_name, sort_order
+    FROM provider_models
+    WHERE provider = ? AND id = ?
+  `).get(provider, recordId) as CustomProviderModelRow | undefined;
+
+  return row ?? null;
+};
+
+/**
+ * Custom provider-model persistence API consumed by the Providers module.
+ *
+ * Every row in this repository is user-created. Predefined models are owned by
+ * provider adapters and never enter this table. Model-id changes are propagated
+ * to stored sessions, while deletion replaces affected session selections with
+ * the predefined default supplied by the Providers service.
+ */
+export const providerModelsDb = {
+  listCustomProviderModels(provider: LLMProvider): CustomProviderModelRecord[] {
+    const rows = getConnection().prepare(`
+      SELECT id, provider, model_id, model_name, sort_order
+      FROM provider_models
+      WHERE provider = ?
+      ORDER BY sort_order ASC, lower(model_name) ASC, id ASC
+    `).all(provider) as CustomProviderModelRow[];
+
+    return rows.map(toCustomProviderModelRecord);
+  },
+
+  getCustomProviderModel(
+    provider: LLMProvider,
+    recordId: number,
+  ): CustomProviderModelRecord | null {
+    const row = readCustomProviderModelRow(provider, recordId);
+    return row ? toCustomProviderModelRecord(row) : null;
+  },
+
+  findCustomProviderModelByModelId(
+    provider: LLMProvider,
+    modelId: string,
+  ): CustomProviderModelRecord | null {
+    const row = getConnection().prepare(`
+      SELECT id, provider, model_id, model_name, sort_order
+      FROM provider_models
+      WHERE provider = ? AND model_id = ?
+    `).get(provider, modelId) as CustomProviderModelRow | undefined;
+
+    return row ? toCustomProviderModelRecord(row) : null;
+  },
+
+  createCustomProviderModel(
+    provider: LLMProvider,
+    input: CustomProviderModelInput,
+  ): CustomProviderModelRecord {
+    const db = getConnection();
+    const nextOrder = db.prepare(`
+      SELECT COALESCE(MAX(sort_order), -1) + 1 AS next_order
+      FROM provider_models
+      WHERE provider = ?
+    `).get(provider) as { next_order: number };
+
+    const result = db.prepare(`
+      INSERT INTO provider_models (provider, model_id, model_name, sort_order)
+      VALUES (?, ?, ?, ?)
+    `).run(provider, input.id, input.model, nextOrder.next_order);
+
+    const row = readCustomProviderModelRow(provider, Number(result.lastInsertRowid));
+    if (!row) {
+      throw new Error('Created provider model could not be read back.');
+    }
+
+    return toCustomProviderModelRecord(row);
+  },
+
+  updateCustomProviderModel(
+    provider: LLMProvider,
+    recordId: number,
+    input: CustomProviderModelInput,
+  ): CustomProviderModelRecord | null {
+    const db = getConnection();
+    const update = db.transaction(() => {
+      const previous = readCustomProviderModelRow(provider, recordId);
+      if (!previous) {
+        return null;
+      }
+
+      db.prepare(`
+        UPDATE provider_models
+        SET model_id = ?, model_name = ?, updated_at = CURRENT_TIMESTAMP
+        WHERE provider = ? AND id = ?
+      `).run(input.id, input.model, provider, recordId);
+
+      if (previous.model_id !== input.id) {
+        db.prepare(`
+          UPDATE sessions
+          SET model = ?, updated_at = CURRENT_TIMESTAMP
+          WHERE provider = ? AND model = ?
+        `).run(input.id, provider, previous.model_id);
+      }
+
+      return readCustomProviderModelRow(provider, recordId);
+    });
+
+    const row = update();
+    return row ? toCustomProviderModelRecord(row) : null;
+  },
+
+  deleteCustomProviderModel(
+    provider: LLMProvider,
+    recordId: number,
+    fallbackModelId: string,
+  ): CustomProviderModelRecord | null {
+    const db = getConnection();
+    const remove = db.transaction(() => {
+      const row = readCustomProviderModelRow(provider, recordId);
+      if (!row) {
+        return null;
+      }
+
+      // The effort is cleared alongside the model because it was chosen for the
+      // deleted model; replaced sessions resume on the fallback model's default.
+      db.prepare(`
+        UPDATE sessions
+        SET model = ?, effort = NULL, updated_at = CURRENT_TIMESTAMP
+        WHERE provider = ? AND model = ?
+      `).run(fallbackModelId, provider, row.model_id);
+
+      db.prepare(`
+        DELETE FROM provider_models
+        WHERE provider = ? AND id = ?
+      `).run(provider, recordId);
+
+      return row;
+    });
+
+    const row = remove();
+    return row ? toCustomProviderModelRecord(row) : null;
+  },
+};

package/server/modules/database/repositories/push-subscriptions.ts

@@ -0,0 +1,80 @@
+/**
+ * Push subscriptions repository.
+ *
+ * Persists browser push subscription endpoints and keys per user.
+ */
+
+import { getConnection } from '@/modules/database/connection.js';
+
+type PushSubscriptionLookupRow = {
+  endpoint: string;
+  keys_p256dh: string;
+  keys_auth: string;
+};
+
+export const pushSubscriptionsDb = {
+  /** Upserts a push subscription endpoint for a user. */
+  createPushSubscription(
+    userId: number,
+    endpoint: string,
+    keysP256dh: string,
+    keysAuth: string
+  ): void {
+    const db = getConnection();
+    db.prepare(
+      `INSERT INTO push_subscriptions (user_id, endpoint, keys_p256dh, keys_auth)
+       VALUES (?, ?, ?, ?)
+       ON CONFLICT(endpoint) DO UPDATE SET
+         user_id = excluded.user_id,
+         keys_p256dh = excluded.keys_p256dh,
+         keys_auth = excluded.keys_auth`
+    ).run(userId, endpoint, keysP256dh, keysAuth);
+  },
+
+  /** Returns all subscriptions for a user. */
+  getPushSubscriptions(userId: number): PushSubscriptionLookupRow[] {
+    const db = getConnection();
+    return db
+      .prepare(
+        'SELECT endpoint, keys_p256dh, keys_auth FROM push_subscriptions WHERE user_id = ?'
+      )
+      .all(userId) as PushSubscriptionLookupRow[];
+  },
+
+  /** Deletes one subscription by endpoint. */
+  deletePushSubscription(endpoint: string): void {
+    const db = getConnection();
+    db.prepare('DELETE FROM push_subscriptions WHERE endpoint = ?').run(endpoint);
+  },
+
+  /** Deletes all subscriptions for a user. */
+  deletePushSubscriptionsForUser(userId: number): void {
+    const db = getConnection();
+    db.prepare('DELETE FROM push_subscriptions WHERE user_id = ?').run(userId);
+  },
+
+  // Legacy aliases used by existing services/routes
+  saveSubscription(
+    userId: number,
+    endpoint: string,
+    keysP256dh: string,
+    keysAuth: string
+  ): void {
+    pushSubscriptionsDb.createPushSubscription(
+      userId,
+      endpoint,
+      keysP256dh,
+      keysAuth
+    );
+  },
+  getSubscriptions(userId: number): PushSubscriptionLookupRow[] {
+    return pushSubscriptionsDb.getPushSubscriptions(userId);
+  },
+  removeSubscription(endpoint: string): void {
+    pushSubscriptionsDb.deletePushSubscription(endpoint);
+  },
+  removeAllForUser(userId: number): void {
+    pushSubscriptionsDb.deletePushSubscriptionsForUser(userId);
+  },
+};
+

package/server/modules/database/repositories/scan-state.db.ts

@@ -0,0 +1,42 @@
+import { getConnection } from '@/modules/database/connection.js';
+
+type ScanStateRow = {
+  last_scanned_at: string;
+};
+
+export const scanStateDb = {
+    getLastScannedAt() {
+        const db = getConnection();
+
+        const row = db
+            .prepare(`SELECT last_scanned_at FROM scan_state WHERE id = 1`)
+            .get() as ScanStateRow;
+
+        if (!row) {
+            return null; // Before any scan, the row is undefined.
+        }
+
+        let lastScannedDate: Date | null = null;
+        const lastScannedStr = row.last_scanned_at;
+
+        if (lastScannedStr) {
+            // SQLite CURRENT_TIMESTAMP returns UTC in "YYYY-MM-DD HH:MM:SS" format.
+            // Replace space with 'T' and append 'Z' to parse reliably in JS across all platforms.
+            lastScannedDate = new Date(lastScannedStr.replace(' ', 'T') + 'Z');
+        }
+
+        return lastScannedDate;
+    },
+
+    updateLastScannedAt(scannedAt: Date = new Date()) {
+        const db = getConnection();
+        const sqliteTimestamp = scannedAt.toISOString().slice(0, 19).replace('T', ' ');
+
+        db.prepare(`
+            INSERT INTO scan_state (id, last_scanned_at)
+            VALUES (1, ?)
+            ON CONFLICT (id)
+            DO UPDATE SET last_scanned_at = excluded.last_scanned_at
+        `).run(sqliteTimestamp);
+    }
+};