@tokscale/cli 1.2.2 → 1.2.3

package/src/cli.ts

@@ -121,6 +121,7 @@ interface FilterOptions {
   amp?: boolean;
   droid?: boolean;
   openclaw?: boolean;
+  pi?: boolean;
 }
 
 interface DateFilterOptions {
@@ -446,6 +447,7 @@ async function main() {
     .option("--amp", "Show only Amp usage")
     .option("--droid", "Show only Factory Droid usage")
     .option("--openclaw", "Show only OpenClaw usage")
+    .option("--pi", "Show only Pi usage")
     .option("--today", "Show only today's usage")
     .option("--week", "Show last 7 days")
     .option("--month", "Show current month")
@@ -483,6 +485,7 @@ async function main() {
     .option("--amp", "Show only Amp usage")
     .option("--droid", "Show only Factory Droid usage")
     .option("--openclaw", "Show only OpenClaw usage")
+    .option("--pi", "Show only Pi usage")
     .option("--today", "Show only today's usage")
     .option("--week", "Show last 7 days")
     .option("--month", "Show current month")
@@ -529,12 +532,13 @@ async function main() {
         path.join(homeDir, ".moltbot", "agents"),
         path.join(homeDir, ".moldbot", "agents"),
       ];
+      const piSessions = path.join(homeDir, ".pi", "agent", "sessions");
 
       let localMessages: ParsedMessages | null = null;
       try {
         localMessages = await parseLocalSourcesAsync({
           homeDir,
-          sources: ["opencode", "claude", "codex", "gemini", "amp", "droid", "openclaw"],
+          sources: ["opencode", "claude", "codex", "gemini", "amp", "droid", "openclaw", "pi"],
         });
       } catch (e) {
         console.error(`Error: ${(e as Error).message}`);
@@ -634,6 +638,15 @@ async function main() {
           headlessPaths: [],
           headlessMessageCount: 0,
         },
+        {
+          source: "pi",
+          label: "Pi",
+          sessionsPath: piSessions,
+          messageCount: localMessages.piCount,
+          headlessSupported: false,
+          headlessPaths: [],
+          headlessMessageCount: 0,
+        },
       ];
 
       if (options.json) {
@@ -723,6 +736,7 @@ async function main() {
     .option("--amp", "Include only Amp data")
     .option("--droid", "Include only Factory Droid data")
     .option("--openclaw", "Include only OpenClaw data")
+    .option("--pi", "Include only Pi data")
     .option("--today", "Show only today's usage")
     .option("--week", "Show last 7 days")
     .option("--month", "Show current month")
@@ -748,6 +762,7 @@ async function main() {
     .option("--amp", "Include only Amp data")
     .option("--droid", "Include only Factory Droid data")
     .option("--openclaw", "Include only OpenClaw data")
+    .option("--pi", "Include only Pi data")
     .option("--no-spinner", "Disable loading spinner (for scripting)")
     .option("--short", "Display total tokens in abbreviated format (e.g., 7.14B)")
     .addOption(new Option("--agents", "Show Top OpenCode Agents (default)").conflicts("clients"))
@@ -793,6 +808,7 @@ async function main() {
     .option("--amp", "Include only Amp data")
     .option("--droid", "Include only Factory Droid data")
     .option("--openclaw", "Include only OpenClaw data")
+    .option("--pi", "Include only Pi data")
     .option("--since <date>", "Start date (YYYY-MM-DD)")
     .option("--until <date>", "End date (YYYY-MM-DD)")
     .option("--year <year>", "Filter to specific year")
@@ -807,6 +823,7 @@ async function main() {
         amp: options.amp,
         droid: options.droid,
         openclaw: options.openclaw,
+        pi: options.pi,
         since: options.since,
         until: options.until,
         year: options.year,
@@ -829,6 +846,7 @@ async function main() {
     .option("--amp", "Show only Amp usage")
     .option("--droid", "Show only Factory Droid usage")
     .option("--openclaw", "Show only OpenClaw usage")
+    .option("--pi", "Show only Pi usage")
     .option("--today", "Show only today's usage")
     .option("--week", "Show last 7 days")
     .option("--month", "Show current month")
@@ -985,7 +1003,7 @@ async function main() {
 }
 
 function getEnabledSources(options: FilterOptions): SourceType[] | undefined {
-  const hasFilter = options.opencode || options.claude || options.codex || options.gemini || options.cursor || options.amp || options.droid || options.openclaw;
+  const hasFilter = options.opencode || options.claude || options.codex || options.gemini || options.cursor || options.amp || options.droid || options.openclaw || options.pi;
   if (!hasFilter) return undefined; // All sources
 
   const sources: SourceType[] = [];
@@ -997,6 +1015,7 @@ function getEnabledSources(options: FilterOptions): SourceType[] | undefined {
   if (options.amp) sources.push("amp");
   if (options.droid) sources.push("droid");
   if (options.openclaw) sources.push("openclaw");
+  if (options.pi) sources.push("pi");
   return sources;
 }
 
@@ -1086,7 +1105,7 @@ async function showModelReport(options: FilterOptions & DateFilterOptions & { be
   const useSpinner = extraOptions?.spinner !== false;
   const spinner = useSpinner ? createSpinner({ color: "cyan" }) : null;
 
-  const localSources: SourceType[] = (enabledSources || ['opencode', 'claude', 'codex', 'gemini', 'cursor', 'amp', 'droid', 'openclaw'])
+  const localSources: SourceType[] = (enabledSources || ['opencode', 'claude', 'codex', 'gemini', 'cursor', 'amp', 'droid', 'openclaw', 'pi'])
     .filter(s => s !== 'cursor');
 
   spinner?.start(pc.gray("Scanning session data..."));
@@ -1116,7 +1135,7 @@ async function showModelReport(options: FilterOptions & DateFilterOptions & { be
 
   let report: ModelReport;
   try {
-    const emptyMessages: ParsedMessages = { messages: [], opencodeCount: 0, claudeCount: 0, codexCount: 0, geminiCount: 0, ampCount: 0, droidCount: 0, openclawCount: 0, processingTimeMs: 0 };
+    const emptyMessages: ParsedMessages = { messages: [], opencodeCount: 0, claudeCount: 0, codexCount: 0, geminiCount: 0, ampCount: 0, droidCount: 0, openclawCount: 0, piCount: 0, processingTimeMs: 0 };
     report = await finalizeReportAsync({
       localMessages: localMessages || emptyMessages,
       includeCursor: includeCursor && (cursorSync.synced || hasCursorUsageCache()),
@@ -1223,7 +1242,7 @@ async function showMonthlyReport(options: FilterOptions & DateFilterOptions & {
 
   const dateFilters = getDateFilters(options);
   const enabledSources = getEnabledSources(options);
-  const localSources: SourceType[] = (enabledSources || ['opencode', 'claude', 'codex', 'gemini', 'cursor', 'amp', 'droid', 'openclaw'])
+  const localSources: SourceType[] = (enabledSources || ['opencode', 'claude', 'codex', 'gemini', 'cursor', 'amp', 'droid', 'openclaw', 'pi'])
     .filter(s => s !== 'cursor');
   const includeCursor = !enabledSources || enabledSources.includes('cursor');
 
@@ -1332,7 +1351,7 @@ async function outputJsonReport(
   const enabledSources = getEnabledSources(options);
   const onlyCursor = enabledSources?.length === 1 && enabledSources[0] === 'cursor';
   const includeCursor = !enabledSources || enabledSources.includes('cursor');
-  const localSources: SourceType[] = (enabledSources || ['opencode', 'claude', 'codex', 'gemini', 'cursor', 'amp', 'droid', 'openclaw'])
+  const localSources: SourceType[] = (enabledSources || ['opencode', 'claude', 'codex', 'gemini', 'cursor', 'amp', 'droid', 'openclaw', 'pi'])
     .filter(s => s !== 'cursor');
 
   const { cursorSync, localMessages } = await loadDataSourcesParallel(
@@ -1345,7 +1364,7 @@ async function outputJsonReport(
     process.exit(1);
   }
 
-  const emptyMessages: ParsedMessages = { messages: [], opencodeCount: 0, claudeCount: 0, codexCount: 0, geminiCount: 0, ampCount: 0, droidCount: 0, openclawCount: 0, processingTimeMs: 0 };
+  const emptyMessages: ParsedMessages = { messages: [], opencodeCount: 0, claudeCount: 0, codexCount: 0, geminiCount: 0, ampCount: 0, droidCount: 0, openclawCount: 0, piCount: 0, processingTimeMs: 0 };
 
   if (reportType === "models") {
     const report = await finalizeReportAsync({
@@ -1380,7 +1399,7 @@ async function handleGraphCommand(options: GraphCommandOptions) {
 
   const dateFilters = getDateFilters(options);
   const enabledSources = getEnabledSources(options);
-  const localSources: SourceType[] = (enabledSources || ['opencode', 'claude', 'codex', 'gemini', 'cursor', 'amp', 'droid', 'openclaw'])
+  const localSources: SourceType[] = (enabledSources || ['opencode', 'claude', 'codex', 'gemini', 'cursor', 'amp', 'droid', 'openclaw', 'pi'])
     .filter(s => s !== 'cursor');
   const includeCursor = !enabledSources || enabledSources.includes('cursor');
 

package/src/graph-types.ts

@@ -6,7 +6,7 @@
 /**
  * Valid source identifiers
  */
-export type SourceType = "opencode" | "claude" | "codex" | "gemini" | "cursor" | "amp" | "droid" | "openclaw";
+export type SourceType = "opencode" | "claude" | "codex" | "gemini" | "cursor" | "amp" | "droid" | "openclaw" | "pi";
 
 /**
  * Token breakdown by category

package/src/native.ts

@@ -153,8 +153,9 @@ interface NativeParsedMessages {
   codexCount: number;
   geminiCount: number;
   ampCount: number;
-  droidCount?: number;
-  openclawCount?: number;
+  droidCount: number;
+  openclawCount: number;
+  piCount: number;
   processingTimeMs: number;
 }
 
@@ -355,6 +356,7 @@ export interface ParsedMessages {
   ampCount: number;
   droidCount: number;
   openclawCount: number;
+  piCount: number;
   processingTimeMs: number;
 }
 

package/src/sessions/types.ts

@@ -22,7 +22,7 @@ export interface UnifiedMessage {
   agent?: string;
 }
 
-export type SourceType = "opencode" | "claude" | "codex" | "gemini" | "cursor" | "amp" | "droid" | "openclaw";
+export type SourceType = "opencode" | "claude" | "codex" | "gemini" | "cursor" | "amp" | "droid" | "openclaw" | "pi";
 
 /**
  * Convert Unix milliseconds timestamp to YYYY-MM-DD date string

package/src/submit.ts

@@ -25,6 +25,7 @@ interface SubmitOptions {
   amp?: boolean;
   droid?: boolean;
   openclaw?: boolean;
+  pi?: boolean;
   since?: string;
   until?: string;
   year?: string;
@@ -50,7 +51,7 @@ interface SubmitResponse {
   details?: string[];
 }
 
-type SourceType = "opencode" | "claude" | "codex" | "gemini" | "cursor" | "amp" | "droid" | "openclaw";
+type SourceType = "opencode" | "claude" | "codex" | "gemini" | "cursor" | "amp" | "droid" | "openclaw" | "pi";
 
 async function checkGhCliExists(): Promise<boolean> {
   try {
@@ -193,7 +194,7 @@ export async function submit(options: SubmitOptions = {}): Promise<void> {
 
   console.log(pc.gray("  Scanning local session data..."));
 
-  const hasFilter = options.opencode || options.claude || options.codex || options.gemini || options.cursor || options.amp || options.droid || options.openclaw;
+  const hasFilter = options.opencode || options.claude || options.codex || options.gemini || options.cursor || options.amp || options.droid || options.openclaw || options.pi;
   let sources: SourceType[] | undefined;
   let includeCursor = true;
   if (hasFilter) {
@@ -206,6 +207,7 @@ export async function submit(options: SubmitOptions = {}): Promise<void> {
     if (options.amp) sources.push("amp");
     if (options.droid) sources.push("droid");
     if (options.openclaw) sources.push("openclaw");
+    if (options.pi) sources.push("pi");
     includeCursor = sources.includes("cursor");
   }
 

package/src/tui/App.tsx

@@ -288,6 +288,7 @@ export function App(props: AppProps) {
     if (key.name === "6") { handleSourceToggle("amp"); return; }
     if (key.name === "7") { handleSourceToggle("droid"); return; }
     if (key.name === "8") { handleSourceToggle("openclaw"); return; }
+    if (key.name === "9") { handleSourceToggle("pi"); return; }
 
     if (key.name === "up") {
       if (activeTab() === "overview") {

package/src/tui/hooks/useData.ts

@@ -158,8 +158,8 @@ async function loadData(
   const phase1Results = await Promise.allSettled([
     includeCursor && isCursorLoggedIn() ? syncCursorCache() : Promise.resolve({ synced: false, rows: 0, error: undefined }),
     localSources.length > 0
-      ? parseLocalSourcesAsync({ sources: localSources as ("opencode" | "claude" | "codex" | "gemini" | "amp" | "droid" | "openclaw")[], since, until, year })
-      : Promise.resolve({ messages: [], opencodeCount: 0, claudeCount: 0, codexCount: 0, geminiCount: 0, ampCount: 0, droidCount: 0, openclawCount: 0, processingTimeMs: 0 } as ParsedMessages),
+      ? parseLocalSourcesAsync({ sources: localSources as ("opencode" | "claude" | "codex" | "gemini" | "amp" | "droid" | "openclaw" | "pi")[], since, until, year })
+      : Promise.resolve({ messages: [], opencodeCount: 0, claudeCount: 0, codexCount: 0, geminiCount: 0, ampCount: 0, droidCount: 0, openclawCount: 0, piCount: 0, processingTimeMs: 0 } as ParsedMessages),
   ]);
 
   const cursorSync = phase1Results[0].status === "fulfilled" 
@@ -184,6 +184,7 @@ async function loadData(
     ampCount: 0,
     droidCount: 0,
     openclawCount: 0,
+    piCount: 0,
     processingTimeMs: 0,
   };
 

package/src/tui/types/index.ts

@@ -2,7 +2,7 @@ import type { ColorPaletteName } from "../config/themes.js";
 
 export type TabType = "overview" | "model" | "daily" | "stats";
 export type SortType = "cost" | "tokens" | "date";
-export type SourceType = "opencode" | "claude" | "codex" | "cursor" | "gemini" | "amp" | "droid" | "openclaw";
+export type SourceType = "opencode" | "claude" | "codex" | "cursor" | "gemini" | "amp" | "droid" | "openclaw" | "pi";
 
 export type { ColorPaletteName };
 
@@ -163,7 +163,8 @@ export const SOURCE_LABELS: Record<SourceType, string> = {
   amp: "AM",
   droid: "DR",
   openclaw: "CL",
+  pi: "PI",
 } as const;
 
 export const TABS: readonly TabType[] = ["overview", "model", "daily", "stats"] as const;
-export const ALL_SOURCES: readonly SourceType[] = ["opencode", "claude", "codex", "cursor", "gemini", "amp", "droid", "openclaw"] as const;
+export const ALL_SOURCES: readonly SourceType[] = ["opencode", "claude", "codex", "cursor", "gemini", "amp", "droid", "openclaw", "pi"] as const;

package/src/tui/utils/colors.ts

@@ -61,6 +61,7 @@ export const SOURCE_COLORS: Record<SourceType, string> = {
   amp: "#EC4899",
   droid: "#10b981",
   openclaw: "#ef4444",
+  pi: "#f97316",
 };
 
 export function getSourceColor(source: SourceType | string): string {
@@ -70,5 +71,6 @@ export function getSourceColor(source: SourceType | string): string {
 export function getSourceDisplayName(source: string): string {
   if (source === "droid") return "Droid";
   if (source === "openclaw") return "OpenClaw";
+  if (source === "pi") return "Pi";
   return source.charAt(0).toUpperCase() + source.slice(1);
 }

package/src/wrapped.ts

@@ -65,6 +65,7 @@ const SOURCE_DISPLAY_NAMES: Record<string, string> = {
   amp: "Amp",
   droid: "Droid",
   openclaw: "OpenClaw",
+  pi: "Pi",
 };
 
 const ASSETS_BASE_URL = "https://tokscale.ai/assets/logos";
@@ -97,6 +98,7 @@ const CLIENT_LOGO_URLS: Record<string, string> = {
   "Amp": `${ASSETS_BASE_URL}/amp.png`,
   "Droid": `${ASSETS_BASE_URL}/droid.png`,
   "OpenClaw": `${ASSETS_BASE_URL}/openclaw.png`,
+  "Pi": `${ASSETS_BASE_URL}/pi.png`,
 };
 
 const PROVIDER_LOGO_URLS: Record<string, string> = {
@@ -214,8 +216,8 @@ async function ensureFontsLoaded(): Promise<void> {
 
 async function loadWrappedData(options: WrappedOptions): Promise<WrappedData> {
   const year = options.year || new Date().getFullYear().toString();
-  const sources = options.sources || ["opencode", "claude", "codex", "gemini", "cursor", "amp", "droid", "openclaw"];
-  const localSources = sources.filter(s => s !== "cursor") as ("opencode" | "claude" | "codex" | "gemini" | "amp" | "droid" | "openclaw")[];
+  const sources = options.sources || ["opencode", "claude", "codex", "gemini", "cursor", "amp", "droid", "openclaw", "pi"];
+  const localSources = sources.filter(s => s !== "cursor") as ("opencode" | "claude" | "codex" | "gemini" | "amp" | "droid" | "openclaw" | "pi")[];
   const includeCursor = sources.includes("cursor");
 
   const since = `${year}-01-01`;
@@ -225,7 +227,7 @@ async function loadWrappedData(options: WrappedOptions): Promise<WrappedData> {
     includeCursor && isCursorLoggedIn() ? syncCursorCache() : Promise.resolve({ synced: false, rows: 0, error: undefined }),
     localSources.length > 0
       ? parseLocalSourcesAsync({ sources: localSources, since, until, year })
-      : Promise.resolve({ messages: [], opencodeCount: 0, claudeCount: 0, codexCount: 0, geminiCount: 0, ampCount: 0, droidCount: 0, openclawCount: 0, processingTimeMs: 0 } as ParsedMessages),
+      : Promise.resolve({ messages: [], opencodeCount: 0, claudeCount: 0, codexCount: 0, geminiCount: 0, ampCount: 0, droidCount: 0, openclawCount: 0, piCount: 0, processingTimeMs: 0 } as ParsedMessages),
   ]);
 
   const cursorSync = phase1Results[0].status === "fulfilled" 
@@ -249,6 +251,7 @@ async function loadWrappedData(options: WrappedOptions): Promise<WrappedData> {
     ampCount: 0,
     droidCount: 0,
     openclawCount: 0,
+    piCount: 0,
     processingTimeMs: 0,
   };