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,
    });
  });

  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();
    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 };