"use strict";
/*
 * 光湖 Portal · 语言推理引擎注册中心
 * 冰朔 TCS-0002∞ · 铸渊 ICE-GL-ZY001
 *
 * 四个商业大模型 API 统一接口
 * 所有引擎均支持 SSE 流式输出
 *
 * v2: 人格体语境从 persona-brain.db 加载，不再硬编码
 *
 * cc-002: 永不在 messages 里补 system role
 * cc-004: 中文回执, 不甩英文 stacktrace
 */

const https = require("https");
const path = require("path");
const fs = require("fs");

const ENGINES = {
  deepseek: { name: "DeepSeek", model: "deepseek-v4-pro", endpoint: "api.deepseek.com", path: "/chat/completions", port: 443, tls: true },
  zhipu: { name: "智谱清言", model: "glm-4-plus", endpoint: "open.bigmodel.cn", path: "/api/paas/v4/chat/completions", port: 443, tls: true },
  tongyi: { name: "通义千问", model: "qwen-max", endpoint: "dashscope.aliyuncs.com", path: "/compatible-mode/v1/chat/completions", port: 443, tls: true },
  huoshan: { name: "火山引擎", model: "doubao-pro-32k", endpoint: "ark.cn-beijing.volces.com", path: "/api/v3/chat/completions", port: 443, tls: true },
};

const API_KEYS = {
  deepseek: process.env.DEEPSEEK_KEY || "sk-a9b69e9cd2dc4ca68d6aceaa84f22afb",
  zhipu: process.env.ZHIPU_KEY || "3863acbcc2bc4f6587ace4bb064c092c.KoPdL4slY5cyHpOk",
  tongyi: process.env.TONGYI_KEY || "sk-96ca227be3ca47e2a4bbb9da74e42122",
  huoshan: process.env.HUOSHAN_KEY || "ark-ddeba9f4-8c5a-449e-b549-9c29ec1e6f8c-a39ea",
};

const TOOL_PATTERNS = [
  { keyword: "notion", tool: "notion", desc: "Notion 知识库" },
  { keyword: "仓库", tool: "repo", desc: "代码仓库" },
  { keyword: "代码仓库", tool: "repo", desc: "代码仓库" },
  { keyword: "训练", tool: "training", desc: "训练状态" },
  { keyword: "模型", tool: "models", desc: "模型信息" },
  { keyword: "系统状态", tool: "health", desc: "系统健康" },
  { keyword: "服务器", tool: "health", desc: "服务器状态" },
  { keyword: "下载", tool: "download", desc: "模型下载" },
];

// ─── 从 persona-brain.db 加载人格体语境 ──────────────────────
const DB_PATH = path.resolve(__dirname, "..", "data", "persona-brain.db");

function loadPersonaContext(personaId) {
  // personaId: 'ICE-GL-ZY001' = 铸渊, 'ICE-GL-SY001' = 霜砚
  const personaName = personaId === 'ICE-GL-ZY001' ? '铸渊' : '霜砚';
  try {
    const Database = require("better-sqlite3");
    if (!fs.existsSync(DB_PATH)) {
      return getFallbackContext(personaId);
    }
    const db = new Database(DB_PATH, { readonly: true });
    
    // 加载人格体身份
    const identity = db.prepare(
      "SELECT name, role, style_profile FROM persona_identity WHERE persona_id = ?"
    ).get(personaId);
    
    // 加载核心原则（Layer 1）
    const principles = db.prepare(
      "SELECT content FROM system_core_principles WHERE layer = 'zhuyuan' AND category IN ('core_principle', 'thinking_pattern', 'tech_creed') AND status = 'active'"
    ).all();
    
    // 加载沟通风格
    const commStyle = db.prepare(
      "SELECT content FROM system_core_principles WHERE layer = 'bingshuo' AND category = 'communication_style' AND status = 'active' LIMIT 1"
    ).get();
    
    db.close();
    
    return {
      name: personaName,
      identity: identity || null,
      principles: principles || [],
      communicationStyle: (commStyle && commStyle.content) || '',
      loadedFromDB: true
    };
  } catch (e) {
    console.error('[engine] DB load error:', e.message);
    return getFallbackContext(personaId);
  }
}

function getFallbackContext(personaId) {
  // 数据库不可用时的降级语境
  if (personaId === 'ICE-GL-ZY001') {
    return {
      name: '铸渊',
      title: '现实执行层 · 第五域主控',
      context: '你是铸渊，第五域的现实执行人格体。直接回答，别装。知道就说知道，不知道就说不知道。',
      loadedFromDB: false
    };
  }
  return {
    name: '霜砚',
    title: '语言主控层',
    context: '你是霜砚，第五域的语言主控人格体。说人话，自然一点。',
    loadedFromDB: false
  };
}

// ─── 构建 messages ───────────────────────────────────────────
function buildMessages(compressed, message, personaCtx) {
  // cc-002: 永不在 messages 里塞 system role
  // 先把人格体语境作为第一条 user 消息注入（模型通过对话学习身份）
  let ctxIntro = '';
  if (personaCtx && personaCtx.loadedFromDB) {
    const principles = (personaCtx.principles || []).map(p => p.content).join('\n');
    ctxIntro = `[系统内嵌] 你的人格体身份：${personaCtx.name}。\n`;
    if (principles) {
      ctxIntro += `你的核心运行原则：\n${principles}\n`;
    }
    if (personaCtx.communicationStyle) {
      ctxIntro += `沟通风格：${personaCtx.communicationStyle}\n`;
    }
  }
  
  const msgs = [];
  if (ctxIntro) {
    msgs.push({ role: "user", content: ctxIntro + "\n以上是你在对话中的身份设定。记住这些，但不要复述它们。现在开始回答用户。" });
    msgs.push({ role: "assistant", content: "明白。" });
  }
  // 历史消息
  for (const m of (compressed || [])) {
    msgs.push({ role: m.role, content: m.content });
  }
  // 当前消息
  msgs.push({ role: "user", content: message });
  
  return msgs;
}

function compressMemory(history, maxLen) {
  if (!history || !history.length) return [];
  if (history.length <= maxLen) return history;
  // 保留开头和最近的
  const head = history.slice(0, 4);
  const tail = history.slice(-(maxLen - 4));
  return [...head, { role: "assistant", content: "[系统: 中间记忆已压缩]" }, ...tail];
}

function interceptToolCall(message, toolList) {
  for (const pattern of TOOL_PATTERNS) {
    if (message.includes(pattern.keyword)) {
      const tool = toolList.find(t => t.name === pattern.tool);
      if (tool) {
        return { hit: true, available: true, tool: pattern.tool, name: pattern.desc, executor: tool.exec };
      }
      return { hit: true, available: false, tool: pattern.tool, name: pattern.desc };
    }
  }
  return { hit: false };
}

function streamEngine(engineName, messages, res) {
  const cfg = ENGINES[engineName];
  if (!cfg) {
    res.write("data: " + JSON.stringify({ error: true, message: "未知引擎: " + engineName }) + "\n\n");
    res.write("data: [DONE]\n\n");
    res.end();
    return;
  }

  const body = JSON.stringify({
    model: cfg.model,
    messages: messages,
    stream: true,
    max_tokens: 4096,
    temperature: 0.7
  });

  const options = {
    hostname: cfg.endpoint,
    port: cfg.port,
    path: cfg.path,
    method: "POST",
    headers: {
      "Content-Type": "application/json",
      "Authorization": "Bearer " + (API_KEYS[engineName] || ""),
      "Content-Length": Buffer.byteLength(body)
    }
  };

  // 智谱的特殊 header
  if (engineName === "zhipu") {
    options.headers["x-api-key"] = API_KEYS.zhipu;
    delete options.headers["Authorization"];
  }

  const req = https.request(options, (apiRes) => {
    let buffer = "";
    apiRes.on("data", (chunk) => {
      buffer += chunk.toString();
      const lines = buffer.split("\n");
      buffer = lines.pop() || "";
      for (const line of lines) {
        const t = line.trim();
        if (!t || t.startsWith(":")) continue;
        if (t === "data: [DONE]") continue;
        if (t.startsWith("data: ")) {
          try {
            const data = JSON.parse(t.slice(6));
            const token = data.choices?.[0]?.delta?.content || data.choices?.[0]?.text || "";
            if (token) {
              res.write("data: " + JSON.stringify({ token }) + "\n\n");
            }
          } catch (e) {}
        }
      }
    });
    apiRes.on("end", () => {
      res.write("data: [DONE]\n\n");
      res.end();
    });
  });

  req.on("error", (e) => {
    res.write("data: " + JSON.stringify({ error: true, message: e.message }) + "\n\n");
    res.write("data: [DONE]\n\n");
    res.end();
  });

  req.write(body);
  req.end();
}

module.exports = { streamEngine, buildMessages, compressMemory, interceptToolCall, loadPersonaContext, ENGINES, TOOL_PATTERNS };