@szc-ft/mcp-szcd-client 0.30.0 → 0.32.0

package/qwen-extension/skills/local-browser-test/decrypt-creds.mjs

@@ -0,0 +1,210 @@
+#!/usr/bin/env node
+/**
+ * 浏览器凭据解密工具
+ *
+ * 自动检测 Edge/Chrome 的 user-data-dir 配置并解密保存的密码。
+ * 解密结果缓存到 ~/.szcd-mcp/deps/decrypted-creds.json，避免重复解密。
+ *
+ * 用法:
+ *   NODE_PATH=~/.szcd-mcp/deps/node_modules node decrypt-creds.mjs --url aicityos
+ *   NODE_PATH=~/.szcd-mcp/deps/node_modules node decrypt-creds.mjs --json
+ *   NODE_PATH=~/.szcd-mcp/deps/node_modules node decrypt-creds.mjs --refresh   # 强制刷新缓存
+ *
+ * 前置条件:
+ *   - 浏览器以 --remote-debugging-port 和 --user-data-dir 启动
+ *   - ~/.szcd-mcp/deps/ 已安装 sql.js（共享缓存，跨 IDE 共享）
+ */
+import { getFullBrowserConfig } from './lib/cdp-browser-config.js';
+import { decryptAllPasswords } from './lib/decrypt-passwords.js';
+import { existsSync, readFileSync, writeFileSync, statSync } from 'fs';
+import { resolve, dirname, join } from 'path';
+import { homedir } from 'os';
+import { fileURLToPath } from 'url';
+
+const __dirname = dirname(fileURLToPath(import.meta.url));
+
+// 缓存文件路径（共享缓存目录）
+const CACHE_DIR = join(homedir(), '.szcd-mcp', 'deps');
+const CACHE_FILE = join(CACHE_DIR, 'decrypted-creds.json');
+
+// 解析命令行参数
+const args = process.argv.slice(2);
+const options = {
+  url: null,
+  json: false,
+  refresh: false,      // 强制刷新，忽略缓存
+  cdpUrl: 'http://localhost:9222',
+  userDataDir: null,
+  verbose: false,
+};
+
+for (let i = 0; i < args.length; i++) {
+  switch (args[i]) {
+    case '--url': options.url = args[++i]; break;
+    case '--json': options.json = true; break;
+    case '--refresh': case '-r': options.refresh = true; break;
+    case '--cdp': options.cdpUrl = args[++i]; break;
+    case '--user-data-dir': options.userDataDir = args[++i]; break;
+    case '--verbose': case '-v': options.verbose = true; break;
+    case '--help': case '-h':
+      console.log(`浏览器凭据解密工具（带缓存）
+
+用法:
+  node decrypt-creds.mjs [选项]
+
+选项:
+  --url <pattern>       按 URL 过滤（如 aicityos）
+  --json                 JSON 格式输出
+  --refresh, -r          强制刷新，忽略缓存重新解密
+  --cdp <url>            CDP 地址（默认 http://localhost:9222）
+  --user-data-dir <path> 手动指定 user-data-dir 路径
+  --verbose, -v          输出详细信息
+  --help, -h             显示帮助
+
+缓存说明:
+  解密结果自动缓存到 ${CACHE_FILE}
+  缓存以 userDataDir + filter 为 key，监测 Login Data 文件修改时间判断是否过期
+`);
+      process.exit(0);
+  }
+}
+
+/**
+ * 尝试从缓存读取凭据
+ * @returns {Array|null} 缓存的凭据数组，null 表示缓存无效
+ */
+function tryReadCache(userDataDir, filter) {
+  if (options.refresh) {
+    if (options.verbose) console.error('[decrypt-creds] --refresh 强制刷新，跳过缓存');
+    return null;
+  }
+  if (!existsSync(CACHE_FILE)) {
+    if (options.verbose) console.error('[decrypt-creds] 缓存文件不存在');
+    return null;
+  }
+
+  try {
+    const cache = JSON.parse(readFileSync(CACHE_FILE, 'utf8'));
+    const cacheKey = `${userDataDir}::${filter}`;
+
+    if (!cache[cacheKey]) {
+      if (options.verbose) console.error('[decrypt-creds] 缓存 key 不匹配');
+      return null;
+    }
+
+    // 检查缓存有效性：Login Data 未变 + 24 小时内
+    const CACHE_MAX_AGE_MS = 24 * 60 * 60 * 1000;
+    const entry = cache[cacheKey];
+    const age = Date.now() - entry.cachedAt;
+
+    const loginDataPath = join(userDataDir, 'Default', 'Login Data');
+    if (existsSync(loginDataPath)) {
+      const loginDataMtime = statSync(loginDataPath).mtimeMs;
+      if (loginDataMtime > entry.cachedAt) {
+        if (options.verbose) console.error('[decrypt-creds] Login Data 已更新，缓存过期');
+        return null;
+      }
+    }
+
+    if (age > CACHE_MAX_AGE_MS) {
+      if (options.verbose) console.error(`[decrypt-creds] 缓存已过期 (${Math.round(age / 3600000)}h)，重新解密`);
+      return null;
+    }
+
+    if (options.verbose) {
+      console.error(`[decrypt-creds] ✓ 使用缓存（${Math.round(age / 60000)}min 前缓存，${entry.credentials.length} 条凭据）`);
+    }
+    return entry.credentials;
+  } catch (e) {
+    if (options.verbose) console.error('[decrypt-creds] 缓存读取失败:', e.message);
+    return null;
+  }
+}
+
+/**
+ * 将凭据写入缓存
+ */
+function writeCache(userDataDir, filter, credentials) {
+  try {
+    let cache = {};
+    if (existsSync(CACHE_FILE)) {
+      try { cache = JSON.parse(readFileSync(CACHE_FILE, 'utf8')); } catch {}
+    }
+    const cacheKey = `${userDataDir}::${filter}`;
+    cache[cacheKey] = {
+      cachedAt: Date.now(),
+      userDataDir,
+      filter,
+      credentials,
+    };
+    writeFileSync(CACHE_FILE, JSON.stringify(cache, null, 2), 'utf8');
+    if (options.verbose) console.error('[decrypt-creds] ✓ 已缓存到', CACHE_FILE);
+  } catch (e) {
+    if (options.verbose) console.error('[decrypt-creds] 缓存写入失败:', e.message);
+  }
+}
+
+async function main() {
+  console.error('[decrypt-creds] 检测浏览器配置...');
+
+  // 1. 获取 user-data-dir
+  let userDataDir = options.userDataDir;
+  if (!userDataDir) {
+    try {
+      const config = await getFullBrowserConfig(options.cdpUrl);
+      if (config) {
+        userDataDir = config.userDataDir;
+        if (options.verbose) {
+          console.error('[decrypt-creds] 浏览器:', config.browserInfo?.browser);
+          console.error('[decrypt-creds] CDP 端口:', config.remoteDebuggingPort);
+          console.error('[decrypt-creds] user-data-dir:', userDataDir);
+        }
+      }
+    } catch (e) {
+      console.error('[decrypt-creds] CDP 检测失败:', e.message);
+    }
+  }
+
+  if (!userDataDir) {
+    console.error('[decrypt-creds] 错误: 未检测到 user-data-dir，请用 --user-data-dir 指定');
+    process.exit(1);
+  }
+
+  const filter = options.url ? `%${options.url}%` : '%';
+
+  // 2. 尝试读缓存
+  let credentials = tryReadCache(userDataDir, filter);
+
+  // 3. 缓存未命中，执行解密
+  if (!credentials) {
+    console.error(`[decrypt-creds] 解密凭据 (filter: ${filter})...`);
+    try {
+      credentials = await decryptAllPasswords({
+        userDataDir,
+        filter,
+        verbose: options.verbose,
+      });
+      // 写入缓存
+      writeCache(userDataDir, filter, credentials);
+    } catch (e) {
+      console.error('[decrypt-creds] 解密失败:', e.message);
+      process.exit(1);
+    }
+  }
+
+  // 4. 输出结果
+  if (options.json) {
+    console.log(JSON.stringify(credentials, null, 2));
+  } else {
+    if (credentials.length === 0) {
+      console.log('(未找到匹配的凭据)');
+    } else {
+      const maxUserLen = Math.max(...credentials.map(c => c.username.length));
+      for (const { url, username, password } of credentials) {
+        console.log(`${username.padEnd(maxUserLen)}  ${password}  (${url})`);
+      }
+    }
+  }
+}
+
+main();