vibeusage 0.2.21 → 0.2.22

package/src/lib/opencode-usage-audit.js

@@ -1,24 +1,24 @@
-const fs = require('node:fs/promises');
-const os = require('node:os');
-const path = require('node:path');
+const fs = require("node:fs/promises");
+const os = require("node:os");
+const path = require("node:path");
 
-const { listOpencodeMessageFiles, parseOpencodeIncremental } = require('./rollout');
+const { listOpencodeMessageFiles, parseOpencodeIncremental } = require("./rollout");
 
-const BUCKET_SEPARATOR = '|';
+const BUCKET_SEPARATOR = "|";
 const DAY_RE = /^\d{4}-\d{2}-\d{2}$/;
 
 function formatHourKey(date) {
-  return `${date.getUTCFullYear()}-${String(date.getUTCMonth() + 1).padStart(2, '0')}-${String(
-    date.getUTCDate()
-  ).padStart(2, '0')}T${String(date.getUTCHours()).padStart(2, '0')}:${String(
-    date.getUTCMinutes() >= 30 ? 30 : 0
-  ).padStart(2, '0')}:00`;
+  return `${date.getUTCFullYear()}-${String(date.getUTCMonth() + 1).padStart(2, "0")}-${String(
+    date.getUTCDate(),
+  ).padStart(2, "0")}T${String(date.getUTCHours()).padStart(2, "0")}:${String(
+    date.getUTCMinutes() >= 30 ? 30 : 0,
+  ).padStart(2, "0")}:00`;
 }
 
 function toBig(value) {
-  if (typeof value === 'bigint') return value;
-  if (typeof value === 'number') return BigInt(value);
-  if (typeof value === 'string' && value.trim()) return BigInt(value);
+  if (typeof value === "bigint") return value;
+  if (typeof value === "number") return BigInt(value);
+  if (typeof value === "string" && value.trim()) return BigInt(value);
   return 0n;
 }
 
@@ -30,9 +30,12 @@ function addTotals(target, delta) {
   target.total_tokens += toBig(delta.total_tokens);
 }
 
-async function buildLocalHourlyTotals({ storageDir, source = 'opencode' }) {
+async function buildLocalHourlyTotals({ storageDir, source = "opencode" }) {
   const messageFiles = await listOpencodeMessageFiles(storageDir);
-  const queuePath = path.join(os.tmpdir(), `vibeusage-opencode-audit-${process.pid}-${Date.now()}.jsonl`);
+  const queuePath = path.join(
+    os.tmpdir(),
+    `vibeusage-opencode-audit-${process.pid}-${Date.now()}.jsonl`,
+  );
   const cursors = { version: 1, files: {}, hourly: null, opencode: null };
 
   await parseOpencodeIncremental({ messageFiles, cursors, queuePath, source });
@@ -58,7 +61,7 @@ async function buildLocalHourlyTotals({ storageDir, source = 'opencode' }) {
       cached_input_tokens: 0n,
       output_tokens: 0n,
       reasoning_output_tokens: 0n,
-      total_tokens: 0n
+      total_tokens: 0n,
     };
     addTotals(totals, bucket.totals || {});
     byHour.set(hourKey, totals);
@@ -78,8 +81,8 @@ function normalizeServerRows(rows) {
         cached_input_tokens: toBig(row.cached_input_tokens),
         output_tokens: toBig(row.output_tokens),
         reasoning_output_tokens: toBig(row.reasoning_output_tokens),
-        total_tokens: toBig(row.total_tokens)
-      }
+        total_tokens: toBig(row.total_tokens),
+      },
     });
   }
   return map;
@@ -91,7 +94,7 @@ function diffTotals(local, server) {
     cached_input_tokens: local.cached_input_tokens - server.cached_input_tokens,
     output_tokens: local.output_tokens - server.output_tokens,
     reasoning_output_tokens: local.reasoning_output_tokens - server.reasoning_output_tokens,
-    total_tokens: local.total_tokens - server.total_tokens
+    total_tokens: local.total_tokens - server.total_tokens,
   };
 }
 
@@ -101,7 +104,7 @@ function maxAbsDelta(delta) {
     delta.cached_input_tokens,
     delta.output_tokens,
     delta.reasoning_output_tokens,
-    delta.total_tokens
+    delta.total_tokens,
   ].reduce((acc, value) => {
     const abs = value < 0n ? -value : value;
     return abs > acc ? abs : acc;
@@ -121,25 +124,25 @@ function listDays(from, to) {
   const start = new Date(`${from}T00:00:00.000Z`);
   const end = new Date(`${to}T00:00:00.000Z`);
   for (let dt = start; dt <= end; dt = new Date(dt.getTime() + 24 * 60 * 60 * 1000)) {
-    const day = `${dt.getUTCFullYear()}-${String(dt.getUTCMonth() + 1).padStart(2, '0')}-${String(
-      dt.getUTCDate()
-    ).padStart(2, '0')}`;
+    const day = `${dt.getUTCFullYear()}-${String(dt.getUTCMonth() + 1).padStart(2, "0")}-${String(
+      dt.getUTCDate(),
+    ).padStart(2, "0")}`;
     out.push(day);
   }
   return out;
 }
 
 async function auditOpencodeUsage({ storageDir, from, to, fetchHourly, includeMissing = false }) {
-  const local = await buildLocalHourlyTotals({ storageDir, source: 'opencode' });
+  const local = await buildLocalHourlyTotals({ storageDir, source: "opencode" });
   if (!local.minDay || !local.maxDay) {
-    throw new Error('No local opencode data found');
+    throw new Error("No local opencode data found");
   }
 
   const fromDay = from || local.minDay;
   const toDay = to || local.maxDay;
   const days = listDays(fromDay, toDay);
   if (days.length === 0) {
-    throw new Error('Invalid date range for audit');
+    throw new Error("Invalid date range for audit");
   }
 
   const diffs = [];
@@ -153,13 +156,13 @@ async function auditOpencodeUsage({ storageDir, from, to, fetchHourly, includeMi
     const serverByHour = normalizeServerRows(server?.data || []);
     for (let h = 0; h < 24; h++) {
       for (const m of [0, 30]) {
-        const hourKey = `${day}T${String(h).padStart(2, '0')}:${String(m).padStart(2, '0')}:00`;
+        const hourKey = `${day}T${String(h).padStart(2, "0")}:${String(m).padStart(2, "0")}:00`;
         const localTotals = local.byHour.get(hourKey) || {
           input_tokens: 0n,
           cached_input_tokens: 0n,
           output_tokens: 0n,
           reasoning_output_tokens: 0n,
-          total_tokens: 0n
+          total_tokens: 0n,
         };
         const serverEntry = serverByHour.get(hourKey) || null;
         if (serverEntry?.missing && !includeMissing) {
@@ -171,7 +174,7 @@ async function auditOpencodeUsage({ storageDir, from, to, fetchHourly, includeMi
           cached_input_tokens: 0n,
           output_tokens: 0n,
           reasoning_output_tokens: 0n,
-          total_tokens: 0n
+          total_tokens: 0n,
         };
         const delta = diffTotals(localTotals, serverTotals);
         const deltaMax = maxAbsDelta(delta);
@@ -193,9 +196,9 @@ async function auditOpencodeUsage({ storageDir, from, to, fetchHourly, includeMi
       matched,
       mismatched,
       incomplete,
-      maxDelta
+      maxDelta,
     },
-    diffs
+    diffs,
   };
 }
 

package/src/lib/progress.js

@@ -1,32 +1,32 @@
 function createProgress({ stream } = {}) {
   const out = stream || process.stdout;
   const enabled = Boolean(out && out.isTTY);
-  const frames = ['|', '/', '-', '\\'];
+  const frames = ["|", "/", "-", "\\"];
   const intervalMs = 90;
 
   let timer = null;
-  let text = '';
+  let text = "";
   let frame = 0;
   let lastLen = 0;
 
   function render() {
     if (!enabled) return;
     const line = `${frames[frame++ % frames.length]} ${text}`;
-    const pad = lastLen > line.length ? ' '.repeat(lastLen - line.length) : '';
+    const pad = lastLen > line.length ? " ".repeat(lastLen - line.length) : "";
     lastLen = line.length;
     out.write(`\r${line}${pad}`);
   }
 
   function start(initialText) {
     if (!enabled) return;
-    text = initialText || '';
+    text = initialText || "";
     if (timer) clearInterval(timer);
     timer = setInterval(render, intervalMs);
     render();
   }
 
   function update(nextText) {
-    text = nextText || '';
+    text = nextText || "";
     render();
   }
 
@@ -34,7 +34,7 @@ function createProgress({ stream } = {}) {
     if (!enabled) return;
     if (timer) clearInterval(timer);
     timer = null;
-    out.write(`\r${' '.repeat(lastLen)}\r`);
+    out.write(`\r${" ".repeat(lastLen)}\r`);
     lastLen = 0;
   }
 
@@ -45,19 +45,19 @@ function renderBar(progress, width = 20) {
   const p = Number.isFinite(progress) ? Math.min(1, Math.max(0, progress)) : 0;
   const filled = Math.round(p * width);
   const empty = Math.max(0, width - filled);
-  return `[${'='.repeat(filled)}${'-'.repeat(empty)}] ${Math.round(p * 100)}%`;
+  return `[${"=".repeat(filled)}${"-".repeat(empty)}] ${Math.round(p * 100)}%`;
 }
 
 function formatNumber(n) {
   const v = Number(n);
-  if (!Number.isFinite(v)) return '0';
-  return Math.trunc(v).toLocaleString('en-US');
+  if (!Number.isFinite(v)) return "0";
+  return Math.trunc(v).toLocaleString("en-US");
 }
 
 function formatBytes(bytes) {
   const n = Number(bytes);
-  if (!Number.isFinite(n) || n <= 0) return '0 B';
-  const units = ['B', 'KB', 'MB', 'GB', 'TB'];
+  if (!Number.isFinite(n) || n <= 0) return "0 B";
+  const units = ["B", "KB", "MB", "GB", "TB"];
   let v = n;
   let i = 0;
   while (v >= 1024 && i < units.length - 1) {
@@ -72,6 +72,5 @@ module.exports = {
   createProgress,
   renderBar,
   formatNumber,
-  formatBytes
+  formatBytes,
 };
-

package/src/lib/project-usage-purge.js

@@ -1,15 +1,20 @@
-const fs = require('node:fs');
-const fsp = require('node:fs/promises');
-const path = require('node:path');
-const readline = require('node:readline');
+const fs = require("node:fs");
+const fsp = require("node:fs/promises");
+const path = require("node:path");
+const readline = require("node:readline");
 
-async function purgeProjectUsage({ projectKey, projectQueuePath, projectQueueStatePath, projectState }) {
+async function purgeProjectUsage({
+  projectKey,
+  projectQueuePath,
+  projectQueueStatePath,
+  projectState,
+}) {
   if (!projectKey || !projectQueuePath || !projectQueueStatePath || !projectState) {
     return { removed: 0, kept: 0, removedBuckets: 0 };
   }
 
   let previousOffset = 0;
-  const previousState = await fsp.readFile(projectQueueStatePath, 'utf8').catch(() => null);
+  const previousState = await fsp.readFile(projectQueueStatePath, "utf8").catch(() => null);
   if (previousState) {
     try {
       const parsed = JSON.parse(previousState);
@@ -20,7 +25,8 @@ async function purgeProjectUsage({ projectKey, projectQueuePath, projectQueueSta
     }
   }
 
-  const buckets = projectState.buckets && typeof projectState.buckets === 'object' ? projectState.buckets : {};
+  const buckets =
+    projectState.buckets && typeof projectState.buckets === "object" ? projectState.buckets : {};
   let removedBuckets = 0;
   const prefix = `${projectKey}|`;
   for (const key of Object.keys(buckets)) {
@@ -38,24 +44,24 @@ async function purgeProjectUsage({ projectKey, projectQueuePath, projectQueueSta
   if (st && st.isFile()) {
     const tmpPath = `${projectQueuePath}.tmp`;
     await fsp.mkdir(path.dirname(projectQueuePath), { recursive: true });
-    const input = fs.createReadStream(projectQueuePath, 'utf8');
-    const output = fs.createWriteStream(tmpPath, { encoding: 'utf8' });
+    const input = fs.createReadStream(projectQueuePath, "utf8");
+    const output = fs.createWriteStream(tmpPath, { encoding: "utf8" });
     const rl = readline.createInterface({ input, crlfDelay: Infinity });
 
     let inputOffset = 0;
     let outputOffset = 0;
     for await (const line of rl) {
       const trimmed = line.trim();
-      const inputBytes = Buffer.byteLength(line, 'utf8') + 1;
+      const inputBytes = Buffer.byteLength(line, "utf8") + 1;
       inputOffset += inputBytes;
       if (!trimmed) continue;
       let parsed = null;
       try {
         parsed = JSON.parse(trimmed);
       } catch (_err) {
-        const entry = trimmed + '\n';
+        const entry = trimmed + "\n";
         output.write(entry);
-        outputOffset += Buffer.byteLength(entry, 'utf8');
+        outputOffset += Buffer.byteLength(entry, "utf8");
         kept += 1;
         if (inputOffset <= previousOffset) {
           nextOffset = outputOffset;
@@ -69,9 +75,9 @@ async function purgeProjectUsage({ projectKey, projectQueuePath, projectQueueSta
         }
         continue;
       }
-      const entry = JSON.stringify(parsed) + '\n';
+      const entry = JSON.stringify(parsed) + "\n";
       output.write(entry);
-      outputOffset += Buffer.byteLength(entry, 'utf8');
+      outputOffset += Buffer.byteLength(entry, "utf8");
       kept += 1;
       if (inputOffset <= previousOffset) {
         nextOffset = outputOffset;
@@ -80,7 +86,7 @@ async function purgeProjectUsage({ projectKey, projectQueuePath, projectQueueSta
 
     await new Promise((resolve, reject) => {
       output.end(resolve);
-      output.on('error', reject);
+      output.on("error", reject);
     });
 
     await fsp.rename(tmpPath, projectQueuePath);
@@ -88,11 +94,11 @@ async function purgeProjectUsage({ projectKey, projectQueuePath, projectQueueSta
       nextOffset = outputOffset;
     }
   } else {
-    await fsp.writeFile(projectQueuePath, '', 'utf8');
+    await fsp.writeFile(projectQueuePath, "", "utf8");
     nextOffset = 0;
   }
 
-  await fsp.writeFile(projectQueueStatePath, JSON.stringify({ offset: nextOffset }), 'utf8');
+  await fsp.writeFile(projectQueueStatePath, JSON.stringify({ offset: nextOffset }), "utf8");
 
   return { removed, kept, removedBuckets };
 }

package/src/lib/prompt.js

@@ -1,20 +1,24 @@
-const readline = require('node:readline');
+const readline = require("node:readline");
 
 async function prompt(label) {
   const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
   const value = await new Promise((resolve) => rl.question(label, resolve));
   rl.close();
-  return String(value || '').trim();
+  return String(value || "").trim();
 }
 
 async function promptHidden(label) {
-  const rl = readline.createInterface({ input: process.stdin, output: process.stdout, terminal: true });
+  const rl = readline.createInterface({
+    input: process.stdin,
+    output: process.stdout,
+    terminal: true,
+  });
   const value = await new Promise((resolve) => {
     rl._writeToOutput = function _writeToOutput() {};
     rl.question(label, (answer) => resolve(answer));
   });
   rl.close();
-  return String(value || '').trim();
+  return String(value || "").trim();
 }
 
 module.exports = { prompt, promptHidden };