hyreal-gen-pic/server/routes/settings.js

'use strict';

// 系统设置：模型列表、配置读写（脱敏）、生图/对话 API 连通性测试

const fs = require('fs');
const path = require('path');
const https = require('https');
const { URL } = require('url');
const { sendJson, readBody } = require('../util.js');
const { DEFAULT_CHAT_MODEL } = require('../context.js');

function register(app, ctx) {
  app.get('/api/models', async ({ res }) => {
    sendJson(res, 200, { models: ctx.models.MODELS, default: ctx.config.model });
  });

  app.get('/api/settings', async ({ res }) => {
    sendJson(res, 200, {
      apiKey: ctx.config.apiKey ? ctx.config.apiKey.slice(0, 8) + '••••••••' : '',
      apiBase: ctx.config.apiBase,
      model: ctx.config.model,
      rootDir: ctx.workspace.getWorkspaceRoot(),
      pageSize: ctx.config.pageSize,
      delayMs: ctx.config.delayMs,
      cookieConfigured: !!ctx.session.cookie && !!ctx.session.getKsid(),
      port: ctx.config.port,
      chatModel: ctx.config.chatModel || '',
      chatApiBase: ctx.config.chatApiBase || ctx.config.apiBase,
      chatApiKey: ctx.config.chatApiKey ? ctx.config.chatApiKey.slice(0, 8) + '••••••••' : '',
      chatKeyFallback: !ctx.config.chatApiKey && !!ctx.config.apiKey,
      autoSyncEnabled: !!(ctx.config.autoSync && ctx.config.autoSync.enabled),
      autoSyncTime: (ctx.config.autoSync && ctx.config.autoSync.time) || '06:00',
      accessTokenSet: !!ctx.config.accessToken,
      aitableEnabled: !!(ctx.config.aitable && ctx.config.aitable.enabled),
      aitableAppKey: (ctx.config.aitable && ctx.config.aitable.appKey) || '',
      aitableBaseId: (ctx.config.aitable && ctx.config.aitable.baseId) || '',
      aitableSheetName: (ctx.config.aitable && ctx.config.aitable.sheetName) || '经营数据',
      aitableSecretSet: !!(ctx.config.aitable && ctx.config.aitable.appSecret),
    });
  });

  app.post('/api/settings', async ({ req, res }) => {
    const body = await readBody(req);
    const config = ctx.config;
    const previousApiKey = config.apiKey;
    const previousCookie = ctx.session.cookie;
    if (body.apiKey && !body.apiKey.includes('••••')) config.apiKey = body.apiKey;
    if (body.model && ctx.models.MODEL_IDS.has(body.model)) config.model = body.model;
    if (body.rootDir) {
      const rootDir = path.resolve(String(body.rootDir).trim());
      if (!fs.existsSync(rootDir) || !fs.statSync(rootDir).isDirectory()) {
        return sendJson(res, 400, { error: '输出根目录不存在: ' + rootDir });
      }
      config.rootDir = rootDir;
    }
    if (body.pageSize !== undefined) {
      const pageSize = Number(body.pageSize);
      if (!Number.isInteger(pageSize) || pageSize < 1 || pageSize > 200) {
        return sendJson(res, 400, { error: '每页条数必须为 1-200 的整数' });
      }
      config.pageSize = pageSize;
    }
    if (body.delayMs !== undefined) {
      const delayMs = Number(body.delayMs);
      if (!Number.isFinite(delayMs) || delayMs < 0 || delayMs > 60000) {
        return sendJson(res, 400, { error: '间隔必须为 0-60000 毫秒' });
      }
      config.delayMs = delayMs;
    }
    if (body.chatModel !== undefined) config.chatModel = String(body.chatModel).trim();
    if (body.chatApiBase !== undefined) config.chatApiBase = String(body.chatApiBase).trim();
    if (body.chatApiKey !== undefined && !String(body.chatApiKey).includes('••••')) config.chatApiKey = String(body.chatApiKey).trim();
    if (body.autoSyncEnabled !== undefined || body.autoSyncTime !== undefined) {
      if (!config.autoSync || typeof config.autoSync !== 'object') config.autoSync = { enabled: false, time: '06:00' };
      if (body.autoSyncEnabled !== undefined) config.autoSync.enabled = !!body.autoSyncEnabled;
      if (body.autoSyncTime !== undefined) {
        const t = String(body.autoSyncTime).trim();
        const hm = /^\d{1,2}:\d{2}$/.test(t) ? t.split(':').map(Number) : null;
        if (!hm || hm[0] > 23 || hm[1] > 59) {
          return sendJson(res, 400, { error: '自动同步时间格式应为 HH:mm' });
        }
        config.autoSync.time = t;
      }
    }
    if (body.accessToken !== undefined) {
      const t = String(body.accessToken).trim();
      if (t === '-') config.accessToken = ''; // 约定：输入 - 清除口令
      else if (t) config.accessToken = t;
    }
    if (body.aitableEnabled !== undefined || body.aitableAppKey !== undefined || body.aitableBaseId !== undefined || body.aitableSheetName !== undefined || body.aitableAppSecret !== undefined) {
      if (!config.aitable || typeof config.aitable !== 'object') config.aitable = { enabled: false, appKey: '', appSecret: '', baseId: '', sheetName: '经营数据' };
      if (body.aitableEnabled !== undefined) config.aitable.enabled = !!body.aitableEnabled;
      if (body.aitableAppKey !== undefined) config.aitable.appKey = String(body.aitableAppKey).trim();
      if (body.aitableBaseId !== undefined) {
        // 支持直接粘贴 AI 表格分享链接：自动提取 /i/nodes/<ID> 里的节点 ID 作为 Base ID
        const raw = String(body.aitableBaseId).trim();
        const m = /i\/nodes\/([A-Za-z0-9]+)/.exec(raw) || /[?&]baseId=([A-Za-z0-9]+)/.exec(raw);
        config.aitable.baseId = m ? m[1] : raw;
      }
      if (body.aitableSheetName !== undefined) config.aitable.sheetName = String(body.aitableSheetName).trim() || '经营数据';
      if (body.aitableAppSecret !== undefined) {
        const sec = String(body.aitableAppSecret).trim();
        if (sec === '-') config.aitable.appSecret = ''; // 约定：输入 - 清除
        else if (sec) config.aitable.appSecret = sec;
      }
    }
    ctx.saveConfig();
    if (body.cookie !== undefined) {
      ctx.session.cookie = body.cookie;
      ctx.session.saveSession();
    }
    ctx.logs.pushSystemLog(
      `系统设置已保存：模型=${config.model}，根目录=${config.rootDir}，每页=${config.pageSize}，间隔=${config.delayMs}ms` +
      (previousApiKey !== config.apiKey ? '，API Key 已更新' : '') +
      (body.cookie !== undefined && previousCookie !== ctx.session.cookie ? '，Cookie 已更新' : ''),
      'success'
    );
    sendJson(res, 200, { ok: true });
  });

  // 生图 API 连通性测试：GET /v1/models（轻量、不产生生图费用）
  app.post('/api/image/test', async ({ res }) => {
    return new Promise((resolve) => {
      let base;
      try { base = new URL(ctx.config.apiBase); } catch { return sendJson(res, 400, { error: 'API 地址不合法: ' + ctx.config.apiBase }); }
      const options = {
        hostname: base.hostname, port: base.port || 443, path: '/v1/models', method: 'GET',
        headers: { Authorization: `Bearer ${ctx.config.apiKey}` }, timeout: 15000,
      };
      const up = https.request(options, (upRes) => {
        const chunks = [];
        upRes.on('data', (c) => chunks.push(c));
        upRes.on('end', () => {
          let json = null;
          try { json = JSON.parse(Buffer.concat(chunks).toString('utf8')); } catch { /* 保留空 */ }
          const models = json && Array.isArray(json.data) ? json.data.map((m) => (m && m.id) || '').filter(Boolean) : null;
          sendJson(res, 200, { statusCode: upRes.statusCode, models, modelFound: !!(models && models.includes(ctx.config.model)), model: ctx.config.model });
        });
      });
      up.on('error', (e) => sendJson(res, 200, { statusCode: 0, error: e.message || String(e) }));
      up.on('timeout', () => up.destroy(new Error('请求超时')));
      up.end();
    });
  });

  app.post('/api/chat/test', async ({ req, res }) => {
    const body = await readBody(req);
    const nlQuery = require('../../nl-query.js');
    const cfg = {
      chatApiBase: body.chatApiBase || ctx.config.chatApiBase || ctx.config.apiBase,
      chatApiKey: (body.chatApiKey && !String(body.chatApiKey).includes('••••')) ? body.chatApiKey : (ctx.config.chatApiKey || ctx.config.apiKey),
      chatModel: body.chatModel || ctx.config.chatModel || DEFAULT_CHAT_MODEL,
    };
    try {
      const reply = await nlQuery.ping(cfg);
      sendJson(res, 200, { ok: true, reply });
    } catch (e) {
      sendJson(res, 502, { error: e.message });
    }
  });
}

module.exports = { register };