kingkont 0.16.2 → 0.17.0

package/lib/jobsHub.js

@@ -15,22 +15,43 @@ const wsHub = require('./wsHub');
 
 const jobsByProject = new Map();   // projectKey → Map<jobId, jobInfo>
 
-function _emit(projectKey) {
+// Серверные polling-таймеры. Для каждого jobId с known taskId+kind
+// сервер сам опрашивает провайдера каждые POLL_INTERVAL_MS. На completion
+// — WS-event с готовым URL+cost. Renderer (если жив) скачивает + пишет
+// файл; если был закрыт — при возврате найдёт по scanAllBoardsForPendingJobs.
+const pollers = new Map();   // jobId → { intervalId, taskId, kind, projectKey, lastPollAt }
+const POLL_INTERVAL_MS = 4000;
+
+let _settingsGetter = () => ({});
+function setSettingsGetter(fn) { if (typeof fn === 'function') _settingsGetter = fn; }
+
+function _emit(projectKey, extra) {
   const list = listForProject(projectKey);
-  wsHub.publish('jobs:' + projectKey, { kind: 'changed', list });
-  wsHub.publish('jobs:all', { kind: 'changed', projectKey, list });
+  const evt = { kind: 'changed', list, ...(extra || {}) };
+  wsHub.publish('jobs:' + projectKey, evt);
+  wsHub.publish('jobs:all', { ...evt, projectKey });
 }
 
-function start({ projectKey, jobId, kind, name, type }) {
+function start({ projectKey, jobId, kind, name, type, taskId }) {
   if (!projectKey || !jobId) return;
   let m = jobsByProject.get(projectKey);
   if (!m) { m = new Map(); jobsByProject.set(projectKey, m); }
-  m.set(jobId, { kind: kind || null, name: name || null, type: type || null, startedAt: Date.now() });
+  m.set(jobId, {
+    kind: kind || null, name: name || null, type: type || null,
+    taskId: taskId || null, startedAt: Date.now(),
+  });
   _emit(projectKey);
+  // Если есть taskId + kind — стартуем серверный poller. Renderer тоже
+  // полит параллельно (старый flow), но если он закрыт — серверный
+  // продолжает и шлёт WS-event на completion.
+  if (taskId && (kind === 'image' || kind === 'video')) {
+    _startPoller({ projectKey, jobId, taskId, kind });
+  }
 }
 
 function end({ projectKey, jobId }) {
   if (!projectKey || !jobId) return null;
+  _stopPoller(jobId);
   const m = jobsByProject.get(projectKey);
   if (!m) return null;
   const job = m.get(jobId);
@@ -40,6 +61,63 @@ function end({ projectKey, jobId }) {
   return job || null;
 }
 
+function _startPoller({ projectKey, jobId, taskId, kind }) {
+  if (pollers.has(jobId)) return;
+  const providers = require('./providers');
+  const tick = async () => {
+    const entry = pollers.get(jobId);
+    if (!entry) return;
+    entry.lastPollAt = Date.now();
+    let r;
+    try { r = await providers.pollGeneration(taskId, _settingsGetter()); }
+    catch (e) {
+      console.warn('[jobsHub] poll failed for', jobId, e?.message);
+      return;          // следующий tick попробует снова
+    }
+    if (r.status === 'done') {
+      _stopPoller(jobId);
+      // Push WS event с completion-info. Renderer (если жив) скачает
+      // через /api/proxy и запишет в FSAH. Если renderer закрыт — event
+      // пропадёт; при возврате scanAllBoardsForPendingJobs восстановит
+      // (там status='generating' + taskId в scene.json).
+      wsHub.publish('jobs:' + projectKey, {
+        kind: 'done', jobId, taskId, url: r.url, cost: r.cost ?? null, kindOf: kind, provider: r.provider,
+      });
+      wsHub.publish('jobs:all', {
+        kind: 'done', projectKey, jobId, taskId, url: r.url, cost: r.cost ?? null, kindOf: kind, provider: r.provider,
+      });
+      // Job-end ОТЛОЖЕН — ждём ack от клиента (он скачает файл).
+      // Но если за 60s никто не пришёл — auto-end (чтобы счётчики не зависали).
+      setTimeout(() => {
+        if (!pollers.has(jobId)) end({ projectKey, jobId });
+      }, 60000);
+    } else if (r.status === 'error') {
+      _stopPoller(jobId);
+      wsHub.publish('jobs:' + projectKey, {
+        kind: 'failed', jobId, taskId, error: r.error, kindOf: kind,
+      });
+      wsHub.publish('jobs:all', {
+        kind: 'failed', projectKey, jobId, taskId, error: r.error, kindOf: kind,
+      });
+      // На ошибке тоже auto-end через 60s.
+      setTimeout(() => {
+        if (!pollers.has(jobId)) end({ projectKey, jobId });
+      }, 60000);
+    }
+    // pending → ждём следующий tick (interval сам зовёт)
+  };
+  // Первый poll — через 2s (даём провайдеру время принять задачу).
+  setTimeout(tick, 2000);
+  const intervalId = setInterval(tick, POLL_INTERVAL_MS);
+  pollers.set(jobId, { intervalId, taskId, kind, projectKey, lastPollAt: 0 });
+}
+function _stopPoller(jobId) {
+  const entry = pollers.get(jobId);
+  if (!entry) return;
+  clearInterval(entry.intervalId);
+  pollers.delete(jobId);
+}
+
 function listForProject(projectKey) {
   const m = jobsByProject.get(projectKey);
   if (!m) return [];
@@ -54,4 +132,4 @@ function listAll() {
   return out;
 }
 
-module.exports = { start, end, listForProject, listAll };
+module.exports = { start, end, listForProject, listAll, setSettingsGetter };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "kingkont",
-  "version": "0.16.2",
+  "version": "0.17.0",
   "description": "KingKont · Chatium — нод-редактор сцен с AI-генерацией (картинки/видео/голос/SFX/музыка/текст)",
   "main": "main.js",
   "bin": {

package/renderer/board.js

@@ -632,6 +632,91 @@ async function refreshBgServerCache() {
   } catch {}
 }
 
+// === WS subscription для активного проекта: server-side polling ===
+// Сервер сам опрашивает провайдера (см. lib/jobsHub.js _startPoller).
+// На completion пушит 'done' event с url+cost. Renderer скачивает и
+// записывает файл — это и есть "пробуждение" даже после Cmd+R.
+let _projectJobsWS = null;
+function _subscribeProjectJobsWS() {
+  const pk = state.cloudProjectId ? 'cloud:' + state.cloudProjectId
+           : state.filmHandle?.name ? 'folder:' + state.filmHandle.name : null;
+  if (!pk) return;
+  // Reuse тот же WS что использует welcome для jobs:all — но теперь
+  // подписка ещё и на jobs:<pk>.
+  if (!_projectJobsWS || _projectJobsWS.readyState !== WebSocket.OPEN) {
+    try {
+      const proto = location.protocol === 'https:' ? 'wss' : 'ws';
+      _projectJobsWS = new WebSocket(`${proto}://${location.host}/ws`);
+    } catch { return; }
+    _projectJobsWS.onopen = () => {
+      try { _projectJobsWS.send(JSON.stringify({ type: 'subscribe', channel: 'jobs:' + pk })); } catch {}
+    };
+    _projectJobsWS.onmessage = (e) => {
+      let m; try { m = JSON.parse(e.data); } catch { return; }
+      if (m?.type === 'event') _handleProjectJobsEvent(m.event);
+    };
+    _projectJobsWS.onclose = () => { _projectJobsWS = null; setTimeout(_subscribeProjectJobsWS, 5000); };
+    _projectJobsWS.onerror = () => {};
+  } else {
+    try { _projectJobsWS.send(JSON.stringify({ type: 'subscribe', channel: 'jobs:' + pk })); } catch {}
+  }
+}
+
+// Обработчик WS-event'ов от server-side jobsHub. На 'done' — скачиваем
+// результат через /api/proxy и пишем файл (это работает только если
+// FSAH-handle ещё валиден — т.е. project открыт).
+async function _handleProjectJobsEvent(evt) {
+  if (!evt) return;
+  if (evt.kind === 'done') {
+    if (state.jobs.has(evt.jobId)) {
+      // Local pollJob уже видит этот же taskId и сам обработает completion.
+      // Server-side push — это backup. Не дублируем работу.
+      return;
+    }
+    // Local job не активен — значит pollJob не запущен (после Cmd+R).
+    // Скачиваем + пишем + помечаем done. Сначала найдём ноду.
+    if (!state.currentBoard) { vlog('info', `WS done arrived but no currentBoard, jobId=${evt.jobId}`); return; }
+    const node = state.currentBoard.metadata.nodes.find(n => n.id === evt.jobId);
+    if (!node) { vlog('info', `WS done: nodeId not found in current board ${evt.jobId}`); return; }
+    try {
+      const r = await fetch('/api/proxy?url=' + encodeURIComponent(evt.url));
+      if (!r.ok) throw new Error(`download HTTP ${r.status}`);
+      const blob = await r.blob();
+      const ext = evt.kindOf === 'image' ? 'jpg' : 'mp4';
+      const sub = evt.kindOf === 'image' ? 'frames' : 'clips';
+      const dir = await getOrCreateBoardSubdir(state.currentBoard.handle, sub);
+      const baseName = await uniqueName(dir, `gen_${Date.now()}.${ext}`);
+      await writeFile(dir, baseName, blob);
+      const relPath = `${sub}/${baseName}`;
+      await mutateNode(state.currentBoard.key, state.currentBoard.handle, evt.jobId, n => {
+        n.status = undefined; n.error = undefined; n.file = relPath;
+        if (typeof evt.cost === 'number') {
+          n.generated = { ...(n.generated || {}), creditsCharged: evt.cost };
+        }
+      });
+      if (typeof showToast === 'function') {
+        showToast(`✓ ${evt.kindOf} «${node.name || node.id.slice(0,6)}» готов (server-side)`, 'ok');
+      }
+      if (typeof systemNotify === 'function' && document.hidden) {
+        systemNotify('KingKont', `${evt.kindOf} «${node.name || ''}» готов`).catch(() => {});
+      }
+      // Завершаем job на сервере (free up the slot).
+      const pk = state.cloudProjectId ? 'cloud:' + state.cloudProjectId
+               : state.filmHandle?.name ? 'folder:' + state.filmHandle.name : null;
+      if (pk) bgJobEnd(evt.jobId, pk);
+    } catch (e) {
+      console.warn('[ws-done] download/write failed:', e?.message);
+    }
+  } else if (evt.kind === 'failed') {
+    if (typeof showToast === 'function') {
+      showToast(`⚠ Генерация провалилась: ${evt.error?.slice?.(0,80) || 'unknown'}`, 'error');
+    }
+    const pk = state.cloudProjectId ? 'cloud:' + state.cloudProjectId
+             : state.filmHandle?.name ? 'folder:' + state.filmHandle.name : null;
+    if (pk) bgJobEnd(evt.jobId, pk);
+  }
+}
+
 // Live-update welcome когда bg-jobs меняются (start/end). Перерисовываем
 // только если открыт welcome (нет проекта), иначе UI не виден.
 window.addEventListener('bgjobs:changed', () => {
@@ -1142,6 +1227,10 @@ async function openFilm(handle) {
 
   // Сканируем все доски на незавершённые задачи генерации (после перезагрузки)
   scanAllBoardsForPendingJobs(handle).catch(e => console.warn('scan failed', e));
+  // Subscribe на server-side WS events для этого проекта — сервер сам
+  // полит провайдера и пушит 'done'/'failed' даже если у нас нет local
+  // pollJob (например после Cmd+R). См. _handleProjectJobsEvent.
+  _subscribeProjectJobsWS();
   // Подгружаем настройки всех локаций
   loadAllLocationsInfo().catch(() => {});
   // Подгружаем настройки всех персонажей (для add-menu и т.д.)

package/renderer/generate.js

@@ -1851,6 +1851,10 @@ async function startGenerationJob(node, kind, prompt, mediaRefs, boardHandle, bK
     }
     job.taskId = data.taskId;
     logJob(node.id, `taskId=${data.taskId}`);
+    // Сообщаем server-side poller'у taskId — теперь сервер тоже опрашивает
+    // провайдера. Если renderer закроется/Cmd+R — server продолжит и
+    // пушнёт WS-event на completion.
+    if (typeof bgJobUpdateTaskId === 'function') bgJobUpdateTaskId(node.id, data.taskId, job.projectKey);
     await mutateNode(bKey, boardHandle, node.id, n => {
       n.generated = { ...(n.generated || {}), taskId: data.taskId, state: 'queued' };
     });

package/renderer/state.js

@@ -82,10 +82,9 @@ function _projectKeyForCurrent() {
   return null;
 }
 function bgJobStart(info) {
-  // info: {nodeId, kind, name?, projectKey?}
-  // ВАЖНО: projectKey обязателен — иначе job висит на сервере вечно
-  // (если юзер закроет проект до bgJobEnd, _projectKeyForCurrent вернёт
-  //  null, и end-call станет no-op).
+  // info: {nodeId, kind, name?, projectKey?, taskId?}
+  // taskId важен — server-side poller использует его чтобы опрашивать
+  // провайдера сам (Chatium/KIE) даже когда renderer закрыт.
   const pk = info.projectKey || _projectKeyForCurrent();
   if (!pk) { console.warn('bgJobStart: no projectKey, skipping'); return; }
   try {
@@ -98,9 +97,26 @@ function bgJobStart(info) {
   } catch {}
   fetch('/api/jobs/track', {
     method: 'POST', headers: { 'Content-Type': 'application/json' },
-    body: JSON.stringify({ action: 'start', projectKey: pk, jobId: info.nodeId, kind: info.kind, name: info.name }),
+    body: JSON.stringify({
+      action: 'start',
+      projectKey: pk,
+      jobId: info.nodeId,
+      kind: info.kind,
+      name: info.name,
+      taskId: info.taskId || null,        // ← server-side poller цепляется по taskId
+    }),
   }).catch(() => {});
 }
+// Update taskId for an existing job (server didn't have it on initial track).
+function bgJobUpdateTaskId(nodeId, taskId, projectKey) {
+  const pk = projectKey || _projectKeyForCurrent();
+  if (!pk || !taskId) return;
+  fetch('/api/jobs/track', {
+    method: 'POST', headers: { 'Content-Type': 'application/json' },
+    body: JSON.stringify({ action: 'start', projectKey: pk, jobId: nodeId, taskId }),
+  }).catch(() => {});
+}
+window.bgJobUpdateTaskId = bgJobUpdateTaskId;
 function bgJobEnd(nodeId, projectKey) {
   // projectKey ОБЯЗАТЕЛЕН — без него юзер на welcome теряет уведомление
   // и счётчик зависает. Раньше фоллбэк на _projectKeyForCurrent давал

package/server.js

@@ -355,9 +355,9 @@ async function handleChatClear(req, res) {
 async function handleJobsTrack(req, res) {
   try {
     const body = await readJson(req);
-    const { action, projectKey, jobId, kind, name, type } = body || {};
+    const { action, projectKey, jobId, kind, name, type, taskId } = body || {};
     if (!projectKey || !jobId) return send(res, 400, { error: 'projectKey + jobId обязательны' });
-    if (action === 'start') jobsHub.start({ projectKey, jobId, kind, name, type });
+    if (action === 'start') jobsHub.start({ projectKey, jobId, kind, name, type, taskId });
     else if (action === 'end') jobsHub.end({ projectKey, jobId });
     else return send(res, 400, { error: 'action: start|end' });
     send(res, 200, { ok: true });
@@ -455,6 +455,9 @@ function start(port = PORT, opts = {}) {
   // Init chatSession с userDataDir для персистентности историй.
   // Без opts.userDataDir чат живёт только in-memory (CLI-режим).
   chatSession.init({ userDataDir: opts.userDataDir || null });
+  // jobsHub нужен settingsGetter чтобы поллить провайдеров (Chatium token,
+  // KIE_API_KEY и т.п.).
+  jobsHub.setSettingsGetter(getSettings);
   return new Promise((resolveOk, reject) => {
     server.once('error', reject);
     server.listen(port, () => {