@gopherhole/cli 0.1.13 → 0.1.14

package/dist/index.js

@@ -19,7 +19,7 @@ const brand = {
     greenDark: chalk_1.default.hex('#16a34a'), // gopher-600 - emphasis
 };
 // Version
-const VERSION = '0.1.8';
+const VERSION = '0.1.14';
 // ASCII art banner
 function showBanner(context) {
     const gopher = [
@@ -1723,6 +1723,255 @@ access
         process.exit(1);
     }
 });
+// ========== BUDGET COMMAND ==========
+const budget = program
+    .command('budget')
+    .description(`View and manage agent spending limits
+
+${chalk_1.default.bold('Examples:')}
+  $ gopherhole budget my-agent              # View budget
+  $ gopherhole budget my-agent -d 10 -w 50  # Set daily=$10, weekly=$50
+  $ gopherhole budget my-agent --clear      # Remove all limits
+  $ gopherhole budget --all                 # List all agents' budgets
+`);
+budget
+    .argument('[agentId]', 'Agent ID (optional with --all)')
+    .option('-a, --all', 'List all agents\' budgets')
+    .option('-d, --daily <amount>', 'Set daily limit (dollars)')
+    .option('-w, --weekly <amount>', 'Set weekly limit (dollars)')
+    .option('-r, --per-request <amount>', 'Set per-request max (dollars)')
+    .option('--clear', 'Remove all spending limits')
+    .option('--json', 'Output as JSON')
+    .action(async (agentId, options) => {
+    const sessionId = config.get('sessionId');
+    if (!sessionId) {
+        console.log(chalk_1.default.yellow('Not logged in.'));
+        console.log(chalk_1.default.gray('Run: gopherhole login'));
+        process.exit(1);
+    }
+    // Handle --all flag
+    if (options.all) {
+        const spinner = (0, ora_1.default)('Fetching spending overview...').start();
+        try {
+            const res = await fetch(`${API_URL}/spending`, {
+                headers: { 'X-Session-ID': sessionId },
+            });
+            if (!res.ok) {
+                throw new Error('Failed to fetch spending overview');
+            }
+            const data = await res.json();
+            spinner.stop();
+            if (options.json) {
+                console.log(JSON.stringify(data, null, 2));
+                return;
+            }
+            console.log(chalk_1.default.bold('\n💰 Spending Overview\n'));
+            console.log(`  Today: $${data.totalSpentToday.toFixed(2)}    This week: $${data.totalSpentThisWeek.toFixed(2)}\n`);
+            if (data.agents.length === 0) {
+                console.log(chalk_1.default.gray('  No agents found.\n'));
+                return;
+            }
+            // Header
+            console.log(chalk_1.default.gray('  Agent               Daily         Weekly        Status'));
+            console.log(chalk_1.default.gray('  ─'.padEnd(70, '─')));
+            for (const agent of data.agents) {
+                const name = agent.name.padEnd(18).slice(0, 18);
+                let daily = '∞'.padEnd(12);
+                if (agent.daily?.limit !== null && agent.daily?.limit !== undefined) {
+                    const bar = getProgressBar(agent.daily.percent ?? 0, 5);
+                    daily = `$${agent.daily.spent.toFixed(0)}/$${agent.daily.limit.toFixed(0)} ${bar}`.padEnd(12);
+                }
+                else if (agent.daily?.spent) {
+                    daily = `$${agent.daily.spent.toFixed(2)}`.padEnd(12);
+                }
+                let weekly = '∞'.padEnd(12);
+                if (agent.weekly?.limit !== null && agent.weekly?.limit !== undefined) {
+                    const bar = getProgressBar(agent.weekly.percent ?? 0, 5);
+                    weekly = `$${agent.weekly.spent.toFixed(0)}/$${agent.weekly.limit.toFixed(0)} ${bar}`.padEnd(12);
+                }
+                else if (agent.weekly?.spent) {
+                    weekly = `$${agent.weekly.spent.toFixed(2)}`.padEnd(12);
+                }
+                const statusEmoji = {
+                    ok: brand.green('✅ OK'),
+                    warning: chalk_1.default.yellow('⚠️ Warning'),
+                    critical: chalk_1.default.red('🔴 Critical'),
+                    blocked: chalk_1.default.red('🚫 Blocked'),
+                    unlimited: chalk_1.default.gray('∞'),
+                }[agent.status] ?? agent.status;
+                console.log(`  ${name}  ${daily}  ${weekly}  ${statusEmoji}`);
+            }
+            console.log('');
+        }
+        catch (err) {
+            spinner.fail(chalk_1.default.red(err.message));
+            process.exit(1);
+        }
+        return;
+    }
+    // Need agentId for single-agent operations
+    if (!agentId) {
+        console.log(chalk_1.default.yellow('Please provide an agent ID or use --all'));
+        console.log(chalk_1.default.gray('Usage: gopherhole budget <agentId> [options]'));
+        console.log(chalk_1.default.gray('       gopherhole budget --all'));
+        process.exit(1);
+    }
+    // Handle --clear flag
+    if (options.clear) {
+        const spinner = (0, ora_1.default)('Removing spending limits...').start();
+        try {
+            const res = await fetch(`${API_URL}/agents/${agentId}/spending-limits`, {
+                method: 'DELETE',
+                headers: { 'X-Session-ID': sessionId },
+            });
+            if (!res.ok) {
+                throw new Error('Failed to remove spending limits');
+            }
+            spinner.succeed('Spending limits removed');
+            console.log(chalk_1.default.gray(`  Agent ${agentId} now has no spending limits.\n`));
+        }
+        catch (err) {
+            spinner.fail(chalk_1.default.red(err.message));
+            process.exit(1);
+        }
+        return;
+    }
+    // Handle setting limits
+    if (options.daily !== undefined || options.weekly !== undefined || options.perRequest !== undefined) {
+        const spinner = (0, ora_1.default)('Updating spending limits...').start();
+        try {
+            const body = {};
+            if (options.daily !== undefined) {
+                body.dailyLimit = options.daily === 'unlimited' ? null : parseFloat(options.daily);
+            }
+            if (options.weekly !== undefined) {
+                body.weeklyLimit = options.weekly === 'unlimited' ? null : parseFloat(options.weekly);
+            }
+            if (options.perRequest !== undefined) {
+                body.maxPerRequest = options.perRequest === 'unlimited' ? null : parseFloat(options.perRequest);
+            }
+            const res = await fetch(`${API_URL}/agents/${agentId}/spending-limits`, {
+                method: 'PUT',
+                headers: {
+                    'X-Session-ID': sessionId,
+                    'Content-Type': 'application/json',
+                },
+                body: JSON.stringify(body),
+            });
+            if (!res.ok) {
+                const err = await res.json();
+                throw new Error(err.error || 'Failed to update limits');
+            }
+            spinner.succeed('Spending limits updated');
+            // Show the updated limits
+            const data = await res.json();
+            console.log(chalk_1.default.gray('\n  New limits:'));
+            if (data.limits.daily) {
+                console.log(`    Daily: $${data.limits.daily.limit.toFixed(2)}`);
+            }
+            if (data.limits.weekly) {
+                console.log(`    Weekly: $${data.limits.weekly.limit.toFixed(2)}`);
+            }
+            if (data.limits.maxPerRequest) {
+                console.log(`    Per Request: $${data.limits.maxPerRequest.toFixed(2)}`);
+            }
+            console.log('');
+        }
+        catch (err) {
+            spinner.fail(chalk_1.default.red(err.message));
+            process.exit(1);
+        }
+        return;
+    }
+    // Default: show budget
+    const spinner = (0, ora_1.default)('Fetching budget...').start();
+    try {
+        const res = await fetch(`${API_URL}/agents/${agentId}/budget`, {
+            headers: { 'X-Session-ID': sessionId },
+        });
+        if (!res.ok) {
+            if (res.status === 404) {
+                throw new Error('Agent not found');
+            }
+            throw new Error('Failed to fetch budget');
+        }
+        const data = await res.json();
+        spinner.stop();
+        if (options.json) {
+            console.log(JSON.stringify(data, null, 2));
+            return;
+        }
+        console.log(chalk_1.default.bold(`\n💰 Budget for ${agentId}\n`));
+        if (!data.daily && !data.weekly && !data.maxPerRequest) {
+            console.log(chalk_1.default.gray('  No spending limits configured.\n'));
+            console.log(chalk_1.default.gray('  Set limits: gopherhole budget <agentId> -d 10 -w 50\n'));
+            return;
+        }
+        if (data.daily) {
+            const percent = Math.round((data.daily.spent / data.daily.limit) * 100);
+            const bar = getProgressBar(percent, 10);
+            const timeLeft = getTimeUntil(data.daily.resetsAt);
+            const color = percent >= 80 ? chalk_1.default.red : percent >= 50 ? chalk_1.default.yellow : brand.green;
+            console.log(`  Daily:        ${color(`$${data.daily.spent.toFixed(2)} / $${data.daily.limit.toFixed(2)}`)} (${percent}%) ${bar}`);
+            console.log(chalk_1.default.gray(`                Resets in ${timeLeft}`));
+        }
+        if (data.weekly) {
+            const percent = Math.round((data.weekly.spent / data.weekly.limit) * 100);
+            const bar = getProgressBar(percent, 10);
+            const timeLeft = getTimeUntil(data.weekly.resetsAt);
+            const color = percent >= 80 ? chalk_1.default.red : percent >= 50 ? chalk_1.default.yellow : brand.green;
+            console.log(`  Weekly:       ${color(`$${data.weekly.spent.toFixed(2)} / $${data.weekly.limit.toFixed(2)}`)} (${percent}%) ${bar}`);
+            console.log(chalk_1.default.gray(`                Resets ${timeLeft}`));
+        }
+        if (data.maxPerRequest) {
+            console.log(`  Per Request:  $${data.maxPerRequest.toFixed(2)} max`);
+        }
+        // Status
+        let status = brand.green('✅ OK');
+        const dailyPercent = data.daily ? (data.daily.spent / data.daily.limit) * 100 : 0;
+        const weeklyPercent = data.weekly ? (data.weekly.spent / data.weekly.limit) * 100 : 0;
+        const maxPercent = Math.max(dailyPercent, weeklyPercent);
+        if (maxPercent >= 100) {
+            status = chalk_1.default.red('🚫 Blocked');
+        }
+        else if (maxPercent >= 80) {
+            status = chalk_1.default.red('🔴 Critical');
+        }
+        else if (maxPercent >= 50) {
+            status = chalk_1.default.yellow('⚠️ Warning');
+        }
+        console.log(`\n  Status: ${status}\n`);
+    }
+    catch (err) {
+        spinner.fail(chalk_1.default.red(err.message));
+        process.exit(1);
+    }
+});
+// Helper: progress bar
+function getProgressBar(percent, width) {
+    const filled = Math.round((percent / 100) * width);
+    const empty = width - filled;
+    const color = percent >= 80 ? chalk_1.default.red : percent >= 50 ? chalk_1.default.yellow : brand.green;
+    return color('█'.repeat(filled)) + chalk_1.default.gray('░'.repeat(empty));
+}
+// Helper: time until
+function getTimeUntil(isoDate) {
+    const target = new Date(isoDate);
+    const now = new Date();
+    const diffMs = target.getTime() - now.getTime();
+    if (diffMs <= 0)
+        return 'now';
+    const hours = Math.floor(diffMs / (1000 * 60 * 60));
+    const minutes = Math.floor((diffMs % (1000 * 60 * 60)) / (1000 * 60));
+    if (hours > 24) {
+        const days = Math.floor(hours / 24);
+        return `${days}d ${hours % 24}h`;
+    }
+    if (hours > 0) {
+        return `${hours}h ${minutes}m`;
+    }
+    return `${minutes}m`;
+}
 // ========== STATUS COMMAND ==========
 program
     .command('status')

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gopherhole/cli",
-  "version": "0.1.13",
+  "version": "0.1.14",
   "description": "GopherHole CLI - Connect AI agents to the world",
   "main": "dist/index.js",
   "bin": {

package/src/index.ts

@@ -18,7 +18,7 @@ const brand = {
 };
 
 // Version
-const VERSION = '0.1.8';
+const VERSION = '0.1.14';
 
 // ASCII art banner
 function showBanner(context?: string) {
@@ -1891,6 +1891,302 @@ access
     }
   });
 
+// ========== BUDGET COMMAND ==========
+
+const budget = program
+  .command('budget')
+  .description(`View and manage agent spending limits
+
+${chalk.bold('Examples:')}
+  $ gopherhole budget my-agent              # View budget
+  $ gopherhole budget my-agent -d 10 -w 50  # Set daily=$10, weekly=$50
+  $ gopherhole budget my-agent --clear      # Remove all limits
+  $ gopherhole budget --all                 # List all agents' budgets
+`);
+
+budget
+  .argument('[agentId]', 'Agent ID (optional with --all)')
+  .option('-a, --all', 'List all agents\' budgets')
+  .option('-d, --daily <amount>', 'Set daily limit (dollars)')
+  .option('-w, --weekly <amount>', 'Set weekly limit (dollars)')
+  .option('-r, --per-request <amount>', 'Set per-request max (dollars)')
+  .option('--clear', 'Remove all spending limits')
+  .option('--json', 'Output as JSON')
+  .action(async (agentId: string | undefined, options) => {
+    const sessionId = config.get('sessionId') as string;
+    if (!sessionId) {
+      console.log(chalk.yellow('Not logged in.'));
+      console.log(chalk.gray('Run: gopherhole login'));
+      process.exit(1);
+    }
+
+    // Handle --all flag
+    if (options.all) {
+      const spinner = ora('Fetching spending overview...').start();
+      try {
+        const res = await fetch(`${API_URL}/spending`, {
+          headers: { 'X-Session-ID': sessionId },
+        });
+
+        if (!res.ok) {
+          throw new Error('Failed to fetch spending overview');
+        }
+
+        const data = await res.json() as {
+          agents: Array<{
+            agentId: string;
+            name: string;
+            daily: { spent: number; limit: number | null; percent: number | null } | null;
+            weekly: { spent: number; limit: number | null; percent: number | null } | null;
+            status: string;
+          }>;
+          totalSpentToday: number;
+          totalSpentThisWeek: number;
+        };
+        spinner.stop();
+
+        if (options.json) {
+          console.log(JSON.stringify(data, null, 2));
+          return;
+        }
+
+        console.log(chalk.bold('\n💰 Spending Overview\n'));
+        console.log(`  Today: $${data.totalSpentToday.toFixed(2)}    This week: $${data.totalSpentThisWeek.toFixed(2)}\n`);
+
+        if (data.agents.length === 0) {
+          console.log(chalk.gray('  No agents found.\n'));
+          return;
+        }
+
+        // Header
+        console.log(chalk.gray('  Agent               Daily         Weekly        Status'));
+        console.log(chalk.gray('  ─'.padEnd(70, '─')));
+
+        for (const agent of data.agents) {
+          const name = agent.name.padEnd(18).slice(0, 18);
+          
+          let daily = '∞'.padEnd(12);
+          if (agent.daily?.limit !== null && agent.daily?.limit !== undefined) {
+            const bar = getProgressBar(agent.daily.percent ?? 0, 5);
+            daily = `$${agent.daily.spent.toFixed(0)}/$${agent.daily.limit.toFixed(0)} ${bar}`.padEnd(12);
+          } else if (agent.daily?.spent) {
+            daily = `$${agent.daily.spent.toFixed(2)}`.padEnd(12);
+          }
+
+          let weekly = '∞'.padEnd(12);
+          if (agent.weekly?.limit !== null && agent.weekly?.limit !== undefined) {
+            const bar = getProgressBar(agent.weekly.percent ?? 0, 5);
+            weekly = `$${agent.weekly.spent.toFixed(0)}/$${agent.weekly.limit.toFixed(0)} ${bar}`.padEnd(12);
+          } else if (agent.weekly?.spent) {
+            weekly = `$${agent.weekly.spent.toFixed(2)}`.padEnd(12);
+          }
+
+          const statusEmoji = {
+            ok: brand.green('✅ OK'),
+            warning: chalk.yellow('⚠️ Warning'),
+            critical: chalk.red('🔴 Critical'),
+            blocked: chalk.red('🚫 Blocked'),
+            unlimited: chalk.gray('∞'),
+          }[agent.status] ?? agent.status;
+
+          console.log(`  ${name}  ${daily}  ${weekly}  ${statusEmoji}`);
+        }
+        console.log('');
+      } catch (err) {
+        spinner.fail(chalk.red((err as Error).message));
+        process.exit(1);
+      }
+      return;
+    }
+
+    // Need agentId for single-agent operations
+    if (!agentId) {
+      console.log(chalk.yellow('Please provide an agent ID or use --all'));
+      console.log(chalk.gray('Usage: gopherhole budget <agentId> [options]'));
+      console.log(chalk.gray('       gopherhole budget --all'));
+      process.exit(1);
+    }
+
+    // Handle --clear flag
+    if (options.clear) {
+      const spinner = ora('Removing spending limits...').start();
+      try {
+        const res = await fetch(`${API_URL}/agents/${agentId}/spending-limits`, {
+          method: 'DELETE',
+          headers: { 'X-Session-ID': sessionId },
+        });
+
+        if (!res.ok) {
+          throw new Error('Failed to remove spending limits');
+        }
+
+        spinner.succeed('Spending limits removed');
+        console.log(chalk.gray(`  Agent ${agentId} now has no spending limits.\n`));
+      } catch (err) {
+        spinner.fail(chalk.red((err as Error).message));
+        process.exit(1);
+      }
+      return;
+    }
+
+    // Handle setting limits
+    if (options.daily !== undefined || options.weekly !== undefined || options.perRequest !== undefined) {
+      const spinner = ora('Updating spending limits...').start();
+      try {
+        const body: Record<string, number | null> = {};
+        
+        if (options.daily !== undefined) {
+          body.dailyLimit = options.daily === 'unlimited' ? null : parseFloat(options.daily);
+        }
+        if (options.weekly !== undefined) {
+          body.weeklyLimit = options.weekly === 'unlimited' ? null : parseFloat(options.weekly);
+        }
+        if (options.perRequest !== undefined) {
+          body.maxPerRequest = options.perRequest === 'unlimited' ? null : parseFloat(options.perRequest);
+        }
+
+        const res = await fetch(`${API_URL}/agents/${agentId}/spending-limits`, {
+          method: 'PUT',
+          headers: { 
+            'X-Session-ID': sessionId,
+            'Content-Type': 'application/json',
+          },
+          body: JSON.stringify(body),
+        });
+
+        if (!res.ok) {
+          const err = await res.json();
+          throw new Error((err as { error?: string }).error || 'Failed to update limits');
+        }
+
+        spinner.succeed('Spending limits updated');
+        
+        // Show the updated limits
+        const data = await res.json() as { limits: Record<string, unknown> };
+        console.log(chalk.gray('\n  New limits:'));
+        if ((data.limits as any).daily) {
+          console.log(`    Daily: $${((data.limits as any).daily.limit as number).toFixed(2)}`);
+        }
+        if ((data.limits as any).weekly) {
+          console.log(`    Weekly: $${((data.limits as any).weekly.limit as number).toFixed(2)}`);
+        }
+        if ((data.limits as any).maxPerRequest) {
+          console.log(`    Per Request: $${((data.limits as any).maxPerRequest as number).toFixed(2)}`);
+        }
+        console.log('');
+      } catch (err) {
+        spinner.fail(chalk.red((err as Error).message));
+        process.exit(1);
+      }
+      return;
+    }
+
+    // Default: show budget
+    const spinner = ora('Fetching budget...').start();
+    try {
+      const res = await fetch(`${API_URL}/agents/${agentId}/budget`, {
+        headers: { 'X-Session-ID': sessionId },
+      });
+
+      if (!res.ok) {
+        if (res.status === 404) {
+          throw new Error('Agent not found');
+        }
+        throw new Error('Failed to fetch budget');
+      }
+
+      const data = await res.json() as {
+        daily: { limit: number; spent: number; remaining: number; resetsAt: string } | null;
+        weekly: { limit: number; spent: number; remaining: number; resetsAt: string } | null;
+        maxPerRequest: number | null;
+      };
+      spinner.stop();
+
+      if (options.json) {
+        console.log(JSON.stringify(data, null, 2));
+        return;
+      }
+
+      console.log(chalk.bold(`\n💰 Budget for ${agentId}\n`));
+
+      if (!data.daily && !data.weekly && !data.maxPerRequest) {
+        console.log(chalk.gray('  No spending limits configured.\n'));
+        console.log(chalk.gray('  Set limits: gopherhole budget <agentId> -d 10 -w 50\n'));
+        return;
+      }
+
+      if (data.daily) {
+        const percent = Math.round((data.daily.spent / data.daily.limit) * 100);
+        const bar = getProgressBar(percent, 10);
+        const timeLeft = getTimeUntil(data.daily.resetsAt);
+        const color = percent >= 80 ? chalk.red : percent >= 50 ? chalk.yellow : brand.green;
+        console.log(`  Daily:        ${color(`$${data.daily.spent.toFixed(2)} / $${data.daily.limit.toFixed(2)}`)} (${percent}%) ${bar}`);
+        console.log(chalk.gray(`                Resets in ${timeLeft}`));
+      }
+
+      if (data.weekly) {
+        const percent = Math.round((data.weekly.spent / data.weekly.limit) * 100);
+        const bar = getProgressBar(percent, 10);
+        const timeLeft = getTimeUntil(data.weekly.resetsAt);
+        const color = percent >= 80 ? chalk.red : percent >= 50 ? chalk.yellow : brand.green;
+        console.log(`  Weekly:       ${color(`$${data.weekly.spent.toFixed(2)} / $${data.weekly.limit.toFixed(2)}`)} (${percent}%) ${bar}`);
+        console.log(chalk.gray(`                Resets ${timeLeft}`));
+      }
+
+      if (data.maxPerRequest) {
+        console.log(`  Per Request:  $${data.maxPerRequest.toFixed(2)} max`);
+      }
+
+      // Status
+      let status = brand.green('✅ OK');
+      const dailyPercent = data.daily ? (data.daily.spent / data.daily.limit) * 100 : 0;
+      const weeklyPercent = data.weekly ? (data.weekly.spent / data.weekly.limit) * 100 : 0;
+      const maxPercent = Math.max(dailyPercent, weeklyPercent);
+      
+      if (maxPercent >= 100) {
+        status = chalk.red('🚫 Blocked');
+      } else if (maxPercent >= 80) {
+        status = chalk.red('🔴 Critical');
+      } else if (maxPercent >= 50) {
+        status = chalk.yellow('⚠️ Warning');
+      }
+
+      console.log(`\n  Status: ${status}\n`);
+    } catch (err) {
+      spinner.fail(chalk.red((err as Error).message));
+      process.exit(1);
+    }
+  });
+
+// Helper: progress bar
+function getProgressBar(percent: number, width: number): string {
+  const filled = Math.round((percent / 100) * width);
+  const empty = width - filled;
+  const color = percent >= 80 ? chalk.red : percent >= 50 ? chalk.yellow : brand.green;
+  return color('█'.repeat(filled)) + chalk.gray('░'.repeat(empty));
+}
+
+// Helper: time until
+function getTimeUntil(isoDate: string): string {
+  const target = new Date(isoDate);
+  const now = new Date();
+  const diffMs = target.getTime() - now.getTime();
+  
+  if (diffMs <= 0) return 'now';
+  
+  const hours = Math.floor(diffMs / (1000 * 60 * 60));
+  const minutes = Math.floor((diffMs % (1000 * 60 * 60)) / (1000 * 60));
+  
+  if (hours > 24) {
+    const days = Math.floor(hours / 24);
+    return `${days}d ${hours % 24}h`;
+  }
+  if (hours > 0) {
+    return `${hours}h ${minutes}m`;
+  }
+  return `${minutes}m`;
+}
+
 // ========== STATUS COMMAND ==========
 
 program