token-usage-sync 1.3.0 → 1.5.0

package/index.js

@@ -20,6 +20,7 @@ const flags = {
   help: args.includes('--help') || args.includes('-h'),
   json: args.includes('--json'),
   send: args.includes('--send'),
+  fetchApi: args.includes('--fetch-api'),
   verbose: args.includes('--verbose') || args.includes('-v'),
 };
 
@@ -27,6 +28,10 @@ const flags = {
 const API_URL = process.env.SYNC_API_URL || getArgValue('api-url');
 const USER_ID = process.env.SYNC_USER_ID || getArgValue('user-id') || 'default-user';
 
+// Claude API認証情報
+const CLAUDE_SESSION_KEY = process.env.CLAUDE_SESSION_KEY || getArgValue('session-key');
+const CLAUDE_ORG_ID = process.env.CLAUDE_ORG_ID || getArgValue('org-id');
+
 // ヘルプ表示
 if (flags.help) {
   console.log(`
@@ -38,30 +43,34 @@ token-usage-sync - Claude Code使用量をダッシュボードに同期
 オプション:
   --json              JSON形式で出力
   --send              ダッシュボードに送信（--api-url必須）
+  --fetch-api         Claude APIから使用率を取得（--session-key, --org-id必須）
   --api-url=URL       送信先APIのURL
   --user-id=ID        ユーザーID（デフォルト: default-user）
+  --session-key=KEY   Claude セッションキー（ブラウザCookieから取得）
+  --org-id=ID         Claude 組織ID（ブラウザCookieから取得）
   --verbose           詳細ログ出力
   --help              このヘルプを表示
 
 環境変数:
   SYNC_API_URL        送信先URL（--api-urlの代わり）
   SYNC_USER_ID        ユーザーID（--user-idの代わり）
+  CLAUDE_SESSION_KEY  Claudeセッションキー（--session-keyの代わり）
+  CLAUDE_ORG_ID       Claude組織ID（--org-idの代わり）
 
 例:
-  # 使用量を表示
+  # 使用量を表示（ローカルJSONLから計算）
   npx token-usage-sync
 
-  # 自分のダッシュボードに送信
-  npx token-usage-sync --send --api-url=https://your-worker.workers.dev/api/sync-usage
+  # Claude APIから使用率を取得して表示
+  npx token-usage-sync --fetch-api --session-key=sk-xxx --org-id=xxx
 
-  # 環境変数を使用
-  export SYNC_API_URL=https://your-worker.workers.dev/api/sync-usage
-  npx token-usage-sync --send
+  # APIから取得してダッシュボードに送信
+  npx token-usage-sync --fetch-api --send
 
-セットアップ:
-  1. リポジトリをフォーク: https://github.com/your-repo/token-usage-dashboard
-  2. Cloudflare Workersにデプロイ
-  3. 上記コマンドで同期
+セッションキー/組織IDの取得方法:
+  1. claude.ai にログイン
+  2. 開発者ツール > Application > Cookies
+  3. sessionKey と lastActiveOrg の値をコピー
 `);
   process.exit(0);
 }
@@ -331,8 +340,123 @@ function aggregateUsage(entries) {
   };
 }
 
+// Claude APIから使用率を取得
+async function fetchClaudeApiUsage(sessionKey, orgId) {
+  const url = `https://claude.ai/api/organizations/${orgId}/usage`;
+
+  const response = await fetch(url, {
+    method: 'GET',
+    headers: {
+      'Accept': 'application/json',
+      'Cookie': `sessionKey=${sessionKey}`,
+    },
+  });
+
+  if (!response.ok) {
+    throw new Error(`Claude API error: ${response.status} ${response.statusText}`);
+  }
+
+  const data = await response.json();
+
+  // APIレスポンスをパース
+  // utilization は 0.0〜1.0 の値（0%〜100%）
+  const fiveHour = data.five_hour || {};
+  const sevenDay = data.seven_day || {};
+
+  return {
+    fiveHourPercent: fiveHour.utilization != null ? Math.round(fiveHour.utilization * 100) : null,
+    fiveHourResetAt: fiveHour.resets_at || null,
+    weeklyPercent: sevenDay.utilization != null ? Math.round(sevenDay.utilization * 100) : null,
+    weeklyResetAt: sevenDay.resets_at || null,
+    raw: data,
+  };
+}
+
 // メイン処理
 async function main() {
+  // --fetch-api モード: Claude APIから使用率を取得
+  if (flags.fetchApi) {
+    if (!CLAUDE_SESSION_KEY || !CLAUDE_ORG_ID) {
+      console.error('✗ --fetch-api requires --session-key and --org-id');
+      console.error('');
+      console.error('Usage:');
+      console.error('  npx token-usage-sync --fetch-api --session-key=YOUR_KEY --org-id=YOUR_ORG');
+      console.error('');
+      console.error('Or set environment variables:');
+      console.error('  export CLAUDE_SESSION_KEY=YOUR_KEY');
+      console.error('  export CLAUDE_ORG_ID=YOUR_ORG');
+      process.exit(1);
+    }
+
+    if (flags.verbose) {
+      console.error('Fetching usage from Claude API...');
+    }
+
+    try {
+      const apiUsage = await fetchClaudeApiUsage(CLAUDE_SESSION_KEY, CLAUDE_ORG_ID);
+
+      if (flags.json) {
+        console.log(JSON.stringify(apiUsage, null, 2));
+        return;
+      }
+
+      // 表示
+      console.log('\n📊 Claude Code Usage (from API)\n');
+      console.log('=== Usage Limits ===');
+      console.log(`5-Hour:   ${apiUsage.fiveHourPercent ?? '--'}%`);
+      console.log(`Weekly:   ${apiUsage.weeklyPercent ?? '--'}%`);
+
+      if (apiUsage.fiveHourResetAt) {
+        const resetTime = new Date(apiUsage.fiveHourResetAt);
+        const remaining = Math.max(0, Math.round((resetTime - new Date()) / 60000));
+        console.log(`\n5H Reset: ${Math.floor(remaining / 60)}h ${remaining % 60}m remaining`);
+      }
+
+      // 送信
+      if (flags.send) {
+        if (!API_URL) {
+          console.error('\n✗ --api-url is required for --send');
+          process.exit(1);
+        }
+
+        const payload = {
+          source: 'api',
+          apiUsage: {
+            fiveHourPercent: apiUsage.fiveHourPercent,
+            fiveHourResetAt: apiUsage.fiveHourResetAt,
+            weeklyPercent: apiUsage.weeklyPercent,
+            weeklyResetAt: apiUsage.weeklyResetAt,
+          },
+          syncedAt: new Date().toISOString(),
+        };
+
+        const response = await fetch(API_URL, {
+          method: 'POST',
+          headers: {
+            'Content-Type': 'application/json',
+            'X-User-Id': USER_ID,
+          },
+          body: JSON.stringify(payload),
+        });
+
+        if (response.ok) {
+          console.log('\n✓ Usage data synced successfully!');
+        } else {
+          console.error(`\n✗ Failed to sync: ${response.status} ${response.statusText}`);
+          process.exit(1);
+        }
+      } else {
+        console.log('\n💡 Run with --send to sync to dashboard');
+      }
+
+      return;
+    } catch (e) {
+      console.error(`✗ Failed to fetch from Claude API: ${e.message}`);
+      process.exit(1);
+    }
+  }
+
+  // ローカルJSONLモード（デフォルト）
   // ファイル検索
   if (flags.verbose) {
     console.error(`Searching for JSONL files in: ${PROJECTS_DIR}`);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "token-usage-sync",
-  "version": "1.3.0",
+  "version": "1.5.0",
   "description": "Sync Claude Code token usage to your self-hosted dashboard",
   "type": "module",
   "bin": {