@bismawy/pi-arnative 0.1.0

package/lib/usage-store.ts

@@ -0,0 +1,354 @@
+/**
+ * Token usage data layer — the single source for /usage and the Model tab.
+ * - Scans local Pi session history (~/.pi/agent/sessions/*.jsonl).
+ * - mtime+size keyed cache in ~/.pi/agent/arnative-usage-cache.json, read into
+ *   memory once (parsing 4MB per header frame cost ~50ms and lagged long chats).
+ * - No double counting: the active session is already in its own file, so it is
+ *   NOT re-added from sessionManager (that inflated the ↑↓ totals).
+ */
+import { existsSync, mkdirSync, readFileSync, readdirSync, statSync, writeFileSync } from "node:fs";
+import { dirname, join } from "node:path";
+import { getAgentDir } from "@earendil-works/pi-coding-agent";
+
+export interface UsageEntry {
+	provider: string;
+	model: string;
+	input: number;
+	output: number;
+	cacheRead: number;
+	cost: number;
+}
+
+export interface FileCacheItem {
+	mtimeMs: number;
+	size: number;
+	v?: number;
+	entries: UsageEntry[];
+}
+
+export interface UsageModelStats {
+	name: string;
+	sessions: number;
+	msgs: number;
+	input: number;
+	output: number;
+	cacheRead: number;
+	totalTokens: number;
+	cost: number;
+}
+
+export interface UsageProviderStats {
+	name: string;
+	sessions: number;
+	msgs: number;
+	input: number;
+	output: number;
+	cacheRead: number;
+	totalTokens: number;
+	cost: number;
+	models: UsageModelStats[];
+}
+
+export interface UsageSummary {
+	providers: UsageProviderStats[];
+	totals: {
+		sessions: number;
+		msgs: number;
+		input: number;
+		output: number;
+		cacheRead: number;
+		totalTokens: number;
+		cost: number;
+	};
+}
+
+/** Token count with k/M/B units; 0 or negative = "-". */
+export function formatTokens(n: number): string {
+	if (!n || n <= 0) return "-";
+	if (n < 1000) return String(n);
+	if (n < 1_000_000) {
+		const k = n / 1000;
+		return `${k < 10 ? k.toFixed(1) : Math.round(k)}k`;
+	}
+	if (n < 1_000_000_000) {
+		const m = n / 1_000_000;
+		return `${m.toFixed(1).replace(/\.0$/, "")}M`;
+	}
+	const b = n / 1_000_000_000;
+	return `${b.toFixed(1).replace(/\.0$/, "")}B`;
+}
+
+const CACHE_FILE_PATH = join(getAgentDir(), "arnative-usage-cache.json");
+const CACHE_VERSION = 2; // v2: per-entry cost
+
+// Disk cache is memoized; save writes memory + disk together.
+let diskMemo: Map<string, FileCacheItem> | null = null;
+
+function loadDiskCache(): Map<string, FileCacheItem> {
+	if (diskMemo) return diskMemo;
+	diskMemo = new Map<string, FileCacheItem>();
+	try {
+		if (existsSync(CACHE_FILE_PATH)) {
+			const raw = JSON.parse(readFileSync(CACHE_FILE_PATH, "utf8")) as Record<string, FileCacheItem>;
+			for (const [k, v] of Object.entries(raw)) {
+				diskMemo.set(k, v);
+			}
+		}
+	} catch {
+		// ignore
+	}
+	return diskMemo;
+}
+
+function saveDiskCache(map: Map<string, FileCacheItem>): void {
+	try {
+		const obj: Record<string, FileCacheItem> = {};
+		for (const [k, v] of map.entries()) {
+			obj[k] = v;
+		}
+		const dir = dirname(CACHE_FILE_PATH);
+		if (!existsSync(dir)) mkdirSync(dir, { recursive: true });
+		writeFileSync(CACHE_FILE_PATH, JSON.stringify(obj), "utf8");
+	} catch {
+		// ignore
+	}
+}
+
+function scanSessionFiles(dir: string): string[] {
+	const files: string[] = [];
+	try {
+		if (!existsSync(dir)) return files;
+		const list = readdirSync(dir, { withFileTypes: true });
+		for (const ent of list) {
+			const full = join(dir, ent.name);
+			if (ent.isDirectory()) {
+				files.push(...scanSessionFiles(full));
+			} else if (ent.isFile() && ent.name.endsWith(".jsonl")) {
+				files.push(full);
+			}
+		}
+	} catch {
+		// ignore
+	}
+	return files;
+}
+
+function parseSessionFile(filePath: string): UsageEntry[] {
+	const entries: UsageEntry[] = [];
+	try {
+		const content = readFileSync(filePath, "utf8");
+		const lines = content.split("\n");
+		for (const line of lines) {
+			if (!line || !line.includes('"role":"assistant"')) continue;
+			try {
+				const obj = JSON.parse(line);
+				const m = obj?.message;
+				if (m?.role === "assistant") {
+					const provider = String(m.provider || "unknown");
+					const model = String(m.model || "unknown");
+					const u = m.usage || {};
+					entries.push({
+						provider,
+						model,
+						input: Number(u.input) || 0,
+						output: Number(u.output) || 0,
+						cacheRead: Number(u.cacheRead) || 0,
+						cost: Number(u.cost?.total) || 0,
+					});
+				}
+			} catch {
+				// unparsable line
+			}
+		}
+	} catch {
+		// unreadable file
+	}
+	return entries;
+}
+
+let cachedSummary: UsageSummary | null = null;
+let lastScanMs = 0;
+
+export function collectUsageSummary(forceScan = false): UsageSummary {
+	const now = Date.now();
+	if (!forceScan && cachedSummary && now - lastScanMs < 5000) {
+		return cachedSummary;
+	}
+
+	const sessionsDir = join(getAgentDir(), "sessions");
+	const files = scanSessionFiles(sessionsDir);
+	const diskCache = loadDiskCache();
+	let cacheDirty = false;
+
+	const providerMap = new Map<
+		string,
+		{
+			sessions: Set<string>;
+			msgs: number;
+			input: number;
+			output: number;
+			cacheRead: number;
+			cost: number;
+			models: Map<string, { sessions: Set<string>; msgs: number; input: number; output: number; cacheRead: number; cost: number }>;
+		}
+	>();
+
+	for (const file of files) {
+		let stat;
+		try {
+			stat = statSync(file);
+		} catch {
+			continue;
+		}
+
+		let fileEntries: UsageEntry[];
+		const cached = diskCache.get(file);
+		if (cached && cached.v === CACHE_VERSION && cached.mtimeMs === stat.mtimeMs && cached.size === stat.size) {
+			fileEntries = cached.entries;
+		} else {
+			fileEntries = parseSessionFile(file);
+			diskCache.set(file, {
+				mtimeMs: stat.mtimeMs,
+				size: stat.size,
+				v: CACHE_VERSION,
+				entries: fileEntries,
+			});
+			cacheDirty = true;
+		}
+
+		for (const ent of fileEntries) {
+			if (!providerMap.has(ent.provider)) {
+				providerMap.set(ent.provider, {
+					sessions: new Set(),
+					msgs: 0,
+					input: 0,
+					output: 0,
+					cacheRead: 0,
+					cost: 0,
+					models: new Map(),
+				});
+			}
+			const p = providerMap.get(ent.provider)!;
+			p.sessions.add(file);
+			p.msgs++;
+			p.input += ent.input;
+			p.output += ent.output;
+			p.cacheRead += ent.cacheRead;
+			p.cost += ent.cost;
+
+			if (!p.models.has(ent.model)) {
+				p.models.set(ent.model, {
+					sessions: new Set(),
+					msgs: 0,
+					input: 0,
+					output: 0,
+					cacheRead: 0,
+					cost: 0,
+				});
+			}
+			const m = p.models.get(ent.model)!;
+			m.sessions.add(file);
+			m.msgs++;
+			m.input += ent.input;
+			m.output += ent.output;
+			m.cacheRead += ent.cacheRead;
+			m.cost += ent.cost;
+		}
+	}
+
+	if (cacheDirty) {
+		saveDiskCache(diskCache);
+	}
+
+	// Fold per-provider results into totals
+	let totalMsgs = 0;
+	let totalInp = 0;
+	let totalOut = 0;
+	let totalCache = 0;
+	let totalCost = 0;
+
+	const allSessions = new Set<string>();
+	const providers: UsageProviderStats[] = [];
+
+	for (const [pName, pData] of providerMap.entries()) {
+		for (const s of pData.sessions) allSessions.add(s);
+		totalMsgs += pData.msgs;
+		totalInp += pData.input;
+		totalOut += pData.output;
+		totalCache += pData.cacheRead;
+		totalCost += pData.cost;
+
+		const models: UsageModelStats[] = [];
+		for (const [mName, mData] of pData.models.entries()) {
+			models.push({
+				name: mName,
+				sessions: mData.sessions.size,
+				msgs: mData.msgs,
+				input: mData.input,
+				output: mData.output,
+				cacheRead: mData.cacheRead,
+				totalTokens: mData.input + mData.output,
+				cost: mData.cost,
+			});
+		}
+		models.sort((a, b) => b.totalTokens - a.totalTokens);
+
+		providers.push({
+			name: pName,
+			sessions: pData.sessions.size,
+			msgs: pData.msgs,
+			input: pData.input,
+			output: pData.output,
+			cacheRead: pData.cacheRead,
+			totalTokens: pData.input + pData.output,
+			cost: pData.cost,
+			models,
+		});
+	}
+
+	providers.sort((a, b) => b.totalTokens - a.totalTokens);
+
+	cachedSummary = {
+		providers,
+		totals: {
+			sessions: allSessions.size,
+			msgs: totalMsgs,
+			input: totalInp,
+			output: totalOut,
+			cacheRead: totalCache,
+			totalTokens: totalInp + totalOut,
+			cost: totalCost,
+		},
+	};
+	lastScanMs = now;
+	return cachedSummary;
+}
+
+// Match model names across id shapes ("gemini-3.8-flash", "vendor/gemini-…").
+const modelMatches = (mNameLower: string, targetLower: string): boolean =>
+	!targetLower ||
+	mNameLower === targetLower ||
+	mNameLower.endsWith("/" + targetLower) ||
+	mNameLower.split(":")[0] === targetLower ||
+	targetLower.endsWith("/" + mNameLower);
+
+/** All-time usage for one model (same numbers as /usage; the active session is already in its own file). */
+export function getModelAllTimeUsage(modelId?: string): { input: number; output: number; cacheRead: number } {
+	const summary = collectUsageSummary();
+	let inp = 0;
+	let out = 0;
+	let read = 0;
+
+	const target = (modelId || "").toLowerCase();
+	for (const p of summary.providers) {
+		for (const m of p.models) {
+			if (modelMatches(m.name.toLowerCase(), target)) {
+				inp += m.input;
+				out += m.output;
+				read += m.cacheRead;
+			}
+		}
+	}
+
+	return { input: inp, output: out, cacheRead: read };
+}

package/package.json

@@ -0,0 +1,43 @@
+{
+  "name": "@bismawy/pi-arnative",
+  "version": "0.1.0",
+  "description": "Arnative look for Pi: 11 warm-neutral themes, rounded tool boxes, branded header with tabs, 2-line footer with git/model/status, transcript clock, and a /usage dashboard.",
+  "keywords": [
+    "pi-package",
+    "pi-theme",
+    "pi-extension"
+  ],
+  "type": "module",
+  "license": "MIT",
+  "publishConfig": {
+    "access": "public"
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/bismawy/pi-arnative.git"
+  },
+  "homepage": "https://github.com/bismawy/pi-arnative#readme",
+  "files": [
+    "extensions",
+    "lib",
+    "themes",
+    "README.md"
+  ],
+  "pi": {
+    "extensions": [
+      "./extensions/*.ts"
+    ],
+    "themes": [
+      "./themes/*.json"
+    ]
+  },
+  "peerDependencies": {
+    "@earendil-works/pi-ai": "*",
+    "@earendil-works/pi-coding-agent": "*",
+    "@earendil-works/pi-tui": "*"
+  },
+  "scripts": {
+    "test": "node lib/box.ts && node lib/ansi.ts && node themes/gen-themes.mjs --check && node extensions/tools.ts && node extensions/footer.ts && node extensions/timestamps.ts && node extensions/section-headers.ts && node extensions/ui-render-tweaks.ts && node extensions/usage.ts",
+    "themes": "node themes/gen-themes.mjs"
+  }
+}

package/themes/arnative-cyberpunk.json

@@ -0,0 +1,94 @@
+{
+	"$schema": "https://raw.githubusercontent.com/earendil-works/pi/main/packages/coding-agent/src/modes/interactive/theme/theme-schema.json",
+	"name": "arnative-cyberpunk",
+	"vars": {
+		"cyan": "#00f0ff",
+		"blue": "#2a7fd4",
+		"green": "#00ff9f",
+		"red": "#ff4d6d",
+		"yellow": "#ffe600",
+		"text": "#eaeaf5",
+		"gray": "#8b88a8",
+		"dimGray": "#6f6a9a",
+		"darkGray": "#4a4763",
+		"accent": "#ff2e97",
+		"softCyan": "#7fd8f0",
+		"searchBg": "#5f3349",
+		"selectedBg": "#2b2350",
+		"userMsgBg": "#16131f",
+		"toolPendingBg": "#0d0b16",
+		"toolSuccessBg": "#0d2430",
+		"toolErrorBg": "#3a1524",
+		"customMsgBg": "#1a1329"
+	},
+	"colors": {
+		"accent": "accent",
+		"tint": "softCyan",
+		"border": "blue",
+		"borderAccent": "cyan",
+		"borderMuted": "darkGray",
+		"success": "green",
+		"error": "red",
+		"warning": "yellow",
+		"muted": "gray",
+		"dim": "dimGray",
+		"text": "text",
+		"thinkingText": "gray",
+
+		"selectedBg": "selectedBg",
+		"scrollbarTrack": "darkGray",
+		"scrollbarThumb": "softCyan",
+		"searchMatchBg": "searchBg",
+		"searchMatchText": "text",
+		"userMessageBg": "userMsgBg",
+		"userMessageText": "text",
+		"customMessageBg": "customMsgBg",
+		"customMessageText": "text",
+		"customMessageLabel": "#00f0ff",
+		"toolPendingBg": "toolPendingBg",
+		"toolSuccessBg": "toolSuccessBg",
+		"toolErrorBg": "toolErrorBg",
+		"toolTitle": "text",
+		"toolOutput": "gray",
+
+		"mdHeading": "#ffe600",
+		"mdLink": "#00f0ff",
+		"mdLinkUrl": "dimGray",
+		"mdCode": "accent",
+		"mdCodeBlock": "green",
+		"mdCodeBlockBorder": "gray",
+		"mdQuote": "gray",
+		"mdQuoteBorder": "gray",
+		"mdHr": "gray",
+		"mdListBullet": "accent",
+
+		"toolDiffAdded": "green",
+		"toolDiffRemoved": "red",
+		"toolDiffContext": "gray",
+
+		"syntaxComment": "#6f6a9a",
+		"syntaxKeyword": "#00f0ff",
+		"syntaxFunction": "#ffe600",
+		"syntaxVariable": "#ffd9f2",
+		"syntaxString": "#ff2e97",
+		"syntaxNumber": "#ff9a3c",
+		"syntaxType": "#00ff9f",
+		"syntaxOperator": "#c8c8e8",
+		"syntaxPunctuation": "#9a96c0",
+
+		"thinkingOff": "darkGray",
+		"thinkingMinimal": "#33285c",
+		"thinkingLow": "#4d3a94",
+		"thinkingMedium": "#7a4fd6",
+		"thinkingHigh": "#b84fe8",
+		"thinkingXhigh": "#ff4fc3",
+		"thinkingMax": "accent",
+
+		"bashMode": "green"
+	},
+	"export": {
+		"pageBg": "#0d0b16",
+		"cardBg": "#16131f",
+		"infoBg": "#3a2a10"
+	}
+}

package/themes/arnative-dracula.json

@@ -0,0 +1,94 @@
+{
+	"$schema": "https://raw.githubusercontent.com/earendil-works/pi/main/packages/coding-agent/src/modes/interactive/theme/theme-schema.json",
+	"name": "arnative-dracula",
+	"vars": {
+		"cyan": "#8be9fd",
+		"blue": "#6272a4",
+		"green": "#50fa7b",
+		"red": "#ff6363",
+		"yellow": "#f1fa8c",
+		"text": "#f8f8f2",
+		"gray": "#8e96ba",
+		"dimGray": "#6272a4",
+		"darkGray": "#4a4e63",
+		"accent": "#bd93f9",
+		"softCyan": "#c9b8f2",
+		"searchBg": "#59427b",
+		"selectedBg": "#44475a",
+		"userMsgBg": "#343746",
+		"toolPendingBg": "#21222c",
+		"toolSuccessBg": "#26332e",
+		"toolErrorBg": "#412936",
+		"customMsgBg": "#383a52"
+	},
+	"colors": {
+		"accent": "accent",
+		"tint": "softCyan",
+		"border": "blue",
+		"borderAccent": "cyan",
+		"borderMuted": "darkGray",
+		"success": "green",
+		"error": "red",
+		"warning": "yellow",
+		"muted": "gray",
+		"dim": "dimGray",
+		"text": "text",
+		"thinkingText": "gray",
+
+		"selectedBg": "selectedBg",
+		"scrollbarTrack": "darkGray",
+		"scrollbarThumb": "softCyan",
+		"searchMatchBg": "searchBg",
+		"searchMatchText": "text",
+		"userMessageBg": "userMsgBg",
+		"userMessageText": "text",
+		"customMessageBg": "customMsgBg",
+		"customMessageText": "text",
+		"customMessageLabel": "#ff79c6",
+		"toolPendingBg": "toolPendingBg",
+		"toolSuccessBg": "toolSuccessBg",
+		"toolErrorBg": "toolErrorBg",
+		"toolTitle": "text",
+		"toolOutput": "gray",
+
+		"mdHeading": "#f1fa8c",
+		"mdLink": "#8be9fd",
+		"mdLinkUrl": "dimGray",
+		"mdCode": "accent",
+		"mdCodeBlock": "green",
+		"mdCodeBlockBorder": "gray",
+		"mdQuote": "gray",
+		"mdQuoteBorder": "gray",
+		"mdHr": "gray",
+		"mdListBullet": "accent",
+
+		"toolDiffAdded": "green",
+		"toolDiffRemoved": "red",
+		"toolDiffContext": "gray",
+
+		"syntaxComment": "#6272a4",
+		"syntaxKeyword": "#ff79c6",
+		"syntaxFunction": "#50fa7b",
+		"syntaxVariable": "#8be9fd",
+		"syntaxString": "#f1fa8c",
+		"syntaxNumber": "#bd93f9",
+		"syntaxType": "#8be9fd",
+		"syntaxOperator": "#ff79c6",
+		"syntaxPunctuation": "#f8f8f2",
+
+		"thinkingOff": "darkGray",
+		"thinkingMinimal": "#464a5d",
+		"thinkingLow": "#6272a4",
+		"thinkingMedium": "#7a6fd0",
+		"thinkingHigh": "#bd93f9",
+		"thinkingXhigh": "#ff79c6",
+		"thinkingMax": "accent",
+
+		"bashMode": "green"
+	},
+	"export": {
+		"pageBg": "#21222c",
+		"cardBg": "#282a36",
+		"infoBg": "#44475a"
+	}
+}

package/themes/arnative-emerald.json

@@ -0,0 +1,94 @@
+{
+	"$schema": "https://raw.githubusercontent.com/earendil-works/pi/main/packages/coding-agent/src/modes/interactive/theme/theme-schema.json",
+	"name": "arnative-emerald",
+	"vars": {
+		"cyan": "#a7f3d0",
+		"blue": "#34b98a",
+		"green": "#b5bd68",
+		"red": "#d07272",
+		"yellow": "#ffff00",
+		"text": "#d4d4d4",
+		"gray": "#8d8d8d",
+		"dimGray": "#666666",
+		"darkGray": "#505050",
+		"accent": "#6ee7b7",
+		"softCyan": "#7eceaf",
+		"searchBg": "#356351",
+		"selectedBg": "#324840",
+		"userMsgBg": "#25312d",
+		"toolPendingBg": "#1f2925",
+		"toolSuccessBg": "#202828",
+		"toolErrorBg": "#342222",
+		"customMsgBg": "#1f3028"
+	},
+	"colors": {
+		"accent": "accent",
+		"tint": "softCyan",
+		"border": "blue",
+		"borderAccent": "cyan",
+		"borderMuted": "darkGray",
+		"success": "green",
+		"error": "red",
+		"warning": "yellow",
+		"muted": "gray",
+		"dim": "dimGray",
+		"text": "text",
+		"thinkingText": "gray",
+
+		"selectedBg": "selectedBg",
+		"scrollbarTrack": "darkGray",
+		"scrollbarThumb": "softCyan",
+		"searchMatchBg": "searchBg",
+		"searchMatchText": "text",
+		"userMessageBg": "userMsgBg",
+		"userMessageText": "text",
+		"customMessageBg": "customMsgBg",
+		"customMessageText": "text",
+		"customMessageLabel": "#9575cd",
+		"toolPendingBg": "toolPendingBg",
+		"toolSuccessBg": "toolSuccessBg",
+		"toolErrorBg": "toolErrorBg",
+		"toolTitle": "text",
+		"toolOutput": "gray",
+
+		"mdHeading": "#f0c674",
+		"mdLink": "#81a2be",
+		"mdLinkUrl": "dimGray",
+		"mdCode": "accent",
+		"mdCodeBlock": "green",
+		"mdCodeBlockBorder": "gray",
+		"mdQuote": "gray",
+		"mdQuoteBorder": "gray",
+		"mdHr": "gray",
+		"mdListBullet": "accent",
+
+		"toolDiffAdded": "green",
+		"toolDiffRemoved": "red",
+		"toolDiffContext": "gray",
+
+		"syntaxComment": "#6A9955",
+		"syntaxKeyword": "#569CD6",
+		"syntaxFunction": "#DCDCAA",
+		"syntaxVariable": "#9CDCFE",
+		"syntaxString": "#CE9178",
+		"syntaxNumber": "#B5CEA8",
+		"syntaxType": "#4EC9B0",
+		"syntaxOperator": "#D4D4D4",
+		"syntaxPunctuation": "#D4D4D4",
+
+		"thinkingOff": "darkGray",
+		"thinkingMinimal": "#305f4d",
+		"thinkingLow": "#3c8066",
+		"thinkingMedium": "#50aa87",
+		"thinkingHigh": "#41e1a4",
+		"thinkingXhigh": "#4af2b2",
+		"thinkingMax": "accent",
+
+		"bashMode": "green"
+	},
+	"export": {
+		"pageBg": "#18181e",
+		"cardBg": "#1e1e24",
+		"infoBg": "#3c3728"
+	}
+}