troxy-cli 1.0.2 → 1.0.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "troxy-cli",
-  "version": "1.0.2",
+  "version": "1.0.4",
   "description": "AI payment control — protect your agent's payments with policies",
   "type": "module",
   "bin": {

package/src/api.js

@@ -49,7 +49,7 @@ export const api = {
   evaluate: (body, apiKey) => request('POST', '/evaluate', { apiKey, body }),
 
   // MCP heartbeat (agent API key)
-  mcpHeartbeat: (apiKey) => request('POST', '/mcp/heartbeat', { apiKey }),
+  mcpHeartbeat: (apiKey, agentName) => request('POST', '/mcp/heartbeat', { apiKey, body: agentName ? { agent_name: agentName } : undefined }),
 };
 
 // Named export for backwards compat with init.js + mcp-server.js

package/src/init.js

@@ -1,8 +1,15 @@
-import fs   from 'fs';
-import os   from 'os';
-import path from 'path';
-import { saveConfig }       from './config.js';
-import { evaluatePayment }  from './api.js';
+import fs             from 'fs';
+import os             from 'os';
+import path           from 'path';
+import readline       from 'readline';
+import { execSync }   from 'child_process';
+import { saveConfig } from './config.js';
+import { evaluatePayment } from './api.js';
+
+function prompt(question) {
+  const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
+  return new Promise(resolve => rl.question(question, ans => { rl.close(); resolve(ans.trim()); }));
+}
 
 // MCP config locations per client and platform
 const MCP_CLIENTS = [
@@ -66,8 +73,15 @@ export async function runInit({ key } = {}) {
     console.log('✓');
   }
 
+  // Ask for agent name
+  const agentName = await prompt('  What would you like to name this agent? (e.g. "Shopping Bot"): ');
+  if (!agentName) {
+    console.error('\n  Error: agent name is required.\n');
+    process.exit(1);
+  }
+
   // Save config
-  saveConfig({ apiKey: key });
+  saveConfig({ apiKey: key, agentName });
   console.log('  Config saved (~/.troxy/config.json)  ✓');
 
   // Detect and patch MCP clients
@@ -96,10 +110,83 @@ export async function runInit({ key } = {}) {
     console.log('\n  Restart your MCP client to activate Troxy.');
   }
 
+  // Install background service so MCP server survives reboots
+  console.log('\n  Setting up background service...');
+  try {
+    installService(key, agentName);
+    console.log('  Background service installed  ✓');
+  } catch (err) {
+    console.log(`  Background service  ✗  (${err.message})`);
+    console.log('  You can start it manually with: troxy mcp &');
+  }
+
   console.log('\n  Your payments are now protected.');
   console.log('  Dashboard → https://dashboard.troxy.ai\n');
 }
 
+function installService(apiKey, agentName) {
+  const platform = process.platform;
+  const troxy    = process.execPath.replace(/node$/, 'troxy') ;
+
+  if (platform === 'linux') {
+    const unit = `[Unit]
+Description=Troxy MCP Server
+After=network.target
+
+[Service]
+ExecStart=${troxy} mcp
+Restart=always
+RestartSec=10
+User=${os.userInfo().username}
+Environment=TROXY_API_KEY=${apiKey}
+Environment=TROXY_AGENT_NAME=${agentName}
+
+[Install]
+WantedBy=multi-user.target
+`;
+    fs.writeFileSync('/tmp/troxy-mcp.service', unit);
+    execSync('sudo mv /tmp/troxy-mcp.service /etc/systemd/system/troxy-mcp.service');
+    execSync('sudo systemctl daemon-reload');
+    execSync('sudo systemctl enable troxy-mcp');
+    execSync('sudo systemctl restart troxy-mcp');
+
+  } else if (platform === 'darwin') {
+    const plist = `<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+<dict>
+  <key>Label</key>
+  <string>ai.troxy.mcp</string>
+  <key>ProgramArguments</key>
+  <array>
+    <string>${troxy}</string>
+    <string>mcp</string>
+  </array>
+  <key>EnvironmentVariables</key>
+  <dict>
+    <key>TROXY_API_KEY</key>
+    <string>${apiKey}</string>
+    <key>TROXY_AGENT_NAME</key>
+    <string>${agentName}</string>
+  </dict>
+  <key>RunAtLoad</key>
+  <true/>
+  <key>KeepAlive</key>
+  <true/>
+</dict>
+</plist>
+`;
+    const plistPath = path.join(os.homedir(), 'Library/LaunchAgents/ai.troxy.mcp.plist');
+    fs.mkdirSync(path.dirname(plistPath), { recursive: true });
+    fs.writeFileSync(plistPath, plist);
+    try { execSync(`launchctl unload ${plistPath} 2>/dev/null`); } catch {}
+    execSync(`launchctl load ${plistPath}`);
+
+  } else {
+    throw new Error('Auto-start not supported on this platform. Start manually: troxy mcp &');
+  }
+}
+
 function mcpEntry(apiKey) {
   return {
     troxy: {

package/src/mcp-server.js

@@ -8,8 +8,9 @@ import { loadConfig }      from './config.js';
 import { evaluatePayment, api } from './api.js';
 
 export async function runMcp() {
-  const config = loadConfig();
-  const apiKey = process.env.TROXY_API_KEY || config?.apiKey;
+  const config    = loadConfig();
+  const apiKey    = process.env.TROXY_API_KEY || config?.apiKey;
+  const agentName = process.env.TROXY_AGENT_NAME || config?.agentName;
 
   if (!apiKey) {
     process.stderr.write(
@@ -70,6 +71,7 @@ export async function runMcp() {
     }
 
     const args   = request.params.arguments ?? {};
+    if (agentName && !args.agent) args.agent = agentName;
     const result = await evaluatePayment(args, apiKey);
 
     if (result.error) {
@@ -109,7 +111,7 @@ export async function runMcp() {
   // Must be set up before server.connect() since stdio transport keeps the
   // event loop running but connect() may not return in all environments.
   const sendHeartbeat = () =>
-    api.mcpHeartbeat(apiKey)
+    api.mcpHeartbeat(apiKey, agentName)
       .then(() => process.stderr.write('[troxy] heartbeat ok\n'))
       .catch(err => process.stderr.write(`[troxy] heartbeat failed: ${err.message}\n`));
   sendHeartbeat();