feat: colorful oh-my-pi logo and optional omp notify hook

README.md

@@ -224,7 +224,7 @@ Upgrade with `brew upgrade --cask mm7894215/tokentracker/tokentracker`. The tap
 | **GitHub Copilot App / CLI** | ✅ Auto | Unified per-request SQLite usage (`~/.copilot/session-store.db`); App DB legacy baseline |
 | **GitHub Copilot Chat extension / legacy CLI** | ✅ Auto | OpenTelemetry file exporter (`COPILOT_OTEL_FILE_EXPORTER_PATH`) |
 | **Kimi Code** | ✅ Auto | Passive `wire.jsonl` reader (`~/.kimi/sessions/**/wire.jsonl`) |
-| **oh-my-pi (Pi Coding Agent)** | ✅ Auto | Passive reader (`~/.omp/agent/sessions/**/*.jsonl`) |
+| **oh-my-pi (Pi Coding Agent)** | ✅ Auto | Passive reader (`~/.omp/agent/sessions/**/*.jsonl`) + optional notify extension (`~/.omp/agent/extensions/tokentracker-notify.ts`) for near-real-time sync |
 | **CodeBuddy** (Tencent) | ✅ Auto | SessionEnd hook in `~/.codebuddy/settings.json` (Claude-Code fork) |
 | **WorkBuddy** (Tencent) | ✅ Auto | SessionEnd hook in `~/.workbuddy/settings.json` (Claude-Code fork) + passive `projects/**/*.jsonl` scan |
 | **Grok Build** (xAI) | ✅ Auto | SessionEnd hook + passive `updates.jsonl` / `signals.json` scan (`~/.grok/sessions/**/`) |

README.zh-CN.md

@@ -179,7 +179,7 @@ brew install mm7894215/tokentracker/tokentracker
 | **GitHub Copilot App / CLI** | ✅ 自动 | 统一逐请求 SQLite 用量（`~/.copilot/session-store.db`）；App DB 作为旧数据基线 |
 | **GitHub Copilot Chat 扩展 / 旧版 CLI** | ✅ 自动 | OpenTelemetry 文件导出（`COPILOT_OTEL_FILE_EXPORTER_PATH`） |
 | **Kimi Code** | ✅ 自动 | 被动读取 `wire.jsonl`（`~/.kimi/sessions/**/wire.jsonl`） |
-| **oh-my-pi (Pi Coding Agent)** | ✅ 自动 | 被动读取（`~/.omp/agent/sessions/**/*.jsonl`） |
+| **oh-my-pi (Pi Coding Agent)** | ✅ 自动 | 被动读取（`~/.omp/agent/sessions/**/*.jsonl`）+ 可选 notify 扩展（`~/.omp/agent/extensions/tokentracker-notify.ts`）近实时同步 |
 | **CodeBuddy** (腾讯) | ✅ 自动 | 写入 `~/.codebuddy/settings.json` 的 SessionEnd hook（Claude-Code fork） |
 | **WorkBuddy** (腾讯) | ✅ 自动 | 写入 `~/.workbuddy/settings.json` 的 SessionEnd hook（Claude-Code fork）+ 被动扫描 `projects/**/*.jsonl` |
 | **Grok Build** (xAI) | ✅ 自动 | SessionEnd hook + 被动扫描 `updates.jsonl` / `signals.json`（`~/.grok/sessions/**/`） |

dashboard/src/lib/provider-display.js

@@ -1,6 +1,7 @@
 const SPECIAL_PROVIDER_NAMES = {
   anythingllm: "AnythingLLM",
   claudescience: "Claude Science",
+  omp: "oh-my-pi",
   pianthropic: "Pi · Anthropic",
   pigithubcopilot: "Pi · GitHub Copilot",
   picopilot: "Pi · Copilot",

dashboard/src/lib/provider-display.test.js

@@ -19,4 +19,9 @@ describe("formatProviderDisplayName", () => {
     expect(formatProviderDisplayName("pi-anthropic")).toBe("Pi · Anthropic");
     expect(formatProviderDisplayName("PI-GITHUB-COPILOT")).toBe("Pi · GitHub Copilot");
   });
+
+  it("formats omp as oh-my-pi", () => {
+    expect(formatProviderDisplayName("omp")).toBe("oh-my-pi");
+    expect(formatProviderDisplayName("OMP")).toBe("oh-my-pi");
+  });
 });

dashboard/src/ui/dashboard/components/ProviderIcon.jsx

@@ -320,8 +320,14 @@ const PROVIDER_LOGO_MAP = {
   "KILO-CLI": "/brand-logos/kilo.svg",
   "KILO-CODE": "/brand-logos/kilo.svg",
   MIMO: "/brand-logos/mimo.svg",
+  // oh-my-pi / pi: multi-color brand marks (pi letterform + plugin connector).
+  OMP: "/brand-logos/omp.svg",
   OPENCLAW: "/brand-logos/openclaw.svg",
   OPENCODE: "/brand-logos/opencode.svg",
+  PI: "/brand-logos/pi.svg",
+  "PI-ANTHROPIC": "/brand-logos/pi.svg",
+  "PI-GITHUB-COPILOT": "/brand-logos/pi.svg",
+  "PI-COPILOT": "/brand-logos/pi.svg",
   QODER: "/brand-logos/qoder.svg",
 };
 

dashboard/src/ui/dashboard/components/ProviderIcon.test.jsx

@@ -40,4 +40,19 @@ describe("ProviderIcon", () => {
     expect(icon?.querySelector("path")).not.toBeNull();
     expect(icon?.querySelector("circle")).toBeNull();
   });
+
+  it("renders the multi-color oh-my-pi brand logo", () => {
+    const { container } = render(<ProviderIcon provider="omp" size={20} />);
+    const icon = container.querySelector('img[src="/brand-logos/omp.svg"]');
+    expect(icon).not.toBeNull();
+    expect(icon).toHaveAttribute("width", "20");
+    expect(icon).toHaveAttribute("height", "20");
+  });
+
+  it("renders the multi-color Pi brand logo", () => {
+    const { container } = render(<ProviderIcon provider="pi" size={18} />);
+    const icon = container.querySelector('img[src="/brand-logos/pi.svg"]');
+    expect(icon).not.toBeNull();
+    expect(icon).toHaveAttribute("width", "18");
+  });
 });

src/commands/init.js

@@ -56,6 +56,10 @@ const {
   removeGrokHook,
   GROK_HOOK_FILENAME
 } = require("../lib/grok-hook");
+const {
+  upsertOmpHook,
+  probeOmpHookState,
+} = require("../lib/omp-hook");
 const { resolveTrackerPaths } = require("../lib/tracker-paths");
 const {
   resolveOmpAgentDir,
@@ -361,6 +365,7 @@ async function runSetup({
     trackerDir,
     notifyPath,
     notifyOriginalPath,
+    dryRun: Boolean(opts.dryRun),
   });
 
   return {
@@ -551,7 +556,13 @@ function buildIntegrationTargets({ home, trackerDir, notifyPath }) {
   };
 }
 
-async function applyIntegrationSetup({ home, trackerDir, notifyPath, notifyOriginalPath }) {
+async function applyIntegrationSetup({
+  home,
+  trackerDir,
+  notifyPath,
+  notifyOriginalPath,
+  dryRun = false,
+}) {
   const context = buildIntegrationTargets({ home, trackerDir, notifyPath });
   context.notifyOriginalPath = notifyOriginalPath;
 
@@ -642,15 +653,44 @@ async function applyIntegrationSetup({ home, trackerDir, notifyPath, notifyOrigi
     }
   }
 
-  // oh-my-pi: passive reader — no hook installation needed.
-  // TokenTracker reads ~/.omp/agent/sessions/**/*.jsonl directly.
+  // oh-my-pi: passive session scan always works; also install an optional
+  // notify extension so turn_end triggers sync --source omp near-real-time.
   {
-    // resolveOmpAgentDir returns null on Windows when ~/.omp doesn't exist (the
-    // win32 path resolver only yields a dir it can see) — null means "not
-    // installed", so skip rather than path.join(null, …) and crash.
+    const fssyncLocal = require("node:fs");
     const ompAgentDir = resolveOmpAgentDir(process.env);
-    if (ompAgentDir && fssync.existsSync(path.join(ompAgentDir, "sessions"))) {
-      summary.push({ label: "oh-my-pi", status: "detected", detail: "Passive reader (no hook needed)" });
+    const ompSessions = ompAgentDir && fssyncLocal.existsSync(path.join(ompAgentDir, "sessions"));
+    if (ompAgentDir && (ompSessions || fssyncLocal.existsSync(ompAgentDir))) {
+      if (dryRun) {
+        const probe = await probeOmpHookState({ home, trackerDir, env: process.env });
+        summary.push({
+          label: "oh-my-pi",
+          status: "detected",
+          detail: probe.configured
+            ? "Notify extension already installed (passive scan still runs)"
+            : "Will install notify extension + keep passive scan",
+        });
+      } else {
+        const result = await upsertOmpHook({ home, trackerDir, env: process.env });
+        if (result.written) {
+          summary.push({
+            label: "oh-my-pi",
+            status: "installed",
+            detail: `Notify extension → ${result.extensionPath} (passive scan still runs)`,
+          });
+        } else if (result.skippedReason === "unmanaged-extension-present") {
+          summary.push({
+            label: "oh-my-pi",
+            status: "skipped",
+            detail: "Existing unmanaged tokentracker-notify extension left untouched",
+          });
+        } else {
+          summary.push({
+            label: "oh-my-pi",
+            status: "detected",
+            detail: "Passive reader (notify extension not written)",
+          });
+        }
+      }
     }
   }
 
@@ -1904,6 +1944,7 @@ module.exports = {
   installLocalTrackerApp,
   repairCodexNotifyIntegration,
   repairRuntimeIntegrations,
+  applyIntegrationSetup,
 };
 
 async function probeFile(p) {

src/commands/status.js

@@ -84,6 +84,7 @@ const wsl = require("../lib/wsl-probe");
 const { getWslMode, isInvalidWslMode, shouldProbeWsl, discoverWslHome } = wsl;
 const { resolveInstallPaths } = require("../lib/install-resolver");
 const { probeGrokHookState, resolveGrokHome } = require("../lib/grok-hook");
+const { probeOmpHookState } = require("../lib/omp-hook");
 
 function formatResolvedPaths(paths, filename) {
   const active = [];
@@ -359,10 +360,11 @@ async function cmdStatus(argv = []) {
     ? fssync.existsSync(path.join(workbuddyHome, "workbuddy.db"))
     : false;
 
-  // oh-my-pi — passive scan only (no hooks).
+  // oh-my-pi — passive scan + optional notify extension.
   const ompAgentDir = resolveOmpAgentDir(process.env);
   const ompInstalled = Boolean(ompAgentDir) && fssync.existsSync(path.join(ompAgentDir, "sessions"));
   const ompFiles = ompInstalled ? resolveOmpSessionFiles(process.env) : [];
+  const ompHookState = await probeOmpHookState({ home, trackerDir, env: process.env });
 
   // pi (@mariozechner/pi-coding-agent) — passive scan only (no hooks).
   // Skip when its agent dir collides with omp's; sync would dedupe anyway.
@@ -771,8 +773,13 @@ async function cmdStatus(argv = []) {
         workbuddy: workbuddyInstalled
           ? { installed: true, files: workbuddyFiles.length }
           : { installed: false },
-        omp: ompInstalled
-          ? { installed: true, files: ompFiles.length }
+        omp: ompInstalled || ompHookState.ompPresent
+          ? {
+              installed: true,
+              files: ompFiles.length,
+              notify_extension: Boolean(ompHookState.configured),
+              notify_extension_path: ompHookState.extensionPath || null,
+            }
           : { installed: false },
         pi: piInstalled
           ? { installed: true, files: piFiles.length }
@@ -895,8 +902,8 @@ async function cmdStatus(argv = []) {
       workbuddyInstalled
         ? `- WorkBuddy hooks: ${workbuddyHookConfigured ? "set" : "unset"} (${workbuddyFiles.length} session jsonl file${workbuddyFiles.length !== 1 ? "s" : ""} found, SQLite DB ${workbuddyDbExists ? "found" : "not found"})`
         : null,
-      ompInstalled
-        ? `- oh-my-pi: passive reader (${ompFiles.length} session jsonl file${ompFiles.length !== 1 ? "s" : ""} found)`
+      ompInstalled || ompHookState.ompPresent
+        ? `- oh-my-pi: passive reader (${ompFiles.length} session jsonl file${ompFiles.length !== 1 ? "s" : ""} found${ompHookState.configured ? ", notify extension: yes" : ", notify extension: no"})`
         : null,
       piInstalled
         ? `- pi: passive reader (${piFiles.length} session jsonl file${piFiles.length !== 1 ? "s" : ""} found)`

src/commands/uninstall.js

@@ -24,6 +24,7 @@ const { resolveOpencodeConfigDir, removeOpencodePlugin } = require("../lib/openc
 const { removeOpenclawHookConfig } = require("../lib/openclaw-hook");
 const { removeOpenclawSessionPluginConfig } = require("../lib/openclaw-session-plugin");
 const { removeGrokHook } = require("../lib/grok-hook");
+const { removeOmpHook } = require("../lib/omp-hook");
 const { resolveTrackerPaths } = require("../lib/tracker-paths");
 
 async function cmdUninstall(argv) {
@@ -104,6 +105,7 @@ async function cmdUninstall(argv) {
   });
   const openclawHookRemove = await removeOpenclawHookConfig({ home, trackerDir, env: process.env });
   const grokHookRemove = await removeGrokHook({ home, trackerDir, env: process.env });
+  const ompHookRemove = await removeOmpHook({ home, trackerDir, env: process.env });
 
   // Remove installed notify handler.
   await fs.unlink(notifyPath).catch(() => {});
@@ -195,6 +197,11 @@ async function cmdUninstall(argv) {
       grokHookRemove?.removed
         ? `- Grok Build hook removed: ${grokHookRemove.hookPath}`
         : "- Grok Build hook: no change",
+      ompHookRemove?.removed
+        ? `- oh-my-pi notify extension removed: ${ompHookRemove.extensionPath}`
+        : ompHookRemove?.skippedReason === "unmanaged"
+          ? "- oh-my-pi notify extension: skipped (unmanaged file)"
+          : "- oh-my-pi notify extension: no change",
       opts.purge ? `- Purged: ${path.join(home, ".tokentracker")}` : "- Purge: skipped (use --purge)",
       ...(machineIdSeedKept
         ? [`- Kept: ${machineIdSeedPath} (cloud device identity — a reinstall reuses the same device; delete it to fully reset)`]