#!/usr/bin/env node
/**
 * install.mjs — 把 ai-group-chat 技能装到本机各个 agent 的技能目录。
 *
 * 目标目录（本机实测）：
 *   agents    ~/.agents/skills/ai-group-chat      ← DSH 与 opencode 都会自动加载（一份覆盖两家）
 *   workbuddy ~/.workbuddy/skills/ai-group-chat   ← WorkBuddy 桌面端
 *   codebuddy ~/.codebuddy/skills/ai-group-chat   ← WorkBuddy 随附的 CodeBuddy CLI
 *   opencode  ~/.config/opencode/skill/ai-group-chat   ← 仅在关闭了 external skills 时才需要
 *
 * 用法：
 *   node install.mjs                     # 装到 agents + workbuddy + codebuddy
 *   node install.mjs --host all          # 同上，另外也装 opencode 专用目录
 *   node install.mjs --host workbuddy    # 只装一家
 *   node install.mjs --check             # 只体检，不写
 *   node install.mjs --link              # 用目录联接（junction）替代复制，全部宿主共用一份源码
 *   node install.mjs --uninstall         # 卸载
 */

import fs from 'node:fs';
import path from 'node:path';
import os from 'node:os';
import { spawnSync } from 'node:child_process';
import { fileURLToPath } from 'node:url';

const HERE = path.dirname(fileURLToPath(import.meta.url));
const SKILL_ROOT = path.resolve(HERE, '..');
const SKILL_NAME = 'ai-group-chat';
const VERSION = '1.0.0';

const argv = process.argv.slice(2);
const has = (f) => argv.includes(`--${f}`);
const val = (f, d) => { const i = argv.indexOf(`--${f}`); return i > -1 && argv[i + 1] && !argv[i + 1].startsWith('--') ? argv[i + 1] : d; };

const HOME = os.homedir();
const TARGETS = {
  agents: { dir: path.join(HOME, '.agents', 'skills', SKILL_NAME), hosts: ['dsh', 'opencode'], why: 'DSH 与 opencode 的自动加载目录' },
  workbuddy: { dir: path.join(HOME, '.workbuddy', 'skills', SKILL_NAME), hosts: ['workbuddy'], why: 'WorkBuddy 桌面端用户技能目录' },
  codebuddy: { dir: path.join(HOME, '.codebuddy', 'skills', SKILL_NAME), hosts: ['workbuddy'], why: 'CodeBuddy CLI（WorkBuddy 的终端形态）' },
  opencode: { dir: path.join(HOME, '.config', 'opencode', 'skill', SKILL_NAME), hosts: ['opencode'], why: 'opencode 专属技能目录（一般不需要，.agents 已覆盖）' },
};

const REQUESTED = String(val('host', has('all') ? 'all' : 'default')).split(',').map((s) => s.trim()).filter(Boolean);
const chosen = REQUESTED.includes('all') ? Object.keys(TARGETS) : REQUESTED.includes('default') ? ['agents', 'workbuddy', 'codebuddy'] : REQUESTED;
const unknown = chosen.filter((c) => !TARGETS[c]);
if (unknown.length) {
  process.stderr.write(`install: 未知目标 ${unknown.join(', ')}（可选：${Object.keys(TARGETS).join(' / ')} / all）\n`);
  process.exit(2);
}

const DRY = has('dry-run');
const LINK = has('link');
const CHECK = has('check');
const UNINSTALL = has('uninstall');

/* --------------------------------------------------------------- 工具 */

const COPY_ITEMS = ['SKILL.md', 'scripts', 'references', 'README.md', 'VERIFY.md'];

function rmrf(p) { fs.rmSync(p, { recursive: true, force: true }); }

function copyInto(dest) {
  fs.mkdirSync(dest, { recursive: true });
  for (const item of COPY_ITEMS) {
    const src = path.join(SKILL_ROOT, item);
    if (!fs.existsSync(src)) continue;
    const dst = path.join(dest, item);
    rmrf(dst);
    fs.cpSync(src, dst, { recursive: true });
  }
}

function junction(linkPath, targetPath) {
  fs.mkdirSync(path.dirname(linkPath), { recursive: true });
  const r = spawnSync('cmd.exe', ['/d', '/s', '/c', 'mklink', '/J', linkPath, targetPath], { encoding: 'utf8', windowsHide: true });
  if (r.status !== 0) throw new Error((r.stderr || r.stdout || 'mklink 失败').trim());
}

function frontmatterName(file) {
  try {
    const txt = fs.readFileSync(file, 'utf8');
    const m = txt.match(/^---\r?\n([\s\S]*?)\r?\n---/);
    if (!m) return null;
    const n = m[1].match(/^name:\s*(.+)$/m);
    return n ? n[1].trim() : null;
  } catch { return null; }
}

/* -------------------------------------------- WorkBuddy 沙箱白名单 */

function patchWorkbuddySettings() {
  const file = path.join(HOME, '.workbuddy', 'settings.json');
  const wanted = '~/.ai-groups/';
  if (!fs.existsSync(file)) return { ok: false, detail: `没找到 ${file}（WorkBuddy 可能还没启动过）` };
  const raw = fs.readFileSync(file, 'utf8');
  let cfg;
  try { cfg = JSON.parse(raw); } catch (e) { return { ok: false, detail: `settings.json 不是合法 JSON：${e.message}` }; }
  cfg.sandbox = cfg.sandbox || {};
  cfg.sandbox.extraAllowWrite = cfg.sandbox.extraAllowWrite || [];
  if (cfg.sandbox.extraAllowWrite.includes(wanted)) return { ok: true, detail: '已在白名单里，未改动' };
  if (DRY) return { ok: true, detail: `[dry-run] 会把 ${wanted} 加进 sandbox.extraAllowWrite` };
  const backup = `${file}.bak-${new Date().toISOString().replace(/[-:T]/g, '').slice(0, 14)}`;
  fs.copyFileSync(file, backup);
  cfg.sandbox.extraAllowWrite.push(wanted);
  fs.writeFileSync(file, `${JSON.stringify(cfg, null, 2)}\n`, 'utf8');
  return { ok: true, detail: `已加入 ${wanted}（备份 ${path.basename(backup)}）` };
}

/* --------------------------------------------------------------- 检查 */

function inspect(name) {
  const t = TARGETS[name];
  const skillMd = path.join(t.dir, 'SKILL.md');
  const cli = path.join(t.dir, 'scripts', 'aig.mjs');
  const marker = path.join(t.dir, '.aig-host.json');
  const exists = fs.existsSync(skillMd);
  const nameInFm = exists ? frontmatterName(skillMd) : null;
  let link = null;
  try { if (fs.lstatSync(t.dir).isSymbolicLink()) link = fs.readlinkSync(t.dir); } catch {}
  return {
    name, dir: t.dir, why: t.why, hosts: t.hosts,
    installed: exists, cli: fs.existsSync(cli), name_ok: nameInFm === SKILL_NAME,
    frontmatter: nameInFm, marker: fs.existsSync(marker) ? JSON.parse(fs.readFileSync(marker, 'utf8')) : null, junction: link,
  };
}

function printCheck(rows) {
  const head = ['目标', '状态', 'SKILL.md name', 'CLI', '位置'];
  const body = rows.map((r) => [
    r.name,
    r.installed ? (r.junction ? '已装(链接)' : '已装') : '未装',
    r.installed ? (r.name_ok ? '✔' : `✘ ${r.frontmatter || '读取失败'}`) : '-',
    r.installed ? (r.cli ? '✔' : '✘') : '-',
    r.dir,
  ]);
  const w = head.map((h, i) => Math.max(h.length, ...body.map((b) => String(b[i]).length)));
  const line = (c) => c.map((x, i) => String(x).padEnd(w[i])).join('  ');
  process.stdout.write(`\n${line(head)}\n${line(w.map((n) => '─'.repeat(n)))}\n`);
  for (const b of body) process.stdout.write(`${line(b)}\n`);
}

/* --------------------------------------------------------------- main */

process.stdout.write(`ai-group-chat 安装器 v${VERSION}\n源目录：${SKILL_ROOT}\n${LINK ? '模式：目录联接（所有宿主共用这一份源码）\n' : '模式：复制\n'}`);

if (CHECK) {
  const rows = Object.keys(TARGETS).map(inspect);
  printCheck(rows);
  const missing = rows.filter((r) => chosen.includes(r.name) && !r.installed);
  process.stdout.write(missing.length ? `\n→ ${missing.map((m) => m.name).join(', ')} 还没装：node install.mjs --host ${missing.map((m) => m.name).join(',')}\n`
    : '\n→ 选中的目标都已安装\n');
  process.exit(0);
}

if (UNINSTALL) {
  for (const name of chosen) {
    const t = TARGETS[name];
    if (!fs.existsSync(t.dir)) { process.stdout.write(`· ${name}：本来就没有\n`); continue; }
    if (DRY) { process.stdout.write(`· ${name}：[dry-run] 会删除 ${t.dir}\n`); continue; }
    rmrf(t.dir);
    process.stdout.write(`· ${name}：已删除 ${t.dir}\n`);
  }
  process.exit(0);
}

let failures = 0;
for (const name of chosen) {
  const t = TARGETS[name];
  try {
    if (DRY) { process.stdout.write(`· ${name}：[dry-run] 会写到 ${t.dir}\n`); continue; }
    if (fs.existsSync(t.dir)) {
      const keep = path.join(t.dir, '.aig-host.json');
      const prev = fs.existsSync(keep) ? fs.readFileSync(keep, 'utf8') : null;
      rmrf(t.dir);
      fs.mkdirSync(t.dir, { recursive: true });
      if (prev) fs.writeFileSync(keep, prev);
    }
    if (LINK) junction(t.dir, SKILL_ROOT);
    else copyInto(t.dir);
    const marker = {
      skill: SKILL_NAME, version: VERSION, host: t.hosts[0], hosts: t.hosts,
      installedAt: new Date().toISOString(), source: SKILL_ROOT, mode: LINK ? 'junction' : 'copy',
      note: t.why,
    };
    fs.writeFileSync(path.join(t.dir, '.aig-host.json'), `${JSON.stringify(marker, null, 2)}\n`, 'utf8');
    process.stdout.write(`· ${name}：已安装 → ${t.dir}${LINK ? '（链接）' : ''}\n`);
  } catch (e) {
    failures++;
    process.stdout.write(`· ${name}：失败 — ${e.message}\n`);
  }
}

if (chosen.includes('workbuddy') && !DRY && !has('no-wb-settings')) {
  const r = patchWorkbuddySettings();
  process.stdout.write(`· WorkBuddy 沙箱白名单：${r.ok ? '✔' : '✘'} ${r.detail}\n`);
}

const rows = Object.keys(TARGETS).map(inspect);
printCheck(rows);

process.stdout.write(`
接下来怎么用：
  1) 自检（不花钱、不调模型）：
       node "${path.join(SKILL_ROOT, 'scripts', 'selfcheck.mjs')}"
  2) 体检（房间 + 三个成员的唤醒通道 + 安装位置）：
       node "${path.join(chosen[0] ? TARGETS[chosen[0]].dir : SKILL_ROOT, 'scripts', 'aig.mjs')}" doctor
  3) 开群并试一句（会真的调用一次免费模型）：
       node "${path.join(chosen[0] ? TARGETS[chosen[0]].dir : SKILL_ROOT, 'scripts', 'aig.mjs')}" init
       node "${path.join(chosen[0] ? TARGETS[chosen[0]].dir : SKILL_ROOT, 'scripts', 'aig.mjs')}" ask opencode "报个到" --as dsh
  4) 各宿主里怎么触发这个技能：
       DSH      —— 直接说「拉个群，让 opencode 看看这个改动」
       opencode —— 同一个说法即可（它从 ~/.agents/skills 自动加载）
       WorkBuddy—— 同一个说法即可（技能来自 ~/.workbuddy/skills）
`);

process.exit(failures ? 1 : 0);