kingkont 0.7.39 → 0.7.41

package/lib/cli.js

@@ -0,0 +1,504 @@
+// CLI команды для KingKont. Используется из bin/kingkont.js.
+// Все команды работают с выключенным GUI — правят scene.json напрямую через
+// FS, генерации запускают через lib/providers (тот же роутинг что и server.js).
+
+'use strict';
+
+const fs = require('node:fs');
+const fsp = require('node:fs/promises');
+const path = require('node:path');
+
+const settingsLib = require('./settings');
+const providers = require('./providers');
+const fsLib = require('./projectFs');
+
+// =============================================================================
+// CLI-arg парсер (минимальный, без зависимостей).
+// =============================================================================
+
+/**
+ * Парсит хвост argv в { positional: [...], flags: {...} }.
+ *   --kind=image    → flags.kind = 'image'
+ *   --kind image    → flags.kind = 'image'
+ *   --json          → flags.json = true
+ *   --refs=a,b,c    → flags.refs = ['a', 'b', 'c']
+ *   foo bar         → positional = ['foo', 'bar']
+ */
+function parseArgs(argv) {
+  const positional = [];
+  const flags = {};
+  for (let i = 0; i < argv.length; i++) {
+    const a = argv[i];
+    if (a.startsWith('--')) {
+      const eq = a.indexOf('=');
+      let key, val;
+      if (eq >= 0) {
+        key = a.slice(2, eq);
+        val = a.slice(eq + 1);
+      } else {
+        key = a.slice(2);
+        const next = argv[i + 1];
+        if (next !== undefined && !next.startsWith('--')) { val = next; i++; }
+        else val = true;
+      }
+      // Авто-распак списков по запятой для известных полей.
+      if (typeof val === 'string' && (key === 'refs' || key === 'imageInputs' || key === 'videoInputs')) {
+        flags[key] = val.split(',').map(s => s.trim()).filter(Boolean);
+      } else {
+        flags[key] = val;
+      }
+    } else {
+      positional.push(a);
+    }
+  }
+  return { positional, flags };
+}
+
+function parseProjectArg(positional) {
+  if (!positional.length) throw new Error('Не указан путь к проекту');
+  const p = path.resolve(positional[0]);
+  if (!fs.existsSync(p)) throw new Error(`Не существует: ${p}`);
+  if (!fs.statSync(p).isDirectory()) throw new Error(`Не папка: ${p}`);
+  return p;
+}
+
+function asJson(obj) { return JSON.stringify(obj, null, 2); }
+
+// Определяем формат вывода: --json → строгий JSON; иначе человеко-читаемый.
+function out(flags, machine, human) {
+  if (flags.json) console.log(asJson(machine));
+  else console.log(human);
+}
+
+// =============================================================================
+// Команды.
+// =============================================================================
+
+/** kingkont open <project> — печатает JSON со структурой проекта. */
+async function cmdOpen({ positional, flags }) {
+  const root = parseProjectArg(positional);
+  const boards = await fsLib.listBoards(root);
+  const all = [];
+  for (const name of boards.scenes)     all.push({ ref: name,                  kind: 'scene',     name });
+  for (const name of boards.characters) all.push({ ref: `_characters/${name}`, kind: 'character', name });
+  for (const name of boards.locations)  all.push({ ref: `_locations/${name}`,  kind: 'location',  name });
+  // Для каждой — короткая сводка.
+  const summary = [];
+  for (const b of all) {
+    const scene = await fsLib.loadScene(root, b);
+    summary.push({
+      ref: b.ref, kind: b.kind, name: b.name,
+      nodesCount: scene.nodes.length,
+      connectionsCount: (scene.connections || []).length,
+    });
+  }
+  const result = { root, boards: summary };
+  if (flags.json) console.log(asJson(result));
+  else {
+    console.log(`KingKont проект: ${root}`);
+    console.log(`Досок: ${summary.length}`);
+    for (const b of summary) {
+      console.log(`  ${b.kind.padEnd(9)} ${b.ref.padEnd(40)} (${b.nodesCount} нод, ${b.connectionsCount} связей)`);
+    }
+  }
+}
+
+/** kingkont list <project> — перечисление досок. */
+async function cmdList({ positional, flags }) {
+  const root = parseProjectArg(positional);
+  const boards = await fsLib.listBoards(root);
+  if (flags.json) return console.log(asJson(boards));
+  console.log('Сцены:');
+  for (const n of boards.scenes) console.log(`  ${n}`);
+  console.log('Персонажи:');
+  for (const n of boards.characters) console.log(`  ${n}`);
+  console.log('Локации:');
+  for (const n of boards.locations) console.log(`  ${n}`);
+}
+
+/** kingkont board <project> <board> — JSON одной доски. */
+async function cmdBoard({ positional, flags }) {
+  const root = parseProjectArg(positional);
+  if (positional.length < 2) throw new Error('Нужно: kingkont board <project> <board>');
+  const ref = fsLib.parseBoardRef(positional[1]);
+  const scene = await fsLib.loadScene(root, ref);
+  if (flags.json) return console.log(asJson(scene));
+  console.log(`Доска ${fsLib.boardKey(ref)}`);
+  console.log(`  ${scene.nodes.length} нод, ${(scene.connections || []).length} связей`);
+  for (const n of scene.nodes) {
+    const tag = n.name ? `@${n.name}` : '';
+    console.log(`  - ${n.id.slice(0, 8)} ${n.type.padEnd(6)} ${tag.padEnd(20)} ${n.file || ''}`);
+  }
+}
+
+/** kingkont add-node <project> <board> --kind=... [--name=...] [--file=...] [--text="..."] [--x=...] [--y=...] */
+async function cmdAddNode({ positional, flags }) {
+  const root = parseProjectArg(positional);
+  if (positional.length < 2) throw new Error('Нужно: kingkont add-node <project> <board> --kind=...');
+  const ref = fsLib.parseBoardRef(positional[1]);
+  const partial = {
+    type: flags.kind,
+    name: flags.name,
+    file: flags.file,
+    text: flags.text,
+    x: flags.x != null ? Number(flags.x) : undefined,
+    y: flags.y != null ? Number(flags.y) : undefined,
+  };
+  const scene = await fsLib.loadScene(root, ref);
+  const node = fsLib.addNode(scene, partial);
+  await fsLib.saveScene(root, ref, scene);
+  if (flags.json) console.log(asJson(node));
+  else console.log(`✓ создана нода ${node.id} (${node.type})`);
+}
+
+/** kingkont gen <project> <board> --kind=... --prompt=... [--model=...] [--refs=@a,@b]
+ *  Совмещает add-node + generate. Самая частая команда. */
+async function cmdGen({ positional, flags }) {
+  const root = parseProjectArg(positional);
+  if (positional.length < 2) throw new Error('Нужно: kingkont gen <project> <board> --kind=... --prompt="..."');
+  const ref = fsLib.parseBoardRef(positional[1]);
+  const kind = flags.kind || 'image';
+  const prompt = flags.prompt;
+  if (!prompt && kind !== 'video') throw new Error('--prompt обязателен');
+
+  // 1) Грузим сетки.
+  const settings = await loadSettingsForCli();
+
+  // 2) Резолвим refs → загружаем их в storage.
+  const refNames = Array.isArray(flags.refs) ? flags.refs : [];
+  const resolved = await fsLib.resolveRefs(root, ref, refNames);
+  const imageInputs = [];
+  const videoInputs = [];
+  for (const r of resolved) {
+    const buf = await fsp.readFile(r.abs);
+    const up = await providers.uploadFile({
+      buffer: buf,
+      filename: path.basename(r.file),
+      mime: guessMime(r.file),
+      settings,
+    });
+    if (r.type === 'image') imageInputs.push(up.url);
+    else if (r.type === 'video') videoInputs.push(up.url);
+    console.error(`[ref] @${r.name} → ${up.url}`);
+  }
+
+  // 3) Создаём ноду в статусе 'generating'.
+  const scene = await fsLib.loadScene(root, ref);
+  const node = fsLib.addNode(scene, {
+    type: kind,
+    name: flags.name,
+    x: flags.x != null ? Number(flags.x) : undefined,
+    y: flags.y != null ? Number(flags.y) : undefined,
+    status: 'generating',
+    generated: {
+      kind,
+      prompt,
+      model: flags.model,
+      modelKey: flags.model,
+      refs: resolved.map(r => ({ name: r.name, type: r.type, file: r.file })),
+      state: 'submitting',
+    },
+  });
+  await fsLib.saveScene(root, ref, scene);
+  console.error(`[node] создана ${node.id} (status=generating)`);
+
+  // 4) Запускаем генерацию через провайдер.
+  if (kind === 'text') return await runTextGeneration(root, ref, node, { prompt, model: flags.model, settings, flags });
+  if (kind === 'audio') {
+    const subKind = flags['sub-kind'] || flags.subKind || (flags.tts ? 'tts' : null);
+    if (subKind === 'sfx')   return await runSfxGeneration(root, ref, node, { text: prompt, durationSeconds: flags.duration, settings, flags });
+    if (subKind === 'music') return await runMusicGeneration(root, ref, node, { prompt, durationMs: flags['duration-ms'] || flags.durationMs, settings, flags });
+    return await runTtsGeneration(root, ref, node, { text: prompt, voice: flags.voice, ttsModel: flags['tts-model'] || flags.ttsModel, settings, flags });
+  }
+  // image | video
+  return await runMediaGeneration(root, ref, node, {
+    kind, prompt, modelKey: flags.model, imageInputs, videoInputs,
+    aspectRatio: flags['aspect-ratio'] || flags.aspectRatio,
+    resolution: flags.resolution,
+    duration: flags.duration,
+    firstFrame: flags['first-frame'] || flags.firstFrame,
+    lastFrame: flags['last-frame'] || flags.lastFrame,
+    settings, flags,
+  });
+}
+
+/** kingkont generate <project> <board> <nodeId> — для существующей draft-ноды.
+ *  Берёт generated.prompt/model и перезапускает генерацию. */
+async function cmdGenerate({ positional, flags }) {
+  const root = parseProjectArg(positional);
+  if (positional.length < 3) throw new Error('Нужно: kingkont generate <project> <board> <nodeId>');
+  const ref = fsLib.parseBoardRef(positional[1]);
+  const scene = await fsLib.loadScene(root, ref);
+  const node = fsLib.findNode(scene, positional[2]);
+  if (!node) throw new Error(`Нода ${positional[2]} не найдена`);
+  if (!node.generated?.prompt) throw new Error(`У ноды нет generated.prompt — добавь сначала через add-node или сразу gen`);
+
+  const settings = await loadSettingsForCli();
+  const args = {
+    kind: node.type, prompt: node.generated.prompt, modelKey: node.generated.model || node.generated.modelKey,
+    imageInputs: [], videoInputs: [], settings, flags,
+  };
+  // Если у ноды есть refs — резолвим их сейчас.
+  if (Array.isArray(node.generated.refs)) {
+    for (const r of node.generated.refs) {
+      const refRes = await fsLib.resolveRefs(root, ref, [r.name]);
+      if (!refRes[0]) continue;
+      const buf = await fsp.readFile(refRes[0].abs);
+      const up = await providers.uploadFile({ buffer: buf, filename: path.basename(refRes[0].file), mime: guessMime(refRes[0].file), settings });
+      if (r.type === 'image') args.imageInputs.push(up.url);
+      else if (r.type === 'video') args.videoInputs.push(up.url);
+    }
+  }
+  if (node.type === 'image' || node.type === 'video') return await runMediaGeneration(root, ref, node, args);
+  if (node.type === 'text') return await runTextGeneration(root, ref, node, { prompt: node.generated.prompt, model: node.generated.model, settings, flags });
+  if (node.type === 'audio') return await runTtsGeneration(root, ref, node, { text: node.generated.prompt, voice: node.generated.voiceId, ttsModel: node.generated.model, settings, flags });
+  throw new Error(`unknown node type: ${node.type}`);
+}
+
+/** kingkont status <project> <board> <nodeId> — для in-flight задачи. */
+async function cmdStatus({ positional, flags }) {
+  const root = parseProjectArg(positional);
+  if (positional.length < 3) throw new Error('Нужно: kingkont status <project> <board> <nodeId>');
+  const ref = fsLib.parseBoardRef(positional[1]);
+  const scene = await fsLib.loadScene(root, ref);
+  const node = fsLib.findNode(scene, positional[2]);
+  if (!node) throw new Error(`Нода не найдена`);
+  if (!node.generated?.taskId) {
+    if (flags.json) return console.log(asJson({ status: node.status, file: node.file, error: node.error }));
+    console.log(`status=${node.status || 'ready'} file=${node.file || '—'} ${node.error ? 'error: ' + node.error : ''}`);
+    return;
+  }
+  const settings = await loadSettingsForCli();
+  const r = await providers.pollGeneration(node.generated.taskId, settings);
+  if (flags.json) return console.log(asJson(r));
+  console.log(`status=${r.status} provider=${r.provider} ${r.url ? 'url=' + r.url : ''} ${r.error || ''}`);
+}
+
+/** kingkont connect <project> <board> <fromId> <toId> [--port=...] */
+async function cmdConnect({ positional, flags }) {
+  const root = parseProjectArg(positional);
+  if (positional.length < 4) throw new Error('Нужно: kingkont connect <project> <board> <fromId> <toId>');
+  const ref = fsLib.parseBoardRef(positional[1]);
+  const scene = await fsLib.loadScene(root, ref);
+  const fromId = (fsLib.findNode(scene, positional[2]) || {}).id;
+  const toId   = (fsLib.findNode(scene, positional[3]) || {}).id;
+  if (!fromId || !toId) throw new Error('Не найдена одна из нод');
+  const added = fsLib.addConnection(scene, fromId, toId, { toPort: flags.port });
+  await fsLib.saveScene(root, ref, scene);
+  console.log(added ? `✓ связь ${fromId.slice(0,8)} → ${toId.slice(0,8)}` : 'связь уже была');
+}
+
+/** kingkont rm-node <project> <board> <nodeId> */
+async function cmdRmNode({ positional }) {
+  const root = parseProjectArg(positional);
+  if (positional.length < 3) throw new Error('Нужно: kingkont rm-node <project> <board> <nodeId>');
+  const ref = fsLib.parseBoardRef(positional[1]);
+  const scene = await fsLib.loadScene(root, ref);
+  const node = fsLib.findNode(scene, positional[2]);
+  if (!node) throw new Error(`Нода не найдена`);
+  await fsLib.removeNode(root, ref, node.id);
+  console.log(`✓ удалена ${node.id}`);
+}
+
+/** kingkont voices — список голосов ElevenLabs. */
+async function cmdVoices({ flags }) {
+  const settings = await loadSettingsForCli();
+  if (!process.env.ELEVENLABS_API_KEY) throw new Error('Нет ELEVENLABS_API_KEY (включи ElevenLabs в настройках приложения)');
+  const voices = await providers.listElevenVoices();
+  if (flags.json) return console.log(asJson(voices));
+  for (const v of voices) console.log(`${v.id}  ${v.name}  (${v.category})`);
+}
+
+/** kingkont balance — балансы всех включённых провайдеров. */
+async function cmdBalance({ flags }) {
+  const settings = await loadSettingsForCli();
+  const all = await providers.fetchBalances(settings);
+  if (flags.json) return console.log(asJson(all));
+  for (const [k, v] of Object.entries(all)) {
+    console.log(`${k.padEnd(12)} ${v.amount} ${v.unit}${v.limit ? ` / ${v.limit}` : ''}`);
+  }
+  if (!Object.keys(all).length) console.log('Нет подключённых провайдеров (см. настройки приложения)');
+}
+
+/** kingkont upload <file> — загрузить файл в storage, печатает URL. */
+async function cmdUpload({ positional, flags }) {
+  if (!positional.length) throw new Error('Нужно: kingkont upload <file>');
+  const file = path.resolve(positional[0]);
+  const settings = await loadSettingsForCli();
+  const buf = await fsp.readFile(file);
+  const r = await providers.uploadFile({
+    buffer: buf, filename: path.basename(file), mime: guessMime(file), settings,
+  });
+  if (flags.json) return console.log(asJson(r));
+  console.log(r.url);
+}
+
+// =============================================================================
+// Generation runners (детали).
+// =============================================================================
+
+async function runMediaGeneration(root, ref, node, args) {
+  const { kind, prompt, modelKey, imageInputs, videoInputs, aspectRatio, resolution, duration, firstFrame, lastFrame, settings, flags } = args;
+  const start = await providers.startGeneration({
+    kind, prompt, modelKey, imageInputs, videoInputs, aspectRatio, resolution, duration, firstFrame, lastFrame, settings,
+  });
+  console.error(`[task] ${start.taskId} provider=${start.provider}`);
+  await updateNode(root, ref, node.id, n => {
+    n.generated = { ...(n.generated || {}), taskId: start.taskId, state: 'queued' };
+  });
+
+  const result = await providers.waitForGeneration(start.taskId, settings, {
+    onProgress: r => console.error(`[poll] state=${r.state} status=${r.status}`),
+  });
+
+  // Скачиваем и сохраняем.
+  const buf = await providers.downloadUrl(result.url);
+  const subdir = fsLib.defaultSubdir(kind);
+  const ext = fsLib.defaultExt(kind, null, guessMimeFromUrl(result.url));
+  const fileRel = await fsLib.uniqueFilename(root, ref, subdir, `${kind}_${node.id.slice(0,8)}${ext}`);
+  await fsLib.saveBoardFile(root, ref, fileRel, buf);
+
+  await updateNode(root, ref, node.id, n => {
+    n.status = undefined;
+    n.error = undefined;
+    n.file = fileRel;
+    n.generated = {
+      ...(n.generated || {}), state: 'success',
+      ...(typeof result.cost === 'number' ? { creditsCharged: result.cost } : {}),
+    };
+  });
+  if (flags.json) console.log(asJson({ id: node.id, file: fileRel, url: result.url, cost: result.cost ?? null }));
+  else console.log(`✓ ${node.id.slice(0,8)} → ${fileRel}${result.cost != null ? ` (cost=${result.cost})` : ''}`);
+}
+
+async function runTextGeneration(root, ref, node, { prompt, model, settings, flags }) {
+  const r = await providers.generateText({ prompt, model, settings });
+  // Текст сохраняем в .md (создаём имя если нет).
+  let fileRel = node.file;
+  if (!fileRel) {
+    const stem = (node.name || 'text').replace(/[^\wЀ-ӿ.-]+/g, '_').slice(0, 60) || 'text';
+    fileRel = await fsLib.uniqueFilename(root, ref, 'texts', `${stem}_${node.id.slice(0,8)}.md`);
+  }
+  await fsLib.saveBoardFile(root, ref, fileRel, r.text);
+  await updateNode(root, ref, node.id, n => {
+    n.status = undefined;
+    n.error = undefined;
+    n.file = fileRel;
+    n.text = r.text;
+    n.generated = { ...(n.generated || {}), state: 'success', model: r.model, ...(typeof r.cost === 'number' ? { creditsCharged: r.cost } : {}) };
+  });
+  if (flags.json) console.log(asJson({ id: node.id, file: fileRel, text: r.text, cost: r.cost ?? null }));
+  else console.log(`✓ ${node.id.slice(0,8)} → ${fileRel} (${r.text.length}ch)`);
+}
+
+async function runTtsGeneration(root, ref, node, { text, voice, ttsModel, settings, flags }) {
+  const r = await providers.generateTts({ text, voice, ttsModel, settings });
+  const fileRel = await fsLib.uniqueFilename(root, ref, 'audio', `tts_${node.id.slice(0,8)}.mp3`);
+  await fsLib.saveBoardFile(root, ref, fileRel, r.buffer);
+  await updateNode(root, ref, node.id, n => {
+    n.status = undefined; n.error = undefined; n.file = fileRel;
+    n.generated = { ...(n.generated || {}), state: 'success', ...(typeof r.cost === 'number' ? { creditsCharged: r.cost } : {}) };
+  });
+  if (flags.json) console.log(asJson({ id: node.id, file: fileRel, cost: r.cost ?? null }));
+  else console.log(`✓ ${node.id.slice(0,8)} → ${fileRel}`);
+}
+
+async function runSfxGeneration(root, ref, node, { text, durationSeconds, settings, flags }) {
+  const r = await providers.generateSfx({ text, durationSeconds, settings });
+  const fileRel = await fsLib.uniqueFilename(root, ref, 'audio', `sfx_${node.id.slice(0,8)}.mp3`);
+  await fsLib.saveBoardFile(root, ref, fileRel, r.buffer);
+  await updateNode(root, ref, node.id, n => {
+    n.status = undefined; n.error = undefined; n.file = fileRel;
+    n.generated = { ...(n.generated || {}), state: 'success', subKind: 'sfx', ...(typeof r.cost === 'number' ? { creditsCharged: r.cost } : {}) };
+  });
+  if (flags.json) console.log(asJson({ id: node.id, file: fileRel }));
+  else console.log(`✓ ${node.id.slice(0,8)} → ${fileRel}`);
+}
+
+async function runMusicGeneration(root, ref, node, { prompt, durationMs, settings, flags }) {
+  const r = await providers.generateMusic({ prompt, durationMs, settings });
+  const fileRel = await fsLib.uniqueFilename(root, ref, 'audio', `music_${node.id.slice(0,8)}.mp3`);
+  await fsLib.saveBoardFile(root, ref, fileRel, r.buffer);
+  await updateNode(root, ref, node.id, n => {
+    n.status = undefined; n.error = undefined; n.file = fileRel;
+    n.generated = { ...(n.generated || {}), state: 'success', subKind: 'music', ...(typeof r.cost === 'number' ? { creditsCharged: r.cost } : {}) };
+  });
+  if (flags.json) console.log(asJson({ id: node.id, file: fileRel }));
+  else console.log(`✓ ${node.id.slice(0,8)} → ${fileRel}`);
+}
+
+// =============================================================================
+// Helpers.
+// =============================================================================
+
+async function loadSettingsForCli() {
+  const settings = settingsLib.loadSettings();
+  settingsLib.applySettingsToEnv(settings);
+  // Подгружаем .env из cwd как fallback (для разработчиков).
+  const envPath = path.join(process.cwd(), '.env');
+  if (fs.existsSync(envPath)) {
+    for (const raw of fs.readFileSync(envPath, 'utf-8').split('\n')) {
+      const line = raw.trim();
+      if (!line || line.startsWith('#')) continue;
+      const eq = line.indexOf('=');
+      if (eq < 0) continue;
+      const key = line.slice(0, eq).trim();
+      let val = line.slice(eq + 1).trim();
+      if ((val.startsWith('"') && val.endsWith('"')) || (val.startsWith("'") && val.endsWith("'"))) val = val.slice(1, -1);
+      if (!(key in process.env)) process.env[key] = val;
+    }
+  }
+  return settings;
+}
+
+/** Перечитать scene, мутировать ноду через mutator, сохранить. Атомарно (для CLI). */
+async function updateNode(root, ref, nodeId, mutator) {
+  const scene = await fsLib.loadScene(root, ref);
+  const n = scene.nodes.find(x => x.id === nodeId);
+  if (!n) throw new Error(`нода ${nodeId} пропала между шагами генерации`);
+  mutator(n);
+  await fsLib.saveScene(root, ref, scene);
+}
+
+function guessMime(filename) {
+  const ext = path.extname(filename).toLowerCase();
+  return {
+    '.jpg': 'image/jpeg', '.jpeg': 'image/jpeg', '.png': 'image/png',
+    '.webp': 'image/webp', '.gif': 'image/gif', '.svg': 'image/svg+xml',
+    '.mp4': 'video/mp4', '.webm': 'video/webm', '.mov': 'video/quicktime',
+    '.mp3': 'audio/mpeg', '.wav': 'audio/wav', '.m4a': 'audio/mp4',
+    '.md':  'text/markdown', '.txt': 'text/plain', '.json': 'application/json',
+  }[ext] || 'application/octet-stream';
+}
+
+function guessMimeFromUrl(url) {
+  try { return guessMime(new URL(url).pathname); } catch { return 'application/octet-stream'; }
+}
+
+// =============================================================================
+// Dispatch.
+// =============================================================================
+
+const COMMANDS = {
+  open: cmdOpen,
+  list: cmdList,
+  board: cmdBoard,
+  'add-node': cmdAddNode,
+  gen: cmdGen,
+  generate: cmdGenerate,
+  status: cmdStatus,
+  connect: cmdConnect,
+  'rm-node': cmdRmNode,
+  voices: cmdVoices,
+  balance: cmdBalance,
+  upload: cmdUpload,
+};
+
+async function run(argv) {
+  const cmd = argv[0];
+  const fn = COMMANDS[cmd];
+  if (!fn) throw new Error(`unknown command: ${cmd}`);
+  return await fn(parseArgs(argv.slice(1)));
+}
+
+module.exports = { run, COMMANDS, parseArgs };