kingkont 0.7.39 → 0.7.40

package/lib/projectFs.js

@@ -0,0 +1,391 @@
+// Операции с папкой KingKont-проекта (FS-уровень).
+// Используется CLI (bin/kingkont.js). UI работает через File System Access
+// API, но формат файлов идентичный — поэтому эти операции совместимы.
+//
+// Формат проекта (см. skill/SKILL.md):
+//   <root>/
+//   ├── _characters/<name>/scene.json + clips/ images/ audio/ texts/
+//   ├── _locations/<name>/scene.json
+//   ├── <scene-name>/scene.json
+//   └── _deleted/
+
+'use strict';
+
+const fs = require('node:fs');
+const fsp = require('node:fs/promises');
+const path = require('node:path');
+const crypto = require('node:crypto');
+
+// =============================================================================
+// Board references — парсинг и резолв путей.
+// =============================================================================
+
+/**
+ * Распарсить board-reference в нормализованный вид.
+ *   '_characters/Anna'  →  { kind: 'character', name: 'Anna' }
+ *   'characters/Anna'   →  { kind: 'character', name: 'Anna' }
+ *   '_locations/Office' →  { kind: 'location',  name: 'Office' }
+ *   'locations/Office'  →  { kind: 'location',  name: 'Office' }
+ *   'Scene1'            →  { kind: 'scene',     name: 'Scene1' }
+ */
+function parseBoardRef(s) {
+  if (!s || typeof s !== 'string') throw new Error('board ref пустой');
+  const parts = s.split(/[\\/]/).filter(Boolean);
+  if (parts.length === 1) return { kind: 'scene', name: parts[0] };
+  const head = parts[0].replace(/^_/, '').toLowerCase();
+  const tail = parts.slice(1).join('/');
+  if (head === 'characters' || head === 'character') return { kind: 'character', name: tail };
+  if (head === 'locations'  || head === 'location')  return { kind: 'location',  name: tail };
+  // Иначе: считаем что это сцена с подпапкой (редко, но возможно).
+  return { kind: 'scene', name: parts.join('/') };
+}
+
+/** Абсолютный путь к папке доски. */
+function boardDir(root, ref) {
+  if (typeof ref === 'string') ref = parseBoardRef(ref);
+  switch (ref.kind) {
+    case 'character': return path.join(root, '_characters', ref.name);
+    case 'location':  return path.join(root, '_locations',  ref.name);
+    case 'scene':     return path.join(root, ref.name);
+  }
+  throw new Error(`unknown board kind: ${ref.kind}`);
+}
+
+/** Уникальный «ключ» (как в renderer'е): 'scene/Scene1' / 'character/Anna'. */
+function boardKey(ref) {
+  if (typeof ref === 'string') ref = parseBoardRef(ref);
+  return `${ref.kind}/${ref.name}`;
+}
+
+// =============================================================================
+// Listing.
+// =============================================================================
+
+/** Все доски проекта. */
+async function listBoards(root) {
+  const out = { scenes: [], characters: [], locations: [] };
+  const entries = await fsp.readdir(root, { withFileTypes: true }).catch(() => []);
+  for (const e of entries) {
+    if (!e.isDirectory()) continue;
+    if (e.name === '_characters') {
+      const sub = await fsp.readdir(path.join(root, '_characters'), { withFileTypes: true }).catch(() => []);
+      for (const c of sub) if (c.isDirectory() && (await hasSceneJson(path.join(root, '_characters', c.name)))) {
+        out.characters.push(c.name);
+      }
+    } else if (e.name === '_locations') {
+      const sub = await fsp.readdir(path.join(root, '_locations'), { withFileTypes: true }).catch(() => []);
+      for (const l of sub) if (l.isDirectory() && (await hasSceneJson(path.join(root, '_locations', l.name)))) {
+        out.locations.push(l.name);
+      }
+    } else if (e.name === '_deleted' || e.name.startsWith('.') || e.name === 'node_modules') {
+      continue;
+    } else if (await hasSceneJson(path.join(root, e.name))) {
+      out.scenes.push(e.name);
+    }
+  }
+  return out;
+}
+
+async function hasSceneJson(dir) {
+  try { return (await fsp.stat(path.join(dir, 'scene.json'))).isFile(); } catch { return false; }
+}
+
+// =============================================================================
+// Scene IO.
+// =============================================================================
+
+/**
+ * Прочитать scene.json и распаковать text-ноды (текст подгружается из .md).
+ * @returns {Promise<object>} { nodes, connections, view, character, location, timeline, history }
+ */
+async function loadScene(root, ref) {
+  const dir = boardDir(root, ref);
+  const sceneFile = path.join(dir, 'scene.json');
+  let raw;
+  try { raw = await fsp.readFile(sceneFile, 'utf-8'); }
+  catch (e) {
+    if (e.code === 'ENOENT') {
+      // Доски ещё нет — вернём пустую (для генерации новой).
+      return { nodes: [], connections: [], view: null, character: null, location: null, timeline: null, history: null };
+    }
+    throw e;
+  }
+  const data = JSON.parse(raw);
+  const scene = {
+    nodes: Array.isArray(data.nodes) ? data.nodes : [],
+    connections: Array.isArray(data.connections) ? data.connections : [],
+    view: data.view || null,
+    character: data.character || null,
+    location: data.location || null,
+    timeline: data.timeline || null,
+    history: data.history || null,
+  };
+  // Подгрузить .md для text-нод (как делает renderer в loadBoardMetadata).
+  for (const n of scene.nodes) {
+    if (n.type === 'text' && n.file) {
+      try { n.text = await fsp.readFile(path.join(dir, n.file), 'utf-8'); }
+      catch { n.text = ''; }
+    }
+  }
+  return scene;
+}
+
+/**
+ * Записать scene.json. Текст text-нод вынесет в .md (или обновит).
+ * Создаёт папку доски если её нет.
+ */
+async function saveScene(root, ref, scene) {
+  const dir = boardDir(root, ref);
+  await fsp.mkdir(dir, { recursive: true });
+  // Сохранить .md для text-нод.
+  for (const n of scene.nodes) {
+    if (n.type === 'text' && n.file && typeof n.text === 'string') {
+      const full = path.join(dir, n.file);
+      await fsp.mkdir(path.dirname(full), { recursive: true });
+      await fsp.writeFile(full, n.text);
+    }
+  }
+  const nodesForJson = scene.nodes.map(n => {
+    const out = { ...n };
+    delete out.text;
+    return out;
+  });
+  const payload = {
+    nodes: nodesForJson,
+    connections: scene.connections || [],
+    view: scene.view || null,
+    character: scene.character || null,
+    location: scene.location || null,
+    timeline: scene.timeline || null,
+    history: scene.history && (scene.history.past?.length || scene.history.future?.length) ? scene.history : null,
+  };
+  await fsp.writeFile(path.join(dir, 'scene.json'), JSON.stringify(payload, null, 2));
+}
+
+// =============================================================================
+// Node mutations.
+// =============================================================================
+
+const NODE_KINDS = new Set(['image', 'video', 'audio', 'text']);
+
+/**
+ * Добавить ноду в scene. Мутирует scene.nodes. Возвращает новую ноду.
+ * @param {object} scene
+ * @param {object} partial { type, name?, x?, y?, file?, text?, generated?, ... }
+ */
+function addNode(scene, partial) {
+  if (!partial || typeof partial !== 'object') throw new Error('node partial обязателен');
+  if (!NODE_KINDS.has(partial.type)) throw new Error(`unknown node type: ${partial.type}`);
+  // Авто-координаты: если не задано, ставим справа от последней.
+  let x = partial.x, y = partial.y;
+  if (typeof x !== 'number' || typeof y !== 'number') {
+    const last = scene.nodes[scene.nodes.length - 1];
+    x = last ? (last.x + 320) : 100;
+    y = last ? last.y : 100;
+  }
+  const node = {
+    id: crypto.randomUUID(),
+    type: partial.type,
+    x, y,
+    ...(partial.name ? { name: partial.name } : {}),
+    ...(partial.file ? { file: partial.file } : {}),
+    ...(partial.width ? { width: partial.width } : {}),
+    ...(partial.height ? { height: partial.height } : {}),
+    ...(partial.generated ? { generated: partial.generated } : {}),
+    ...(partial.status ? { status: partial.status } : {}),
+  };
+  if (partial.type === 'text' && typeof partial.text === 'string') {
+    node.text = partial.text;
+    // .md-файл создадим при saveScene если ещё нет.
+    if (!node.file) {
+      const slug = (partial.name || 'text').replace(/[^\wЀ-ӿ.-]+/g, '_').slice(0, 60) || 'text';
+      node.file = `texts/${slug}_${node.id.slice(0, 8)}.md`;
+    }
+  }
+  scene.nodes.push(node);
+  return node;
+}
+
+/** Найти ноду по id или имени. */
+function findNode(scene, idOrName) {
+  return scene.nodes.find(n => n.id === idOrName)
+      || scene.nodes.find(n => n.name === idOrName);
+}
+
+/** Добавить связь. Дубликаты игнорируются. */
+function addConnection(scene, fromId, toId, opts = {}) {
+  if (!scene.connections) scene.connections = [];
+  const exists = scene.connections.some(c => c.from === fromId && c.to === toId);
+  if (exists) return false;
+  const conn = { from: fromId, to: toId };
+  if (opts.toPort) conn.toPort = opts.toPort;
+  scene.connections.push(conn);
+  return true;
+}
+
+/** Удалить ноду + связанные коннекты + переместить файл в _deleted/. */
+async function removeNode(root, ref, nodeId) {
+  const scene = await loadScene(root, ref);
+  const idx = scene.nodes.findIndex(n => n.id === nodeId);
+  if (idx < 0) throw new Error(`node ${nodeId} не найдена`);
+  const node = scene.nodes[idx];
+  scene.nodes.splice(idx, 1);
+  scene.connections = (scene.connections || []).filter(c => c.from !== nodeId && c.to !== nodeId);
+  // Переместить файл в _deleted/.
+  if (node.file) {
+    try {
+      const src = path.join(boardDir(root, ref), node.file);
+      const trash = path.join(root, '_deleted', boardKey(ref).replace('/', '_'));
+      await fsp.mkdir(trash, { recursive: true });
+      const dst = path.join(trash, `${nodeId.slice(0, 8)}_${path.basename(node.file)}`);
+      await fsp.rename(src, dst).catch(() => {});
+    } catch {}
+  }
+  await saveScene(root, ref, scene);
+  return node;
+}
+
+// =============================================================================
+// Reference resolution для генерации.
+// =============================================================================
+
+/**
+ * Резолвит @-ссылки из промпта/refs в массив абсолютных путей к файлам и
+ * meta-инфу. Используется CLI перед загрузкой в провайдера.
+ *
+ * Поиск:
+ *   1) В текущей доске — по имени ноды.
+ *   2) Если имя совпадает с character/location — берём всю доску
+ *      (для character — `characterSheet`/первую image-ноду).
+ *
+ * @returns {Promise<Array<{ name, type, file, abs, char?, board? }>>}
+ */
+async function resolveRefs(root, ref, refNames) {
+  if (!Array.isArray(refNames) || !refNames.length) return [];
+  const out = [];
+  const scene = await loadScene(root, ref);
+  const boards = await listBoards(root);
+  const boardDirAbs = boardDir(root, ref);
+
+  for (const raw of refNames) {
+    const name = raw.replace(/^@/, '');
+    // 1) Локальная нода в текущей доске
+    const local = scene.nodes.find(n => n.name === name);
+    if (local && local.file) {
+      out.push({
+        name, type: local.type, file: local.file,
+        abs: path.join(boardDirAbs, local.file),
+      });
+      continue;
+    }
+    // 2) Character / location доска
+    if (boards.characters.includes(name)) {
+      const charScene = await loadScene(root, { kind: 'character', name });
+      const sheet = charScene.character?.characterSheet
+        || charScene.nodes.find(n => n.type === 'image')?.file;
+      if (sheet) {
+        out.push({
+          name, type: 'image', file: sheet,
+          abs: path.join(boardDir(root, { kind: 'character', name }), sheet),
+          char: name,
+        });
+      }
+      continue;
+    }
+    if (boards.locations.includes(name)) {
+      const locScene = await loadScene(root, { kind: 'location', name });
+      const first = locScene.nodes.find(n => n.type === 'image');
+      if (first?.file) {
+        out.push({
+          name, type: 'image', file: first.file,
+          abs: path.join(boardDir(root, { kind: 'location', name }), first.file),
+        });
+      }
+      continue;
+    }
+    // Не нашли — пропускаем (молча, но логируем).
+    console.warn(`[refs] не нашёл @${name} (ни ноды, ни character/location)`);
+  }
+  return out;
+}
+
+// =============================================================================
+// Helpers для CLI генерации (file naming).
+// =============================================================================
+
+/**
+ * Сгенерировать уникальное имя файла в субпапке доски.
+ * `images/img_001.jpg` → `images/img_002.jpg` если 001 занят.
+ */
+async function uniqueFilename(root, ref, subdir, baseName) {
+  const dir = path.join(boardDir(root, ref), subdir);
+  await fsp.mkdir(dir, { recursive: true });
+  const ext = path.extname(baseName);
+  const stem = path.basename(baseName, ext);
+  let candidate = baseName;
+  let i = 1;
+  while (true) {
+    try { await fsp.access(path.join(dir, candidate)); }
+    catch { return path.join(subdir, candidate); } // не существует — занимаем
+    i++;
+    candidate = `${stem}_${String(i).padStart(3, '0')}${ext}`;
+  }
+}
+
+/** Сохранить буфер в файл доски. Возвращает относительный путь. */
+async function saveBoardFile(root, ref, relPath, buffer) {
+  const abs = path.join(boardDir(root, ref), relPath);
+  await fsp.mkdir(path.dirname(abs), { recursive: true });
+  await fsp.writeFile(abs, buffer);
+  return relPath;
+}
+
+/** Прочитать файл из доски. */
+async function readBoardFile(root, ref, relPath) {
+  return await fsp.readFile(path.join(boardDir(root, ref), relPath));
+}
+
+// =============================================================================
+// Дефолтный субдир для kind.
+// =============================================================================
+function defaultSubdir(kind, subKind) {
+  if (kind === 'image') return 'images';
+  if (kind === 'video') return 'clips';
+  if (kind === 'audio') return 'audio';
+  if (kind === 'text')  return 'texts';
+  return 'misc';
+}
+
+function defaultExt(kind, subKind, mime) {
+  if (mime?.includes('jpeg') || mime?.includes('jpg')) return '.jpg';
+  if (mime?.includes('png')) return '.png';
+  if (mime?.includes('webp')) return '.webp';
+  if (mime?.includes('mpeg') && kind === 'audio') return '.mp3';
+  if (mime?.includes('wav')) return '.wav';
+  if (mime?.includes('mp4')) return '.mp4';
+  if (mime?.includes('webm')) return '.webm';
+  if (kind === 'image') return '.jpg';
+  if (kind === 'video') return '.mp4';
+  if (kind === 'audio') return '.mp3';
+  if (kind === 'text')  return '.md';
+  return '.bin';
+}
+
+module.exports = {
+  parseBoardRef,
+  boardDir,
+  boardKey,
+  listBoards,
+  loadScene,
+  saveScene,
+  addNode,
+  findNode,
+  addConnection,
+  removeNode,
+  resolveRefs,
+  uniqueFilename,
+  saveBoardFile,
+  readBoardFile,
+  defaultSubdir,
+  defaultExt,
+};