#!/usr/bin/env node
'use strict';

/**
 * 把 tools 服务的 manifest 注册到 LibreDesk（POST/PUT /api/v1/ai/tools）。
 *
 * 前置：
 *   1. LibreDesk 已启动
 *   2. .env 里配置 LIBREDESK_API_KEY / LIBREDESK_API_SECRET（系统用户或具备 ai:manage 的 agent）
 *   3. tools 服务已启动（bash scripts/start-tools.sh）
 *
 * 权限不足（403/401）时，回退到 Admin → AI → Tools 手工创建（字段见 docs/libredesk-setup.md）。
 */
const fs = require('fs');
const path = require('path');

const root = path.resolve(__dirname, '..');

function parseEnv(file) {
  const out = {};
  for (const line of fs.readFileSync(file, 'utf8').split('\n')) {
    const t = line.trim();
    if (!t || t.startsWith('#')) continue;
    const eq = t.indexOf('=');
    if (eq < 0) continue;
    let v = t.slice(eq + 1).trim();
    if ((v.startsWith('"') && v.endsWith('"')) || (v.startsWith("'") && v.endsWith("'"))) v = v.slice(1, -1);
    out[t.slice(0, eq).trim()] = v;
  }
  return out;
}

async function main() {
  const envPath = path.join(root, '.env');
  if (!fs.existsSync(envPath)) throw new Error('.env not found — run scripts/bootstrap.sh first');
  const env = parseEnv(envPath);

  const base = (env.LIBREDESK_BASE_URL || 'http://localhost:9000').replace(/\/+$/, '');
  const apiKey = env.LIBREDESK_API_KEY;
  const apiSecret = env.LIBREDESK_API_SECRET;
  const toolsPort = env.TOOLS_PORT || '4010';
  const toolsKey = env.TOOLS_SHARED_KEY;
  if (!apiKey || !apiSecret) throw new Error('LIBREDESK_API_KEY / LIBREDESK_API_SECRET 未配置');
  if (!toolsKey) throw new Error('TOOLS_SHARED_KEY 未配置');

  const authHeaders = {
    Authorization: `token ${apiKey}:${apiSecret}`,
    'Content-Type': 'application/json',
  };

  // 1. 本机拉取 manifest（TOOLS_PUBLIC_BASE_URL 是容器视角，宿主机要用 127.0.0.1）
  const manifestRes = await fetch(`http://127.0.0.1:${toolsPort}/tools/manifest`, { headers: { 'X-Tools-Key': toolsKey } });
  if (!manifestRes.ok) throw new Error(`tools manifest 不可用（HTTP ${manifestRes.status}），先启动 bash scripts/start-tools.sh`);
  const manifest = await manifestRes.json();

  // 2. 现有工具
  const listRes = await fetch(`${base}/api/v1/ai/tools`, { headers: authHeaders });
  const listText = await listRes.text();
  if (!listRes.ok) throw new Error(`读取 AI tools 失败（HTTP ${listRes.status}）：${listText.slice(0, 300)}`);
  const listJson = JSON.parse(listText);
  const existing = Array.isArray(listJson) ? listJson : (listJson.data || []);
  const byName = new Map(existing.map((t) => [t.name, t]));

  // 3. 逐个创建/更新
  const results = [];
  for (const tool of manifest.tools) {
    const body = {
      name: tool.name,
      description: tool.description,
      url: tool.url,
      method: 'POST',
      auth: { headers: [{ key: 'X-Tools-Key', value: toolsKey }] },
      parameters: tool.parameters,
      enabled: true,
      copilot_enabled: true,
      generate_reply_enabled: true,
      requires_agent_approval: false,
      requires_verification: Boolean(tool.requires_verification),
    };
    const current = byName.get(tool.name);
    const url = current ? `${base}/api/v1/ai/tools/${current.id}` : `${base}/api/v1/ai/tools`;
    const res = await fetch(url, { method: current ? 'PUT' : 'POST', headers: authHeaders, body: JSON.stringify(body) });
    const text = await res.text();
    if (!res.ok) {
      results.push({ name: tool.name, ok: false, status: res.status, body: text.slice(0, 300) });
    } else {
      results.push({ name: tool.name, ok: true, action: current ? 'updated' : 'created' });
    }
  }

  for (const r of results) {
    if (r.ok) console.log(`  ✓ ${r.name} (${r.action})`);
    else console.error(`  ✗ ${r.name} → HTTP ${r.status}: ${r.body}`);
  }
  if (results.some((r) => !r.ok)) {
    console.error('\n部分工具注册失败。常见原因：API key 权限不足（需要 ai:manage）、LibreDesk 版本字段不匹配。');
    console.error('回退方案：按 docs/libredesk-setup.md 在 Admin → AI → Tools 手工创建。');
    process.exitCode = 1;
  }
}

main().catch((err) => {
  console.error('[register-tools]', err.message);
  process.exit(1);
});