aihezu 2.1.0 → 2.3.1

package/README.md

@@ -18,6 +18,7 @@
   - Claude：默认 `https://cc.aihezu.dev/api`，企业可用 `--api` / `--api-url` 指定独立域名
   - Codex：默认 `https://cc.aihezu.dev/openai`，企业可用 `--api` / `--api-url` 指定独立域名；写入 `~/.codex/config.toml` 和 `auth.json`，使用 `AIHEZU_OAI_KEY`
 - `ccclear`：清理 Claude Code 缓存和配置
+- `ccusage`：查看 Claude Code API 用量统计（读取 `ANTHROPIC_BASE_URL` / `ANTHROPIC_AUTH_TOKEN`）
 
 ## 清理内容
 
@@ -84,6 +85,14 @@ npm install -g aihezu
 sudo ccclear
 ```
 
+### 查看 Claude Code 用量（ccusage）
+
+无需管理员权限：
+
+```bash
+npx aihezu ccusage
+```
+
 ## 运行效果
 
 ```

package/bin/aihezu.js

@@ -6,7 +6,8 @@ const readline = require('readline');
 
 const commands = {
   install: require('../commands/install'),
-  clear: require('../commands/clear')
+  clear: require('../commands/clear'),
+  ccusage: require('../commands/ccusage')
 };
 
 const services = {
@@ -22,6 +23,7 @@ function showHelp() {
   console.log('\n命令:');
   console.log('  install <service>   配置服务 (API Key, URL, 网络设置)');
   console.log('  clear <service>     清理缓存并刷新网络设置');
+  console.log('  ccusage             查看 Claude Code 用量统计');
   console.log('\n服务:');
   console.log('  claude              Claude Code');
   console.log('  codex               Codex');
@@ -29,6 +31,7 @@ function showHelp() {
   console.log('\n示例:');
   console.log('  aihezu install claude');
   console.log('  aihezu clear codex');
+  console.log('  aihezu ccusage');
   console.log('  aihezu install      (交互式安装)');
   console.log('  aihezu help         (显示帮助信息)');
 }
@@ -38,73 +41,97 @@ async function askQuestion(rl, question) {
 }
 
 async function main() {
-  // Removed global rl creation
-
   try {
-    if (args.length === 0 || args[0] === 'help') { // Handle 'aihezu' or 'aihezu help'
+    if (args.length === 0 || args[0] === 'help') {
       showHelp();
       return;
     }
 
     const commandName = args[0];
-    let serviceName = args[1];
-
     if (!commands[commandName]) {
-      console.error(`❌ 未知命��: ${commandName}`);
+      console.error(`❌ 未知命令: ${commandName}`);
       showHelp();
       process.exit(1);
     }
 
+    // Commands that don't require a service
+    if (commandName === 'ccusage') {
+      await commands[commandName](args.slice(1));
+      return;
+    }
+
+    let serviceName = '';
+    let commandArgs = [];
+
+    // Check if the second argument is a service name or an option/flag
+    if (args.length > 1 && !args[1].startsWith('-')) {
+      serviceName = args[1].toLowerCase();
+      commandArgs = args.slice(2);
+    } else {
+      // No service name provided, or an option was provided instead.
+      // The command arguments will be all arguments after the command itself.
+      commandArgs = args.slice(1);
+    }
+
+    // If service name is not provided, and command is 'install', enter interactive mode.
     if (!serviceName) {
       if (commandName === 'install') {
-        // Init rl ONLY here for selection
         const rl = readline.createInterface({
           input: process.stdin,
           output: process.stdout
         });
 
         try {
-          // Interactive service selection for 'install' command
           console.log('\n请选择要安装的服务:');
           const serviceKeys = Object.keys(services);
+          const claudeIndex = serviceKeys.indexOf('claude');
+
           serviceKeys.forEach((key, index) => {
-            console.log(`  [${index + 1}] ${services[key].displayName}`);
+            const isDefault = (key === 'claude');
+            console.log(`  [${index + 1}] ${services[key].displayName}${isDefault ? ' (默认)' : ''}`);
           });
 
-          let chosenIndex;
+          let chosenIndex = claudeIndex !== -1 ? claudeIndex : 0; // Default to Claude
           let validChoice = false;
+
           while (!validChoice) {
-            const answer = await askQuestion(rl, '请输入选项编号: ');
-            chosenIndex = parseInt(answer, 10) - 1;
-            if (chosenIndex >= 0 && chosenIndex < serviceKeys.length) {
-              serviceName = serviceKeys[chosenIndex];
-              validChoice = true;
+            const answer = await askQuestion(rl, '请输入选项编号 (直接回车使用默认值): ');
+            if (answer === '') {
+              validChoice = true; // Use the default index
             } else {
-              console.log('无效的选择，请输入列表中的数字。');
+              const selectedIdx = parseInt(answer, 10) - 1;
+              if (selectedIdx >= 0 && selectedIdx < serviceKeys.length) {
+                chosenIndex = selectedIdx;
+                validChoice = true;
+              } else {
+                console.log('无效的选择，请输入列表中的数字。');
+              }
             }
           }
+          serviceName = serviceKeys[chosenIndex];
           console.log(`✅ 已选择: ${services[serviceName].displayName}`);
+
         } finally {
-          // Close rl immediately after selection and BEFORE calling the command
           rl.close();
         }
       } else {
-        // For 'clear' or other commands, service name is mandatory
+        // For other commands like 'clear', service name is mandatory.
         console.error(`❌ 请为 '${commandName}' 命令指定一个服务名称。`);
         showHelp();
         process.exit(1);
       }
     }
 
-    const service = services[serviceName.toLowerCase()];
+    const service = services[serviceName];
     if (!service) {
       console.error(`❌ 未知服务: ${serviceName}`);
       console.log('可用服务: ' + Object.keys(services).map(key => services[key].displayName).join(', '));
       process.exit(1);
     }
 
-    // Execute
-    await commands[commandName](service, args.slice(2));
+    // Execute the command with the correct service and arguments
+    await commands[commandName](service, commandArgs);
+    
   } catch (error) {
     console.error('❌ 发生错误:', error);
     process.exit(1);

package/commands/ccusage.js

@@ -0,0 +1,278 @@
+const fs = require('fs');
+const os = require('os');
+const path = require('path');
+const http = require('http');
+const https = require('https');
+
+function normalizeHttpUrl(url) {
+  if (!url) return '';
+  const trimmed = String(url).trim();
+  if (!trimmed) return '';
+  if (/^https?:\/\//i.test(trimmed)) return trimmed;
+  return `https://${trimmed}`;
+}
+
+function readClaudeEnvFromSettingsFile() {
+  const settingsPath = path.join(os.homedir(), '.claude', 'settings.json');
+  if (!fs.existsSync(settingsPath)) {
+    return { env: null, settingsPath, error: null };
+  }
+
+  try {
+    const content = fs.readFileSync(settingsPath, 'utf8');
+    const settings = JSON.parse(content);
+    const env = settings && settings.env ? settings.env : null;
+    return { env, settingsPath, error: null };
+  } catch (error) {
+    return { env: null, settingsPath, error };
+  }
+}
+
+function postJson(urlString, body, options = {}) {
+  const { timeoutMs = 15000 } = options;
+  const url = new URL(urlString);
+  const payload = JSON.stringify(body ?? {});
+
+  const transport = url.protocol === 'https:' ? https : http;
+
+  return new Promise((resolve, reject) => {
+    const req = transport.request(
+      {
+        method: 'POST',
+        hostname: url.hostname,
+        port: url.port || (url.protocol === 'https:' ? 443 : 80),
+        path: `${url.pathname}${url.search || ''}`,
+        headers: {
+          'Content-Type': 'application/json',
+          'Accept': 'application/json',
+          'Content-Length': Buffer.byteLength(payload),
+          'User-Agent': 'aihezu-cli'
+        }
+      },
+      res => {
+        res.setEncoding('utf8');
+        let raw = '';
+
+        res.on('data', chunk => {
+          raw += chunk;
+        });
+
+        res.on('end', () => {
+          let json = null;
+          try {
+            json = raw ? JSON.parse(raw) : null;
+          } catch (error) {
+            // keep json as null, return raw for debugging
+          }
+
+          resolve({
+            statusCode: res.statusCode || 0,
+            raw,
+            json
+          });
+        });
+      }
+    );
+
+    req.on('error', reject);
+    req.setTimeout(timeoutMs, () => {
+      req.destroy(new Error(`Request timed out after ${timeoutMs}ms`));
+    });
+
+    req.write(payload);
+    req.end();
+  });
+}
+
+function asNumber(value) {
+  if (typeof value === 'number' && Number.isFinite(value)) return value;
+  if (typeof value === 'string' && value.trim() !== '' && Number.isFinite(Number(value))) {
+    return Number(value);
+  }
+  return null;
+}
+
+function formatCost(value) {
+  const n = asNumber(value);
+  if (n === null) return '-';
+  return n.toFixed(2);
+}
+
+function formatPercent(current, limit) {
+  const c = asNumber(current);
+  const l = asNumber(limit);
+  if (c === null || l === null || l <= 0) return '-';
+  return `${((c / l) * 100).toFixed(1)}%`;
+}
+
+function renderBar(current, limit, width = 18) {
+  const c = asNumber(current);
+  const l = asNumber(limit);
+  if (c === null || l === null || l <= 0) return '';
+
+  const ratio = Math.max(0, Math.min(1, c / l));
+  let filled = Math.round(ratio * width);
+  if (ratio > 0 && filled === 0) filled = 1;
+  const empty = Math.max(0, width - filled);
+  return `[${'█'.repeat(filled)}${'░'.repeat(empty)}]`;
+}
+
+function usageHint(current, limit) {
+  const c = asNumber(current);
+  const l = asNumber(limit);
+  if (c === null || l === null || l <= 0) return '';
+  const ratio = c / l;
+  if (ratio >= 1) return '⚠️ 已超出限制';
+  if (ratio >= 0.9) return '⚠️ 接近上限';
+  return '';
+}
+
+function showUsageHelp() {
+  console.log('用法: aihezu ccusage [--json]');
+  console.log('');
+  console.log('说明:');
+  console.log('  - 读取当前 ANTHROPIC_BASE_URL / ANTHROPIC_AUTH_TOKEN（优先环境变量，其次 ~/.claude/settings.json）');
+  console.log('  - 自动请求 /apiStats/api/get-key-id 和 /apiStats/api/user-stats 并美观输出用量');
+  console.log('');
+  console.log('示例:');
+  console.log('  npx aihezu ccusage');
+  console.log('  npx aihezu ccusage --json');
+}
+
+async function ccusageCommand(args = []) {
+  const wantsHelp = args.includes('help') || args.includes('--help') || args.includes('-h');
+  if (wantsHelp) {
+    showUsageHelp();
+    return;
+  }
+
+  const outputJson = args.includes('--json');
+
+  let baseUrl = process.env.ANTHROPIC_BASE_URL || '';
+  let authToken = process.env.ANTHROPIC_AUTH_TOKEN || '';
+  let sourceHint = '环境变量';
+
+  if (!baseUrl || !authToken) {
+    const { env, settingsPath, error } = readClaudeEnvFromSettingsFile();
+    if (error) {
+      console.log(`⚠️  读取配置文件失败: ${settingsPath}`);
+      console.log(`   错误: ${error.message}`);
+    }
+
+    if (env) {
+      baseUrl = baseUrl || env.ANTHROPIC_BASE_URL || '';
+      authToken = authToken || env.ANTHROPIC_AUTH_TOKEN || '';
+      sourceHint = `配置文件 (${settingsPath})`;
+    }
+  }
+
+  baseUrl = normalizeHttpUrl(baseUrl);
+  authToken = String(authToken || '').trim();
+
+  if (!baseUrl || !authToken) {
+    console.error('❌ 未找到 ANTHROPIC_BASE_URL 或 ANTHROPIC_AUTH_TOKEN。');
+    console.error('   请先运行: npx aihezu install claude');
+    console.error('   或设置环境变量: ANTHROPIC_BASE_URL / ANTHROPIC_AUTH_TOKEN');
+    process.exit(1);
+  }
+
+  let origin = '';
+  try {
+    origin = new URL(baseUrl).origin;
+  } catch (error) {
+    console.error('❌ ANTHROPIC_BASE_URL 不是合法的 URL: ' + baseUrl);
+    process.exit(1);
+  }
+
+  const keyIdUrl = `${origin}/apiStats/api/get-key-id`;
+  const statsUrl = `${origin}/apiStats/api/user-stats`;
+
+  console.log('');
+  console.log('📊 Claude Code 用量统计');
+  console.log(`🌐 域名: ${origin}`);
+  console.log(`🔎 配置来源: ${sourceHint}`);
+  console.log('');
+
+  const keyIdRes = await postJson(keyIdUrl, { apiKey: authToken });
+  if (keyIdRes.statusCode < 200 || keyIdRes.statusCode >= 300) {
+    console.error(`❌ 获取 API Key ID 失败 (HTTP ${keyIdRes.statusCode})`);
+    if (keyIdRes.raw) console.error(keyIdRes.raw);
+    process.exit(1);
+  }
+
+  const apiId =
+    keyIdRes?.json?.data?.id ||
+    keyIdRes?.json?.id ||
+    null;
+
+  if (!apiId) {
+    console.error('❌ 返回值中未找到 API Key ID。');
+    if (keyIdRes.raw) console.error(keyIdRes.raw);
+    process.exit(1);
+  }
+
+  const statsRes = await postJson(statsUrl, { apiId });
+  if (statsRes.statusCode < 200 || statsRes.statusCode >= 300) {
+    console.error(`❌ 获取用量失败 (HTTP ${statsRes.statusCode})`);
+    if (statsRes.raw) console.error(statsRes.raw);
+    process.exit(1);
+  }
+
+  if (!statsRes.json || typeof statsRes.json !== 'object') {
+    console.error('❌ 获取用量失败：返回内容不是合法的 JSON。');
+    if (statsRes.raw) console.error(statsRes.raw);
+    process.exit(1);
+  }
+
+  const stats = (statsRes && statsRes.json && statsRes.json.data) ? statsRes.json.data : (statsRes.json || {});
+
+  if (outputJson) {
+    console.log(JSON.stringify(stats, null, 2));
+    return;
+  }
+
+  const limits = (stats && typeof stats === 'object' && stats.limits && typeof stats.limits === 'object')
+    ? stats.limits
+    : {};
+
+  const currentDailyCost = (stats.currentDailyCost !== undefined) ? stats.currentDailyCost : limits.currentDailyCost;
+  const dailyCostLimit = (stats.dailyCostLimit !== undefined) ? stats.dailyCostLimit : limits.dailyCostLimit;
+  const currentWindowCost = (stats.currentWindowCost !== undefined) ? stats.currentWindowCost : limits.currentWindowCost;
+  const rateLimitCost = (stats.rateLimitCost !== undefined) ? stats.rateLimitCost : limits.rateLimitCost;
+
+  const dailyRemaining = (() => {
+    const c = asNumber(currentDailyCost);
+    const l = asNumber(dailyCostLimit);
+    if (c === null || l === null) return null;
+    return l - c;
+  })();
+
+  const windowRemaining = (() => {
+    const c = asNumber(currentWindowCost);
+    const l = asNumber(rateLimitCost);
+    if (c === null || l === null) return null;
+    return l - c;
+  })();
+
+  console.log('=== 关键指标 ===');
+  console.log(
+    `日额度: ${formatCost(currentDailyCost)} / ${formatCost(dailyCostLimit)} (${formatPercent(currentDailyCost, dailyCostLimit)}) ` +
+    `${renderBar(currentDailyCost, dailyCostLimit)} ${usageHint(currentDailyCost, dailyCostLimit)}`.trimEnd()
+  );
+  if (dailyRemaining !== null) {
+    console.log(`日剩余: ${formatCost(dailyRemaining)}`);
+  }
+
+  console.log('');
+  console.log(
+    `窗口:   ${formatCost(currentWindowCost)} / ${formatCost(rateLimitCost)} (${formatPercent(currentWindowCost, rateLimitCost)}) ` +
+    `${renderBar(currentWindowCost, rateLimitCost)} ${usageHint(currentWindowCost, rateLimitCost)}`.trimEnd()
+  );
+  if (windowRemaining !== null) {
+    console.log(`窗口剩余: ${formatCost(windowRemaining)}`);
+  }
+
+  console.log('');
+}
+
+module.exports = ccusageCommand;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aihezu",
-  "version": "2.1.0",
+  "version": "2.3.1",
   "description": "AI 开发环境配置工具 - 支持 Claude Code, Codex, Google Gemini 的本地化配置、代理设置与缓存清理",
   "main": "bin/aihezu.js",
   "bin": {
@@ -8,6 +8,14 @@
     "ccclear": "bin/ccclear.js",
     "ccinstall": "bin/ccinstall.js"
   },
+  "files": [
+    "bin/",
+    "commands/",
+    "services/",
+    "lib/",
+    "README.md",
+    "INSTALL_USAGE.md"
+  ],
   "scripts": {
     "test": "node bin/aihezu.js help"
   },