neuroalpha 0.3.17 → 0.5.1

package/server/instance/agent.js

@@ -0,0 +1,259 @@
+// Copyright (c) 2026 Roman A. Perepelkin. All rights reserved. Proprietary software.
+// Агент арендованной машины клиента: раз в минуту отправляет в центр техническое состояние
+// и получает подписанную лицензию (тариф, срок, разрешение торговли). Без действующей лицензии
+// новые заявки не отправляются; защитные стоп-заявки по открытым позициям продолжают работать
+// (кроме аварийной остановки «halted» из центра).
+import fs from 'node:fs';
+import os from 'node:os';
+import path from 'node:path';
+import { ROOT, DATA_DIR } from '../config.js';
+import { db, kvGet, kvSet, userKvGet } from '../db.js';
+import { IS_INSTANCE, IS_DESKTOP, INSTANCE_KIND } from '../role.js';
+import { verifyToken } from './signing.js';
+import { getIdentity, isActivated, saveDesktopIdentity, installId } from './identity.js';
+
+const HEARTBEAT_MS = 60000;
+const VERSION = (() => {
+  try { return JSON.parse(fs.readFileSync(path.join(ROOT, 'package.json'), 'utf8')).version || ''; } catch { return ''; }
+})();
+const startedAt = Date.now();
+
+let license = null; // проверенная полезная нагрузка лицензии
+let lastError = '';
+const recentErrors = [];
+
+function noteError(msg) {
+  const m = String(msg || '').slice(0, 200);
+  if (m && m !== lastError) console.warn(`⚠ Связь с центром: ${m}`);
+  lastError = m;
+  if (m) {
+    recentErrors.unshift(`${new Date().toISOString()} ${m}`);
+    recentErrors.length = Math.min(recentErrors.length, 5);
+  }
+}
+
+// ---------- команды из мессенджера (кнопки «Исполнить / Отклонить»), доставленные в подписанной лицензии ----------
+let commandHandler = null;
+export function setInstanceCommandHandler(fn) { commandHandler = fn; }
+const CMD_DONE_KEY = 'instanceCmdsDone';
+const doneCmds = () => (kvGet(CMD_DONE_KEY) || []).filter((x) => Date.now() - x.ts < 3 * 3600000);
+
+async function runCommands(cmds = []) {
+  if (!Array.isArray(cmds) || !cmds.length || !commandHandler) return;
+  const done = doneCmds();
+  const seen = new Set(done.map((x) => x.id));
+  for (const cmd of cmds.slice(0, 20)) {
+    if (!cmd?.id || seen.has(cmd.id)) continue;
+    // отмечаем до выполнения: повторная доставка той же команды не исполнит пакет дважды
+    done.push({ id: cmd.id, ts: Date.now() });
+    seen.add(cmd.id);
+    kvSet(CMD_DONE_KEY, done);
+    let reply;
+    try { reply = await commandHandler(cmd); } catch (e) { reply = `Ошибка выполнения команды: ${e.message}`; }
+    if (reply) relayNotifyToCentral(String(reply)).catch(() => {});
+  }
+}
+
+function acceptLicenseToken(token, { runCmds = false } = {}) {
+  const p = verifyToken(token, { typ: 'license', aud: getIdentity().id });
+  license = p;
+  if (runCmds && p.cmds?.length) setImmediate(() => runCommands(p.cmds).catch(() => {}));
+  kvSet('instanceLicenseToken', token);
+  // локальный пользователь машины — клиент (не администратор платформы)
+  db.prepare("UPDATE users SET role = 'user', email = ?, name = ?, plan = ?, plan_until = ?, offer_accepted_at = ? WHERE id = 1")
+    .run(String(p.email || 'client@instance.local'), String(p.name || 'Клиент'), String(p.plan || 'none'), Number(p.planUntil) || 0, Number(p.offerAcceptedAt) || 0);
+  // ключ ИИ-прокси центра (провайдер «НейроАльфа Облако» берёт apiKey отсюда)
+  kvSet('cloudLicense', {
+    apiKey: p.apiKey || '',
+    cloudUrl: getIdentity().centralUrl,
+    email: p.email,
+    name: p.name,
+    plan: p.plan,
+    planName: p.planName,
+    planUntil: p.planUntil || 0,
+    verifiedAt: Date.now(),
+  });
+  return p;
+}
+
+export function currentInstanceLicense() {
+  if (!license) return null;
+  if (!(license.exp * 1000 > Date.now())) return null;
+  return license;
+}
+
+// Разрешения торговли. Вне режима instance ограничений нет.
+export function tradingGate() {
+  if (!IS_INSTANCE) return { allowNew: true, allowProtective: true, reason: '' };
+  if (!isActivated()) return { allowNew: false, allowProtective: true, reason: 'Программа не активирована — введите код активации из личного кабинета' };
+  const lic = currentInstanceLicense();
+  if (!lic) return { allowNew: false, allowProtective: true, reason: 'Нет действующей лицензии центра — новые заявки приостановлены (защитные стопы работают)' };
+  if (lic.trading === 'halted') return { allowNew: false, allowProtective: false, reason: 'Торговля остановлена администратором платформы' };
+  if (lic.revoked) return { allowNew: false, allowProtective: true, reason: 'Аренда машины завершена — новые заявки не отправляются' };
+  if (lic.trading === 'paused') return { allowNew: false, allowProtective: true, reason: 'Торговля приостановлена администратором платформы (защитные стопы работают)' };
+  if (!lic.plan || lic.plan === 'none') return { allowNew: false, allowProtective: true, reason: 'Подписка не активна — новые заявки приостановлены' };
+  return { allowNew: true, allowProtective: true, reason: '' };
+}
+
+// ---------- техническое состояние (без сумм, позиций, токенов и персональных данных) ----------
+function collectStatus() {
+  let list = [];
+  try {
+    list = db.prepare('SELECT data FROM broker_accounts WHERE user_id = 1').all().map((r) => JSON.parse(r.data));
+  } catch {
+    const accs = userKvGet('brokerAccounts', [], 1);
+    list = Array.isArray(accs) ? accs : [];
+  }
+  const brokers = {};
+  for (const a of list) brokers[a.broker || 'unknown'] = (brokers[a.broker || 'unknown'] || 0) + 1;
+  const syncTs = list.map((a) => Number(a.lastSync?.at) || 0).filter(Boolean);
+  let dbMb = 0;
+  try { dbMb = Math.round(fs.statSync(path.join(DATA_DIR, 'trading.db')).size / 1048576); } catch { /* ignore */ }
+  const fcAt = Number(kvGet('lastForecastSnap')?.result?.at || 0);
+  const gate = tradingGate();
+  const lic = currentInstanceLicense();
+  return {
+    version: VERSION,
+    node: process.version,
+    uptimeSec: Math.round((Date.now() - startedAt) / 1000),
+    rssMb: Math.round(process.memoryUsage().rss / 1048576),
+    dbMb,
+    accounts: list.length,
+    autoAccounts: list.filter((a) => a.mode === 'auto').length,
+    pausedAccounts: list.filter((a) => a.autoSchedule === false || a.mode === 'readonly').length,
+    haltedAccounts: list.filter((a) => a.haltedReason).length,
+    brokers,
+    lastSyncAgoSec: syncTs.length ? Math.round((Date.now() - Math.max(...syncTs)) / 1000) : null,
+    forecastAgoSec: fcAt ? Math.round((Date.now() - fcAt) / 1000) : null,
+    tradingAllowed: gate.allowNew,
+    tradingReason: gate.reason,
+    licenseLeftSec: lic ? Math.round(lic.exp - Date.now() / 1000) : 0,
+    errors: recentErrors.slice(0, 5),
+    host: os.hostname().slice(0, 60),
+  };
+}
+
+async function centralPost(pathname, body, timeoutMs = 20000, { withKey = true } = {}) {
+  const id = getIdentity();
+  const r = await fetch(`${id.centralUrl}${pathname}`, {
+    method: 'POST',
+    headers: { 'Content-Type': 'application/json', ...(withKey ? { Authorization: `Bearer ${id.key}` } : {}) },
+    body: JSON.stringify(body || {}),
+    signal: AbortSignal.timeout(timeoutMs),
+  });
+  const j = await r.json().catch(() => ({}));
+  if (!r.ok) throw Object.assign(new Error(j.error || `HTTP ${r.status}`), { status: r.status });
+  return j;
+}
+
+// Адрес терминала на компьютере клиента (для входа из кабинета центра одноразовой ссылкой)
+const localUrl = () => `http://localhost:${Number(process.env.PORT || 3000)}`;
+
+let updateInfo = null; // { latestVersion, downloadUrl } — если центр сообщил о новой версии
+
+export function instanceInfo() {
+  const id = getIdentity();
+  const lic = currentInstanceLicense();
+  return {
+    kind: INSTANCE_KIND,
+    activated: isActivated(),
+    id: id.id,
+    centralUrl: id.centralUrl,
+    version: VERSION,
+    licenseValidTill: lic ? lic.exp * 1000 : 0,
+    lastError,
+    update: updateInfo,
+  };
+}
+
+let hbBusy = false;
+export async function heartbeatNow() {
+  if (hbBusy || !isActivated()) return license;
+  hbBusy = true;
+  try {
+    const j = await centralPost('/api/instance/heartbeat', {
+      status: collectStatus(),
+      installId: installId(),
+      kind: INSTANCE_KIND,
+      ack: doneCmds().map((x) => x.id),
+      ...(IS_DESKTOP ? { localUrl: localUrl() } : {}),
+    });
+    if (!j.license) throw new Error('Центр не выдал лицензию');
+    acceptLicenseToken(j.license, { runCmds: true });
+    updateInfo = j.latestVersion && j.latestVersion !== VERSION ? { latestVersion: j.latestVersion, downloadUrl: j.downloadUrl || '' } : null;
+    if (lastError) console.log('✓ Связь с центром восстановлена, лицензия обновлена');
+    lastError = '';
+  } catch (e) {
+    // центр отверг ключ (отвязан, заменён или машина удалена) — лицензия отзывается сразу, без ожидания срока
+    if (e.status === 401 || e.status === 409) {
+      license = null;
+      kvSet('instanceLicenseToken', null);
+    }
+    noteError(e.message);
+  } finally {
+    hbBusy = false;
+  }
+  return license;
+}
+
+// ---------- одноразовый вход с центра (SSO) ----------
+const usedSso = new Map(); // jti -> exp
+export function consumeSsoToken(token) {
+  const p = verifyToken(token, { typ: 'sso', aud: getIdentity().id, leewaySec: 30 });
+  const now = Math.floor(Date.now() / 1000);
+  for (const [k, exp] of usedSso) if (exp < now - 300) usedSso.delete(k);
+  if (!p.jti || usedSso.has(p.jti)) throw new Error('Ссылка входа уже использована — откройте терминал из кабинета ещё раз');
+  usedSso.set(p.jti, p.exp);
+  return p;
+}
+
+// ---------- уведомления клиенту через центр (боты Telegram / MAX живут только в центре) ----------
+// после уведомления с кнопками 15 минут связываемся с центром каждые 5 с, чтобы нажатие выполнилось быстро
+let fastUntil = 0;
+export async function relayNotifyToCentral(text, buttons = []) {
+  if (buttons?.length) fastUntil = Date.now() + 15 * 60000;
+  try {
+    const j = await centralPost('/api/instance/notify', { text: String(text || '').slice(0, 3500), buttons: buttons || [] }, 15000);
+    return { tg: Boolean(j.tg), max: Boolean(j.max) };
+  } catch (e) {
+    noteError(`уведомление: ${e.message}`);
+    return { tg: false, max: false };
+  }
+}
+
+// ---------- активация установки на компьютере кодом из личного кабинета ----------
+export async function activateDesktop(code) {
+  if (!IS_DESKTOP) throw new Error('Активация кодом доступна только для установки на компьютер');
+  const clean = String(code || '').trim().toUpperCase();
+  if (!/^[A-Z0-9-]{8,24}$/.test(clean)) throw new Error('Введите код активации из личного кабинета (вида NA-XXXX-XXXX)');
+  const j = await centralPost('/api/instance/activate', {
+    code: clean,
+    installId: installId(),
+    localUrl: localUrl(),
+    hostname: os.hostname().slice(0, 60),
+    version: VERSION,
+  }, 20000, { withKey: false });
+  if (!j.instanceId || !j.instanceKey) throw new Error('Центр не выдал ключ установки');
+  saveDesktopIdentity({ id: j.instanceId, key: j.instanceKey, centralUrl: getIdentity().centralUrl, publicKeyPem: j.publicKey || '' });
+  license = null;
+  await heartbeatNow();
+  if (!currentInstanceLicense()) throw new Error(lastError || 'Ключ получен, но лицензия не выдана — проверьте подписку в кабинете');
+  return instanceInfo();
+}
+
+export function startInstanceAgent() {
+  if (!IS_INSTANCE) return;
+  db.prepare("UPDATE users SET role = 'user' WHERE id = 1 AND role = 'admin'").run();
+  const saved = kvGet('instanceLicenseToken');
+  if (saved) {
+    try { acceptLicenseToken(saved); } catch (e) { noteError(`сохранённая лицензия: ${e.message}`); }
+  }
+  const loop = async () => {
+    await heartbeatNow().catch(() => {});
+    const t = setTimeout(loop, Date.now() < fastUntil ? 5000 : HEARTBEAT_MS);
+    t.unref?.();
+  };
+  loop();
+}
+
+export const INSTANCE_VERSION = VERSION;