@hasna/economy 0.2.10 → 0.2.11

package/dist/mcp/index.js

@@ -1,7 +1,6 @@
 #!/usr/bin/env bun
 // @bun
 var __defProp = Object.defineProperty;
-var __commonJS = (cb, mod) => () => (mod || cb((mod = { exports: {} }).exports, mod), mod.exports);
 var __returnValue = (v) => v;
 function __exportSetter(name, newValue) {
   this[name] = __returnValue.bind(null, newValue);
@@ -16,7 +15,6 @@ var __export = (target, all) => {
     });
 };
 var __esm = (fn, res) => () => (fn && (res = fn(fn = 0)), res);
-var __require = import.meta.require;
 
 // src/lib/pricing.ts
 var exports_pricing = {};
@@ -507,81 +505,13 @@ function seedModelPricing(db, defaults) {
 }
 var init_database = () => {};
 
-// package.json
-var require_package = __commonJS((exports, module) => {
-  module.exports = {
-    name: "@hasna/economy",
-    version: "0.2.10",
-    description: "AI coding cost tracker \u2014 CLI + MCP server + REST API + web dashboard for Claude Code, Codex, and Gemini",
-    type: "module",
-    main: "dist/index.js",
-    types: "dist/index.d.ts",
-    bin: {
-      economy: "dist/cli/index.js",
-      "economy-mcp": "dist/mcp/index.js",
-      "economy-serve": "dist/server/index.js"
-    },
-    exports: {
-      ".": {
-        types: "./dist/index.d.ts",
-        import: "./dist/index.js"
-      }
-    },
-    files: [
-      "dist",
-      "LICENSE"
-    ],
-    scripts: {
-      build: "cd dashboard && bun run build && cd .. && bun build src/cli/index.ts --outdir dist/cli --target bun --packages external && bun build src/mcp/index.ts --outdir dist/mcp --target bun --packages external && bun build src/server/index.ts --outdir dist/server --target bun --packages external && bun build src/index.ts --outdir dist --target bun --packages external && tsc --emitDeclarationOnly --outDir dist",
-      "build:cli": "bun build src/cli/index.ts --outdir dist/cli --target bun --packages external",
-      "build:mcp": "bun build src/mcp/index.ts --outdir dist/mcp --target bun --packages external",
-      "build:server": "bun build src/server/index.ts --outdir dist/server --target bun --packages external",
-      "build:lib": "bun build src/index.ts --outdir dist --target bun --packages external",
-      "build:dashboard": "cd dashboard && bun run build",
-      typecheck: "tsc --noEmit",
-      test: "bun test",
-      "dev:cli": "bun run src/cli/index.ts",
-      "dev:mcp": "bun run src/mcp/index.ts",
-      "dev:serve": "bun run src/server/index.ts",
-      postinstall: "mkdir -p $HOME/.hasna/economy/training 2>/dev/null || true"
-    },
-    keywords: [
-      "economy",
-      "cost",
-      "ai",
-      "claude",
-      "codex",
-      "gemini",
-      "mcp",
-      "cli",
-      "budget",
-      "tracking"
-    ],
-    author: "hasna",
-    license: "Apache-2.0",
-    publishConfig: {
-      registry: "https://registry.npmjs.org",
-      access: "public"
-    },
-    dependencies: {
-      "@hasna/cloud": "^0.1.0",
-      "@modelcontextprotocol/sdk": "^1.12.1",
-      chalk: "^5.4.1",
-      commander: "^13.1.0"
-    },
-    devDependencies: {
-      "@types/bun": "latest",
-      "bun-types": "latest",
-      typescript: "^5.7.2"
-    }
-  };
-});
-
 // src/mcp/index.ts
 init_database();
-import { Server } from "@modelcontextprotocol/sdk/server/index.js";
+import { randomUUID } from "crypto";
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
-import { CallToolRequestSchema, ListToolsRequestSchema } from "@modelcontextprotocol/sdk/types.js";
+import { registerCloudTools } from "@hasna/cloud";
+import { z } from "zod";
 
 // src/ingest/claude.ts
 init_database();
@@ -727,7 +657,7 @@ init_database();
 import { existsSync as existsSync3, readFileSync as readFileSync2 } from "fs";
 import { homedir as homedir3 } from "os";
 import { join as join3, basename as basename2 } from "path";
-import { Database as Database2 } from "bun:sqlite";
+import { Database as BunDatabase } from "bun:sqlite";
 var CODEX_DB_PATH = join3(homedir3(), ".codex", "state_5.sqlite");
 var CODEX_CONFIG_PATH = join3(homedir3(), ".codex", "config.toml");
 async function ingestCodex(db, verbose = false) {
@@ -739,7 +669,7 @@ async function ingestCodex(db, verbose = false) {
   let codexDb = null;
   let ingested = 0;
   try {
-    codexDb = new Database2(CODEX_DB_PATH, { readonly: true });
+    codexDb = new BunDatabase(CODEX_DB_PATH, { readonly: true });
     const threads = codexDb.prepare(`SELECT id, cwd, created_at, updated_at, tokens_used, title FROM threads WHERE tokens_used > 0`).all();
     for (const thread of threads) {
       const stateKey = thread.id;
@@ -850,11 +780,91 @@ async function ingestGemini(db, verbose) {
   return { sessions: totalSessions };
 }
 
+// src/lib/package-metadata.ts
+import { readFileSync as readFileSync4 } from "fs";
+var cachedMetadata = null;
+function getPackageMetadata() {
+  if (cachedMetadata)
+    return cachedMetadata;
+  const raw = readFileSync4(new URL("../../package.json", import.meta.url), "utf8");
+  const parsed = JSON.parse(raw);
+  cachedMetadata = {
+    name: parsed.name ?? "@hasna/economy",
+    version: parsed.version ?? "0.0.0"
+  };
+  return cachedMetadata;
+}
+var packageMetadata = getPackageMetadata();
+
 // src/mcp/index.ts
 init_pricing();
+function printHelp() {
+  console.log(`Usage: economy-mcp [options]
+
+Runs the ${packageMetadata.name} MCP stdio server.
+
+Options:
+  -V, --version  output the version number
+  -h, --help     display help for command`);
+}
+var args = process.argv.slice(2);
+if (args.includes("--help") || args.includes("-h")) {
+  printHelp();
+  process.exit(0);
+}
+if (args.includes("--version") || args.includes("-V")) {
+  console.log(packageMetadata.version);
+  process.exit(0);
+}
 var db = openDatabase();
 ensurePricingSeeded(db);
-var server = new Server({ name: "economy", version: "0.2.2" }, { capabilities: { tools: {} } });
+var server = new McpServer({
+  name: "economy",
+  version: packageMetadata.version
+});
+var _econAgents = new Map;
+var TOOL_NAMES = [
+  "get_cost_summary",
+  "get_sessions",
+  "get_top_sessions",
+  "get_model_breakdown",
+  "get_project_breakdown",
+  "get_budget_status",
+  "get_daily",
+  "get_session_detail",
+  "sync",
+  "search_tools",
+  "describe_tools",
+  "get_goals",
+  "set_goal",
+  "remove_goal",
+  "register_agent",
+  "heartbeat",
+  "set_focus",
+  "list_agents",
+  "send_feedback"
+];
+var TOOL_DESCRIPTIONS = {
+  get_cost_summary: "period(today|week|month|year|all) -> {total_usd, sessions, requests, tokens, summary}",
+  get_sessions: "agent(claude|codex|gemini), project(partial), limit(20) -> compact session table",
+  get_top_sessions: "n(10), agent(claude|codex|gemini) -> top sessions by cost",
+  get_model_breakdown: "no params -> model, requests, tokens, cost",
+  get_project_breakdown: "no params -> project_name, sessions, cost",
+  get_budget_status: "no params -> budget limits, current spend, percent_used, is_over_alert",
+  get_daily: "days(30) -> daily cost table grouped by date and agent",
+  get_session_detail: "session_id(prefix ok) -> per-request breakdown with model, tokens, cost",
+  sync: "sources(all|claude|codex|gemini) -> ingest latest cost data",
+  search_tools: "query substring -> tool name list",
+  describe_tools: "names[] -> one-line parameter hints",
+  get_goals: "no params -> goal progress summary",
+  set_goal: "period(day|week|month|year), limit_usd, project_path?, agent? -> create goal",
+  remove_goal: "id -> delete goal",
+  register_agent: "name, session_id? -> register agent session",
+  heartbeat: "agent_id -> update last_seen_at",
+  set_focus: "agent_id, project_id? -> set active project context",
+  list_agents: "no params -> registered agent list",
+  send_feedback: "message, email?, category? -> save feedback locally"
+};
 var fmtUsd = (n) => "$" + n.toLocaleString("en-US", { minimumFractionDigits: 2, maximumFractionDigits: 2 });
 var fmtTok = (n) => n >= 1e9 ? `${(n / 1e9).toFixed(1)}B` : n >= 1e6 ? `${(n / 1e6).toFixed(1)}M` : n >= 1000 ? `${(n / 1000).toFixed(1)}k` : String(n);
 function fmtSession(s) {
@@ -865,262 +875,226 @@ function fmtSession(s) {
   const tok = fmtTok(Number(s["total_tokens"] ?? 0));
   return `${id} ${agent.padEnd(6)} ${cost.padEnd(10)} ${tok.padEnd(8)} ${proj}`;
 }
-var TOOLS = [
-  { name: "get_cost_summary", description: "Cost summary (total_usd, sessions, requests, tokens, human summary). period: today|week|month|year|all", inputSchema: { type: "object", properties: { period: { type: "string", enum: ["today", "week", "month", "year", "all"] } } } },
-  { name: "get_sessions", description: "List sessions. Returns compact table. Params: agent, project, limit(20)", inputSchema: { type: "object", properties: { agent: { type: "string" }, project: { type: "string" }, limit: { type: "number" } } } },
-  { name: "get_top_sessions", description: "Top sessions by cost. Params: n(10), agent", inputSchema: { type: "object", properties: { n: { type: "number" }, agent: { type: "string" } } } },
-  { name: "get_model_breakdown", description: "Cost per model. No params.", inputSchema: { type: "object", properties: {} } },
-  { name: "get_project_breakdown", description: "Cost per project. No params.", inputSchema: { type: "object", properties: {} } },
-  { name: "get_budget_status", description: "Budget limits vs spend, percent used, alert flags. No params.", inputSchema: { type: "object", properties: {} } },
-  { name: "get_daily", description: "Daily cost table by agent. Params: days(30)", inputSchema: { type: "object", properties: { days: { type: "number" } } } },
-  { name: "get_session_detail", description: "Per-request breakdown of a single session. Params: session_id (prefix ok)", inputSchema: { type: "object", properties: { session_id: { type: "string" } }, required: ["session_id"] } },
-  { name: "sync", description: "Ingest new cost data. sources: all|claude|codex|gemini", inputSchema: { type: "object", properties: { sources: { type: "string", enum: ["all", "claude", "codex", "gemini"] } } } },
-  { name: "search_tools", description: "List tool names matching query. Use first to find relevant tools.", inputSchema: { type: "object", properties: { query: { type: "string" } } } },
-  { name: "describe_tools", description: "Get param hints for specific tools by name.", inputSchema: { type: "object", properties: { names: { type: "array", items: { type: "string" } } }, required: ["names"] } },
-  { name: "get_goals", description: "All spending goals with current progress. No params.", inputSchema: { type: "object", properties: {} } },
-  { name: "set_goal", description: "Create/update a spending goal. period(day|week|month|year), limit_usd, project_path?, agent?", inputSchema: { type: "object", properties: { period: { type: "string" }, limit_usd: { type: "number" }, project_path: { type: "string" }, agent: { type: "string" } }, required: ["period", "limit_usd"] } },
-  { name: "remove_goal", description: "Delete a goal by id.", inputSchema: { type: "object", properties: { id: { type: "string" } }, required: ["id"] } },
-  { name: "register_agent", description: "Register agent session.", inputSchema: { type: "object", properties: { name: { type: "string" }, session_id: { type: "string" } }, required: ["name"] } },
-  { name: "heartbeat", description: "Update last_seen_at.", inputSchema: { type: "object", properties: { agent_id: { type: "string" } }, required: ["agent_id"] } },
-  { name: "set_focus", description: "Set active project context.", inputSchema: { type: "object", properties: { agent_id: { type: "string" }, project_id: { type: "string" } }, required: ["agent_id"] } },
-  { name: "list_agents", description: "List all registered agents.", inputSchema: { type: "object", properties: {} } },
-  { name: "send_feedback", description: "Send feedback about this service.", inputSchema: { type: "object", properties: { message: { type: "string" }, email: { type: "string" }, category: { type: "string", enum: ["bug", "feature", "general"] } }, required: ["message"] } }
-];
-var TOOL_DESCRIPTIONS = {
-  get_cost_summary: "period(today|week|month|year|all) \u2192 {total_usd, sessions, requests, tokens, summary}",
-  get_sessions: "agent(claude|codex), project(partial), limit(20) \u2192 compact session table",
-  get_top_sessions: "n(10), agent(claude|codex) \u2192 top sessions by cost",
-  get_model_breakdown: "no params \u2192 model, requests, tokens, cost",
-  get_project_breakdown: "no params \u2192 project_name, sessions, cost",
-  get_budget_status: "no params \u2192 budget limits, current spend, percent_used, is_over_alert",
-  get_daily: "days(30) \u2192 daily cost table grouped by date and agent",
-  get_session_detail: "session_id(prefix ok) \u2192 per-request breakdown with model, tokens, cost",
-  sync: "sources(all|claude|codex|gemini) \u2192 {files, requests, sessions} ingested",
-  get_goals: "no params \u2192 period, scope, limit, spent, percent, status(ON TRACK/AT RISK/OVER)",
-  set_goal: "period(day|week|month|year), limit_usd, project_path?, agent? \u2192 creates/updates goal",
-  remove_goal: "id \u2192 deletes goal"
-};
-server.setRequestHandler(ListToolsRequestSchema, async () => ({ tools: TOOLS }));
-server.setRequestHandler(CallToolRequestSchema, async (req) => {
-  const { name, arguments: args } = req.params;
-  const a = args ?? {};
-  try {
-    switch (name) {
-      case "search_tools": {
-        const q = a["query"]?.toLowerCase();
-        const names = TOOLS.map((t) => t.name);
-        const matches = q ? names.filter((n) => n.includes(q)) : names;
-        return { content: [{ type: "text", text: matches.join(", ") }] };
-      }
-      case "describe_tools": {
-        const names = a["names"] ?? [];
-        const result = names.map((n) => `${n}: ${TOOL_DESCRIPTIONS[n] ?? "see tool schema"}`).join(`
-`);
-        return { content: [{ type: "text", text: result }] };
-      }
-      case "get_cost_summary": {
-        const period = a["period"] ?? "today";
-        const s = querySummary(db, period);
-        const text = [
-          `period: ${period}`,
-          `cost: ${fmtUsd(s.total_usd)}`,
-          `sessions: ${s.sessions}`,
-          `requests: ${s.requests.toLocaleString()}`,
-          `tokens: ${fmtTok(s.tokens)}`,
-          `summary: You've spent ${fmtUsd(s.total_usd)} ${period === "all" ? "total" : period} across ${s.sessions} sessions (${s.requests.toLocaleString()} requests, ${fmtTok(s.tokens)} tokens)`
-        ].join(`
+function text(text2) {
+  return { content: [{ type: "text", text: text2 }] };
+}
+function textError(message) {
+  return { content: [{ type: "text", text: message }], isError: true };
+}
+server.tool("search_tools", "List tool names matching query. Use first to find relevant tools.", { query: z.string().optional() }, async ({ query }) => {
+  const q = query?.toLowerCase();
+  const matches = q ? TOOL_NAMES.filter((name) => name.includes(q)) : [...TOOL_NAMES];
+  return text(matches.join(", "));
+});
+server.tool("describe_tools", "Get param hints for specific tools by name.", { names: z.array(z.string()) }, async ({ names }) => {
+  const result = names.map((name) => `${name}: ${TOOL_DESCRIPTIONS[name] ?? "see tool schema"}`).join(`
 `);
-        return { content: [{ type: "text", text }] };
-      }
-      case "get_sessions": {
-        const sessions = querySessions(db, {
-          agent: a["agent"],
-          project: a["project"],
-          limit: Number(a["limit"] ?? 20)
-        });
-        const lines = ["id       agent  cost       tokens   project"];
-        for (const s of sessions)
-          lines.push(fmtSession(s));
-        return { content: [{ type: "text", text: lines.join(`
-`) }] };
-      }
-      case "get_top_sessions": {
-        const sessions = queryTopSessions(db, Number(a["n"] ?? 10), a["agent"]);
-        const lines = ["rank  id       agent  cost       tokens   project"];
-        sessions.forEach((s, i) => lines.push(`${String(i + 1).padEnd(5)} ${fmtSession(s)}`));
-        return { content: [{ type: "text", text: lines.join(`
-`) }] };
-      }
-      case "get_model_breakdown": {
-        const rows = queryModelBreakdown(db);
-        const lines = ["model                          reqs    tokens   cost"];
-        for (const r of rows) {
-          lines.push(`${String(r["model"]).slice(0, 30).padEnd(31)}${String(r["requests"]).padEnd(8)}${fmtTok(Number(r["total_tokens"])).padEnd(9)}${fmtUsd(Number(r["cost_usd"]))}`);
-        }
-        return { content: [{ type: "text", text: lines.join(`
-`) }] };
-      }
-      case "get_project_breakdown": {
-        const rows = queryProjectBreakdown(db);
-        const lines = ["project              sessions tokens   cost"];
-        for (const r of rows) {
-          const name2 = String(r["project_name"] || r["project_path"] || "\u2014").slice(0, 20);
-          lines.push(`${name2.padEnd(21)}${String(r["sessions"]).padEnd(9)}${fmtTok(Number(r["total_tokens"])).padEnd(9)}${fmtUsd(Number(r["cost_usd"]))}`);
-        }
-        return { content: [{ type: "text", text: lines.join(`
-`) }] };
-      }
-      case "get_budget_status": {
-        const budgets = getBudgetStatuses(db);
-        if (budgets.length === 0)
-          return { content: [{ type: "text", text: "No budgets set." }] };
-        const lines = ["scope                period   spent      limit      used%  status"];
-        for (const b of budgets) {
-          const scope = String(b["project_path"] ?? "global").slice(0, 20);
-          const pct = Number(b["percent_used"]).toFixed(1);
-          const status = b["is_over_limit"] ? "OVER" : b["is_over_alert"] ? "ALERT" : "OK";
-          lines.push(`${scope.padEnd(21)}${String(b["period"]).padEnd(9)}${fmtUsd(Number(b["current_spend_usd"])).padEnd(11)}${fmtUsd(Number(b["limit_usd"])).padEnd(11)}${pct}%`.padEnd(49) + `  ${status}`);
-        }
-        return { content: [{ type: "text", text: lines.join(`
-`) }] };
-      }
-      case "get_daily": {
-        const days = Number(a["days"] ?? 30);
-        const rows = queryDailyBreakdown(db, days);
-        const lines = ["date        claude     codex      gemini     total"];
-        const byDate = new Map;
-        for (const r of rows) {
-          const d = String(r["date"]);
-          const entry = byDate.get(d) ?? { claude: 0, codex: 0, gemini: 0 };
-          if (r["agent"] === "claude")
-            entry.claude += Number(r["cost_usd"]);
-          else if (r["agent"] === "codex")
-            entry.codex += Number(r["cost_usd"]);
-          else if (r["agent"] === "gemini")
-            entry.gemini += Number(r["cost_usd"]);
-          byDate.set(d, entry);
-        }
-        for (const [date, costs] of [...byDate.entries()].sort()) {
-          const total = costs.claude + costs.codex + costs.gemini;
-          lines.push(`${date}  ${fmtUsd(costs.claude).padEnd(11)}${fmtUsd(costs.codex).padEnd(11)}${fmtUsd(costs.gemini).padEnd(11)}${fmtUsd(total)}`);
-        }
-        return { content: [{ type: "text", text: lines.join(`
-`) }] };
-      }
-      case "get_session_detail": {
-        const sid = String(a["session_id"] ?? "");
-        const session = db.prepare(`SELECT * FROM sessions WHERE id = ? OR id LIKE ?`).get(sid, `${sid}%`);
-        if (!session)
-          return { content: [{ type: "text", text: `Session not found: ${sid}` }], isError: true };
-        const requests = db.prepare(`SELECT * FROM requests WHERE session_id = ? ORDER BY timestamp ASC LIMIT 50`).all(session["id"]);
-        const lines = [
-          `session: ${String(session["id"]).slice(0, 16)}`,
-          `agent: ${session["agent"]}  project: ${session["project_name"] || "\u2014"}`,
-          `cost: ${fmtUsd(Number(session["total_cost_usd"]))}  tokens: ${fmtTok(Number(session["total_tokens"]))}  requests: ${session["request_count"]}`,
-          "",
-          "time      model                  input    output   cost"
-        ];
-        for (const r of requests) {
-          lines.push(`${String(r["timestamp"]).slice(11, 19)}  ${String(r["model"]).slice(0, 22).padEnd(23)}${fmtTok(Number(r["input_tokens"])).padEnd(9)}${fmtTok(Number(r["output_tokens"])).padEnd(9)}${fmtUsd(Number(r["cost_usd"]))}`);
-        }
-        return { content: [{ type: "text", text: lines.join(`
-`) }] };
-      }
-      case "sync": {
-        const sources = a["sources"] ?? "all";
-        const parts = [];
-        if (sources === "all" || sources === "claude") {
-          const r = await ingestClaude(db);
-          parts.push(`claude: ${r["files"]} files, ${r["requests"]} requests, ${r["sessions"]} sessions`);
-        }
-        if (sources === "all" || sources === "codex") {
-          const r = await ingestCodex(db);
-          parts.push(`codex: ${r["sessions"]} sessions`);
-        }
-        if (sources === "all" || sources === "gemini") {
-          const r = await ingestGemini(db);
-          parts.push(`gemini: ${r["sessions"]} sessions`);
-        }
-        return { content: [{ type: "text", text: parts.join(`
-`) || "done" }] };
-      }
-      case "get_goals": {
-        const goals = getGoalStatuses(db);
-        if (goals.length === 0)
-          return { content: [{ type: "text", text: "No goals set." }] };
-        const lines = ["period   scope                limit      spent      used%  status"];
-        for (const g of goals) {
-          const scope = String(g["project_path"] ?? g["agent"] ?? "global").slice(0, 20);
-          const pct = Number(g["percent_used"]).toFixed(1);
-          const status = g["is_over"] ? "OVER" : g["is_at_risk"] ? "AT RISK" : "ON TRACK";
-          lines.push(`${String(g["period"]).padEnd(9)}${scope.padEnd(21)}${fmtUsd(Number(g["limit_usd"])).padEnd(11)}${fmtUsd(Number(g["current_spend_usd"])).padEnd(11)}${pct}%  ${status}`);
-        }
-        return { content: [{ type: "text", text: lines.join(`
-`) }] };
-      }
-      case "set_goal": {
-        const { randomUUID } = await import("crypto");
-        const now = new Date().toISOString();
-        upsertGoal(db, {
-          id: randomUUID(),
-          period: String(a["period"] ?? "month"),
-          project_path: a["project_path"] ?? null,
-          agent: a["agent"] ?? null,
-          limit_usd: Number(a["limit_usd"]),
-          created_at: now,
-          updated_at: now
-        });
-        return { content: [{ type: "text", text: `Goal set: ${a["period"]} $${a["limit_usd"]}` }] };
-      }
-      case "remove_goal": {
-        deleteGoal(db, String(a["id"] ?? ""));
-        return { content: [{ type: "text", text: "Goal removed." }] };
-      }
-      case "register_agent": {
-        const n = String(args["name"] ?? "");
-        const ex = [..._econAgents.values()].find((x) => x.name === n);
-        if (ex) {
-          ex.last_seen_at = new Date().toISOString();
-          return { content: [{ type: "text", text: JSON.stringify(ex) }] };
-        }
-        const id = Math.random().toString(36).slice(2, 10);
-        const ag = { id, name: n, last_seen_at: new Date().toISOString() };
-        _econAgents.set(id, ag);
-        return { content: [{ type: "text", text: JSON.stringify(ag) }] };
-      }
-      case "heartbeat": {
-        const ag = _econAgents.get(String(args["agent_id"] ?? ""));
-        if (!ag)
-          return { content: [{ type: "text", text: `Agent not found` }], isError: true };
-        ag.last_seen_at = new Date().toISOString();
-        return { content: [{ type: "text", text: `\u2665 ${ag.name}` }] };
-      }
-      case "set_focus": {
-        const ag = _econAgents.get(String(args["agent_id"] ?? ""));
-        if (!ag)
-          return { content: [{ type: "text", text: `Agent not found` }], isError: true };
-        ag["project_id"] = args["project_id"];
-        return { content: [{ type: "text", text: String(args["project_id"] ? `Focus: ${args["project_id"]}` : "Focus cleared") }] };
-      }
-      case "list_agents": {
-        return { content: [{ type: "text", text: JSON.stringify([..._econAgents.values()]) }] };
-      }
-      case "send_feedback": {
-        try {
-          const pkg = require_package();
-          db.prepare("INSERT INTO feedback (message, email, category, version) VALUES (?, ?, ?, ?)").run(String(a["message"]), a["email"] || null, a["category"] || "general", pkg.version);
-          return { content: [{ type: "text", text: "Feedback saved. Thank you!" }] };
-        } catch (e) {
-          return { content: [{ type: "text", text: String(e) }], isError: true };
-        }
-      }
-      default:
-        return { content: [{ type: "text", text: `Unknown tool: ${name}` }], isError: true };
-    }
-  } catch (e) {
-    return { content: [{ type: "text", text: `Error: ${e instanceof Error ? e.message : String(e)}` }], isError: true };
+  return text(result);
+});
+server.tool("get_cost_summary", "Cost summary (total_usd, sessions, requests, tokens, human summary). period: today|week|month|year|all", { period: z.enum(["today", "week", "month", "year", "all"]).optional() }, async ({ period }) => {
+  const resolved = period ?? "today";
+  const s = querySummary(db, resolved);
+  return text([
+    `period: ${resolved}`,
+    `cost: ${fmtUsd(s.total_usd)}`,
+    `sessions: ${s.sessions}`,
+    `requests: ${s.requests.toLocaleString()}`,
+    `tokens: ${fmtTok(s.tokens)}`,
+    `summary: You've spent ${fmtUsd(s.total_usd)} ${resolved === "all" ? "total" : resolved} across ${s.sessions} sessions (${s.requests.toLocaleString()} requests, ${fmtTok(s.tokens)} tokens)`
+  ].join(`
+`));
+});
+server.tool("get_sessions", "List sessions. Returns compact table. Params: agent, project, limit(20)", {
+  agent: z.enum(["claude", "codex", "gemini"]).optional(),
+  project: z.string().optional(),
+  limit: z.number().int().positive().max(100).optional()
+}, async ({ agent, project, limit }) => {
+  const sessions = querySessions(db, {
+    agent,
+    project,
+    limit: limit ?? 20
+  });
+  const lines = ["id       agent  cost       tokens   project"];
+  for (const session of sessions)
+    lines.push(fmtSession(session));
+  return text(lines.join(`
+`));
+});
+server.tool("get_top_sessions", "Top sessions by cost. Params: n(10), agent", {
+  n: z.number().int().positive().max(100).optional(),
+  agent: z.enum(["claude", "codex", "gemini"]).optional()
+}, async ({ n, agent }) => {
+  const sessions = queryTopSessions(db, n ?? 10, agent);
+  const lines = ["rank  id       agent  cost       tokens   project"];
+  sessions.forEach((session, i) => lines.push(`${String(i + 1).padEnd(5)} ${fmtSession(session)}`));
+  return text(lines.join(`
+`));
+});
+server.tool("get_model_breakdown", "Cost per model. No params.", {}, async () => {
+  const rows = queryModelBreakdown(db);
+  const lines = ["model                          reqs    tokens   cost"];
+  for (const row of rows) {
+    lines.push(`${String(row["model"]).slice(0, 30).padEnd(31)}${String(row["requests"]).padEnd(8)}${fmtTok(Number(row["total_tokens"])).padEnd(9)}${fmtUsd(Number(row["cost_usd"]))}`);
+  }
+  return text(lines.join(`
+`));
+});
+server.tool("get_project_breakdown", "Cost per project. No params.", {}, async () => {
+  const rows = queryProjectBreakdown(db);
+  const lines = ["project              sessions tokens   cost"];
+  for (const row of rows) {
+    const name = String(row["project_name"] || row["project_path"] || "\u2014").slice(0, 20);
+    lines.push(`${name.padEnd(21)}${String(row["sessions"]).padEnd(9)}${fmtTok(Number(row["total_tokens"])).padEnd(9)}${fmtUsd(Number(row["cost_usd"]))}`);
+  }
+  return text(lines.join(`
+`));
+});
+server.tool("get_budget_status", "Budget limits vs spend, percent used, alert flags. No params.", {}, async () => {
+  const budgets = getBudgetStatuses(db);
+  if (budgets.length === 0)
+    return text("No budgets set.");
+  const lines = ["scope                period   spent      limit      used%  status"];
+  for (const budget of budgets) {
+    const scope = String(budget["project_path"] ?? "global").slice(0, 20);
+    const pct = Number(budget["percent_used"]).toFixed(1);
+    const status = budget["is_over_limit"] ? "OVER" : budget["is_over_alert"] ? "ALERT" : "OK";
+    lines.push(`${scope.padEnd(21)}${String(budget["period"]).padEnd(9)}${fmtUsd(Number(budget["current_spend_usd"])).padEnd(11)}${fmtUsd(Number(budget["limit_usd"])).padEnd(11)}${pct}%`.padEnd(49) + `  ${status}`);
+  }
+  return text(lines.join(`
+`));
+});
+server.tool("get_daily", "Daily cost table by agent. Params: days(30)", { days: z.number().int().positive().max(365).optional() }, async ({ days }) => {
+  const rows = queryDailyBreakdown(db, days ?? 30);
+  const byDate = new Map;
+  for (const row of rows) {
+    const date = String(row["date"]);
+    const entry = byDate.get(date) ?? { claude: 0, codex: 0, gemini: 0 };
+    if (row["agent"] === "claude")
+      entry.claude += Number(row["cost_usd"]);
+    else if (row["agent"] === "codex")
+      entry.codex += Number(row["cost_usd"]);
+    else if (row["agent"] === "gemini")
+      entry.gemini += Number(row["cost_usd"]);
+    byDate.set(date, entry);
+  }
+  const lines = ["date        claude     codex      gemini     total"];
+  for (const [date, costs] of [...byDate.entries()].sort()) {
+    const total = costs.claude + costs.codex + costs.gemini;
+    lines.push(`${date}  ${fmtUsd(costs.claude).padEnd(11)}${fmtUsd(costs.codex).padEnd(11)}${fmtUsd(costs.gemini).padEnd(11)}${fmtUsd(total)}`);
+  }
+  return text(lines.join(`
+`));
+});
+server.tool("get_session_detail", "Per-request breakdown of a single session. Params: session_id (prefix ok)", { session_id: z.string() }, async ({ session_id }) => {
+  const session = db.prepare(`SELECT * FROM sessions WHERE id = ? OR id LIKE ?`).get(session_id, `${session_id}%`);
+  if (!session)
+    return textError(`Session not found: ${session_id}`);
+  const requests = db.prepare(`SELECT * FROM requests WHERE session_id = ? ORDER BY timestamp ASC LIMIT 50`).all(session["id"]);
+  const lines = [
+    `session: ${String(session["id"]).slice(0, 16)}`,
+    `agent: ${session["agent"]}  project: ${session["project_name"] || "\u2014"}`,
+    `cost: ${fmtUsd(Number(session["total_cost_usd"]))}  tokens: ${fmtTok(Number(session["total_tokens"]))}  requests: ${session["request_count"]}`,
+    "",
+    "time      model                  input    output   cost"
+  ];
+  for (const request of requests) {
+    lines.push(`${String(request["timestamp"]).slice(11, 19)}  ${String(request["model"]).slice(0, 22).padEnd(23)}${fmtTok(Number(request["input_tokens"])).padEnd(9)}${fmtTok(Number(request["output_tokens"])).padEnd(9)}${fmtUsd(Number(request["cost_usd"]))}`);
+  }
+  return text(lines.join(`
+`));
+});
+server.tool("sync", "Ingest new cost data. sources: all|claude|codex|gemini", { sources: z.enum(["all", "claude", "codex", "gemini"]).optional() }, async ({ sources }) => {
+  const selected = sources ?? "all";
+  const parts = [];
+  if (selected === "all" || selected === "claude") {
+    const result = await ingestClaude(db);
+    parts.push(`claude: ${result["files"]} files, ${result["requests"]} requests, ${result["sessions"]} sessions`);
+  }
+  if (selected === "all" || selected === "codex") {
+    const result = await ingestCodex(db);
+    parts.push(`codex: ${result["sessions"]} sessions`);
+  }
+  if (selected === "all" || selected === "gemini") {
+    const result = await ingestGemini(db);
+    parts.push(`gemini: ${result["sessions"]} sessions`);
+  }
+  return text(parts.join(`
+`) || "done");
+});
+server.tool("get_goals", "All spending goals with current progress. No params.", {}, async () => {
+  const goals = getGoalStatuses(db);
+  if (goals.length === 0)
+    return text("No goals set.");
+  const lines = ["period   scope                limit      spent      used%  status"];
+  for (const goal of goals) {
+    const scope = String(goal["project_path"] ?? goal["agent"] ?? "global").slice(0, 20);
+    const pct = Number(goal["percent_used"]).toFixed(1);
+    const status = goal["is_over"] ? "OVER" : goal["is_at_risk"] ? "AT RISK" : "ON TRACK";
+    lines.push(`${String(goal["period"]).padEnd(9)}${scope.padEnd(21)}${fmtUsd(Number(goal["limit_usd"])).padEnd(11)}${fmtUsd(Number(goal["current_spend_usd"])).padEnd(11)}${pct}%  ${status}`);
+  }
+  return text(lines.join(`
+`));
+});
+server.tool("set_goal", "Create/update a spending goal. period(day|week|month|year), limit_usd, project_path?, agent?", {
+  period: z.enum(["day", "week", "month", "year"]),
+  limit_usd: z.number().nonnegative(),
+  project_path: z.string().optional(),
+  agent: z.string().optional()
+}, async ({ period, limit_usd, project_path, agent }) => {
+  const now = new Date().toISOString();
+  upsertGoal(db, {
+    id: randomUUID(),
+    period,
+    project_path: project_path ?? null,
+    agent: agent ?? null,
+    limit_usd,
+    created_at: now,
+    updated_at: now
+  });
+  return text(`Goal set: ${period} $${limit_usd}`);
+});
+server.tool("remove_goal", "Delete a goal by id.", { id: z.string() }, async ({ id }) => {
+  deleteGoal(db, id);
+  return text("Goal removed.");
+});
+server.tool("register_agent", "Register agent session.", { name: z.string(), session_id: z.string().optional() }, async ({ name }) => {
+  const existing = [..._econAgents.values()].find((agent2) => agent2.name === name);
+  if (existing) {
+    existing.last_seen_at = new Date().toISOString();
+    return text(JSON.stringify(existing));
+  }
+  const id = Math.random().toString(36).slice(2, 10);
+  const agent = { id, name, last_seen_at: new Date().toISOString() };
+  _econAgents.set(id, agent);
+  return text(JSON.stringify(agent));
+});
+server.tool("heartbeat", "Update last_seen_at.", { agent_id: z.string() }, async ({ agent_id }) => {
+  const agent = _econAgents.get(agent_id);
+  if (!agent)
+    return textError("Agent not found");
+  agent.last_seen_at = new Date().toISOString();
+  return text(`\u2665 ${agent.name}`);
+});
+server.tool("set_focus", "Set active project context.", { agent_id: z.string(), project_id: z.string().optional().nullable() }, async ({ agent_id, project_id }) => {
+  const agent = _econAgents.get(agent_id);
+  if (!agent)
+    return textError("Agent not found");
+  agent.project_id = project_id ?? undefined;
+  return text(project_id ? `Focus: ${project_id}` : "Focus cleared");
+});
+server.tool("list_agents", "List all registered agents.", {}, async () => text(JSON.stringify([..._econAgents.values()])));
+server.tool("send_feedback", "Send feedback about this service.", {
+  message: z.string(),
+  email: z.string().optional(),
+  category: z.enum(["bug", "feature", "general"]).optional()
+}, async ({ message, email, category }) => {
+  try {
+    db.prepare("INSERT INTO feedback (message, email, category, version) VALUES (?, ?, ?, ?)").run(message, email ?? null, category ?? "general", packageMetadata.version);
+    return text("Feedback saved. Thank you!");
+  } catch (error) {
+    return textError(String(error));
   }
 });
-var _econAgents = new Map;
 var transport = new StdioServerTransport;
+registerCloudTools(server, "economy");
 await server.connect(transport);