@pify/usage 0.1.0 → 0.2.0

package/README.md

@@ -22,9 +22,13 @@ History (214 local session files)
 By model (all time)
   anthropic/claude-fable-5  $12.30 · 4.1M tok
   openai/gpt-5.5            $9.10 · 5.7M tok
+By project (all time)
+  D--project-pify-plugins   $14.80 · 6.2M tok
+  D--project-shop-api       $6.60 · 3.6M tok
 ```
 
 - **History done right** (tmustier's lessons): counts every usage-bearing entry in pi's session JSONL — assistant turns plus the tool-result/compaction usage pi 0.81+ persists; negative/NaN fields clamp to zero; days are your local calendar days; a per-file mtime cache keeps repeat scans instant.
+- **Per-project spend** (v0.2): pi stores sessions one directory per project, so the dashboard can show where the money actually went — the top 5 projects by cost, all time.
 - **`usage_status` tool**: the agent can check session + today totals before committing to expensive work (subagent fan-outs, large reads).
 
 Provider quota APIs (Codex windows, Copilot allowances, OpenRouter credits…) are deliberately out of v0.1 — they cost ~18k lines of per-provider contract maintenance (see `@narumitw/pi-usage` if you need them today).

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pify/usage",
-  "version": "0.1.0",
+  "version": "0.2.0",
   "description": "Token and cost reporting for pi sessions: live footer, /usage dashboard over local session history, zero network",
   "keywords": [
     "pi-package",

package/src/aggregate.ts

@@ -32,6 +32,7 @@ export function dayKey(timestamp: number): string {
 export function aggregate(records: UsageRecord[], files: number): HistoryAggregate {
   const byDay = new Map<string, UsageTotals>();
   const byModel = new Map<string, UsageTotals>();
+  const byProject = new Map<string, UsageTotals>();
   const total = emptyTotals();
 
   for (const record of records) {
@@ -40,9 +41,13 @@ export function aggregate(records: UsageRecord[], files: number): HistoryAggrega
     addRecord(byDay.get(day) ?? byDay.set(day, emptyTotals()).get(day)!, record);
     const model = `${record.provider}/${record.model}`;
     addRecord(byModel.get(model) ?? byModel.set(model, emptyTotals()).get(model)!, record);
+    if (record.project) {
+      const p = record.project;
+      addRecord(byProject.get(p) ?? byProject.set(p, emptyTotals()).get(p)!, record);
+    }
   }
 
-  return { byDay, byModel, total, files };
+  return { byDay, byModel, byProject, total, files };
 }
 
 /** Sum totals for days within the trailing window (inclusive of today). */
@@ -91,6 +96,7 @@ export function recordFromEntry(entry: unknown): UsageRecord | null {
     timestamp: Number.isFinite(ts) ? ts : 0,
     model: typeof message.model === "string" ? message.model : "unknown",
     provider: typeof message.provider === "string" ? message.provider : "unknown",
+    project: "",
     input: finite(usage.input),
     output: finite(usage.output),
     cacheRead: finite(usage.cacheRead),

package/src/format.ts

@@ -43,16 +43,24 @@ export function historyBlock(history: HistoryAggregate, now: number): string {
     `  30 days  ${formatCost(month.cost)} · ${formatTokens(month.totalTokens)} tok`,
   ];
 
-  const models = [...history.byModel.entries()]
-    .sort((a, b) => b[1].cost - a[1].cost)
-    .slice(0, 6);
-  if (models.length > 0) {
-    lines.push("By model (all time)");
-    const width = models.reduce((m, [name]) => Math.max(m, Math.min(name.length, 40)), 0);
-    for (const [name, totals] of models) {
-      const label = name.length > 40 ? `${name.slice(0, 39)}…` : name;
-      lines.push(`  ${label.padEnd(width)}  ${formatCost(totals.cost)} · ${formatTokens(totals.totalTokens)} tok`);
-    }
-  }
+  pushBreakdown(lines, "By model (all time)", history.byModel, 6);
+  pushBreakdown(lines, "By project (all time)", history.byProject, 5);
   return lines.join("\n");
 }
+
+/** One "name  $cost · N tok" table, biggest spend first. */
+function pushBreakdown(
+  lines: string[],
+  heading: string,
+  totals: Map<string, UsageTotals>,
+  limit: number,
+): void {
+  const rows = [...totals.entries()].sort((a, b) => b[1].cost - a[1].cost).slice(0, limit);
+  if (rows.length === 0) return;
+  lines.push(heading);
+  const width = rows.reduce((m, [name]) => Math.max(m, Math.min(name.length, 40)), 0);
+  for (const [name, t] of rows) {
+    const label = name.length > 40 ? `${name.slice(0, 39)}…` : name;
+    lines.push(`  ${label.padEnd(width)}  ${formatCost(t.cost)} · ${formatTokens(t.totalTokens)} tok`);
+  }
+}

package/src/sessions.ts

@@ -44,27 +44,70 @@ export interface ScanResult {
   files: number;
 }
 
-export function scanSessions(sessionsDir: string): ScanResult {
-  const files = listJsonlFiles(sessionsDir, 0);
-  const records: UsageRecord[] = [];
-  for (const file of files) {
+const MAX_LABEL = 34;
+
+/**
+ * Readable name for a project directory. pi encodes the project path into the
+ * directory name (--D--project-pify-plugins--); the tail is the recognizable
+ * part, so long names keep their end.
+ */
+export function projectLabel(dirName: string): string {
+  const stripped = dirName.replace(/^-+|-+$/g, "");
+  if (!stripped) return "unknown";
+  return stripped.length > MAX_LABEL ? `…${stripped.slice(-(MAX_LABEL - 1))}` : stripped;
+}
+
+/** Session files grouped by the project directory they sit under. */
+function groupByProject(sessionsDir: string): Array<{ project: string; files: string[] }> {
+  let names: string[];
+  try {
+    names = readdirSync(sessionsDir);
+  } catch {
+    return [];
+  }
+  const groups: Array<{ project: string; files: string[] }> = [];
+  const loose: string[] = [];
+  for (const name of names) {
+    const full = join(sessionsDir, name);
     try {
-      const stat = statSync(file);
-      const cached = cache.get(file);
-      if (cached && cached.mtimeMs === stat.mtimeMs && cached.size === stat.size) {
-        records.push(...cached.records);
-        continue;
-      }
-      const fileRecords: UsageRecord[] = [];
-      for (const line of readFileSync(file, "utf8").split("\n")) {
-        const record = recordFromLine(line);
-        if (record) fileRecords.push(record);
+      const stat = statSync(full);
+      if (stat.isDirectory()) {
+        groups.push({ project: projectLabel(name), files: listJsonlFiles(full, 1) });
+      } else if (name.endsWith(".jsonl") && stat.size <= MAX_FILE_BYTES) {
+        loose.push(full);
       }
-      cache.set(file, { mtimeMs: stat.mtimeMs, size: stat.size, records: fileRecords });
-      records.push(...fileRecords);
     } catch {
-      // unreadable — skip
+      // race/permission — skip
+    }
+  }
+  if (loose.length > 0) groups.push({ project: "", files: loose });
+  return groups;
+}
+
+export function scanSessions(sessionsDir: string): ScanResult {
+  const records: UsageRecord[] = [];
+  let fileCount = 0;
+  for (const { project, files } of groupByProject(sessionsDir)) {
+    fileCount += files.length;
+    for (const file of files) {
+      try {
+        const stat = statSync(file);
+        const cached = cache.get(file);
+        if (cached && cached.mtimeMs === stat.mtimeMs && cached.size === stat.size) {
+          records.push(...cached.records);
+          continue;
+        }
+        const fileRecords: UsageRecord[] = [];
+        for (const line of readFileSync(file, "utf8").split("\n")) {
+          const record = recordFromLine(line);
+          if (record) fileRecords.push({ ...record, project });
+        }
+        cache.set(file, { mtimeMs: stat.mtimeMs, size: stat.size, records: fileRecords });
+        records.push(...fileRecords);
+      } catch {
+        // unreadable — skip
+      }
     }
   }
-  return { records, files: files.length };
+  return { records, files: fileCount };
 }

package/src/types.ts

@@ -23,6 +23,8 @@ export interface UsageRecord {
   timestamp: number;
   model: string;
   provider: string;
+  /** Project the session belongs to; "" when unknown (live events). */
+  project: string;
   input: number;
   output: number;
   cacheRead: number;
@@ -34,6 +36,7 @@ export interface UsageRecord {
 export interface HistoryAggregate {
   byDay: Map<string, UsageTotals>;
   byModel: Map<string, UsageTotals>;
+  byProject: Map<string, UsageTotals>;
   total: UsageTotals;
   files: number;
 }