hunter-mate 0.1.2 → 0.1.4

package/dist/commands/doctor.js

@@ -1,51 +1,15 @@
-"use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v });
-}) : function(o, v) {
-    o["default"] = v;
-});
-var __importStar = (this && this.__importStar) || (function () {
-    var ownKeys = function(o) {
-        ownKeys = Object.getOwnPropertyNames || function (o) {
-            var ar = [];
-            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
-            return ar;
-        };
-        return ownKeys(o);
-    };
-    return function (mod) {
-        if (mod && mod.__esModule) return mod;
-        var result = {};
-        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
-        __setModuleDefault(result, mod);
-        return result;
-    };
-})();
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.doctorCommand = void 0;
-exports.runDoctor = runDoctor;
-const commander_1 = require("commander");
-const axios_1 = __importDefault(require("axios"));
-const config_1 = require("../config");
-const init_1 = require("./init");
-const server_1 = require("./server");
-const install_1 = require("./install");
-const extension_1 = require("./extension");
-const native_check_1 = require("./native-check");
+import { Command } from 'commander';
+import axios from 'axios';
+import * as fs from 'fs';
+import * as path from 'path';
+import { loadConfig } from '../config.js';
+import { runInit } from './init.js';
+import { runStart } from './server.js';
+import { runInstall } from './install.js';
+import { runExtensionCheck } from './extension.js';
+import { checkNativeModules } from '../native-check.js';
+import { getCliLogger } from '../logger.js';
+import { getToken } from '../keychain.js';
 const B = {
     reset: '\x1b[0m',
     bold: '\x1b[1m',
@@ -55,14 +19,40 @@ const B = {
     cyan: '\x1b[36m',
     magenta: '\x1b[35m',
 };
+/**
+ * 调用 server /doctor/run 统一入口。
+ * 失败时返回 null,调用方决定如何处理。
+ */
+export async function runDoctorChecks(opts) {
+    const config = loadConfig();
+    const token = await getToken().catch(() => undefined);
+    const checks = opts.checks && opts.checks.length ? opts.checks : undefined;
+    const timeout = opts.timeout ?? 60_000;
+    try {
+        const { data } = await axios.post(`${config.serverUrl}/doctor/run`, {
+            checks,
+            mode: opts.fix ? 'repair' : 'check',
+            dryRun: opts.dryRun === true,
+        }, {
+            timeout,
+            headers: token ? { Authorization: `UTOKEN ${token}` } : {},
+        });
+        return data;
+    }
+    catch (e) {
+        const message = e?.response?.data?.message || e.message || String(e);
+        getCliLogger().warn('doctor:checks:failed', { error: message, fix: opts.fix, checks });
+        return null;
+    }
+}
 /** 对应 ChromePilot doctor: 全链路健康检查 — 带 auto-heal, 无感启动 server/安装扩展 */
-async function runDoctor(opts) {
+export async function runDoctor(opts) {
     const checks = [];
     const hints = [];
     const healed = [];
     const actions = [];
     const autoHeal = opts.autoHeal !== false; // default true
-    const interactive = opts.interactive !== false; // default true
+    const fix = opts.fix === true;
     // ── Banner (human 模式) ──
     if (!opts.json) {
         console.log(`\n${B.bold}${B.cyan}╭─────────────────────────────────────────────╮${B.reset}`);
@@ -72,23 +62,21 @@ async function runDoctor(opts) {
     // ── 1) 目录 & token ──
     let initResult = null;
     try {
-        initResult = (0, init_1.runInit)();
+        initResult = runInit();
         checks.push({ key: 'home_dir', status: 'ok', detail: initResult.dir });
     }
     catch (e) {
         checks.push({ key: 'home_dir', status: 'fail', detail: e.message });
         if (autoHeal && !opts.json) {
             console.log(`${B.yellow}⚠  基础目录缺失, hunter-mate 正在自动创建...${B.reset}`);
-            initResult = (0, init_1.runInit)(true);
+            initResult = runInit(true);
             checks.push({ key: 'home_dir_auto_init', status: 'ok', detail: '已自动创建 ~/.hunter-mate 目录结构' });
             healed.push('init_dirs');
         }
     }
     if (!initResult)
-        initResult = (0, init_1.runInit)();
+        initResult = runInit();
     // ── 2) 扩展安装检测 + 自动安装 ──
-    const fs = require('fs');
-    const path = require('path');
     const manifestPath = path.join(initResult.extensionDir, 'manifest.json');
     let extInstalled = fs.existsSync(manifestPath);
     if (extInstalled) {
@@ -100,7 +88,7 @@ async function runDoctor(opts) {
             if (!opts.json) {
                 console.log(`${B.yellow}⚠  扩展尚未安装, hunter-mate 正在自动执行 install...${B.reset}`);
             }
-            const r = (0, install_1.runInstall)(initResult.extensionDir);
+            const r = runInstall(initResult.extensionDir);
             if (r.source) {
                 checks.push({ key: 'extension_auto_install', status: 'ok', detail: `已从 ${r.source} 复制 ${r.copied} 个文件到 ${r.target}` });
                 healed.push('install_extension');
@@ -118,7 +106,7 @@ async function runDoctor(opts) {
     // ── 2.5) 原生模块 ABI 自检(server 启动前置闸) ──
     // better-sqlite3 按 node ABI 绑定;不匹配时 server 起来即崩且报错被后台 spawn 吞掉。
     // 放在自动启动之前:fail 就不浪费 8s 去启动必然崩的 server。
-    const native = (0, native_check_1.checkNativeModules)();
+    const native = checkNativeModules();
     for (const m of native.modules) {
         if (m.status === 'skip') {
             checks.push({ key: `native_${m.module}`, status: 'skip', detail: m.detail });
@@ -131,7 +119,7 @@ async function runDoctor(opts) {
         });
     }
     // ── 3) Server 自动启动(原生模块 fail 时跳过) ──
-    const config = (0, config_1.loadConfig)();
+    const config = loadConfig();
     let serverUp = false;
     if (!native.ok) {
         checks.push({ key: 'server', status: 'skip', detail: '原生模块检查未通过,跳过自动启动(修好 ABI 后重跑 doctor)' });
@@ -139,7 +127,7 @@ async function runDoctor(opts) {
     }
     else {
         try {
-            const r = await axios_1.default.get(`${config.serverUrl}/health`, { timeout: 3000 });
+            const r = await axios.get(`${config.serverUrl}/health`, { timeout: 3000 });
             serverUp = true;
             checks.push({ key: 'server', status: 'ok', detail: `${config.serverUrl} (HTTP ${r.status})` });
         }
@@ -149,12 +137,12 @@ async function runDoctor(opts) {
                 if (!opts.json) {
                     console.log(`${B.yellow}⚠  Server 未运行, hunter-mate 正在后台自动启动...${B.reset}`);
                 }
-                const r = await (0, server_1.runStart)({ foreground: false });
+                const r = await runStart({ foreground: false });
                 // 等待 server 真正就绪(最多 8s)
                 for (let i = 0; i < 16; i++) {
                     await new Promise((resolve) => setTimeout(resolve, 500));
                     try {
-                        await axios_1.default.get(`${config.serverUrl}/health`, { timeout: 2000 });
+                        await axios.get(`${config.serverUrl}/health`, { timeout: 2000 });
                         serverUp = true;
                         break;
                     }
@@ -180,11 +168,31 @@ async function runDoctor(opts) {
             }
         }
     } // native.ok
-    // ── 4) 扩展 WebSocket 连通 ──
+    // ── 4) 数据库 schema 健康 ──
+    if (serverUp) {
+        try {
+            const r = await axios.get(`${config.serverUrl}/health/db`, { timeout: 3000 });
+            checks.push({
+                key: 'db_schema',
+                status: r.data.status === 'ok' ? 'ok' : 'warn',
+                detail: JSON.stringify(r.data),
+            });
+            if (r.data.status !== 'ok') {
+                hints.push('数据库 schema 异常,server 已尝试自动修复,请检查 server.log');
+            }
+        }
+        catch (e) {
+            checks.push({ key: 'db_schema', status: 'warn', detail: `无法检查 schema: ${e.message}` });
+        }
+    }
+    else {
+        checks.push({ key: 'db_schema', status: 'skip', detail: 'server 未运行,跳过 schema 检查' });
+    }
+    // ── 5) 扩展 WebSocket 连通 ──
     let extConnected = false;
     try {
         if (serverUp) {
-            await (0, extension_1.runExtensionCheck)(5000);
+            await runExtensionCheck(5000);
             extConnected = true;
             checks.push({ key: 'extension_ws', status: 'ok', detail: '扩展已连接, WebSocket 通道就绪' });
         }
@@ -198,12 +206,12 @@ async function runDoctor(opts) {
         // autoHeal: 尝试打开扩展所在目录 + chrome://extensions
         if (autoHeal) {
             try {
-                const { default: open } = await Promise.resolve().then(() => __importStar(require('open')));
+                const { default: open } = await import('open');
                 if (extInstalled) {
                     if (!opts.json) {
                         console.log(`${B.magenta}→  hunter-mate 正在为你打开扩展所在目录...${B.reset}`);
                     }
-                    open(initResult.extensionDir).catch(() => { });
+                    open(path.dirname(initResult.extensionDir)).catch(() => { });
                     healed.push('open_ext_dir');
                 }
                 if (!opts.json) {
@@ -217,7 +225,7 @@ async function runDoctor(opts) {
     }
     // ── 5) 登录态(不强制) ──
     try {
-        const { getToken } = await Promise.resolve().then(() => __importStar(require('../keychain')));
+        const { getToken } = await import('../keychain.js');
         const token = await getToken();
         checks.push({
             key: 'auth',
@@ -230,10 +238,45 @@ async function runDoctor(opts) {
     catch (e) {
         checks.push({ key: 'auth', status: 'skip', detail: `跳过(keychain 读取失败): ${e.message}` });
     }
+    // ── 6) 数据面 doctor checks ──
+    let data = null;
+    if (serverUp) {
+        if (!opts.json) {
+            console.log(`\n${B.bold}${B.cyan}→  正在运行数据面检查${fix ? ' (repair 模式)' : ' (只读 check 模式)'}...${B.reset}`);
+        }
+        data = await runDoctorChecks({
+            fix,
+            checks: opts.checks,
+            json: opts.json,
+            timeout: 120_000,
+        });
+        if (data) {
+            for (const c of data.checks) {
+                const status = c.ok ? 'ok' : c.severity === 'error' ? 'fail' : 'warn';
+                checks.push({
+                    key: `data_${c.id}`,
+                    status,
+                    detail: c.message || c.description,
+                });
+            }
+        }
+        else {
+            checks.push({ key: 'data_doctor', status: 'warn', detail: '无法连接 /doctor/run (server 可能尚未就绪)' });
+        }
+    }
+    else {
+        checks.push({ key: 'data_doctor', status: 'skip', detail: 'server 未运行,跳过数据面检查' });
+    }
     // ── Summary ──
     const anyFail = checks.some((c) => c.status === 'fail');
     const anyWarn = checks.some((c) => c.status === 'warn');
     const summary = anyFail ? 'fail' : anyWarn ? 'warn' : 'pass';
+    getCliLogger().info('doctor:done', {
+        summary,
+        failedCount: checks.filter((c) => c.status === 'fail').length,
+        warnCount: checks.filter((c) => c.status === 'warn').length,
+        healed: healed.length,
+    });
     // ── human output ──
     if (!opts.json) {
         if (healed.length) {
@@ -260,7 +303,7 @@ async function runDoctor(opts) {
             console.log(`\n${B.bold}${B.red}✗  环境未就绪, 请按上方引导完成修复后重试。${B.reset}\n`);
         }
     }
-    return { summary, checks, hints, healed, actions };
+    return { summary, checks, hints, healed, actions, data };
 }
 /** 友好中文引导 — 当环境不完备时给用户清晰的操作步骤 */
 function showFriendlyGuide(extInstalled, extConnected, serverUp, healed, extDir) {
@@ -274,21 +317,24 @@ function showFriendlyGuide(extInstalled, extConnected, serverUp, healed, extDir)
         console.log(`   hunter-mate 已尝试自动启动。若持续失败请手动执行:`);
         console.log(`   ${B.bold}hunter-mate start -f${B.reset}  (前台模式可看日志)\n`);
     }
+    const extParentDir = path.dirname(extDir);
+    const extFolderName = path.basename(extDir);
     if (!extInstalled) {
         console.log(`${B.bold}${B.yellow}⚠  Hunter Mate Chrome 扩展尚未安装${B.reset}`);
         console.log(`   步骤: hunter-mate 已自动把扩展文件复制到:`);
         console.log(`   ${B.cyan}${extDir}${B.reset}`);
+        console.log(`   并已打开扩展所在目录: ${B.cyan}${extParentDir}${B.reset}`);
         console.log(`   ${B.bold}接下来需要你手动在 Chrome 中加载(Chrome 安全策略限制命令行自动化):${B.reset}`);
         console.log(`   ${B.magenta}① 在 Chrome 地址栏输入:${B.reset}  chrome://extensions`);
         console.log(`   ${B.magenta}② 打开右上角「开发者模式」开关${B.reset}`);
-        console.log(`   ${B.magenta}③ 点击「加载已解压的扩展程序」${B.reset}`);
-        console.log(`   ${B.magenta}④ 选择上面的目录:${B.reset}  ${B.cyan}${extDir}${B.reset}`);
-        console.log(`   ${B.magenta}⑤ Chrome 工具栏应出现 Hunter Mate 图标${B.reset}\n`);
+        console.log(`   ${B.magenta}③ 把 ${B.bold}${extFolderName}${B.reset} 文件夹拖到 chrome://extensions 页面里${B.reset}`);
+        console.log(`      (也可以点击「加载已解压的扩展程序」后选择 ${B.cyan}${extDir}${B.reset})`);
+        console.log(`   ${B.magenta}④ Chrome 工具栏应出现 Hunter Mate 图标,侧栏显示「已连接」${B.reset}\n`);
     }
     if (extInstalled && !extConnected) {
         console.log(`${B.bold}${B.yellow}⚠  扩展已安装但尚未连接到 Server${B.reset}`);
         console.log(`   hunter-mate 已自动帮你打开了:`);
-        console.log(`   · 扩展所在目录: ${B.cyan}${extDir}${B.reset}`);
+        console.log(`   · 扩展所在目录: ${B.cyan}${extParentDir}${B.reset}`);
         console.log(`   · Chrome 扩展管理页: ${B.cyan}chrome://extensions${B.reset}`);
         console.log(`   ${B.bold}请确认以下几点:${B.reset}`);
         console.log(`   ${B.magenta}① 扩展已在 chrome://extensions 中「已开启」${B.reset} (灰色开关变蓝色)`);
@@ -312,21 +358,115 @@ function showFriendlyGuide(extInstalled, extConnected, serverUp, healed, extDir)
     }
     console.log(`${B.bold}${B.cyan}━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━${B.reset}\n`);
 }
-exports.doctorCommand = new commander_1.Command('doctor')
+export const doctorCommand = new Command('doctor')
     .description('全链路自检 + 自动修复(对应 chromepilot doctor): 自动启动 server、安装扩展、打开目录引导')
     .option('--json', '仅以 JSON 输出检查结果与建议(跳过引导文案)')
-    .option('--open-extension-folder', '扩展未连接时自动打开扩展所在文件夹')
     .option('--no-auto-heal', '禁用自动修复(仅检查并给出建议, 不自动启动/安装)')
     .option('--no-interactive', '无交互模式(不打印友好引导, 仅输出结构化结果)')
+    .option('--fix', '执行可修复的数据检查(默认只读 check)')
+    .option('--checks <ids>', '只跑指定检查,逗号分隔(如 linkedin-dupes)', (val) => val.split(',').map((s) => s.trim()).filter(Boolean))
+    .option('--list', '列出 server 上注册的数据检查项')
     .action(async (opts) => {
+    if (opts.list) {
+        const config = loadConfig();
+        try {
+            const { data } = await axios.get(`${config.serverUrl}/doctor/checks`, { timeout: 10_000 });
+            if (opts.json) {
+                console.log(JSON.stringify(data, null, 2));
+            }
+            else {
+                console.log(`\n${B.bold}${B.cyan}已注册 Doctor 检查项:${B.reset}`);
+                for (const c of data) {
+                    const fixTag = c.fixable ? `${B.green}[可修复]${B.reset}` : `${B.yellow}[只读]${B.reset}`;
+                    console.log(`  ${fixTag} ${B.bold}${c.id}${B.reset} (${c.severity})`);
+                    console.log(`     ${c.name}`);
+                    console.log(`     ${c.description}`);
+                }
+            }
+            process.exit(0);
+        }
+        catch (e) {
+            const message = e?.response?.data?.message || e.message || String(e);
+            getCliLogger().error('doctor:list failed', { message });
+            if (opts.json) {
+                console.log(JSON.stringify({ error: message }, null, 2));
+            }
+            else {
+                console.log(`${B.red}获取检查列表失败:${B.reset} ${message}`);
+            }
+            process.exit(1);
+        }
+        return;
+    }
     const r = await runDoctor({
         json: opts.json,
-        openExtensionFolder: opts.openExtensionFolder,
         autoHeal: opts.autoHeal !== false,
         interactive: opts.interactive !== false,
+        fix: opts.fix === true,
+        checks: opts.checks,
     });
     if (opts.json) {
         console.log(JSON.stringify(r, null, 2));
     }
     process.exit(r.summary === 'fail' ? 1 : 0);
 });
+export const repairLinkedInDupesCommand = new Command('repair-linkedin-dupes')
+    .description('修复 LinkedIn /in/<slug>/details/... 子页产生的重复 talent 数据')
+    .option('--dry-run', '仅预览,不真正合并/删除')
+    .option('--json', '仅输出 JSON 结果')
+    .action(async (opts) => {
+    const data = await runDoctorChecks({
+        fix: true,
+        checks: ['linkedin-dupes'],
+        dryRun: opts.dryRun === true,
+        json: opts.json,
+        timeout: 120_000,
+    });
+    if (!data) {
+        const message = '无法连接 server /doctor/run, 请确认 server 已启动';
+        getCliLogger().error('repair-linkedin-dupes failed', { message });
+        if (opts.json) {
+            console.log(JSON.stringify({ error: message }, null, 2));
+        }
+        else {
+            console.log(`修复失败: ${message}`);
+        }
+        process.exit(1);
+    }
+    const c = data.checks.find((x) => x.id === 'linkedin-dupes');
+    const details = c?.details || {};
+    const compat = {
+        dryRun: data.dryRun,
+        totalCandidates: details.totalCandidates ?? 0,
+        duplicateGroups: details.duplicateGroups ?? 0,
+        repairedCount: data.repaired?.length ?? 0,
+        merged: data.repaired ?? [],
+        summary: data.summary,
+    };
+    if (opts.json) {
+        console.log(JSON.stringify({ ...compat, raw: data }, null, 2));
+    }
+    else {
+        const B = {
+            reset: '\x1b[0m',
+            bold: '\x1b[1m',
+            green: '\x1b[32m',
+            yellow: '\x1b[33m',
+            cyan: '\x1b[36m',
+        };
+        console.log(`\n${B.bold}${B.cyan}LinkedIn 重复 talent 修复结果${B.reset}`);
+        console.log(`  模式: ${data.dryRun ? B.yellow + 'dry-run' + B.reset : B.green + '正式修复' + B.reset}`);
+        console.log(`  LinkedIn talent 总数: ${compat.totalCandidates}`);
+        console.log(`  归一组数: ${compat.duplicateGroups}`);
+        console.log(`  需合并组数: ${compat.repairedCount}`);
+        if (compat.merged.length) {
+            console.log(`\n${B.bold}合并详情:${B.reset}`);
+            for (const m of compat.merged) {
+                console.log(`  • master #${m.masterId} ← ${m.removedIds.map((id) => `#${id}`).join(', ')}`);
+                console.log(`    ${m.sourceUrl}`);
+            }
+        }
+    }
+    process.exit(0);
+});
+doctorCommand.addCommand(repairLinkedInDupesCommand);