@kairyou/agent-tools 0.3.0 → 0.5.0

package/dist/usage/opencode-plugin.mjs

@@ -0,0 +1,80 @@
+// integrations/usage/opencode-plugin.mjs
+import { queryProviderUsage } from "./core.mjs";
+var DEFAULT_REFRESH_MS = 6e4;
+function toastMessage(text) {
+  return text.replace(/^API \| /, "");
+}
+function firstString(value, keys) {
+  for (const key of keys) {
+    if (typeof value?.[key] === "string" && value[key].trim()) return value[key].trim();
+  }
+  return "";
+}
+function providerContext(input) {
+  const info = input?.provider?.info || {};
+  const options = input?.provider?.options || {};
+  const providerName = String(input?.model?.providerID || info.id || "opencode");
+  const label = String(info.name || providerName);
+  return {
+    providerName,
+    provider: { name: label },
+    label,
+    baseUrl: process.env.PROVIDER_USAGE_BASE_URL || firstString(options, ["baseURL", "baseUrl", "base_url"]),
+    key: process.env.PROVIDER_USAGE_API_KEY || process.env.SUB2API_API_KEY || firstString(options, ["apiKey", "api_key"])
+  };
+}
+function refreshInterval(options) {
+  const configured = Number(options?.refreshMs || process.env.PROVIDER_USAGE_REFRESH_MS);
+  return Number.isFinite(configured) && configured >= 0 ? configured : DEFAULT_REFRESH_MS;
+}
+var AgentToolsUsage = async ({ client }, options = {}) => {
+  const contexts = /* @__PURE__ */ new Map();
+  const refreshedAt = /* @__PURE__ */ new Map();
+  const inFlight = /* @__PURE__ */ new Map();
+  const refreshMs = refreshInterval(options);
+  async function showUsage(text) {
+    if (!text) return;
+    try {
+      await client.tui.showToast({
+        body: {
+          title: "Provider usage",
+          message: toastMessage(text),
+          variant: "info",
+          duration: 8e3
+        }
+      });
+    } catch {
+    }
+  }
+  async function refreshSession(sessionID) {
+    const context = contexts.get(sessionID);
+    if (!context?.baseUrl || !context?.key) return;
+    const cacheKey = context.baseUrl.replace(/\/+$/, "");
+    const now = Date.now();
+    if (now - (refreshedAt.get(cacheKey) || 0) < refreshMs) return;
+    if (inFlight.has(cacheKey)) return await inFlight.get(cacheKey);
+    refreshedAt.set(cacheKey, now);
+    const task = queryProviderUsage(context, {
+      agent: "opencode",
+      rememberSnapshot: true
+    }).then((result) => showUsage(result?.text || "")).catch(() => {
+    }).finally(() => inFlight.delete(cacheKey));
+    inFlight.set(cacheKey, task);
+    return await task;
+  }
+  return {
+    "chat.params": async (input) => {
+      contexts.set(input.sessionID, providerContext(input));
+    },
+    event: async ({ event }) => {
+      if (event.type === "session.idle") {
+        void refreshSession(event.properties.sessionID);
+      } else if (event.type === "session.deleted") {
+        contexts.delete(event.properties.sessionID);
+      }
+    }
+  };
+};
+export {
+  AgentToolsUsage
+};

package/dist/usage/opencode-tui.mjs

@@ -0,0 +1,46 @@
+// integrations/usage/opencode-tui.mjs
+import { readFileSync } from "node:fs";
+import { homedir } from "node:os";
+import { join } from "node:path";
+var AGENT_TOOLS_HOME = process.env.AGENT_TOOLS_HOME || join(homedir(), ".agent-tools");
+var SNAPSHOT_PATH = join(AGENT_TOOLS_HOME, "cache", "usage-snapshot.json");
+function toastMessage(text) {
+  return text.replace(/^API \| /, "");
+}
+function latestUsage() {
+  try {
+    const data = JSON.parse(readFileSync(SNAPSHOT_PATH, "utf8"));
+    return Object.values(data?.items || {}).filter((item) => typeof item?.text === "string" && item.text).sort((a, b) => String(b.updatedAt || "").localeCompare(String(a.updatedAt || "")))[0]?.text || "";
+  } catch {
+    return "";
+  }
+}
+var tui = async (api) => {
+  api.keymap.registerLayer({
+    commands: [
+      {
+        name: "agent-tools.usage",
+        title: "Provider usage",
+        category: "Agent Tools",
+        namespace: "palette",
+        slashName: "at-usage",
+        run() {
+          const message = latestUsage();
+          api.ui.toast({
+            title: "Provider usage",
+            message: message ? toastMessage(message) : "Provider usage is not available yet",
+            variant: message ? "info" : "warning",
+            duration: 8e3
+          });
+        }
+      }
+    ]
+  });
+};
+var opencode_tui_default = {
+  id: "agent-tools-usage",
+  tui
+};
+export {
+  opencode_tui_default as default
+};

package/integrations/statusline/claude-statusline.mjs

@@ -14,6 +14,7 @@ import { execFileSync, spawn } from "node:child_process";
 import fs from "node:fs";
 import { basename, dirname, join } from "node:path";
 import { fileURLToPath } from "node:url";
+import { parse as parseJsonc } from "jsonc-parser";
 
 const SCRIPT_DIR = dirname(fileURLToPath(import.meta.url));
 const AGENT_TOOLS_HOME = process.env.AGENT_TOOLS_HOME || join(SCRIPT_DIR, "..", "..");
@@ -21,8 +22,8 @@ const DEFAULT_CONFIG_FILE = join(AGENT_TOOLS_HOME, "config.jsonc");
 const SNAPSHOT_FILE = join(AGENT_TOOLS_HOME, "cache", "usage-snapshot.json");
 const REFRESH_STATE_FILE = join(AGENT_TOOLS_HOME, "cache", "usage-refresh-state.json");
 // Cross-integration dependency: statusline renders provider usage via the
-// usage integration's query engine.
-const USAGE_RUNTIME = join(AGENT_TOOLS_HOME, "integrations", "usage", "core.mjs");
+// usage integration's query engine (built artifact; see scripts/build.mjs).
+const USAGE_RUNTIME = join(AGENT_TOOLS_HOME, "dist", "usage", "core.mjs");
 const DEFAULT_SNAPSHOT_TTL_MS = 60_000;
 const DEFAULT_REFRESH_COOLDOWN_MS = 30_000;
 const DEFAULT_FAILURE_BACKOFF_MS = 120_000;
@@ -62,47 +63,16 @@ async function readStdin() {
 function readJsonFile(file) {
   try {
     if (!fs.existsSync(file)) return {};
-    const raw = stripJsonComments(fs.readFileSync(file, "utf8").replace(/^\uFEFF/, ""));
-    return raw.trim() ? JSON.parse(raw) : {};
+    const raw = fs.readFileSync(file, "utf8").replace(/^\uFEFF/, "");
+    if (!raw.trim()) return {};
+    const errors = [];
+    const parsed = parseJsonc(raw, errors, { allowTrailingComma: true });
+    return errors.length === 0 && parsed && typeof parsed === "object" ? parsed : {};
   } catch {
     return {};
   }
 }
 
-function stripJsonComments(input) {
-  let out = "";
-  let inString = false;
-  let escaped = false;
-  for (let i = 0; i < input.length; i++) {
-    const ch = input[i];
-    const next = input[i + 1];
-    if (inString) {
-      out += ch;
-      escaped = ch === "\\" ? !escaped : false;
-      if (ch === "\"" && !escaped) inString = false;
-      continue;
-    }
-    if (ch === "\"") {
-      inString = true;
-      out += ch;
-      continue;
-    }
-    if (ch === "/" && next === "/") {
-      while (i < input.length && input[i] !== "\n") i++;
-      out += "\n";
-      continue;
-    }
-    if (ch === "/" && next === "*") {
-      i += 2;
-      while (i < input.length && !(input[i] === "*" && input[i + 1] === "/")) i++;
-      i++;
-      continue;
-    }
-    out += ch;
-  }
-  return out;
-}
-
 function parseArgs(argv) {
   const opts = {};
   for (let i = 0; i < argv.length; i++) {