ima2-gen 1.1.7 → 1.1.9

package/lib/sessionStore.js

@@ -1,272 +1,221 @@
 import { ulid } from "ulid";
 import { getDb } from "./db.js";
-
 function now() {
-  return Date.now();
+    return Date.now();
 }
-
 export function createSession({ title = "Untitled" } = {}) {
-  const db = getDb();
-  const id = "s_" + ulid();
-  const t = now();
-  db.prepare(
-    "INSERT INTO sessions (id, title, created_at, updated_at, graph_version) VALUES (?, ?, ?, ?, 0)",
-  ).run(id, title, t, t);
-  return { id, title, createdAt: t, updatedAt: t, graphVersion: 0 };
+    const db = getDb();
+    const id = "s_" + ulid();
+    const t = now();
+    db.prepare("INSERT INTO sessions (id, title, created_at, updated_at, graph_version) VALUES (?, ?, ?, ?, 0)").run(id, title, t, t);
+    return { id, title, createdAt: t, updatedAt: t, graphVersion: 0 };
 }
-
 export function listSessions() {
-  const db = getDb();
-  const rows = db
-    .prepare(
-      "SELECT id, title, created_at AS createdAt, updated_at AS updatedAt, graph_version AS graphVersion FROM sessions ORDER BY updated_at DESC",
-    )
-    .all();
-  return rows.map((r) => ({
-    ...r,
-    nodeCount: db
-      .prepare("SELECT COUNT(*) AS c FROM nodes WHERE session_id = ?")
-      .get(r.id).c,
-  }));
+    const db = getDb();
+    const rows = db
+        .prepare("SELECT id, title, created_at AS createdAt, updated_at AS updatedAt, graph_version AS graphVersion FROM sessions ORDER BY updated_at DESC")
+        .all();
+    return rows.map((r) => ({
+        ...r,
+        nodeCount: db
+            .prepare("SELECT COUNT(*) AS c FROM nodes WHERE session_id = ?")
+            .get(r.id).c,
+    }));
 }
-
 export function getSession(id) {
-  const db = getDb();
-  const session = db
-    .prepare(
-      "SELECT id, title, created_at AS createdAt, updated_at AS updatedAt, graph_version AS graphVersion FROM sessions WHERE id = ?",
-    )
-    .get(id);
-  if (!session) return null;
-  const nodes = db
-    .prepare("SELECT id, x, y, data FROM nodes WHERE session_id = ?")
-    .all(id)
-    .map((n) => ({ id: n.id, x: n.x, y: n.y, data: safeParse(n.data) }));
-  const edges = db
-    .prepare("SELECT id, source, target, data FROM edges WHERE session_id = ?")
-    .all(id)
-    .map((e) => ({
-      id: e.id,
-      source: e.source,
-      target: e.target,
-      data: safeParse(e.data),
+    const db = getDb();
+    const session = db
+        .prepare("SELECT id, title, created_at AS createdAt, updated_at AS updatedAt, graph_version AS graphVersion FROM sessions WHERE id = ?")
+        .get(id);
+    if (!session)
+        return null;
+    const nodes = db
+        .prepare("SELECT id, x, y, data FROM nodes WHERE session_id = ?")
+        .all(id)
+        .map((n) => ({ id: n.id, x: n.x, y: n.y, data: safeParse(n.data) }));
+    const edges = db
+        .prepare("SELECT id, source, target, data FROM edges WHERE session_id = ?")
+        .all(id)
+        .map((e) => ({
+        id: e.id,
+        source: e.source,
+        target: e.target,
+        data: safeParse(e.data),
     }));
-  return { ...session, nodes, edges };
+    return { ...session, nodes, edges };
 }
-
 export function getSessionTitleMap(ids = []) {
-  const cleanIds = [...new Set(ids.filter((id) => typeof id === "string" && id.length > 0))];
-  if (cleanIds.length === 0) return new Map();
-  const placeholders = cleanIds.map(() => "?").join(", ");
-  const rows = getDb()
-    .prepare(`SELECT id, title FROM sessions WHERE id IN (${placeholders})`)
-    .all(...cleanIds);
-  return new Map(rows.map((row) => [row.id, row.title]));
+    const cleanIds = [...new Set(ids.filter((id) => typeof id === "string" && id.length > 0))];
+    if (cleanIds.length === 0)
+        return new Map();
+    const placeholders = cleanIds.map(() => "?").join(", ");
+    const rows = getDb()
+        .prepare(`SELECT id, title FROM sessions WHERE id IN (${placeholders})`)
+        .all(...cleanIds);
+    return new Map(rows.map((row) => [row.id, row.title]));
 }
-
 export function renameSession(id, title) {
-  const db = getDb();
-  const res = db
-    .prepare("UPDATE sessions SET title = ?, updated_at = ? WHERE id = ?")
-    .run(title, now(), id);
-  return res.changes > 0;
+    const db = getDb();
+    const res = db
+        .prepare("UPDATE sessions SET title = ?, updated_at = ? WHERE id = ?")
+        .run(title, now(), id);
+    return res.changes > 0;
 }
-
 export function deleteSession(id) {
-  const db = getDb();
-  const res = db.prepare("DELETE FROM sessions WHERE id = ?").run(id);
-  return res.changes > 0;
+    const db = getDb();
+    const res = db.prepare("DELETE FROM sessions WHERE id = ?").run(id);
+    return res.changes > 0;
 }
-
 const MAX_STR = 10_000;
-
 function cleanStr(v) {
-  if (typeof v !== "string") return "";
-  return v.length > MAX_STR ? v.slice(0, MAX_STR) : v;
+    if (typeof v !== "string")
+        return "";
+    return v.length > MAX_STR ? v.slice(0, MAX_STR) : v;
 }
-
 function cleanData(v) {
-  try {
-    const json = JSON.stringify(v ?? {});
-    return json.length > MAX_STR * 10 ? "{}" : json;
-  } catch {
-    return "{}";
-  }
+    try {
+        const json = JSON.stringify(v ?? {});
+        return json.length > MAX_STR * 10 ? "{}" : json;
+    }
+    catch {
+        return "{}";
+    }
 }
-
 function normalizeGraphPayload(nodes, edges) {
-  const nodeIds = new Set(nodes.map((n) => n?.id).filter(Boolean).map(String));
-  const cleanEdges = edges.filter(
-    (e) => e?.id && e?.source && e?.target && nodeIds.has(String(e.source)) && nodeIds.has(String(e.target)),
-  );
-  const incomingByTarget = new Map();
-  for (const edge of cleanEdges) {
-    const target = String(edge.target);
-    if (incomingByTarget.has(target)) {
-      const err = new Error(`Node ${target} has multiple parent edges`);
-      err.code = "GRAPH_PARENT_CONFLICT";
-      err.status = 409;
-      throw err;
+    const nodeIds = new Set(nodes.map((n) => n?.id).filter(Boolean).map(String));
+    const cleanEdges = edges.filter((e) => e?.id && e?.source && e?.target && nodeIds.has(String(e.source)) && nodeIds.has(String(e.target)));
+    const incomingByTarget = new Map();
+    for (const edge of cleanEdges) {
+        const target = String(edge.target);
+        if (incomingByTarget.has(target)) {
+            const err = new Error(`Node ${target} has multiple parent edges`);
+            err.code = "GRAPH_PARENT_CONFLICT";
+            err.status = 409;
+            throw err;
+        }
+        incomingByTarget.set(target, edge);
     }
-    incomingByTarget.set(target, edge);
-  }
-
-  const nodeDataById = new Map();
-  for (const node of nodes) {
-    if (!node?.id) continue;
-    const data = node.data && typeof node.data === "object" && !Array.isArray(node.data)
-      ? { ...node.data }
-      : {};
-    nodeDataById.set(String(node.id), data);
-  }
-
-  const normalizedNodes = nodes.map((node) => {
-    if (!node?.id) return node;
-    const id = String(node.id);
-    const data = { ...(nodeDataById.get(id) ?? {}) };
-    const incoming = incomingByTarget.get(id);
-    if (!incoming) {
-      data.parentServerNodeId = null;
-    } else {
-      const parentData = nodeDataById.get(String(incoming.source)) ?? {};
-      data.parentServerNodeId = typeof parentData.serverNodeId === "string"
-        ? parentData.serverNodeId
-        : null;
+    const nodeDataById = new Map();
+    for (const node of nodes) {
+        if (!node?.id)
+            continue;
+        const data = node.data && typeof node.data === "object" && !Array.isArray(node.data)
+            ? { ...node.data }
+            : {};
+        nodeDataById.set(String(node.id), data);
     }
-    return { ...node, data };
-  });
-
-  return { nodes: normalizedNodes, edges: cleanEdges };
+    const normalizedNodes = nodes.map((node) => {
+        if (!node?.id)
+            return node;
+        const id = String(node.id);
+        const data = { ...(nodeDataById.get(id) ?? {}) };
+        const incoming = incomingByTarget.get(id);
+        if (!incoming) {
+            data.parentServerNodeId = null;
+        }
+        else {
+            const parentData = nodeDataById.get(String(incoming.source)) ?? {};
+            data.parentServerNodeId = typeof parentData.serverNodeId === "string"
+                ? parentData.serverNodeId
+                : null;
+        }
+        return { ...node, data };
+    });
+    return { nodes: normalizedNodes, edges: cleanEdges };
 }
-
 export function saveGraph(sessionId, { nodes = [], edges = [], expectedVersion = null }) {
-  const db = getDb();
-  const sessionExists = db
-    .prepare("SELECT 1 FROM sessions WHERE id = ?")
-    .get(sessionId);
-  if (!sessionExists) {
-    const err = new Error(`Session not found: ${sessionId}`);
-    err.code = "SESSION_NOT_FOUND";
-    err.status = 404;
-    throw err;
-  }
-
-  const versionRow = db
-    .prepare("SELECT graph_version AS graphVersion FROM sessions WHERE id = ?")
-    .get(sessionId);
-  const currentVersion = versionRow?.graphVersion ?? 0;
-  if (
-    typeof expectedVersion === "number" &&
-    Number.isFinite(expectedVersion) &&
-    expectedVersion !== currentVersion
-  ) {
-    const err = new Error(
-      `Graph version conflict for session ${sessionId}: expected ${expectedVersion}, got ${currentVersion}`,
-    );
-    err.code = "GRAPH_VERSION_CONFLICT";
-    err.status = 409;
-    err.currentVersion = currentVersion;
-    throw err;
-  }
-
-  const normalized = normalizeGraphPayload(nodes, edges);
-
-  const tx = db.transaction(() => {
-    db.prepare("DELETE FROM nodes WHERE session_id = ?").run(sessionId);
-    db.prepare("DELETE FROM edges WHERE session_id = ?").run(sessionId);
-
-    const insNode = db.prepare(
-      "INSERT INTO nodes (session_id, id, x, y, data) VALUES (?, ?, ?, ?, ?)",
-    );
-    for (const n of normalized.nodes) {
-      if (!n?.id) continue;
-      const x = Number(n.x ?? n.position?.x ?? 0);
-      const y = Number(n.y ?? n.position?.y ?? 0);
-      insNode.run(
-        sessionId,
-        cleanStr(String(n.id)),
-        Number.isFinite(x) ? x : 0,
-        Number.isFinite(y) ? y : 0,
-        cleanData(n.data),
-      );
+    const db = getDb();
+    const sessionExists = db
+        .prepare("SELECT 1 FROM sessions WHERE id = ?")
+        .get(sessionId);
+    if (!sessionExists) {
+        const err = new Error(`Session not found: ${sessionId}`);
+        err.code = "SESSION_NOT_FOUND";
+        err.status = 404;
+        throw err;
     }
-
-    const insEdge = db.prepare(
-      "INSERT INTO edges (session_id, id, source, target, data) VALUES (?, ?, ?, ?, ?)",
-    );
-    for (const e of normalized.edges) {
-      insEdge.run(
-        sessionId,
-        cleanStr(String(e.id)),
-        cleanStr(String(e.source)),
-        cleanStr(String(e.target)),
-        cleanData(e.data),
-      );
+    const versionRow = db
+        .prepare("SELECT graph_version AS graphVersion FROM sessions WHERE id = ?")
+        .get(sessionId);
+    const currentVersion = versionRow?.graphVersion ?? 0;
+    if (typeof expectedVersion === "number" &&
+        Number.isFinite(expectedVersion) &&
+        expectedVersion !== currentVersion) {
+        const err = new Error(`Graph version conflict for session ${sessionId}: expected ${expectedVersion}, got ${currentVersion}`);
+        err.code = "GRAPH_VERSION_CONFLICT";
+        err.status = 409;
+        err.currentVersion = currentVersion;
+        throw err;
     }
-
-    db.prepare("UPDATE sessions SET updated_at = ?, graph_version = graph_version + 1 WHERE id = ?").run(
-      now(),
-      sessionId,
-    );
-
-    return db
-      .prepare("SELECT graph_version AS graphVersion FROM sessions WHERE id = ?")
-      .get(sessionId).graphVersion;
-  });
-
-  const nextVersion = tx();
-  return { ok: true, graphVersion: nextVersion };
+    const normalized = normalizeGraphPayload(nodes, edges);
+    const tx = db.transaction(() => {
+        db.prepare("DELETE FROM nodes WHERE session_id = ?").run(sessionId);
+        db.prepare("DELETE FROM edges WHERE session_id = ?").run(sessionId);
+        const insNode = db.prepare("INSERT INTO nodes (session_id, id, x, y, data) VALUES (?, ?, ?, ?, ?)");
+        for (const n of normalized.nodes) {
+            if (!n?.id)
+                continue;
+            const x = Number(n.x ?? n.position?.x ?? 0);
+            const y = Number(n.y ?? n.position?.y ?? 0);
+            insNode.run(sessionId, cleanStr(String(n.id)), Number.isFinite(x) ? x : 0, Number.isFinite(y) ? y : 0, cleanData(n.data));
+        }
+        const insEdge = db.prepare("INSERT INTO edges (session_id, id, source, target, data) VALUES (?, ?, ?, ?, ?)");
+        for (const e of normalized.edges) {
+            insEdge.run(sessionId, cleanStr(String(e.id)), cleanStr(String(e.source)), cleanStr(String(e.target)), cleanData(e.data));
+        }
+        db.prepare("UPDATE sessions SET updated_at = ?, graph_version = graph_version + 1 WHERE id = ?").run(now(), sessionId);
+        return db
+            .prepare("SELECT graph_version AS graphVersion FROM sessions WHERE id = ?")
+            .get(sessionId).graphVersion;
+    });
+    const nextVersion = tx();
+    return { ok: true, graphVersion: nextVersion };
 }
-
 function safeParse(json) {
-  try {
-    return JSON.parse(json);
-  } catch {
-    return {};
-  }
+    try {
+        return JSON.parse(json);
+    }
+    catch {
+        return {};
+    }
 }
-
 export function ensureDefaultSession() {
-  const sessions = listSessions();
-  if (sessions.length > 0) return sessions[0];
-  return createSession({ title: "My first graph" });
+    const sessions = listSessions();
+    if (sessions.length > 0)
+        return sessions[0];
+    return createSession({ title: "My first graph" });
 }
-
 // ── Style sheet (0.10) ───────────────────────────────────────────────────
 export function getStyleSheet(sessionId) {
-  const db = getDb();
-  const row = db
-    .prepare(
-      "SELECT style_sheet AS styleSheet, style_sheet_enabled AS styleSheetEnabled FROM sessions WHERE id = ?",
-    )
-    .get(sessionId);
-  if (!row) return null;
-  let parsed = null;
-  if (row.styleSheet) {
-    try {
-      parsed = JSON.parse(row.styleSheet);
-    } catch {
-      parsed = null;
+    const db = getDb();
+    const row = db
+        .prepare("SELECT style_sheet AS styleSheet, style_sheet_enabled AS styleSheetEnabled FROM sessions WHERE id = ?")
+        .get(sessionId);
+    if (!row)
+        return null;
+    let parsed = null;
+    if (row.styleSheet) {
+        try {
+            parsed = JSON.parse(row.styleSheet);
+        }
+        catch {
+            parsed = null;
+        }
     }
-  }
-  return { styleSheet: parsed, enabled: !!row.styleSheetEnabled };
+    return { styleSheet: parsed, enabled: !!row.styleSheetEnabled };
 }
-
 export function setStyleSheet(sessionId, sheet) {
-  const db = getDb();
-  const json = sheet == null ? null : JSON.stringify(sheet);
-  const res = db
-    .prepare("UPDATE sessions SET style_sheet = ?, updated_at = ? WHERE id = ?")
-    .run(json, now(), sessionId);
-  return res.changes > 0;
+    const db = getDb();
+    const json = sheet == null ? null : JSON.stringify(sheet);
+    const res = db
+        .prepare("UPDATE sessions SET style_sheet = ?, updated_at = ? WHERE id = ?")
+        .run(json, now(), sessionId);
+    return res.changes > 0;
 }
-
 export function setStyleSheetEnabled(sessionId, enabled) {
-  const db = getDb();
-  const res = db
-    .prepare(
-      "UPDATE sessions SET style_sheet_enabled = ?, updated_at = ? WHERE id = ?",
-    )
-    .run(enabled ? 1 : 0, now(), sessionId);
-  return res.changes > 0;
+    const db = getDb();
+    const res = db
+        .prepare("UPDATE sessions SET style_sheet_enabled = ?, updated_at = ? WHERE id = ?")
+        .run(enabled ? 1 : 0, now(), sessionId);
+    return res.changes > 0;
 }