aihezu 2.6.2 → 2.6.3

package/bin/aihezu.js

@@ -36,6 +36,7 @@ function showHelp() {
   console.log('  aihezu usage cc           只显示 Claude Code 用量');
   console.log('  aihezu usage codex        只显示 Codex 用量');
   console.log('  aihezu usage gemini       只显示 Google Gemini 用量');
+  console.log('  aihezu usage --key sk-xxx 使用指定 API Key 查询');
   console.log('\n服务:');
   console.log('  claude              Claude Code');
   console.log('  codex               Codex');

package/commands/usage.js

@@ -17,6 +17,69 @@ function maskToken(token) {
   return '***' + token.slice(-8);
 }
 
+function parseKeyArg(args) {
+  let overrideKey = null;
+  const remaining = [];
+
+  for (let i = 0; i < args.length; i += 1) {
+    const arg = args[i];
+
+    if (arg === '--key' || arg === '-k') {
+      const next = args[i + 1];
+      if (!next || next.startsWith('-')) {
+        console.error('[错误] --key 需要提供 API Key');
+        process.exit(1);
+      }
+      overrideKey = next;
+      i += 1;
+      continue;
+    }
+
+    if (arg.startsWith('--key=')) {
+      const value = arg.slice('--key='.length);
+      if (!value) {
+        console.error('[错误] --key 需要提供 API Key');
+        process.exit(1);
+      }
+      overrideKey = value;
+      continue;
+    }
+
+    if (arg.startsWith('-k=')) {
+      const value = arg.slice('-k='.length);
+      if (!value) {
+        console.error('[错误] -k 需要提供 API Key');
+        process.exit(1);
+      }
+      overrideKey = value;
+      continue;
+    }
+
+    remaining.push(arg);
+  }
+
+  return { overrideKey, remaining };
+}
+
+function uniqueConfigsByOrigin(configs) {
+  const result = [];
+  const seen = new Set();
+
+  for (const config of configs) {
+    let originKey = config.baseUrl || '';
+    try {
+      originKey = new URL(config.baseUrl).origin;
+    } catch (error) {
+      // fallback to baseUrl string
+    }
+    if (seen.has(originKey)) continue;
+    seen.add(originKey);
+    result.push(config);
+  }
+
+  return result;
+}
+
 // 读取 Claude 配置
 function readClaudeConfig() {
   const configs = [];
@@ -454,12 +517,13 @@ function displayUsageStats(stats, origin, source) {
 }
 
 function showUsageHelp() {
-  console.log('用法: aihezu usage [service] [--json]');
+  console.log('用法: aihezu usage [service] [--json] [--key <apiKey>]');
   console.log('');
   console.log('说明:');
   console.log('  - 自动检测 Claude Code、Codex、Google Gemini 的配置');
   console.log('  - 从环境变量和配置文件读取配置');
   console.log('  - 如果两者都存在且不同，将分别查询两个账号的用量');
+  console.log('  - 使用 --key 可查询指定 API Key 的用量');
   console.log('');
   console.log('可选服务参数:');
   console.log('  cc, claude          只显示 Claude Code 用量');
@@ -467,6 +531,10 @@ function showUsageHelp() {
   console.log('  gemini              只显示 Google Gemini 用量');
   console.log('  (不指定)            显示所有已配置服务的用量');
   console.log('');
+  console.log('可选参数:');
+  console.log('  --key, -k <apiKey>  使用指定 API Key 查询用量');
+  console.log('  --json              JSON 格式输出');
+  console.log('');
   console.log('示例:');
   console.log('  npx aihezu usage              # 显示所有服务');
   console.log('  npx aihezu usage cc           # 只显示 Claude Code');
@@ -474,6 +542,8 @@ function showUsageHelp() {
   console.log('  npx aihezu usage gemini       # 只显示 Gemini');
   console.log('  npx aihezu usage --json       # JSON 格式输出所有服务');
   console.log('  npx aihezu usage cc --json    # JSON 格式输出 Claude Code');
+  console.log('  npx aihezu usage --key sk-xxx # 使用指定 Key 查询');
+  console.log('  npx aihezu usage cc --key sk-xxx');
 }
 
 async function usageCommand(args = []) {
@@ -483,10 +553,11 @@ async function usageCommand(args = []) {
     return;
   }
 
-  const outputJson = args.includes('--json');
+  const { overrideKey, remaining } = parseKeyArg(args);
+  const outputJson = remaining.includes('--json');
 
   // 提取服务过滤参数（第一个非选项参数）
-  const serviceFilter = args.find(arg => !arg.startsWith('-'));
+  const serviceFilter = remaining.find(arg => !arg.startsWith('-'));
 
   // 服务别名映射
   const serviceAliases = {
@@ -538,24 +609,30 @@ async function usageCommand(args = []) {
 
     allResults[service.name] = [];
 
-    for (const config of service.configs) {
-      console.log(`[查询中] ${config.source} - Token: ${maskToken(config.authToken)}`);
+    const configsToUse = overrideKey
+      ? uniqueConfigsByOrigin(service.configs)
+      : service.configs;
+
+    for (const config of configsToUse) {
+      const effectiveKey = overrideKey || config.authToken;
+      const sourceLabel = overrideKey ? `命令行 Key (${config.source})` : config.source;
+      console.log(`[查询中] ${sourceLabel} - Token: ${maskToken(effectiveKey)}`);
 
-      const result = await queryUsage(config.baseUrl, config.authToken);
+      const result = await queryUsage(config.baseUrl, effectiveKey);
 
       if (result.error) {
         console.log(`[错误] ${result.error}`);
         console.log('');
         allResults[service.name].push({
-          source: config.source,
+          source: sourceLabel,
           error: result.error
         });
       } else {
         if (!outputJson) {
-          displayUsageStats(result.stats, result.origin, config.source);
+          displayUsageStats(result.stats, result.origin, sourceLabel);
         }
         allResults[service.name].push({
-          source: config.source,
+          source: sourceLabel,
           origin: result.origin,
           stats: result.stats
         });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aihezu",
-  "version": "2.6.2",
+  "version": "2.6.3",
   "description": "AI 开发环境配置工具 - 支持 Claude Code, Codex, Google Gemini 的本地化配置、代理设置与缓存清理",
   "main": "bin/aihezu.js",
   "bin": {