@vibe-cafe/vibe-usage 0.7.5 → 0.7.6

package/README.md

@@ -4,19 +4,29 @@ Track your AI coding tool token usage and sync to [vibecafe.ai](https://vibecafe
 
 ## Quick Start
 
+Get your API key at [vibecafe.ai/usage](https://vibecafe.ai/usage), then copy the one-liner shown there:
+
+```bash
+npx @vibe-cafe/vibe-usage --key vbu_xxxxxxxxxxxx
+```
+
+Or run without a key and paste it interactively:
+
 ```bash
 npx @vibe-cafe/vibe-usage
 ```
 
-This will:
-1. Ask for your API key (get one at https://vibecafe.ai/usage/setup)
+Either path will:
+1. Save your API key to `~/.vibe-usage/config.json`
 2. Detect installed AI coding tools
 3. Run an initial sync of your usage data
+4. Prompt you to enable the background daemon for continuous syncing (recommended)
 
 ## Commands
 
 ```bash
 npx @vibe-cafe/vibe-usage              # Init (first run) or sync (subsequent runs)
+npx @vibe-cafe/vibe-usage --key <vbu_...>   # One-shot init with a pre-copied key
 npx @vibe-cafe/vibe-usage init         # Re-run setup
 npx @vibe-cafe/vibe-usage sync         # Manual sync
 npx @vibe-cafe/vibe-usage daemon       # Continuous sync (every 30m, foreground)

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vibe-cafe/vibe-usage",
-  "version": "0.7.5",
+  "version": "0.7.6",
   "description": "Track your AI coding tool token usage and sync to vibecafe.ai",
   "type": "module",
   "bin": {

package/src/index.js

@@ -88,13 +88,26 @@ function handleConfig(args) {
   }
 }
 
-export async function run(args) {
+function extractOption(args, name) {
+  const flag = `--${name}`;
+  const idx = args.findIndex(a => a === flag);
+  if (idx === -1) return { args, value: undefined };
+  const value = args[idx + 1];
+  if (value === undefined || value.startsWith('--')) {
+    console.error(`Option ${flag} requires a value.`);
+    process.exit(1);
+  }
+  return { args: [...args.slice(0, idx), ...args.slice(idx + 2)], value };
+}
+
+export async function run(rawArgs) {
+  const { args, value: apiKey } = extractOption(rawArgs, 'key');
   const command = args[0];
 
   switch (command) {
     case 'init': {
       const { runInit } = await import('./init.js');
-      await runInit();
+      await runInit({ apiKey });
       break;
     }
     case 'sync': {
@@ -140,7 +153,9 @@ export async function run(args) {
 
   Usage:
     npx @vibe-cafe/vibe-usage              Init (first run) or sync
-    npx @vibe-cafe/vibe-usage init         Set up API key
+    npx @vibe-cafe/vibe-usage --key <vbu_...>   One-shot init with a pre-copied key
+    npx @vibe-cafe/vibe-usage init         Set up API key (interactive)
+    npx @vibe-cafe/vibe-usage init --key <vbu_...>   Init with key, skip paste prompt
     npx @vibe-cafe/vibe-usage sync         Manually sync usage data
     npx @vibe-cafe/vibe-usage daemon       Continuous sync (every 30m, foreground)
     npx @vibe-cafe/vibe-usage daemon install    Install background service (systemd/launchd)
@@ -162,9 +177,10 @@ export async function run(args) {
     }
     default: {
       const config = loadConfig();
-      if (!config?.apiKey) {
+      if (!config?.apiKey || apiKey) {
+        // First run OR user passed --key for a one-shot setup
         const { runInit } = await import('./init.js');
-        await runInit();
+        await runInit({ apiKey });
       } else {
         const { runSync } = await import('./sync.js');
         await runSync();

package/src/init.js

@@ -23,11 +23,22 @@ function openBrowser(url) {
   execFile(cmd, [url], () => {});
 }
 
-export async function runInit() {
+function isDaemonPlatform() {
+  return process.platform === 'linux' || process.platform === 'darwin';
+}
+
+export async function runInit(options = {}) {
+  const { apiKey: providedKey } = options;
+
   console.log('\n  vibe-usage - Vibe Usage Tracker by VibeCaf\u00e9\n');
 
   const existing = loadConfig();
   if (existing?.apiKey) {
+    if (providedKey && existing.apiKey === providedKey) {
+      console.log('Already configured with this key. Running sync...\n');
+      await runSync();
+      return;
+    }
     const answer = await prompt('Config already exists. Overwrite? (y/N) ');
     if (answer.toLowerCase() !== 'y') {
       console.log('Cancelled.');
@@ -36,14 +47,24 @@ export async function runInit() {
   }
 
   const apiUrl = process.env.VIBE_USAGE_API_URL || 'https://vibecafe.ai';
-  console.log(`Get your API key at: ${apiUrl}/usage/setup\n`);
-  openBrowser(`${apiUrl}/usage/setup`);
 
   let apiKey;
-  while (true) {
-    apiKey = await prompt('Paste your API key: ');
-    if (apiKey.startsWith('vbu_')) break;
-    console.log('Invalid key — must start with "vbu_". Try again.');
+  if (providedKey) {
+    if (!providedKey.startsWith('vbu_')) {
+      console.error('Invalid API key — must start with "vbu_".');
+      process.exit(1);
+    }
+    apiKey = providedKey;
+    console.log(`Using API key ${apiKey.slice(0, 8)}...`);
+  } else {
+    console.log(`Get your API key at: ${apiUrl}/usage\n`);
+    openBrowser(`${apiUrl}/usage`);
+
+    while (true) {
+      apiKey = await prompt('Paste your API key: ');
+      if (apiKey.startsWith('vbu_')) break;
+      console.log('Invalid key — must start with "vbu_". Try again.');
+    }
   }
 
   console.log(`\nVerifying key ${apiKey.slice(0, 8)}...`);
@@ -77,7 +98,20 @@ export async function runInit() {
 
   console.log(`\nSetup complete! View your dashboard at: ${apiUrl}/usage`);
 
-  if (process.platform === 'linux' || process.platform === 'darwin') {
-    console.log('\nTip: Run `npx @vibe-cafe/vibe-usage daemon install` to sync automatically in the background.');
+  if (isDaemonPlatform()) {
+    if (process.stdin.isTTY) {
+      console.log('');
+      const answer = await prompt('开启后台自动同步？(持续上报用量数据,推荐) [Y/n] ');
+      const normalized = answer.toLowerCase();
+      if (normalized === '' || normalized === 'y' || normalized === 'yes') {
+        const { manageDaemon } = await import('./daemon-service.js');
+        await manageDaemon('install');
+      } else {
+        console.log('\n可随时运行 `npx @vibe-cafe/vibe-usage daemon install` 开启后台同步。');
+      }
+    } else {
+      // Non-interactive shell (CI, pipe) — don't block on prompt
+      console.log('\nTip: Run `npx @vibe-cafe/vibe-usage daemon install` to sync automatically in the background.');
+    }
   }
 }