vibeusage 0.3.3 → 0.3.5

package/src/lib/openclaw-session-plugin.js

@@ -337,31 +337,79 @@ function buildSessionPluginMeta() {
   )}\n`;
 }
 
-function buildSessionPluginIndex({ trackerDir }) {
+function buildSessionPluginIndex({ trackerDir, packageName = "vibeusage" }) {
   const ledgerModuleUrl = pathToFileURL(
     path.join(trackerDir, "app", "src", "lib", "openclaw-usage-ledger.js"),
   ).href;
+  const trackerBinPath = path.join(trackerDir, "app", "bin", "tracker.js");
+  const fallbackPkg = packageName || "vibeusage";
 
   return (
+    `import fs from 'node:fs';\n` +
+    `import path from 'node:path';\n` +
+    `import cp from 'node:child_process';\n` +
     `import { appendOpenclawUsageEvent } from ${JSON.stringify(ledgerModuleUrl)};\n` +
     `\n` +
     `const trackerDir = ${JSON.stringify(trackerDir)};\n` +
+    `const trackerBinPath = ${JSON.stringify(trackerBinPath)};\n` +
+    `const fallbackPkg = ${JSON.stringify(fallbackPkg)};\n` +
+    `const depsMarkerPath = path.join(trackerDir, 'app', 'node_modules', '@insforge', 'sdk', 'package.json');\n` +
+    `const triggerStatePath = path.join(trackerDir, 'openclaw.session-sync.trigger-state.json');\n` +
+    `const SESSION_TRIGGER_THROTTLE_MS = 15_000;\n` +
     `\n` +
     `export default function register(api) {\n` +
     `  api.on('llm_output', async (event, ctx) => {\n` +
     `    try {\n` +
     `      const payload = buildPayload(event, ctx);\n` +
     `      if (!payload) return;\n` +
-    `      await appendOpenclawUsageEvent({ trackerDir, payload });\n` +
+    `      const result = await appendOpenclawUsageEvent({ trackerDir, payload });\n` +
+    `      if (!result || result.appended === false) return;\n` +
+    `      const scope = normalize(ctx && ctx.agentId) || 'openclaw';\n` +
+    `      const target = resolveSessionRef(event, ctx) || normalize(payload.eventId) || 'unknown';\n` +
+    `      if (!allowTrigger('llm_output', scope, target)) return;\n` +
+    `      spawnSync({ args: ['sync', '--auto', '--from-openclaw'] });\n` +
     `    } catch (_) {}\n` +
     `  });\n` +
     `}\n` +
     `\n` +
+    `function spawnSync({ args }) {\n` +
+    `  const hasLocalRuntime = fs.existsSync(trackerBinPath);\n` +
+    `  const hasLocalDeps = fs.existsSync(depsMarkerPath);\n` +
+    `  const argv = Array.isArray(args) && args.length > 0 ? args : ['sync', '--auto'];\n` +
+    `  const cmd = hasLocalRuntime && hasLocalDeps\n` +
+    `    ? [process.execPath, trackerBinPath, ...argv]\n` +
+    `    : ['npx', '--yes', fallbackPkg, ...argv];\n` +
+    `  const child = cp.spawn(cmd[0], cmd.slice(1), {\n` +
+    `    detached: true,\n` +
+    `    stdio: 'ignore',\n` +
+    `    env: { ...process.env }\n` +
+    `  });\n` +
+    `  child.unref();\n` +
+    `}\n` +
+    `\n` +
+    `function allowTrigger(kind, scope, target) {\n` +
+    `  const key = [kind, scope || 'na', target || 'na'].join(':');\n` +
+    `  const now = Date.now();\n` +
+    `  let state = {};\n` +
+    `  try {\n` +
+    `    state = JSON.parse(fs.readFileSync(triggerStatePath, 'utf8'));\n` +
+    `    if (!state || typeof state !== 'object') state = {};\n` +
+    `  } catch (_) {}\n` +
+    `  const last = Number(state[key] || 0);\n` +
+    `  if (Number.isFinite(last) && now - last < SESSION_TRIGGER_THROTTLE_MS) return false;\n` +
+    `  state[key] = now;\n` +
+    `  try {\n` +
+    `    fs.mkdirSync(path.dirname(triggerStatePath), { recursive: true });\n` +
+    `    fs.writeFileSync(triggerStatePath, JSON.stringify(state), 'utf8');\n` +
+    `  } catch (_) {}\n` +
+    `  return true;\n` +
+    `}\n` +
+    `\n` +
     `function buildPayload(event, ctx) {\n` +
     `  const usage = normalizeUsage(event);\n` +
     `  if (!usage) return null;\n` +
     `\n` +
-    `  const sessionKey = normalize(ctx && ctx.sessionKey);\n` +
+    `  const sessionKey = resolveSessionRef(event, ctx);\n` +
     `  if (!sessionKey) return null;\n` +
     `\n` +
     `  return {\n` +
@@ -378,6 +426,22 @@ function buildSessionPluginIndex({ trackerDir }) {
     `  };\n` +
     `}\n` +
     `\n` +
+    `function resolveSessionRef(event, ctx) {\n` +
+    `  const explicit = normalize(ctx && ctx.sessionKey);\n` +
+    `  if (explicit) return explicit;\n` +
+    `  const eventSessionId = normalize(event && event.sessionId);\n` +
+    `  if (eventSessionId) {\n` +
+    `    const agentId = normalize(ctx && ctx.agentId) || 'openclaw';\n` +
+    `    return 'agent:' + agentId + ':session:' + eventSessionId;\n` +
+    `  }\n` +
+    `  const ctxSessionId = normalize(ctx && ctx.sessionId);\n` +
+    `  if (ctxSessionId) {\n` +
+    `    const agentId = normalize(ctx && ctx.agentId) || 'openclaw';\n` +
+    `    return 'agent:' + agentId + ':session:' + ctxSessionId;\n` +
+    `  }\n` +
+    `  return null;\n` +
+    `}\n` +
+    `\n` +
     `function normalizeUsage(event) {\n` +
     `  const usage = event && typeof event.usage === 'object' ? event.usage : event || {};\n` +
     `  const normalized = {\n` +

package/src/templates/hermes-vibeusage-plugin/__init__.py

@@ -0,0 +1,75 @@
+# VIBEUSAGE_HERMES_PLUGIN
+# Generated by VibeUsage. Do not edit manually.
+import json
+import os
+from datetime import datetime, timezone
+
+LEDGER_PATH = "__LEDGER_PATH__"
+SOURCE = "hermes"
+VERSION = 1
+
+
+def _iso_now():
+    return datetime.now(timezone.utc).isoformat().replace("+00:00", "Z")
+
+
+def _safe_int(value):
+    try:
+        return max(0, int(value or 0))
+    except Exception:
+        return 0
+
+
+def _append_record(record):
+    try:
+        os.makedirs(os.path.dirname(LEDGER_PATH), exist_ok=True)
+        with open(LEDGER_PATH, "a", encoding="utf-8") as handle:
+            handle.write(json.dumps(record, ensure_ascii=False) + "\n")
+    except Exception:
+        return None
+    return None
+
+
+def _base_record(record_type, session_id="", platform="", model="", provider=""):
+    return {
+        "version": VERSION,
+        "type": str(record_type or ""),
+        "source": SOURCE,
+        "session_id": str(session_id or ""),
+        "platform": str(platform or ""),
+        "model": str(model or ""),
+        "provider": str(provider or ""),
+        "emitted_at": _iso_now(),
+    }
+
+
+def on_session_start(session_id="", model="", platform="", **_kwargs):
+    return _append_record(_base_record("session_start", session_id=session_id, platform=platform, model=model))
+
+
+def post_api_request(session_id="", platform="", model="", provider="", api_mode="", api_call_count=0, finish_reason="", usage=None, **_kwargs):
+    if not isinstance(usage, dict):
+        return None
+    record = _base_record("usage", session_id=session_id, platform=platform, model=model, provider=provider)
+    record.update({
+        "api_mode": str(api_mode or ""),
+        "api_call_count": _safe_int(api_call_count),
+        "input_tokens": _safe_int(usage.get("input_tokens")),
+        "output_tokens": _safe_int(usage.get("output_tokens")),
+        "cache_read_tokens": _safe_int(usage.get("cache_read_tokens")),
+        "cache_write_tokens": _safe_int(usage.get("cache_write_tokens")),
+        "reasoning_tokens": _safe_int(usage.get("reasoning_tokens")),
+        "total_tokens": _safe_int(usage.get("total_tokens")),
+        "finish_reason": str(finish_reason or ""),
+    })
+    return _append_record(record)
+
+
+def on_session_end(session_id="", model="", platform="", **_kwargs):
+    return _append_record(_base_record("session_end", session_id=session_id, platform=platform, model=model))
+
+
+def register(ctx):
+    ctx.register_hook("on_session_start", on_session_start)
+    ctx.register_hook("post_api_request", post_api_request)
+    ctx.register_hook("on_session_end", on_session_end)

package/src/templates/hermes-vibeusage-plugin/plugin.yaml

@@ -0,0 +1,9 @@
+# VIBEUSAGE_HERMES_PLUGIN
+name: vibeusage
+version: "1"
+description: "VibeUsage Hermes usage ledger plugin"
+author: "VibeUsage"
+provides_hooks:
+  - "on_session_start"
+  - "post_api_request"
+  - "on_session_end"