ima2-gen 1.1.21 → 1.1.23

package/lib/apiCachePolicy.ts

@@ -1,11 +0,0 @@
-import type { Express, NextFunction, Request, Response } from "express";
-
-export function setApiNoStoreHeaders(_req: Request, res: Response, next: NextFunction) {
-  res.setHeader("Cache-Control", "no-store, max-age=0");
-  next();
-}
-
-export function configureApiCachePolicy(app: Express) {
-  app.set("etag", false);
-  app.use("/api", setApiNoStoreHeaders);
-}

package/lib/assetLifecycle.ts

@@ -1,146 +0,0 @@
-import { getDb } from "./db.js";
-import { mkdir, rename, unlink, access } from "fs/promises";
-import { resolve, sep } from "path";
-import { moveToSystemTrash } from "./systemTrash.js";
-import { config } from "../config.js";
-
-function resolveInGenerated(rootDir: string, relPath: string): string {
-  void rootDir;
-  if (typeof relPath !== "string" || relPath.length === 0) {
-    const err: any = new Error("filename required");
-    err.status = 400;
-    err.code = "INVALID_FILENAME";
-    throw err;
-  }
-  if (relPath.includes("\0")) {
-    const err: any = new Error("invalid filename");
-    err.status = 400;
-    err.code = "INVALID_FILENAME";
-    throw err;
-  }
-  const baseDir = resolve(config.storage.generatedDir);
-  const target = resolve(baseDir, relPath);
-  if (target !== baseDir && !target.startsWith(baseDir + sep)) {
-    const err: any = new Error("filename escapes generated/");
-    err.status = 400;
-    err.code = "INVALID_FILENAME";
-    throw err;
-  }
-  return target;
-}
-
-function nodesReferencingFilename(filename: string): Array<{ sessionId: string; id: string; data: string }> {
-  // The client stores imageUrl as `/generated/<encoded filename>` in node data JSON.
-  // We scan all sessions' nodes for substring match on the decoded and encoded forms.
-  const db = getDb();
-  const encoded = encodeURIComponent(filename);
-  const rows = db
-    .prepare("SELECT session_id AS sessionId, id, data FROM nodes WHERE data LIKE ? OR data LIKE ?")
-    .all(`%${filename}%`, `%${encoded}%`) as Array<{ sessionId: string; id: string; data: string }>;
-  return rows;
-}
-
-function markNodesAssetMissing(filename: string) {
-  const db = getDb();
-  const rows = nodesReferencingFilename(filename);
-  if (rows.length === 0) return { sessionsTouched: 0, nodesTouched: 0 };
-  const touchedSessions = new Set<string>();
-  const update = db.prepare("UPDATE nodes SET data = ? WHERE session_id = ? AND id = ?");
-  const bumpSession = db.prepare("UPDATE sessions SET graph_version = graph_version + 1, updated_at = ? WHERE id = ?");
-  const tx = db.transaction(() => {
-    for (const r of rows) {
-      let data;
-      try { data = JSON.parse(r.data); } catch { data = {}; }
-      const imgRef = data?.imageUrl || "";
-      if (imgRef.includes(filename) || imgRef.includes(encodeURIComponent(filename))) {
-        data.imageUrl = null;
-        data.status = "asset-missing";
-        update.run(JSON.stringify(data), r.sessionId, r.id);
-        touchedSessions.add(r.sessionId);
-      }
-    }
-    const t = Date.now();
-    for (const sid of touchedSessions) bumpSession.run(t, sid);
-  });
-  tx();
-  return { sessionsTouched: touchedSessions.size, nodesTouched: rows.length };
-}
-
-export async function trashAsset(rootDir: string, filename: string) {
-  const src = resolveInGenerated(rootDir, filename);
-  try {
-    await access(src);
-  } catch {
-    const err: any = new Error("Asset not found");
-    err.status = 404;
-    err.code = "ASSET_NOT_FOUND";
-    throw err;
-  }
-
-  const sidecar = `${src}.json`;
-  const paths = [src];
-  try {
-    await access(sidecar);
-    paths.push(sidecar);
-  } catch {}
-
-  let trashMethod: "system" | "internal" = "system";
-  try {
-    await moveToSystemTrash(paths);
-  } catch {
-    trashMethod = "internal";
-    const trashDir = resolve(config.storage.trashDir);
-    await mkdir(trashDir, { recursive: true });
-    const trashId = `${Date.now()}_${filename}`;
-    for (const p of paths) {
-      const dest = resolve(trashDir, p.endsWith(".json") ? `${trashId}.json` : trashId);
-      await rename(p, dest);
-    }
-  }
-
-  const summary = markNodesAssetMissing(filename);
-  return {
-    ok: true,
-    filename,
-    trash: trashMethod,
-    undoableInApp: false,
-    sessionsTouched: summary.sessionsTouched,
-    nodesTouched: summary.nodesTouched,
-  };
-}
-
-export async function deleteAssetPermanent(rootDir: string, filename: string) {
-  const src = resolveInGenerated(rootDir, filename);
-  try {
-    await access(src);
-  } catch {
-    const err: any = new Error("Asset not found");
-    err.status = 404;
-    err.code = "ASSET_NOT_FOUND";
-    throw err;
-  }
-  await unlink(src);
-  await unlink(src + ".json").catch(() => {});
-  const summary = markNodesAssetMissing(filename);
-  return {
-    ok: true,
-    filename,
-    sessionsTouched: summary.sessionsTouched,
-    nodesTouched: summary.nodesTouched,
-  };
-}
-
-export async function restoreAsset(rootDir: string, trashId: string, originalFilename: string) {
-  void rootDir;
-  const trashDir = resolve(config.storage.trashDir);
-  const src = resolve(trashDir, trashId);
-  if (!src.startsWith(trashDir + sep) && src !== trashDir) {
-    const err: any = new Error("invalid trashId");
-    err.status = 400;
-    throw err;
-  }
-  const dst = resolveInGenerated(rootDir, originalFilename);
-  await rename(src, dst);
-  await rename(src + ".json", dst + ".json").catch(() => {});
-  return { ok: true };
-}

package/lib/canvasVersionStore.ts

@@ -1,223 +0,0 @@
-import { mkdir, writeFile, access, readFile } from "fs/promises";
-import { constants } from "fs";
-import { basename, join, normalize, parse } from "path";
-import { randomBytes } from "crypto";
-import { embedImageMetadataBestEffort } from "./imageMetadataStore.js";
-import { invalidateHistoryIndex } from "./historyIndex.js";
-import type { RuntimeContext } from "./runtimeContext.js";
-
-interface CanvasMeta {
-  kind: string;
-  provider: string;
-  format: string;
-  prompt: string | null | undefined;
-  userPrompt: string | null | undefined;
-  promptMode: string;
-  createdAt: number;
-  canvasMergedAt: number;
-  canvasVersion: boolean;
-  canvasSourceFilename: string | null;
-  canvasEditableFilename: string;
-  size?: string | null;
-  quality?: string | null;
-  model?: string | null;
-  moderation?: string | null;
-  revisedPrompt?: string | null;
-}
-
-interface StoredGeneratedMeta {
-  prompt?: string | null;
-  userPrompt?: string | null;
-  promptMode?: string | null;
-  provider?: string | null;
-  quality?: string | null;
-  size?: string | null;
-  model?: string | null;
-  moderation?: string | null;
-  createdAt?: number;
-  canvasMergedAt?: number;
-  canvasSourceFilename?: string | null;
-}
-
-interface CanvasInput {
-  buffer: unknown;
-  sourceFilename?: string | null;
-  prompt?: string | null;
-}
-
-const PNG_SIGNATURE = "89504e470d0a1a0a";
-
-function assertPngBuffer(buffer: unknown): asserts buffer is Buffer {
-  if (!Buffer.isBuffer(buffer) || buffer.length === 0) {
-    const err: any = new Error("PNG body is required");
-    err.status = 400;
-    err.code = "EMPTY_CANVAS_VERSION";
-    throw err;
-  }
-  if (buffer.subarray(0, 8).toString("hex") !== PNG_SIGNATURE) {
-    const err: any = new Error("Canvas version body must be a PNG image");
-    err.status = 400;
-    err.code = "CANVAS_VERSION_NOT_PNG";
-    throw err;
-  }
-}
-
-function assertSafeFilename(filename: string) {
-  if (
-    typeof filename !== "string" ||
-    filename.length === 0 ||
-    filename !== basename(filename) ||
-    filename.includes("..") ||
-    !/^canvas-[a-zA-Z0-9._-]+\.png$/.test(filename)
-  ) {
-    const err: any = new Error("Invalid canvas version filename");
-    err.status = 400;
-    err.code = "INVALID_CANVAS_VERSION_FILENAME";
-    throw err;
-  }
-}
-
-function safeSourceBase(sourceFilename: string | null | undefined) {
-  const parsed = parse(basename(String(sourceFilename || "image")));
-  return parsed.name.replace(/[^a-zA-Z0-9_-]+/g, "-").replace(/^-+|-+$/g, "").slice(0, 80) || "image";
-}
-
-function ensureInsideGeneratedDir(generatedDir: string, filename: string) {
-  const full = normalize(join(generatedDir, filename));
-  const root = normalize(generatedDir);
-  if (!full.startsWith(root)) {
-    const err: any = new Error("Canvas version path escapes generated directory");
-    err.status = 400;
-    err.code = "CANVAS_VERSION_PATH_ESCAPE";
-    throw err;
-  }
-  return full;
-}
-
-function makeCanvasFilename(sourceFilename: string | null | undefined) {
-  const stamp = new Date().toISOString().slice(0, 19).replace(/[-:T]/g, "");
-  const rand = randomBytes(3).toString("hex");
-  return `canvas-${safeSourceBase(sourceFilename)}-${stamp}-${rand}.png`;
-}
-
-async function writeCanvasPng(ctx: RuntimeContext, filename: string, buffer: Buffer, meta: CanvasMeta) {
-  await mkdir(ctx.config.storage.generatedDir, { recursive: true });
-  const full = ensureInsideGeneratedDir(ctx.config.storage.generatedDir, filename);
-  const embedded = await embedImageMetadataBestEffort(buffer, "png", meta, {
-    version: ctx.packageVersion,
-  });
-  await writeFile(full, embedded.buffer);
-  await writeFile(`${full}.json`, JSON.stringify(meta)).catch(() => {});
-  invalidateHistoryIndex();
-}
-
-async function readGeneratedMetadata(ctx: RuntimeContext, filename: string | null | undefined): Promise<StoredGeneratedMeta | null> {
-  if (!filename) return null;
-  try {
-    const full = ensureInsideGeneratedDir(ctx.config.storage.generatedDir, basename(filename));
-    return JSON.parse(await readFile(`${full}.json`, "utf8")) as StoredGeneratedMeta;
-  } catch {
-    return null;
-  }
-}
-
-function firstString(...values: Array<string | null | undefined>) {
-  return values.find((value) => typeof value === "string" && value.trim().length > 0) ?? null;
-}
-
-function toGenerateItem(filename: string, meta: CanvasMeta) {
-  const url = `/generated/${encodeURIComponent(filename)}`;
-  return {
-    image: url,
-    url,
-    thumb: url,
-    filename,
-    prompt: meta.prompt || undefined,
-    userPrompt: meta.userPrompt || meta.prompt || null,
-    revisedPrompt: null,
-    promptMode: meta.promptMode || "direct",
-    provider: meta.provider || "canvas",
-    quality: meta.quality || null,
-    size: meta.size || null,
-    format: "png",
-    moderation: meta.moderation || null,
-    model: meta.model || null,
-    usage: null,
-    createdAt: meta.createdAt,
-    kind: "edit",
-    canvasMergedAt: meta.canvasMergedAt,
-    canvasVersion: true,
-    canvasSourceFilename: meta.canvasSourceFilename || null,
-    canvasEditableFilename: filename,
-  };
-}
-
-export async function createCanvasVersion(ctx: RuntimeContext, input: CanvasInput) {
-  assertPngBuffer(input.buffer);
-  const sourceFilename = basename(String(input.sourceFilename || ""));
-  if (!sourceFilename) {
-    const err: any = new Error("sourceFilename is required");
-    err.status = 400;
-    err.code = "CANVAS_SOURCE_REQUIRED";
-    throw err;
-  }
-  const filename = makeCanvasFilename(sourceFilename);
-  const now = Date.now();
-  const sourceMeta = await readGeneratedMetadata(ctx, sourceFilename);
-  const prompt = firstString(input.prompt, sourceMeta?.userPrompt, sourceMeta?.prompt);
-  const meta = {
-    kind: "edit",
-    provider: "canvas",
-    format: "png",
-    prompt,
-    userPrompt: prompt,
-    promptMode: sourceMeta?.promptMode || "direct",
-    createdAt: now,
-    canvasMergedAt: now,
-    canvasVersion: true,
-    canvasSourceFilename: sourceFilename,
-    canvasEditableFilename: filename,
-  };
-  await writeCanvasPng(ctx, filename, input.buffer, meta);
-  return toGenerateItem(filename, meta);
-}
-
-export async function updateCanvasVersion(ctx: RuntimeContext, filename: string, input: CanvasInput) {
-  assertSafeFilename(filename);
-  assertPngBuffer(input.buffer);
-  const full = ensureInsideGeneratedDir(ctx.config.storage.generatedDir, filename);
-  await access(full, constants.F_OK).catch(() => {
-    const err: any = new Error("Canvas version not found");
-    err.status = 404;
-    err.code = "CANVAS_VERSION_NOT_FOUND";
-    throw err;
-  });
-  const now = Date.now();
-  const sourceFilename = typeof input.sourceFilename === "string"
-    ? basename(input.sourceFilename)
-    : null;
-  const sourceMeta = await readGeneratedMetadata(ctx, sourceFilename);
-  const previousMeta = await readGeneratedMetadata(ctx, filename);
-  const prompt = firstString(
-    input.prompt,
-    sourceMeta?.userPrompt,
-    sourceMeta?.prompt,
-    previousMeta?.userPrompt,
-    previousMeta?.prompt,
-  );
-  const meta = {
-    kind: "edit",
-    provider: "canvas",
-    format: "png",
-    prompt,
-    userPrompt: prompt,
-    promptMode: sourceMeta?.promptMode || previousMeta?.promptMode || "direct",
-    createdAt: now,
-    canvasMergedAt: now,
-    canvasVersion: true,
-    canvasSourceFilename: sourceFilename,
-    canvasEditableFilename: filename,
-  };
-  await writeCanvasPng(ctx, filename, input.buffer, meta);
-  return toGenerateItem(filename, meta);
-}

package/lib/capabilities.ts

@@ -1,126 +0,0 @@
-import { config as runtimeConfigDefault } from "../config.js";
-import { KEY_TO_ENV, WRITABLE_CONFIG_KEYS } from "./configKeys.js";
-import { DEFAULT_IMAGE_QUALITY, VALID_IMAGE_QUALITIES } from "./oauthNormalize.js";
-import type { AppConfig } from "./runtimeContext.js";
-
-type CapabilitySource = "local" | "server";
-
-const MAX_GENERATED_IMAGES = 8;
-const VALID_MODES = ["auto", "direct"] as const;
-const VALID_PROVIDERS = ["auto", "oauth", "api", "grok"] as const;
-const AGENT_COMMANDS = [
-  "skill",
-  "capabilities",
-  "defaults",
-  "gen",
-  "video",
-  "edit",
-  "multimode",
-  "node generate",
-  "inflight ls",
-  "providers",
-  "oauth status",
-  "grok status",
-  "prompt build",
-];
-
-function toArray<T>(value: Iterable<T> | T[] | undefined): T[] {
-  if (!value) return [];
-  if (Array.isArray(value)) return [...value];
-  return Array.from(value);
-}
-
-export function buildIma2Capabilities({
-  appConfig = runtimeConfigDefault,
-  packageVersion,
-  source,
-  server = null,
-}: {
-  appConfig?: AppConfig;
-  packageVersion: string;
-  source: CapabilitySource;
-  server?: string | null;
-}) {
-  return {
-    ok: true,
-    name: "ima2",
-    source,
-    server,
-    version: packageVersion,
-    commands: AGENT_COMMANDS,
-    defaults: {
-      oauth: {
-        model: appConfig.imageModels.default,
-        reasoningEffort: appConfig.imageModels.reasoningEffort,
-      },
-      api: {
-        model: appConfig.apiProvider.defaultImageModel,
-        reasoningEffort: appConfig.apiProvider.defaultReasoningEffort,
-        size: appConfig.apiProvider.defaultSize,
-        webSearchEnabled: appConfig.apiProvider.allowWebSearch,
-      },
-      grok: {
-        model: appConfig.grokProvider.defaultImageModel,
-        plannerModel: appConfig.grokProvider.plannerModel,
-      },
-    },
-    valid: {
-      imageModels: {
-        supported: toArray(appConfig.imageModels.valid),
-        unsupported: toArray(appConfig.imageModels.unsupported),
-        grokSupported: ["grok-imagine-image", "grok-imagine-image-quality"],
-      },
-      videoModels: {
-        supported: ["grok-imagine-video", "grok-imagine-video-1.5-preview"],
-        resolutions: ["480p", "720p"],
-        aspectRatios: ["1:1", "16:9", "9:16", "4:3", "3:4", "3:2", "2:3", "auto"],
-        durationRange: [1, 15],
-        maxReferences: 7,
-      },
-      reasoningEfforts: toArray(appConfig.imageModels.validReasoningEfforts),
-      quality: toArray(VALID_IMAGE_QUALITIES),
-      moderation: toArray(appConfig.oauth.validModeration),
-      modes: [...VALID_MODES],
-      providers: [...VALID_PROVIDERS],
-    },
-    configKeys: {
-      writable: toArray(WRITABLE_CONFIG_KEYS),
-      envOverrides: { ...KEY_TO_ENV },
-    },
-    defaultsMeta: {
-      quality: DEFAULT_IMAGE_QUALITY,
-    },
-    limits: {
-      maxRefCount: appConfig.limits.maxRefCount,
-      maxGeneratedImages: MAX_GENERATED_IMAGES,
-      maxParallel: {
-        value: appConfig.limits.maxParallel,
-        enforced: false,
-        note: "advisory client-side queue guidance only; server-side semaphore is not enforced",
-      },
-    },
-    promptBuilder: {
-      available: true,
-      route: "/api/prompt-builder/chat",
-      cliCommand: "ima2 prompt build",
-      structuredOutput: ["intentSummary", "finalPrompt.ko", "finalPrompt.en", "notes"],
-      uiOnly: false,
-    },
-    agentMode: {
-      available: true,
-      route: "/api/agent/sessions",
-      allowedTools: ["ima2.get_image_context", "ima2.web_search", "ima2.generate_image"],
-      finalArtifact: "image",
-      uiOnly: true,
-      cliCommand: null,
-    },
-    guidance: {
-      highQuality: "Use --quality high for requests where output fidelity matters.",
-      parallelGeneration: "Run multiple ima2 gen commands as separate queued jobs; no --parallel flag is required.",
-      i2i: "Use --ref for reference generation, or ima2 edit <file> --prompt \"<text>\" for image edits.",
-      defaults: "Use ima2 defaults set model/reasoning for persistent defaults; request flags remain per-call overrides.",
-      promptBuilder: "Use ima2 prompt build --message \"...\" to refine prompt intent. Use ima2 gen / ima2 multimode to generate images. Workspace profile settings are UI-only.",
-      video: "Use ima2 video \"<prompt>\" to generate video. Prompts must describe visual flow, motion, sound/no-music, dialogue/no-dialogue, and ending frame. Use ima2 video continue \"<prompt>\" --video <generated.mp4> for branch-local last-frame continuation; --topic is legacy best-effort series context.",
-    },
-  };
-}