agent-tank 0.9.0

package/src/history-store.js

@@ -0,0 +1,273 @@
+/**
+ * History Store Module
+ *
+ * Stores historical usage snapshots to a local JSON file for trend analysis
+ * and pace evaluation. Automatically prunes records older than the configured
+ * retention period.
+ */
+
+const fs = require('node:fs');
+const path = require('node:path');
+const os = require('node:os');
+
+// Default configuration
+const DEFAULT_RETENTION_DAYS = 14;
+const DEFAULT_HISTORY_DIR = path.join(os.homedir(), '.agent-tank');
+const DEFAULT_HISTORY_FILE = 'history.json';
+
+/**
+ * HistoryStore manages persistent storage of usage snapshots.
+ *
+ * Usage snapshots are stored with timestamps and automatically pruned
+ * based on the configured retention period.
+ */
+class HistoryStore {
+  /**
+   * Create a HistoryStore instance.
+   *
+   * @param {Object} options - Configuration options
+   * @param {number} [options.retentionDays=14] - Number of days to retain history
+   * @param {string} [options.historyDir] - Directory for history file (default: ~/.agent-tank)
+   * @param {string} [options.historyFile] - History filename (default: history.json)
+   */
+  constructor(options = {}) {
+    this.retentionDays = options.retentionDays ?? DEFAULT_RETENTION_DAYS;
+    this.historyDir = options.historyDir ?? DEFAULT_HISTORY_DIR;
+    this.historyFile = options.historyFile ?? DEFAULT_HISTORY_FILE;
+    this.historyPath = path.join(this.historyDir, this.historyFile);
+
+    // In-memory cache of history data
+    this._history = null;
+    this._initialized = false;
+  }
+
+  /**
+   * Ensure the history directory exists.
+   * @private
+   */
+  _ensureDirectory() {
+    if (!fs.existsSync(this.historyDir)) {
+      fs.mkdirSync(this.historyDir, { recursive: true });
+    }
+  }
+
+  /**
+   * Load history from disk into memory.
+   * @private
+   */
+  _loadHistory() {
+    if (this._initialized) {
+      return this._history;
+    }
+
+    try {
+      this._ensureDirectory();
+
+      if (fs.existsSync(this.historyPath)) {
+        const data = fs.readFileSync(this.historyPath, 'utf8');
+        this._history = JSON.parse(data);
+
+        // Validate structure
+        if (!Array.isArray(this._history)) {
+          console.warn('[HistoryStore] Invalid history format, resetting to empty array');
+          this._history = [];
+        }
+      } else {
+        this._history = [];
+      }
+    } catch (err) {
+      console.error('[HistoryStore] Error loading history:', err.message);
+      this._history = [];
+    }
+
+    this._initialized = true;
+    return this._history;
+  }
+
+  /**
+   * Save history from memory to disk.
+   * @private
+   */
+  _saveHistory() {
+    try {
+      this._ensureDirectory();
+      fs.writeFileSync(this.historyPath, JSON.stringify(this._history, null, 2));
+    } catch (err) {
+      console.error('[HistoryStore] Error saving history:', err.message);
+    }
+  }
+
+  /**
+   * Calculate the cutoff timestamp for pruning.
+   * @private
+   * @returns {number} Unix timestamp in milliseconds
+   */
+  _getCutoffTimestamp() {
+    const now = Date.now();
+    const retentionMs = this.retentionDays * 24 * 60 * 60 * 1000;
+    return now - retentionMs;
+  }
+
+  /**
+   * Prune records older than the retention period.
+   * @private
+   */
+  _pruneOldRecords() {
+    const cutoff = this._getCutoffTimestamp();
+    const originalLength = this._history.length;
+
+    this._history = this._history.filter(record => {
+      const recordTime = new Date(record.timestamp).getTime();
+      return recordTime >= cutoff;
+    });
+
+    const prunedCount = originalLength - this._history.length;
+    if (prunedCount > 0) {
+      console.log(`[HistoryStore] Pruned ${prunedCount} old records (>${this.retentionDays} days)`);
+    }
+  }
+
+  /**
+   * Add a usage snapshot to history.
+   *
+   * @param {string} agentName - Name of the agent (e.g., 'claude', 'gemini', 'codex')
+   * @param {Object} usageData - The usage data to store
+   * @param {string} [timestamp] - ISO timestamp (defaults to current time)
+   * @returns {Object} The stored record
+   */
+  addSnapshot(agentName, usageData, timestamp = null) {
+    if (!agentName || typeof agentName !== 'string') {
+      throw new Error('agentName is required and must be a string');
+    }
+
+    if (!usageData || typeof usageData !== 'object') {
+      throw new Error('usageData is required and must be an object');
+    }
+
+    this._loadHistory();
+
+    const record = {
+      timestamp: timestamp || new Date().toISOString(),
+      agent: agentName,
+      usage: usageData
+    };
+
+    this._history.push(record);
+    this._pruneOldRecords();
+    this._saveHistory();
+
+    return record;
+  }
+
+  /**
+   * Get all history records, optionally filtered by agent.
+   *
+   * @param {string} [agentName] - Filter by agent name (optional)
+   * @returns {Array} Array of history records
+   */
+  getHistory(agentName = null) {
+    this._loadHistory();
+    this._pruneOldRecords();
+
+    if (agentName) {
+      return this._history.filter(record => record.agent === agentName);
+    }
+
+    return [...this._history];
+  }
+
+  /**
+   * Get the most recent snapshot for an agent.
+   *
+   * @param {string} agentName - Name of the agent
+   * @returns {Object|null} The most recent record or null
+   */
+  getLatestSnapshot(agentName) {
+    const agentHistory = this.getHistory(agentName);
+
+    if (agentHistory.length === 0) {
+      return null;
+    }
+
+    // Sort by timestamp descending and return the first
+    return agentHistory.sort((a, b) =>
+      new Date(b.timestamp).getTime() - new Date(a.timestamp).getTime()
+    )[0];
+  }
+
+  /**
+   * Get snapshots within a specific time window.
+   *
+   * @param {string} agentName - Name of the agent
+   * @param {number} windowSeconds - Time window in seconds from now
+   * @returns {Array} Array of records within the window
+   */
+  getSnapshotsInWindow(agentName, windowSeconds) {
+    const agentHistory = this.getHistory(agentName);
+    const cutoff = Date.now() - (windowSeconds * 1000);
+
+    return agentHistory.filter(record =>
+      new Date(record.timestamp).getTime() >= cutoff
+    ).sort((a, b) =>
+      new Date(a.timestamp).getTime() - new Date(b.timestamp).getTime()
+    );
+  }
+
+  /**
+   * Clear all history records.
+   *
+   * @param {string} [agentName] - Clear only records for this agent (optional)
+   */
+  clearHistory(agentName = null) {
+    this._loadHistory();
+
+    if (agentName) {
+      this._history = this._history.filter(record => record.agent !== agentName);
+    } else {
+      this._history = [];
+    }
+
+    this._saveHistory();
+  }
+
+  /**
+   * Get statistics about stored history.
+   *
+   * @returns {Object} Statistics object
+   */
+  getStats() {
+    this._loadHistory();
+
+    const stats = {
+      totalRecords: this._history.length,
+      retentionDays: this.retentionDays,
+      historyPath: this.historyPath,
+      agents: {}
+    };
+
+    for (const record of this._history) {
+      if (!stats.agents[record.agent]) {
+        stats.agents[record.agent] = {
+          count: 0,
+          oldest: null,
+          newest: null
+        };
+      }
+
+      const agentStats = stats.agents[record.agent];
+      agentStats.count++;
+
+      const recordTime = new Date(record.timestamp).getTime();
+      if (!agentStats.oldest || recordTime < new Date(agentStats.oldest).getTime()) {
+        agentStats.oldest = record.timestamp;
+      }
+      if (!agentStats.newest || recordTime > new Date(agentStats.newest).getTime()) {
+        agentStats.newest = record.timestamp;
+      }
+    }
+
+    return stats;
+  }
+}
+
+module.exports = { HistoryStore, DEFAULT_RETENTION_DAYS };

package/src/http-handler.js

@@ -0,0 +1,182 @@
+/**
+ * HTTP request handler utilities for AgentTank server.
+ */
+
+const { statusPage } = require('./status-page.js');
+
+/**
+ * Handle GET / - Status HTML page
+ */
+function handleStatusPage(res, tank) {
+  res.setHeader('Content-Type', 'text/html');
+  res.writeHead(200);
+  res.end(statusPage(tank.getStatus()));
+}
+
+/**
+ * Handle GET /status - JSON status for all agents
+ */
+function handleStatusJson(res, tank) {
+  const status = tank.getStatus();
+  console.log(`[HTTP] GET /status - returning status for ${Object.keys(status).length} agents`);
+  res.setHeader('Content-Type', 'application/json');
+  res.writeHead(200);
+  res.end(JSON.stringify(status, null, 2));
+}
+
+/**
+ * Handle GET /config - Server configuration
+ */
+function handleConfig(res, tank) {
+  const config = {
+    autoRefresh: {
+      enabled: tank.autoRefresh.enabled && tank.autoRefresh.interval > 0,
+      interval: tank.autoRefresh.interval,
+    },
+    history: {
+      retentionDays: tank.historyRetentionDays,
+    },
+    lastRefreshedAt: tank.lastRefreshedAt,
+  };
+  res.setHeader('Content-Type', 'application/json');
+  res.writeHead(200);
+  res.end(JSON.stringify(config, null, 2));
+}
+
+/**
+ * Handle GET /history - History statistics
+ */
+function handleHistoryStats(res, tank) {
+  const stats = tank.getHistoryStats();
+  res.setHeader('Content-Type', 'application/json');
+  res.writeHead(200);
+  res.end(JSON.stringify(stats, null, 2));
+}
+
+/**
+ * Handle GET /history/:agent - Agent-specific history
+ */
+function handleAgentHistory(res, tank, agentName) {
+  const history = tank.getHistory(agentName);
+  res.setHeader('Content-Type', 'application/json');
+  res.writeHead(200);
+  res.end(JSON.stringify(history, null, 2));
+}
+
+/**
+ * Handle GET /status/:agent - Agent-specific status
+ */
+function handleAgentStatus(res, tank, agentName) {
+  const status = tank.getAgentStatus(agentName);
+  if (!status) {
+    res.writeHead(404);
+    res.end(JSON.stringify({ error: 'Agent not found' }));
+    return;
+  }
+  res.setHeader('Content-Type', 'application/json');
+  res.writeHead(200);
+  res.end(JSON.stringify(status, null, 2));
+}
+
+/**
+ * Handle POST /refresh - Refresh all agents
+ */
+async function handleRefreshAll(res, tank) {
+  await tank.refreshAll();
+  res.setHeader('Content-Type', 'application/json');
+  res.writeHead(200);
+  res.end(JSON.stringify({ success: true, status: tank.getStatus() }));
+}
+
+/**
+ * Handle POST /refresh/:agent - Refresh specific agent
+ */
+async function handleRefreshAgent(res, tank, agentName) {
+  try {
+    await tank.refreshAgent(agentName);
+    res.setHeader('Content-Type', 'application/json');
+    res.writeHead(200);
+    res.end(JSON.stringify({ success: true, status: tank.getAgentStatus(agentName) }));
+  } catch (err) {
+    res.writeHead(404);
+    res.end(JSON.stringify({ error: err.message }));
+  }
+}
+
+/**
+ * Handle 404 - Not Found
+ */
+function handleNotFound(res) {
+  res.writeHead(404);
+  res.end(JSON.stringify({ error: 'Not found' }));
+}
+
+/**
+ * Route and handle HTTP requests.
+ * @param {Object} req - HTTP request
+ * @param {Object} res - HTTP response
+ * @param {Object} tank - AgentTank instance
+ */
+async function handleRequest(req, res, tank) {
+  const url = new URL(req.url, `http://localhost:${tank.port}`);
+  const path = url.pathname;
+
+  // CORS headers
+  res.setHeader('Access-Control-Allow-Origin', '*');
+  res.setHeader('Access-Control-Allow-Methods', 'GET, POST, OPTIONS');
+
+  if (req.method === 'OPTIONS') {
+    res.writeHead(204);
+    res.end();
+    return;
+  }
+
+  if (!tank.authenticate(req, url)) {
+    res.setHeader('WWW-Authenticate', 'Basic realm="LLM Limit Watcher"');
+    res.writeHead(401);
+    res.end(JSON.stringify({ error: 'Unauthorized' }));
+    return;
+  }
+
+  // Route requests
+  if (req.method === 'GET') {
+    if (path === '/') {
+      handleStatusPage(res, tank);
+    } else if (path === '/status') {
+      handleStatusJson(res, tank);
+    } else if (path === '/config') {
+      handleConfig(res, tank);
+    } else if (path === '/history') {
+      handleHistoryStats(res, tank);
+    } else if (path.startsWith('/history/')) {
+      handleAgentHistory(res, tank, path.slice(9));
+    } else if (path.startsWith('/status/')) {
+      handleAgentStatus(res, tank, path.slice(8));
+    } else {
+      handleNotFound(res);
+    }
+  } else if (req.method === 'POST') {
+    if (path === '/refresh') {
+      await handleRefreshAll(res, tank);
+    } else if (path.startsWith('/refresh/')) {
+      await handleRefreshAgent(res, tank, path.slice(9));
+    } else {
+      handleNotFound(res);
+    }
+  } else {
+    handleNotFound(res);
+  }
+}
+
+module.exports = {
+  handleRequest,
+  handleStatusPage,
+  handleStatusJson,
+  handleConfig,
+  handleHistoryStats,
+  handleAgentHistory,
+  handleAgentStatus,
+  handleRefreshAll,
+  handleRefreshAgent,
+  handleNotFound
+};

package/src/icons.js

@@ -0,0 +1,67 @@
+// Brand icons for each agent - inline SVG for performance and styling flexibility
+
+// Tank logo icon for Agent Tank branding - minimalist U-shaped vertical tank
+const tankIcon = `<svg class="tank-logo" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" xmlns="http://www.w3.org/2000/svg">
+  <path d="M6 4v12a4 4 0 0 0 4 4h4a4 4 0 0 0 4-4V4"/>
+  <line x1="6" y1="4" x2="18" y2="4"/>
+  <line x1="8" y1="8" x2="16" y2="8" opacity="0.3"/>
+  <line x1="8" y1="12" x2="16" y2="12" opacity="0.3"/>
+</svg>`;
+
+const agentIcons = {
+  claude: `<svg class="agent-icon" viewBox="0 0 24 24" fill="currentColor" xmlns="http://www.w3.org/2000/svg">
+    <path d="M4.709 15.955l4.72-2.647.08-.23-.08-.128H9.2l-.79-.048-2.698-.073-2.339-.097-2.266-.122-.571-.121L0 11.784l.055-.352.48-.321.686.06 1.52.103 2.278.158 1.652.097 2.449.255h.389l.055-.157-.134-.098-.103-.097-2.358-1.596-2.552-1.688-1.336-.972-.724-.491-.364-.462-.158-1.008.656-.722.881.06.225.061.893.686 1.908 1.476 2.491 1.833.365.304.145-.103.019-.073-.164-.274-1.355-2.446-1.446-2.49-.644-1.032-.17-.619a2.97 2.97 0 01-.104-.729L6.283.134 6.696 0l.996.134.42.364.62 1.414 1.002 2.229 1.555 3.03.456.898.243.832.091.255h.158V9.01l.128-1.706.237-2.095.23-2.695.08-.76.376-.91.747-.492.584.28.48.685-.067.444-.286 1.851-.559 2.903-.364 1.942h.212l.243-.242.985-1.306 1.652-2.064.73-.82.85-.904.547-.431h1.033l.76 1.129-.34 1.166-1.064 1.347-.881 1.142-1.264 1.7-.79 1.36.073.11.188-.02 2.856-.606 1.543-.28 1.841-.315.833.388.091.395-.328.807-1.969.486-2.309.462-3.439.813-.042.03.049.061 1.549.146.662.036h1.622l3.02.225.79.522.474.638-.079.485-1.215.62-1.64-.389-3.829-.91-1.312-.329h-.182v.11l1.093 1.068 2.006 1.81 2.509 2.33.127.578-.322.455-.34-.049-2.205-1.657-.851-.747-1.926-1.62h-.128v.17l.444.649 2.345 3.521.122 1.08-.17.353-.608.213-.668-.122-1.374-1.925-1.415-2.167-1.143-1.943-.14.08-.674 7.254-.316.37-.729.28-.607-.461-.322-.747.322-1.476.389-1.924.315-1.53.286-1.9.17-.632-.012-.042-.14.018-1.434 1.967-2.18 2.945-1.726 1.845-.414.164-.717-.37.067-.662.401-.589 2.388-3.036 1.44-1.882.93-1.086-.006-.158h-.055L4.132 18.56l-1.13.146-.487-.456.061-.746.231-.243 1.908-1.312-.006.006z" />
+  </svg>`,
+  gemini: `<svg class="agent-icon" viewBox="0 0 24 24" fill="currentColor" xmlns="http://www.w3.org/2000/svg">
+    <path d="M20.616 10.835a14.147 14.147 0 01-4.45-3.001 14.111 14.111 0 01-3.678-6.452.503.503 0 00-.975 0 14.134 14.134 0 01-3.679 6.452 14.155 14.155 0 01-4.45 3.001c-.65.28-1.318.505-2.002.678a.502.502 0 000 .975c.684.172 1.35.397 2.002.677a14.147 14.147 0 014.45 3.001 14.112 14.112 0 013.679 6.453.502.502 0 00.975 0c.172-.685.397-1.351.677-2.003a14.145 14.145 0 013.001-4.45 14.113 14.113 0 016.453-3.678.503.503 0 000-.975 13.245 13.245 0 01-2.003-.678z" />
+  </svg>`,
+  codex: `<svg class="agent-icon" viewBox="0 0 24 24" fill="currentColor" xmlns="http://www.w3.org/2000/svg">
+    <path d="M22.2819 9.8211a5.9847 5.9847 0 0 0-.5157-4.9108 6.0462 6.0462 0 0 0-6.5098-2.9A6.0651 6.0651 0 0 0 4.9807 4.1818a5.9847 5.9847 0 0 0-3.9977 2.9 6.0462 6.0462 0 0 0 .7427 7.0966 5.98 5.98 0 0 0 .511 4.9107 6.051 6.051 0 0 0 6.5146 2.9001A5.9847 5.9847 0 0 0 13.2599 24a6.0557 6.0557 0 0 0 5.7718-4.2058 5.9894 5.9894 0 0 0 3.9977-2.9001 6.0557 6.0557 0 0 0-.7475-7.0729zm-9.022 12.6081a4.4755 4.4755 0 0 1-2.8764-1.0408l.1419-.0804 4.7783-2.7582a.7948.7948 0 0 0 .3927-.6813v-6.7369l2.02 1.1686a.071.071 0 0 1 .038.052v5.5826a4.504 4.504 0 0 1-4.4945 4.4944zm-9.6607-4.1254a4.4708 4.4708 0 0 1-.5346-3.0137l.142.0852 4.783 2.7582a.7712.7712 0 0 0 .7806 0l5.8428-3.3685v2.3324a.0804.0804 0 0 1-.0332.0615L9.74 19.9502a4.4992 4.4992 0 0 1-6.1408-1.6464zM2.3408 7.8956a4.485 4.485 0 0 1 2.3655-1.9728V11.6a.7664.7664 0 0 0 .3879.6765l5.8144 3.3543-2.0201 1.1685a.0757.0757 0 0 1-.071 0l-4.8303-2.7865A4.504 4.504 0 0 1 2.3408 7.872zm16.5963 3.8558L13.1038 8.364 15.1192 7.2a.0757.0757 0 0 1 .071 0l4.8303 2.7913a4.4944 4.4944 0 0 1-.6765 8.1042v-5.6772a.79.79 0 0 0-.407-.667zm2.0107-3.0231l-.142-.0852-4.7735-2.7818a.7759.7759 0 0 0-.7854 0L9.409 9.2297V6.8974a.0662.0662 0 0 1 .0284-.0615l4.8303-2.7866a4.4992 4.4992 0 0 1 6.6802 4.66zM8.3065 12.863l-2.02-1.1638a.0804.0804 0 0 1-.038-.0567V6.0742a4.4992 4.4992 0 0 1 7.3757-3.4537l-.142.0805L8.704 5.459a.7948.7948 0 0 0-.3927.6813zm1.0976-2.3654l2.602-1.4998 2.6069 1.4998v2.9994l-2.5974 1.4997-2.6067-1.4997Z" />
+  </svg>`
+};
+
+// Inline SVG refresh icon for card headers
+const refreshIcon = `<svg viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round">
+  <path d="M21 12a9 9 0 1 1-9-9c2.52 0 4.93 1 6.74 2.74L21 8"/>
+  <path d="M21 3v5h-5"/>
+</svg>`;
+
+// Crosshair/target icon for tracking metrics
+const trackIcon = `<svg viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round">
+  <circle cx="12" cy="12" r="10"/>
+  <circle cx="12" cy="12" r="6"/>
+  <circle cx="12" cy="12" r="2"/>
+  <line x1="12" y1="2" x2="12" y2="6"/>
+  <line x1="12" y1="18" x2="12" y2="22"/>
+  <line x1="2" y1="12" x2="6" y2="12"/>
+  <line x1="18" y1="12" x2="22" y2="12"/>
+</svg>`;
+
+// Monitor icon for top nav toggle
+const monitorIcon = `<svg viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round">
+  <rect x="2" y="3" width="20" height="14" rx="2" ry="2"/>
+  <line x1="8" y1="21" x2="16" y2="21"/>
+  <line x1="12" y1="17" x2="12" y2="21"/>
+</svg>`;
+
+// Sync icon (circular arrows) for Refresh All button
+const syncIcon = `<svg class="sync-icon" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round">
+  <path d="M21 12a9 9 0 1 1-9-9c2.52 0 4.93 1 6.74 2.74L21 8"/>
+  <path d="M21 3v5h-5"/>
+</svg>`;
+
+// Copy icon for copying API endpoint
+const copyIcon = `<svg class="copy-icon" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round">
+  <rect x="9" y="9" width="13" height="13" rx="2" ry="2"/>
+  <path d="M5 15H4a2 2 0 0 1-2-2V4a2 2 0 0 1 2-2h9a2 2 0 0 1 2 2v1"/>
+</svg>`;
+
+module.exports = {
+  agentIcons,
+  refreshIcon,
+  trackIcon,
+  monitorIcon,
+  tankIcon,
+  syncIcon,
+  copyIcon
+};