// claudedesk.js —— 把中转分组接入 Claude 桌面 App（图形客户端），免登录直用中转。
// 关键：走 App 自己的「已保存配置库」(configLibrary)，不是托管偏好(defaults/.mobileconfig)。
//   托管偏好那条路是企业 MDM 通道，会强制 SSO 登录，个人账号卡登录页 —— 早期踩过坑，已弃用。
// 做法对照 cc-switch 源码 src-tauri/src/claude_desktop_config.rs（farion1231/cc-switch）核实。
// 详见 memory/claude-desktop-3p-config.md。
"use strict";
const fs = require("fs");
const os = require("os");
const path = require("path");
const router = require("./router");

const PLATFORM = process.platform; // 'darwin' | 'win32' | ...
const CONFIG_FILE = "claude_desktop_config.json";
const CONFIG_LIBRARY_DIR = "configLibrary";
const PROFILE_ID = "00000000-0000-4000-8000-000000157210";
const PROFILE_NAME = "TokenDun";
// 备份根：复用现有 ~/.codex/backups_state 约定
const BACKUP_DIR = path.join(os.homedir(), ".codex", "backups_state", "claudedesk", "official");

// —— 平台路径 ——
// 返回 { normalConfig, threepConfig, libDir, profilePath, metaPath } 或 null（不支持）
function platformPaths() {
  if (PLATFORM === "darwin") {
    const appSupport = path.join(os.homedir(), "Library", "Application Support");
    return dirsToPaths(path.join(appSupport, "Claude"), path.join(appSupport, "Claude-3p"));
  }
  if (PLATFORM === "win32") {
    const localAppData = process.env.LOCALAPPDATA || path.join(os.homedir(), "AppData", "Local");
    return dirsToPaths(pickWinDir(localAppData, "Claude"), pickWinDir(localAppData, "Claude-3p"));
  }
  return null;
}
// Windows 下 App 目录名可能带后缀，先精确后模糊匹配
function pickWinDir(base, exact) {
  const p = path.join(base, exact);
  if (fs.existsSync(p)) return p;
  try {
    const hit = fs.readdirSync(base).find((n) => {
      const full = path.join(base, n);
      const isDir = (() => { try { return fs.statSync(full).isDirectory(); } catch { return false; } })();
      if (!isDir) return false;
      return exact === "Claude-3p" ? /claude.*3p/i.test(n) : (/claude/i.test(n) && !/3p/i.test(n));
    });
    if (hit) return path.join(base, hit);
  } catch {}
  return p;
}
function dirsToPaths(normalDir, threepDir) {
  const libDir = path.join(threepDir, CONFIG_LIBRARY_DIR);
  return {
    normalConfig: path.join(normalDir, CONFIG_FILE),
    threepConfig: path.join(threepDir, CONFIG_FILE),
    libDir,
    profilePath: path.join(libDir, `${PROFILE_ID}.json`),
    metaPath: path.join(libDir, "_meta.json"),
  };
}

// —— JSON 读写小工具 ——
const readJson = (p) => { try { return JSON.parse(fs.readFileSync(p, "utf-8")); } catch { return null; } };
const readJsonOrEmpty = (p) => { const v = readJson(p); return v && typeof v === "object" && !Array.isArray(v) ? v : {}; };
function writeJson(p, obj) {
  fs.mkdirSync(path.dirname(p), { recursive: true });
  fs.writeFileSync(p, JSON.stringify(obj, null, 2) + "\n");
}
const isSupported = () => PLATFORM === "darwin" || PLATFORM === "win32";

// —— gateway profile 本体（写进 configLibrary/<id>.json）——
function buildGatewayProfile(group) {
  const oneM = new Set(Array.isArray(group.oneMillionModels) ? group.oneMillionModels : []);
  const profile = {
    coworkEgressAllowedHosts: ["*"],
    disableDeploymentModeChooser: true,
    inferenceGatewayApiKey: group.key || "",
    inferenceGatewayAuthScheme: "bearer",
    inferenceGatewayBaseUrl: String(group.baseUrl || "").replace(/\/+$/, ""),
    inferenceProvider: "gateway",
  };
  if (group.model) {
    const item = { id: group.model };
    if (oneM.has(group.model)) item.supports1m = true; // 1M 长上下文标记（不是 beta 头）
    profile.inferenceModels = [item];
  }
  return profile;
}

// —— deploymentMode 写入（就地改，不动其它字段，尤其保留官方账号绑定 preferences）——
function writeDeploymentMode(p, mode) {
  const v = readJsonOrEmpty(p);
  v.deploymentMode = mode;
  writeJson(p, v);
}

// —— _meta.json：登记/清除本 profile ——
function writeMeta(metaPath, applied) {
  const v = readJsonOrEmpty(metaPath);
  let entries = Array.isArray(v.entries) ? v.entries : [];
  entries = entries.filter((e) => !(e && e.id === PROFILE_ID)); // 去重
  if (applied) {
    entries.push({ id: PROFILE_ID, name: PROFILE_NAME });
    v.appliedId = PROFILE_ID;
  } else if (v.appliedId === PROFILE_ID) {
    // 取消：若当前 applied 指向本 profile，切到剩余任一或清空
    const next = entries.find((e) => e && e.id);
    if (next) v.appliedId = next.id; else delete v.appliedId;
  }
  v.entries = entries;
  writeJson(metaPath, v);
}

// ============================================================================
//  备份 / 还原（防「回不去官方登录」）—— 幂等 write-once 快照那 4 个文件
// ============================================================================
const SNAP_FILES = (p) => [
  ["normalConfig", p.normalConfig],
  ["threepConfig", p.threepConfig],
  ["profile", p.profilePath],
  ["meta", p.metaPath],
];

function snapshotOfficial() {
  fs.mkdirSync(BACKUP_DIR, { recursive: true });
  const marker = path.join(BACKUP_DIR, "done.json");
  if (fs.existsSync(marker)) return { firstTime: false, dir: BACKUP_DIR };
  const p = platformPaths();
  const meta = { ts: Date.now(), platform: PLATFORM, files: {} };
  if (p) {
    for (const [key, src] of SNAP_FILES(p)) {
      const dst = path.join(BACKUP_DIR, key + ".json");
      if (fs.existsSync(src)) {
        try { fs.copyFileSync(src, dst); meta.files[key] = { existed: true }; try { fs.chmodSync(dst, 0o600); } catch {} }
        catch { meta.files[key] = { existed: true, copyFailed: true }; }
      } else {
        meta.files[key] = { existed: false }; // 记录原本不存在 → 还原时删除
      }
    }
  }
  fs.writeFileSync(marker, JSON.stringify(meta, null, 2));
  return { firstTime: true, dir: BACKUP_DIR };
}

// 还原官方：把中转配置抹掉，deploymentMode 回 1p，App 恢复官方账号登录。
// 优先用首份快照精确复原；无快照则按规则清理（删 profile、meta 清 entry、config 回 1p）。
function restoreOfficial() {
  if (!isSupported()) return { ok: false, error: "当前系统不支持（仅 macOS / Windows）" };
  const p = platformPaths();
  if (!p) return { ok: false, error: "无法定位 Claude 桌面 App 配置目录" };
  const marker = path.join(BACKUP_DIR, "done.json");
  const meta = readJson(marker);

  try {
    if (meta && meta.files) {
      // 精确复原：有备份写回，标记「原本不存在」的删掉
      for (const [key, src] of SNAP_FILES(p)) {
        const info = meta.files[key] || {};
        const bak = path.join(BACKUP_DIR, key + ".json");
        if (info.existed && fs.existsSync(bak)) {
          fs.mkdirSync(path.dirname(src), { recursive: true });
          fs.copyFileSync(bak, src);
        } else if (!info.existed && fs.existsSync(src)) {
          try { fs.unlinkSync(src); } catch {}
        }
      }
    } else {
      // 无备份：按规则清理回官方
      writeDeploymentMode(p.normalConfig, "1p");
      writeDeploymentMode(p.threepConfig, "1p");
      if (fs.existsSync(p.profilePath)) { try { fs.unlinkSync(p.profilePath); } catch {} }
      writeMeta(p.metaPath, null);
    }
  } catch (e) {
    return { ok: false, error: "还原失败：" + (e.message || e) };
  }
  return { ok: true, message: "已还原 Claude 官方配置，请重启 Claude 桌面 App（会回到账号登录）" };
}

// ============================================================================
//  一键接入（写 configLibrary，免登录）
// ============================================================================
function applyGroup(groupId) {
  if (!isSupported()) return { ok: false, error: "当前系统暂不支持（仅 macOS / Windows）" };
  const g = router.groupById(groupId);
  if (!g) return { ok: false, error: "未找到该分组" };
  if (g.isDefault) return { ok: false, error: "默认直连分组不能接入" };
  if (!g.baseUrl || !g.key) return { ok: false, error: "该分组缺少 API 地址或密钥" };
  if ((g.mode || "responses") !== "claude") {
    return { ok: false, error: "Claude 桌面 App 只接受 Anthropic 协议（/v1/messages）分组，请把该分组的接口类型改成 Claude Code" };
  }
  const p = platformPaths();
  if (!p) return { ok: false, error: "无法定位 Claude 桌面 App 配置目录（未安装？）" };

  // 写前强制备份官方现状（幂等 write-once）
  let backup;
  try { backup = snapshotOfficial(); } catch (e) { return { ok: false, error: "备份官方配置失败：" + (e.message || e) }; }

  // 失败回滚：先记住本次要写的 4 个文件的当前内容
  const rollback = SNAP_FILES(p).map(([k, f]) => ({ f, content: fs.existsSync(f) ? fs.readFileSync(f) : null }));
  try {
    const profile = buildGatewayProfile(g);
    writeDeploymentMode(p.normalConfig, "3p");
    writeDeploymentMode(p.threepConfig, "3p");
    writeJson(p.profilePath, profile);
    writeMeta(p.metaPath, true);
  } catch (e) {
    // 回滚
    for (const r of rollback) {
      try { if (r.content === null) { if (fs.existsSync(r.f)) fs.unlinkSync(r.f); } else fs.writeFileSync(r.f, r.content); } catch {}
    }
    return { ok: false, error: "写入 Claude 桌面配置失败：" + (e.message || e) };
  }
  const has1m = Array.isArray(g.oneMillionModels) && g.model && g.oneMillionModels.includes(g.model);
  return {
    ok: true,
    firstTimeBackup: backup.firstTime,
    has1m: !!has1m,
    message: "已接入 Claude 桌面 App，请重启 App 生效（无需登录，直接用中转）。官方配置已备份，可随时一键还原。",
  };
}

// 导出 profile JSON（供高级用户手动放置 / 备份）
function exportConfig(groupId, format) {
  const g = router.groupById(groupId);
  if (!g) return { ok: false, error: "未找到该分组" };
  if (!g.baseUrl || !g.key) return { ok: false, error: "该分组缺少 API 地址或密钥" };
  const profile = buildGatewayProfile(g);
  const content = JSON.stringify(profile, null, 2) + "\n";
  return { ok: true, content, filename: `${PROFILE_ID}.json`, ext: "json" };
}

// 复制片段（configLibrary profile 内容）
function snippet(groupId) {
  const r = exportConfig(groupId, "json");
  return r.ok ? { ok: true, text: r.content } : r;
}

function status() {
  const marker = path.join(BACKUP_DIR, "done.json");
  const meta = readJson(marker);
  const p = platformPaths();
  let applied = false;
  if (p) {
    const cfg = readJson(p.threepConfig) || readJson(p.normalConfig);
    const m = readJson(p.metaPath);
    applied = !!(cfg && cfg.deploymentMode === "3p" && m && m.appliedId === PROFILE_ID);
  }
  return {
    supported: isSupported(),
    platform: PLATFORM,
    hasBackup: !!meta,
    backupTs: meta ? meta.ts : 0,
    currentApplied: applied,
  };
}

module.exports = {
  status, applyGroup, restoreOfficial, exportConfig, snippet,
  // 供测试
  buildGatewayProfile, platformPaths, snapshotOfficial, PROFILE_ID,
};