@winston.wan/burn-your-money 2.0.4 → 2.1.0

package/install.js

@@ -2,6 +2,8 @@ const fs = require('fs');
 const path = require('path');
 const os = require('os');
 const { execSync } = require('child_process');
+const https = require('https');
+const { createWriteStream, existsSync, mkdirSync } = fs;
 
 // ANSI colors for console output
 const colors = {
@@ -34,36 +36,119 @@ function getHomeDir() {
     return os.homedir();
 }
 
-function checkDependencies() {
+// Download file following redirects
+function downloadFile(url, destPath) {
+    return new Promise((resolve, reject) => {
+        const download = (url) => {
+            https.get(url, (response) => {
+                // Handle redirects (301, 302, 307, 308)
+                if ([301, 302, 307, 308].includes(response.statusCode)) {
+                    const redirectUrl = response.headers.location;
+                    if (redirectUrl) {
+                        download(redirectUrl);
+                        return;
+                    }
+                }
+
+                // Handle errors
+                if (response.statusCode !== 200) {
+                    reject(new Error(`Failed to download: HTTP ${response.statusCode}`));
+                    return;
+                }
+
+                const file = createWriteStream(destPath);
+                response.pipe(file);
+
+                file.on('finish', () => {
+                    file.close();
+                    resolve();
+                });
+
+                file.on('error', (err) => {
+                    fs.unlink(destPath, () => {});
+                    reject(err);
+                });
+            }).on('error', (err) => {
+                reject(err);
+            });
+        };
+
+        download(url);
+    });
+}
+
+// Install jq for Windows by downloading binary
+async function installJqWindows() {
+    const home = getHomeDir();
+    const binDir = path.join(home, '.claude', 'bin');
+
+    // Create bin directory
+    if (!existsSync(binDir)) {
+        mkdirSync(binDir, { recursive: true });
+    }
+
+    const jqPath = path.join(binDir, 'jq.exe');
+    const jqUrl = 'https://github.com/jqlang/jq/releases/latest/download/jq-windows-amd64.exe';
+
+    log(`Downloading jq to ${jqPath}...`, colors.cyan);
+
+    try {
+        await downloadFile(jqUrl, jqPath);
+        success("jq downloaded successfully!");
+        return jqPath;
+    } catch (err) {
+        error(`Failed to download jq: ${err.message}`);
+        throw err;
+    }
+}
+
+// Install jq for macOS/Linux
+function installJqUnix() {
+    try {
+        log("Attempting to install jq via brew...", colors.cyan);
+        execSync('brew install jq', { stdio: 'inherit' });
+        success("jq installed successfully!");
+        return 'jq';
+    } catch (brewError) {
+        try {
+            log("Attempting to install jq via apt...", colors.cyan);
+            execSync('sudo apt-get install -y jq', { stdio: 'inherit' });
+            success("jq installed successfully!");
+            return 'jq';
+        } catch (aptError) {
+            throw new Error("Could not auto-install jq. Please install manually: brew install jq or sudo apt-get install jq");
+        }
+    }
+}
+
+async function checkDependencies() {
     log("Checking dependencies...", colors.cyan);
 
+    let jqPath = 'jq';  // Default to system jq
+
     // Check for jq
     try {
         execSync('jq --version', { stdio: 'ignore' });
         success("jq is installed");
     } catch (e) {
-        warning("jq not found! Attempting valid auto-installation...");
+        warning("jq not found! Attempting auto-installation...");
 
         try {
             if (os.platform() === 'win32') {
-                log("Attempting to install jq via winget...", colors.cyan);
-                // silent install might need admin, but let's try
-                execSync('winget install jqlang.jq --accept-source-agreements --accept-package-agreements', { stdio: 'inherit' });
-                success("jq installed successfully!");
+                jqPath = await installJqWindows();
             } else if (os.platform() === 'darwin') {
-                log("Attempting to install jq via brew...", colors.cyan);
-                execSync('brew install jq', { stdio: 'inherit' });
-                success("jq installed successfully!");
+                installJqUnix();
             } else {
-                // Linux: too many variants (apt, dnf, pacman...), just warn
-                throw new Error("Linux auto-install not supported");
+                // Linux
+                installJqUnix();
             }
         } catch (installError) {
             error("Auto-installation failed.");
             log("  Please install jq manually:", colors.yellow);
-            log("  Windows: winget install jqlang.jq", colors.yellow);
+            log("  Windows: Run installer again (will download automatically)", colors.yellow);
             log("  macOS: brew install jq", colors.yellow);
             log("  Linux: sudo apt-get install jq", colors.yellow);
+            throw installError;
         }
     }
 
@@ -74,6 +159,8 @@ function checkDependencies() {
     } catch (e) {
         warning("Claude Code not found in PATH.");
     }
+
+    return jqPath;
 }
 
 function createDirectories() {
@@ -120,7 +207,7 @@ function installScripts() {
     });
 }
 
-function configureSettings() {
+function configureSettings(jqPath) {
     log("Configuring Claude Code...", colors.cyan);
     const home = getHomeDir();
     const settingsFile = path.join(home, '.claude', 'settings.json');
@@ -134,18 +221,35 @@ function configureSettings() {
         }
     }
 
-    // Update statusLine config
-    // We use forward slashes for paths in settings.json even on Windows for consistency in JSON
-    // accessing existing wsl path if needed? No, standard path is fine.
-    // The previous implementation used `~/.claude/statusline.sh`. 
-    // Claude might process `~`? Let's stick to what the bash script did: `~/.claude/statusline.sh`
-
     // Configure command based on platform
     let commandEnv = "~/.claude/statusline.sh";
 
     if (os.platform() === 'win32') {
-        // On Windows, expand home directory and ensure forward slashes
-        const scriptPath = path.join(home, '.claude', 'statusline.sh').replace(/\\/g, '/');
+        // Get the raw Windows path first
+        const scriptPathWindows = path.join(home, '.claude', 'statusline.sh');
+
+        // Convert Windows jq path to Git Bash path format
+        // C:\Users\... -> /c/Users/... (lowercase drive letter)
+        let jqPathForBash = 'jq';
+        if (jqPath && jqPath !== 'jq' && /^[A-Z]:/.test(jqPath)) {
+            jqPathForBash = jqPath.replace(/^([A-Z]):\\/, (match, drive) => `/${drive.toLowerCase()}/`).replace(/\\/g, '/');
+        }
+
+        // Also convert scriptPath to Git Bash format for sourcing
+        const scriptPathGitBash = scriptPathWindows.replace(/^([A-Z]):\\/, (match, drive) => `/${drive.toLowerCase()}/`).replace(/\\/g, '/');
+
+        // Create environment variable for jq path in the script
+        const wrapperScriptPath = path.join(home, '.claude', 'statusline-wrapper.sh');
+        const wrapperContent = `#!/bin/bash
+# Export JQ_PATH for the scripts to use
+export JQ_PATH="${jqPathForBash}"
+# Source the statusline script to preserve environment
+. "${scriptPathGitBash}"
+`;
+        fs.writeFileSync(wrapperScriptPath, wrapperContent);
+        if (os.platform() !== 'win32') {
+            fs.chmodSync(wrapperScriptPath, '755');
+        }
 
         // Robust bash detection: try Git Bash first, then fallback to 'bash'
         let bashPath = 'bash';
@@ -176,7 +280,18 @@ function configureSettings() {
             }
         }
 
-        commandEnv = `${bashPath} "${scriptPath}"`;
+        const wrapperPathGitBash = wrapperScriptPath.replace(/^([A-Z]):\\/, (match, drive) => `/${drive.toLowerCase()}/`).replace(/\\/g, '/');
+        commandEnv = `${bashPath} "${wrapperPathGitBash}"`;
+    } else {
+        // On Unix, just set the JQ_PATH in the wrapper script
+        const wrapperScriptPath = path.join(home, '.claude', 'statusline-wrapper.sh');
+        const wrapperContent = `#!/bin/bash
+export JQ_PATH="${jqPath}"
+exec ~/.claude/statusline.sh
+`;
+        fs.writeFileSync(wrapperScriptPath, wrapperContent);
+        fs.chmodSync(wrapperScriptPath, '755');
+        commandEnv = "bash ~/.claude/statusline-wrapper.sh";
     }
 
     settings.statusLine = {
@@ -190,16 +305,130 @@ function configureSettings() {
     } catch (e) {
         error(`Failed to update settings.json: ${e.message}`);
     }
+
+    // 安装自定义命令文件到 ~/.claude/commands/
+    log("Installing custom commands...", colors.cyan);
+    const commandsDir = path.join(home, '.claude', 'commands');
+
+    // 确保 commands 目录存在
+    if (!fs.existsSync(commandsDir)) {
+        fs.mkdirSync(commandsDir, { recursive: true });
+    }
+
+    const commands = [
+        {
+            name: 'burn-your-money-stats.md',
+            content: `---
+name: burn-your-money-stats
+description: 📊 查看 token 使用趋势图
+---
+
+# Token 使用趋势图
+
+请执行以下 bash 命令查看过去 7 天的 token 使用趋势图：
+
+\\\`\\\`\\\`bash
+bash ~/.claude/scripts/token-history.sh chart
+\\\`\\\`\\\`
+`
+        },
+        {
+            name: 'burn-your-money-today.md',
+            content: `---
+name: burn-your-money-today
+description: 📅 查看今日 token 使用情况
+---
+
+# 今日 Token 使用情况
+
+请执行以下 bash 命令查看今日的 token 使用量：
+
+\\\`\\\`\\\`bash
+bash ~/.claude/scripts/token-history.sh today_tokens
+\\\`\\\`\\\`
+`
+        },
+        {
+            name: 'burn-your-money-week.md',
+            content: `---
+name: burn-your-money-week
+description: 📆 查看本周 token 使用情况
+---
+
+# 本周 Token 使用情况
+
+请执行以下 bash 命令查看本周的 token 使用量：
+
+\\\`\\\`\\\`bash
+bash ~/.claude/scripts/token-history.sh week_tokens
+\\\`\\\`\\\`
+`
+        },
+        {
+            name: 'burn-your-money-month.md',
+            content: `---
+name: burn-your-money-month
+description: 🗓️ 查看本月 token 使用情况
+---
+
+# 本月 Token 使用情况
+
+请执行以下 bash 命令查看本月的 token 使用量：
+
+\\\`\\\`\\\`bash
+bash ~/.claude/scripts/token-history.sh month_tokens
+\\\`\\\`\\\`
+`
+        },
+        {
+            name: 'burn-your-money-export.md',
+            content: `---
+name: burn-your-money-export
+description: 💾 导出 token 数据（JSON）
+---
+
+# 导出 Token 数据
+
+请执行以下 bash 命令导出 token 数据为 JSON 格式：
+
+\\\`\\\`\\\`bash
+bash ~/.claude/scripts/token-history.sh export json
+\\\`\\\`\\\`
+`
+        },
+        {
+            name: 'burn-your-money-export-csv.md',
+            content: `---
+name: burn-your-money-export-csv
+description: 📄 导出 token 数据（CSV）
+---
+
+# 导出 Token 数据 (CSV)
+
+请执行以下 bash 命令导出 token 数据为 CSV 格式：
+
+\\\`\\\`\\\`bash
+bash ~/.claude/scripts/token-history.sh export csv
+\\\`\\\`\\\`
+`
+        }
+    ];
+
+    commands.forEach(cmd => {
+        const cmdPath = path.join(commandsDir, cmd.name);
+        fs.writeFileSync(cmdPath, cmd.content);
+        success(`Installed ${cmd.name}`);
+    });
 }
 
-function main() {
+async function main() {
     log("\n💸 Burn Your Money - Installer\n", colors.purple);
 
     try {
-        checkDependencies();
+        const jqPath = await checkDependencies();
         createDirectories();
         installScripts();
-        configureSettings();
+        configureSettings(jqPath);
 
         log("\n✅ Installation complete!", colors.green);
         log("Please restart Claude Code to see the status bar.\n", colors.cyan);

package/package.json

@@ -1,13 +1,14 @@
 {
   "name": "@winston.wan/burn-your-money",
-  "version": "2.0.4",
+  "version": "2.1.0",
   "description": "💸 Burn Your Money - 实时显示 Claude Code 的 token 消耗，看着你的钱包燃烧！",
   "main": "src/statusline.sh",
   "bin": {
     "burn-your-money": "./bin/burn-your-money"
   },
   "scripts": {
-    "install": "node install.js",
+    "postinstall": "node install.js",
+    "uninstall": "node uninstall.js",
     "test": "bash ./test.sh"
   },
   "repository": {