ima2-gen 1.1.21 → 1.1.22

package/routes/prompts.ts

@@ -1,428 +0,0 @@
-import type { Express, Request, Response } from "express";
-import { logError, logEvent } from "../lib/logger.js";
-import { getDb } from "../lib/db.js";
-
-import { errInfo } from "../lib/errInfo.js";
-import type { RouteRuntimeContext } from "../lib/runtimeContext.js";
-
-interface PromptRow {
-  id: string;
-  folder_id: string;
-  name: string;
-  text: string;
-  tags: string | null;
-  mode: string | null;
-  is_favorite?: number;
-  favorited_at?: number | null;
-  created_at: number;
-  updated_at: number;
-  folder_name?: string | null;
-}
-
-interface FolderRow {
-  id: string;
-  parent_id: string;
-  name: string;
-  created_at: number;
-  updated_at: number;
-}
-
-type IdParams = { id: string };
-
-function getPromptsDb() {
-  return getDb();
-}
-
-function generateId() {
-  return `${Date.now().toString(36)}_${Math.random().toString(36).slice(2, 8)}`;
-}
-
-export function registerPromptRoutes(app: Express, _ctx: RouteRuntimeContext) {
-  // ── Prompts ───────────────────────────────────────────────────────────────
-
-  app.get("/api/prompts", async (req: Request, res: Response) => {
-    try {
-      const db = getPromptsDb();
-      const search = typeof req.query.search === "string" ? req.query.search.trim() : "";
-      const folderId = typeof req.query.folderId === "string" ? req.query.folderId : null;
-      const favoritesOnly = req.query.favoritesOnly === "1" || req.query.favoritesOnly === "true";
-
-      let where = "WHERE 1=1";
-      const params: unknown[] = [];
-
-      if (folderId) {
-        where += " AND p.folder_id = ?";
-        params.push(folderId);
-      } else {
-        where += " AND p.folder_id != '__trash__'";
-      }
-
-      if (favoritesOnly) {
-        where += " AND p.is_favorite = 1";
-      }
-
-      if (search) {
-        where += " AND (p.name LIKE ? OR p.text LIKE ? OR p.tags LIKE ?)";
-        const like = `%${search}%`;
-        params.push(like, like, like);
-      }
-
-      const prompts = db
-        .prepare(
-          `SELECT p.*, f.name as folder_name
-           FROM prompts p
-           LEFT JOIN prompt_folders f ON p.folder_id = f.id
-           ${where}
-           ORDER BY p.updated_at DESC`
-        )
-        .all(...params) as PromptRow[];
-
-      const folders = db
-        .prepare("SELECT * FROM prompt_folders WHERE id NOT IN ('__root__', '__trash__') ORDER BY name COLLATE NOCASE")
-        .all() as FolderRow[];
-
-      res.json({ prompts: prompts.map(normalizePrompt), folders: folders.map(normalizeFolder) });
-    } catch (e) {
-      const err = errInfo(e);
-      logError("prompts", "list_error", err.raw);
-      res.status(500).json({ error: err.message });
-    }
-  });
-
-  app.post("/api/prompts", async (req: Request, res: Response) => {
-    try {
-      const db = getPromptsDb();
-      const body = (req.body ?? {}) as { name?: unknown; text?: unknown; tags?: unknown; folderId?: unknown; mode?: unknown };
-      const { name, text, tags, folderId, mode } = body;
-
-      if (!text || typeof text !== "string") {
-        return res.status(400).json({ error: "text is required" });
-      }
-
-      const promptName = typeof name === "string" && name.trim() ? name.trim() : text.slice(0, 30);
-      const folder_id = typeof folderId === "string" && folderId ? folderId : "__root__";
-      const tagsJson = Array.isArray(tags) ? JSON.stringify(tags) : null;
-      const id = generateId();
-      const now = Math.floor(Date.now() / 1000);
-
-      db.prepare(
-        `INSERT INTO prompts (id, folder_id, name, text, tags, mode, created_at, updated_at)
-         VALUES (?, ?, ?, ?, ?, ?, ?, ?)`
-      ).run(id, folder_id, promptName, text, tagsJson, (typeof mode === "string" ? mode : null), now, now);
-
-      logEvent("prompts", "created", { id, folder_id });
-      res.status(201).json({ prompt: normalizePrompt(db.prepare("SELECT * FROM prompts WHERE id = ?").get(id) as PromptRow) });
-    } catch (e) {
-      const err = errInfo(e);
-      logError("prompts", "create_error", err.raw);
-      res.status(500).json({ error: err.message });
-    }
-  });
-
-  app.get("/api/prompts/:id", async (req: Request<IdParams>, res: Response) => {
-    try {
-      const db = getPromptsDb();
-      const row = db.prepare("SELECT * FROM prompts WHERE id = ?").get(req.params.id) as PromptRow | undefined;
-      if (!row) return res.status(404).json({ error: "Not found" });
-      res.json({ prompt: normalizePrompt(row) });
-    } catch (e) {
-      const err = errInfo(e);
-      logError("prompts", "get_error", err.raw);
-      res.status(500).json({ error: err.message });
-    }
-  });
-
-  app.patch("/api/prompts/:id", async (req: Request<IdParams>, res: Response) => {
-    try {
-      const db = getPromptsDb();
-      const body = (req.body ?? {}) as { name?: unknown; text?: unknown; tags?: unknown; folderId?: unknown; mode?: unknown };
-      const { name, text, tags, folderId, mode } = body;
-      const sets: string[] = [];
-      const params: unknown[] = [];
-
-      if (typeof name === "string") { sets.push("name = ?"); params.push(name); }
-      if (typeof text === "string") { sets.push("text = ?"); params.push(text); }
-      if (Array.isArray(tags)) { sets.push("tags = ?"); params.push(JSON.stringify(tags)); }
-      if (typeof folderId === "string") { sets.push("folder_id = ?"); params.push(folderId); }
-      if (typeof mode === "string") { sets.push("mode = ?"); params.push(mode); }
-
-      if (sets.length === 0) return res.status(400).json({ error: "No fields to update" });
-
-      sets.push("updated_at = ?");
-      params.push(Math.floor(Date.now() / 1000));
-      params.push(req.params.id);
-
-      db.prepare(`UPDATE prompts SET ${sets.join(", ")} WHERE id = ?`).run(...params);
-
-      const row = db.prepare("SELECT * FROM prompts WHERE id = ?").get(req.params.id) as PromptRow | undefined;
-      if (!row) return res.status(404).json({ error: "Not found" });
-      res.json({ prompt: normalizePrompt(row) });
-    } catch (e) {
-      const err = errInfo(e);
-      logError("prompts", "patch_error", err.raw);
-      res.status(500).json({ error: err.message });
-    }
-  });
-
-  app.delete("/api/prompts/:id", async (req: Request<IdParams>, res: Response) => {
-    try {
-      const db = getPromptsDb();
-      db.prepare("UPDATE prompts SET folder_id = '__trash__', updated_at = ? WHERE id = ?").run(
-        Math.floor(Date.now() / 1000),
-        req.params.id,
-      );
-      logEvent("prompts", "soft_deleted", { id: req.params.id });
-      res.json({ ok: true });
-    } catch (e) {
-      const err = errInfo(e);
-      logError("prompts", "delete_error", err.raw);
-      res.status(500).json({ error: err.message });
-    }
-  });
-
-  app.post("/api/prompts/:id/favorite", async (req: Request<IdParams>, res: Response) => {
-    try {
-      const db = getPromptsDb();
-      const row = db.prepare("SELECT is_favorite FROM prompts WHERE id = ?").get(req.params.id) as { is_favorite?: number } | undefined;
-      if (!row) return res.status(404).json({ error: "Not found" });
-
-      const newVal = row.is_favorite ? 0 : 1;
-      const now = Math.floor(Date.now() / 1000);
-      db.prepare("UPDATE prompts SET is_favorite = ?, favorited_at = ? WHERE id = ?").run(
-        newVal,
-        newVal ? now : null,
-        req.params.id,
-      );
-
-      res.json({ isFavorite: !!newVal, favoritedAt: newVal ? now : null });
-    } catch (e) {
-      const err = errInfo(e);
-      logError("prompts", "favorite_error", err.raw);
-      res.status(500).json({ error: err.message });
-    }
-  });
-
-  // ── Import / Export ───────────────────────────────────────────────────────
-
-  app.post("/api/prompts/import", async (req: Request, res: Response) => {
-    try {
-      const db = getPromptsDb();
-      const body = (req.body ?? {}) as { folders?: unknown; prompts?: unknown };
-      const importFolders: Array<{ id?: string; name?: string; parentId?: string }> = Array.isArray(body.folders) ? body.folders : [];
-      const importPrompts: Array<{ id?: string; name?: string; text?: string; tags?: unknown; folderId?: string; mode?: string; isFavorite?: boolean }> = Array.isArray(body.prompts) ? body.prompts : [];
-
-      const result = { foldersCreated: 0, promptsImported: 0, duplicatesSkipped: 0 };
-      const now = Math.floor(Date.now() / 1000);
-
-      // Build name→id map for existing folders
-      const existingFolders = db.prepare("SELECT * FROM prompt_folders").all() as FolderRow[];
-      const folderMap = new Map<string, FolderRow>(existingFolders.map((f) => [f.id, f]));
-      const namePathMap = new Map<string, string>();
-      for (const f of existingFolders) {
-        const parent = folderMap.get(f.parent_id);
-        const path = parent && parent.id !== "__root__" ? `${parent.name}/${f.name}` : f.name;
-        namePathMap.set(path.toLowerCase(), f.id);
-      }
-
-      // Import folders
-      for (const f of importFolders) {
-        if (!f.name) continue;
-        const path = f.parentId && f.parentId !== "__root__"
-          ? `${folderMap.get(f.parentId)?.name || ""}/${f.name}`
-          : f.name;
-        if (namePathMap.has(path.toLowerCase())) continue;
-
-        const id = f.id || generateId();
-        db.prepare(
-          "INSERT INTO prompt_folders (id, parent_id, name, created_at, updated_at) VALUES (?, ?, ?, ?, ?)"
-        ).run(id, f.parentId || "__root__", f.name, now, now);
-        namePathMap.set(path.toLowerCase(), id);
-        folderMap.set(id, { id, parent_id: f.parentId || "__root__", name: f.name, created_at: now, updated_at: now });
-        result.foldersCreated++;
-      }
-
-      // Import prompts
-      for (const p of importPrompts) {
-        if (!p.text) continue;
-        const folderId = p.folderId && folderMap.has(p.folderId) ? p.folderId : "__root__";
-        // Check duplicate by text + folder
-        const dup = db.prepare("SELECT 1 FROM prompts WHERE text = ? AND folder_id = ? LIMIT 1").get(p.text, folderId);
-        if (dup) {
-          result.duplicatesSkipped++;
-          continue;
-        }
-        const id = p.id || generateId();
-        const tagsJson = Array.isArray(p.tags) ? JSON.stringify(p.tags) : null;
-        db.prepare(
-          `INSERT INTO prompts (id, folder_id, name, text, tags, mode, is_favorite, created_at, updated_at)
-           VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)`
-        ).run(id, folderId, p.name || p.text.slice(0, 30), p.text, tagsJson, p.mode || null, p.isFavorite ? 1 : 0, now, now);
-        result.promptsImported++;
-      }
-
-      logEvent("prompts", "imported", result);
-      res.json(result);
-    } catch (e) {
-      const err = errInfo(e);
-      logError("prompts", "import_error", err.raw);
-      res.status(500).json({ error: err.message });
-    }
-  });
-
-  app.get("/api/prompts/export", async (_req: Request, res: Response) => {
-    try {
-      const db = getPromptsDb();
-      const prompts = db.prepare("SELECT * FROM prompts WHERE folder_id != '__trash__'").all() as PromptRow[];
-      const folders = db.prepare("SELECT * FROM prompt_folders WHERE id NOT IN ('__root__', '__trash__')").all() as FolderRow[];
-
-      res.json({
-        version: 1,
-        exportedAt: new Date().toISOString(),
-        folders: folders.map((f) => ({ id: f.id, name: f.name, parentId: f.parent_id })),
-        prompts: prompts.map((p) => ({
-          id: p.id,
-          name: p.name,
-          text: p.text,
-          tags: p.tags ? JSON.parse(p.tags) : [],
-          folderId: p.folder_id,
-          mode: p.mode,
-          isFavorite: !!p.is_favorite,
-        })),
-      });
-    } catch (e) {
-      const err = errInfo(e);
-      logError("prompts", "export_error", err.raw);
-      res.status(500).json({ error: err.message });
-    }
-  });
-
-  // ── Folders ───────────────────────────────────────────────────────────────
-
-  app.get("/api/prompts/folders", async (_req: Request, res: Response) => {
-    try {
-      const db = getPromptsDb();
-      const rows = db.prepare("SELECT * FROM prompt_folders WHERE id NOT IN ('__root__', '__trash__') ORDER BY name COLLATE NOCASE").all() as FolderRow[];
-      res.json({ folders: rows.map(normalizeFolder) });
-    } catch (e) {
-      const err = errInfo(e);
-      logError("prompts", "folders_list_error", err.raw);
-      res.status(500).json({ error: err.message });
-    }
-  });
-
-  app.post("/api/prompts/folders", async (req: Request, res: Response) => {
-    try {
-      const db = getPromptsDb();
-      const body = (req.body ?? {}) as { name?: unknown; parentId?: unknown };
-      const { name, parentId } = body;
-      if (!name || typeof name !== "string" || !name.trim()) {
-        return res.status(400).json({ error: "name is required" });
-      }
-
-      const parent_id = typeof parentId === "string" && parentId ? parentId : "__root__";
-      const now = Math.floor(Date.now() / 1000);
-      const id = generateId();
-
-      try {
-        db.prepare(
-          "INSERT INTO prompt_folders (id, parent_id, name, created_at, updated_at) VALUES (?, ?, ?, ?, ?)"
-        ).run(id, parent_id, name.trim(), now, now);
-      } catch (e) {
-        const err = errInfo(e);
-        if (err.message && err.message.includes("UNIQUE constraint failed")) {
-          return res.status(409).json({ error: "Folder name already exists in this parent" });
-        }
-        throw err.raw;
-      }
-
-      res.status(201).json({ folder: normalizeFolder(db.prepare("SELECT * FROM prompt_folders WHERE id = ?").get(id) as FolderRow) });
-    } catch (e) {
-      const err = errInfo(e);
-      logError("prompts", "folder_create_error", err.raw);
-      res.status(500).json({ error: err.message });
-    }
-  });
-
-  app.patch("/api/prompts/folders/:id", async (req: Request<IdParams>, res: Response) => {
-    try {
-      const db = getPromptsDb();
-      const body = (req.body ?? {}) as { name?: unknown; parentId?: unknown };
-      const { name, parentId } = body;
-      const sets: string[] = [];
-      const params: unknown[] = [];
-
-      if (typeof name === "string" && name.trim()) { sets.push("name = ?"); params.push(name.trim()); }
-      if (typeof parentId === "string") { sets.push("parent_id = ?"); params.push(parentId); }
-      if (sets.length === 0) return res.status(400).json({ error: "No fields to update" });
-
-      sets.push("updated_at = ?");
-      params.push(Math.floor(Date.now() / 1000));
-      params.push(req.params.id);
-
-      try {
-        db.prepare(`UPDATE prompt_folders SET ${sets.join(", ")} WHERE id = ?`).run(...params);
-      } catch (e) {
-        const err = errInfo(e);
-        if (err.message && err.message.includes("UNIQUE constraint failed")) {
-          return res.status(409).json({ error: "Folder name already exists in this parent" });
-        }
-        throw err.raw;
-      }
-
-      const row = db.prepare("SELECT * FROM prompt_folders WHERE id = ?").get(req.params.id) as FolderRow | undefined;
-      res.json({ folder: row ? normalizeFolder(row) : null });
-    } catch (e) {
-      const err = errInfo(e);
-      logError("prompts", "folder_patch_error", err.raw);
-      res.status(500).json({ error: err.message });
-    }
-  });
-
-  app.delete("/api/prompts/folders/:id", async (req: Request<IdParams>, res: Response) => {
-    try {
-      const db = getPromptsDb();
-      const strategy = req.query.strategy === "deleteItems" ? "deleteItems" : "moveToRoot";
-
-      if (strategy === "moveToRoot") {
-        db.prepare("UPDATE prompts SET folder_id = '__root__' WHERE folder_id = ?").run(req.params.id);
-      } else {
-        db.prepare("UPDATE prompts SET folder_id = '__trash__' WHERE folder_id = ?").run(req.params.id);
-      }
-
-      db.prepare("DELETE FROM prompt_folders WHERE id = ?").run(req.params.id);
-      logEvent("prompts", "folder_deleted", { id: req.params.id, strategy });
-      res.json({ ok: true });
-    } catch (e) {
-      const err = errInfo(e);
-      logError("prompts", "folder_delete_error", err.raw);
-      res.status(500).json({ error: err.message });
-    }
-  });
-}
-
-// ── Helpers ─────────────────────────────────────────────────────────────────
-
-function normalizePrompt(row: PromptRow) {
-  return {
-    id: row.id,
-    folderId: row.folder_id,
-    name: row.name,
-    text: row.text,
-    tags: row.tags ? JSON.parse(row.tags) : [],
-    mode: row.mode,
-    isFavorite: !!row.is_favorite,
-    favoritedAt: row.favorited_at || null,
-    createdAt: row.created_at,
-    updatedAt: row.updated_at,
-  };
-}
-
-function normalizeFolder(row: FolderRow) {
-  return {
-    id: row.id,
-    parentId: row.parent_id,
-    name: row.name,
-    createdAt: row.created_at,
-    updatedAt: row.updated_at,
-  };
-}

package/routes/quota.ts

@@ -1,89 +0,0 @@
-import type { Express } from "express";
-import { readFileSync } from "node:fs";
-import { homedir } from "node:os";
-import { join } from "node:path";
-import type { RouteRuntimeContext } from "../lib/runtimeContext.js";
-
-export interface QuotaWindow {
-  label: string;
-  percent: number;
-  resetsAt: string | null;
-}
-
-export interface QuotaResult {
-  provider: string;
-  account?: { email: string | null; plan: string | null } | null;
-  windows: QuotaWindow[];
-  error?: boolean;
-  authenticated?: boolean;
-}
-
-function readCodexTokens(): { access_token: string; account_id: string } | null {
-  const codexHome = process.env.CODEX_HOME || join(homedir(), ".codex");
-  try {
-    const j = JSON.parse(readFileSync(join(codexHome, "auth.json"), "utf8"));
-    if (j?.tokens?.access_token) {
-      return { access_token: j.tokens.access_token, account_id: j.tokens.account_id ?? "" };
-    }
-  } catch {}
-  return null;
-}
-
-async function fetchCodexUsage(tokens: { access_token: string; account_id: string }): Promise<QuotaResult> {
-  try {
-    const resp = await fetch("https://chatgpt.com/backend-api/wham/usage", {
-      headers: {
-        Authorization: `Bearer ${tokens.access_token}`,
-        "ChatGPT-Account-Id": tokens.account_id,
-      },
-      signal: AbortSignal.timeout(8000),
-    });
-    if (!resp.ok) {
-      if (resp.status === 401 || resp.status === 403) return { provider: "codex", authenticated: false, windows: [] };
-      return { provider: "codex", error: true, windows: [] };
-    }
-    const data = await resp.json() as {
-      email?: string | null;
-      plan_type?: string | null;
-      rate_limit?: {
-        primary_window?: { used_percent?: number; reset_at?: number };
-        secondary_window?: { used_percent?: number; reset_at?: number };
-      };
-    };
-    const account = { email: data.email ?? null, plan: data.plan_type ?? null };
-    const windows: QuotaWindow[] = [];
-    if (data.rate_limit?.primary_window) {
-      windows.push({
-        label: "5h",
-        percent: Math.round(data.rate_limit.primary_window.used_percent ?? 0),
-        resetsAt: data.rate_limit.primary_window.reset_at
-          ? new Date(data.rate_limit.primary_window.reset_at * 1000).toISOString()
-          : null,
-      });
-    }
-    if (data.rate_limit?.secondary_window) {
-      windows.push({
-        label: "7d",
-        percent: Math.round(data.rate_limit.secondary_window.used_percent ?? 0),
-        resetsAt: data.rate_limit.secondary_window.reset_at
-          ? new Date(data.rate_limit.secondary_window.reset_at * 1000).toISOString()
-          : null,
-      });
-    }
-    return { provider: "codex", account, windows };
-  } catch {
-    return { provider: "codex", error: true, windows: [] };
-  }
-}
-
-export function registerQuotaRoutes(app: Express, _ctx: RouteRuntimeContext) {
-  app.get("/api/quota", async (_req, res) => {
-    const tokens = readCodexTokens();
-    if (!tokens) {
-      res.json({ codex: { provider: "codex", authenticated: false, windows: [] } });
-      return;
-    }
-    const codex = await fetchCodexUsage(tokens);
-    res.json({ codex });
-  });
-}