omp-uwu 0.2.0 → 0.3.0

package/README.md

@@ -22,8 +22,9 @@ Restart omp. uwu mode is **on** by default in every session.
 - `/uwu on` / `/uwu off` sets it explicitly
 - `/uwu rewrite` uses omp's finalized-message rewrite hook when available
 - `/uwu prompt` asks the model to write in uwu style while it streams
+- `/uwu colors on` / `/uwu colors off` enables or disables the optional kawaii palette (off by default; preference persists)
 
-The TUI also shows a small, theme-accent-colored `(◕ᴗ◕✿) uwu` status badge. Other clients and plain output are not colorized.
+The palette colors chat markdown and the user-message bubble without putting ANSI codes into message text/history. It is applied as an in-memory TUI theme; components sharing those colors can change too. The host's in-memory theme setter pauses automatic theme detection until omp restarts. The TUI also shows a theme-accent `(◕ᴗ◕✿) uwu` status badge; other clients/plain output are not colorized.
 
 ## What gets uwufied, and what doesn't
 
@@ -43,6 +44,14 @@ Here's the same prompt and model, with the mode off and on:
 
 These are real, unedited replies from `anthropic/claude-opus-5-5`, rendered as images. Both code blocks are identical byte for byte. The transcripts are in [`demo/`](demo).
 
+### kawaii/uwu-bench
+
+On kawaii/uwu-bench, Opus 5.5 scores 0.1% on its own and 98.5% with omp-uwu enabled:
+
+![kawaii/uwu-bench: Opus 5.5 scores 0.1%, Opus 5.5 with omp-uwu scores 98.5%](https://raw.githubusercontent.com/NaC-L/omp-uwu/main/demo/uwu-bench.png)
+
+Only the two scores come from the benchmark run. The other numbers on the chart are computed from them, and the terminal and config panels are decoration.
+
 ## How it works
 
 On the first turn, omp-uwu uses the prompt style until the host demonstrates support for the awaited `assistant_message` hook; after that, the default **rewrite** style deterministically uwufies finalized assistant text before it is added to history and context. This first-turn check makes the experience work on both older and newer omp builds. Only text in existing text blocks is changed; code/tool blocks and their metadata stay untouched. Rewrites are markdown-aware and preserve fenced/inline code, URLs, paths, numbers, quoted text, identifiers, and safety-critical words.
@@ -51,13 +60,13 @@ The hook runs after streaming has finished. The TUI refreshes the current assist
 
 - **Deterministic rewrite.** No style instruction/token overhead once the hook is detected.
 - **Prompt style.** Model-dependent, with live uwu output while streaming.
-- **Session-local toggle.** Mode and enabled state reset for each new session.
+- **Persistent settings.** `/uwu off` is remembered across sessions; `/uwu on` re-enables uwu. The colors preference is remembered too. Settings are stored in `~/.omp/agent/omp-uwu.json`. UwU mode remains on by default until explicitly changed.
 
 ## Install options
 
 ```sh
 omp plugin install omp-uwu          # from npm (recommended)
-omp plugin install omp-uwu@0.1.0    # pin a version
+omp plugin install omp-uwu@0.3.0    # pin a version
 omp plugin uninstall omp-uwu        # remove
 ```
 
@@ -85,6 +94,12 @@ uv run --with pillow python demo/render.py
 
 Model output varies between runs, so re-run `record.sh` until you get a reply that reads well.
 
+The uwu-bench chart is drawn from [`demo/uwu-bench.html`](demo/uwu-bench.html). Edit the HTML, then take a 1536×960 screenshot with headless Chrome:
+
+```sh
+chrome --headless=new --hide-scrollbars --window-size=1536,960 --screenshot="$PWD/demo/uwu-bench.png" "file://$PWD/demo/uwu-bench.html"
+```
+
 ### Releasing
 
 CI (`.github/workflows/ci.yml`) runs `check` and the tests on pushes to `main` and on pull requests. Pushing a `v*` tag runs them again, checks that the tag matches `version` in `package.json`, and publishes to npm with provenance (`.github/workflows/publish.yml`).

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "omp-uwu",
-  "version": "0.2.0",
+  "version": "0.3.0",
   "description": "omp extension for playful uwu chat styling, markdown-safe rewrites, and kawaii kaomoji",
   "type": "module",
   "license": "MIT",

package/src/index.ts

@@ -1,4 +1,8 @@
-import type { ExtensionAPI } from "@oh-my-pi/pi-coding-agent";
+import { readFile, writeFile } from "node:fs/promises";
+import { join } from "node:path";
+import { KawaiiTheme } from "./kawaii.ts";
+
+import type { ExtensionAPI, Theme } from "@oh-my-pi/pi-coding-agent";
 import { uwufy } from "./uwufy.ts";
 
 export const UWU_PROMPT = `
@@ -25,16 +29,32 @@ type UiContext = {
   ui?: {
     notify(message: string, level: "info"): void;
     setStatus?(key: string, text: string | undefined): void;
-    theme?: { fg(color: string, text: string): string };
+    theme?: Theme;
   };
 };
 
 export default function uwuExtension(pi: ExtensionAPI) {
   let enabled = true;
+  let colorsEnabled = false;
+  let stateReady: Promise<void> | undefined;
   let style: "rewrite" | "prompt" = "rewrite";
   let hookSupport: boolean | undefined;
   let promptAddedThisTurn = false;
   let notifiedFallback = false;
+  const statePath = join(pi.pi.getAgentDir(), "omp-uwu.json");
+  const kawaiiTheme = new KawaiiTheme(pi.pi);
+  const loadState = () => {
+    stateReady ??= readFile(statePath, "utf8")
+      .then((raw) => {
+        const state: unknown = JSON.parse(raw);
+        if (typeof state !== "object" || state === null) return;
+        if ("enabled" in state && typeof state.enabled === "boolean") enabled = state.enabled;
+        if ("colors" in state && typeof state.colors === "boolean") colorsEnabled = state.colors;
+      })
+      .catch(() => {});
+    return stateReady;
+  };
+  const saveState = () => writeFile(statePath, `${JSON.stringify({ enabled, colors: colorsEnabled }, null, 2)}\n`, "utf8");
 
   const updateBadge = (ctx: UiContext) => {
     if (ctx.mode !== "tui" || ctx.agent?.kind === "sub" || !ctx.ui?.setStatus) return;
@@ -42,11 +62,37 @@ export default function uwuExtension(pi: ExtensionAPI) {
     ctx.ui.setStatus("omp-uwu", badge && ctx.ui.theme ? ctx.ui.theme.fg("accent", badge) : badge);
   };
 
+  const syncColors = (ctx: UiContext) => {
+    if (ctx.mode !== "tui" || ctx.agent?.kind === "sub" || !ctx.ui?.theme) return;
+    if (enabled && colorsEnabled) kawaiiTheme.enable(ctx.ui.theme);
+    else kawaiiTheme.disable(ctx.ui.theme);
+  };
+
   pi.registerCommand("uwu", {
-    description: "UwU chat style and mode (usage: /uwu [on|off|prompt|rewrite])",
+    description: "UwU chat style and mode (usage: /uwu [on|off|prompt|rewrite|colors [on|off]])",
     handler: async (args, rawCtx) => {
+      await loadState();
       const ctx = rawCtx as UiContext;
       const arg = String(args ?? "").trim().toLowerCase();
+      if (arg.startsWith("colors")) {
+        const option = arg.slice("colors".length).trim();
+        if (option === "on") colorsEnabled = true;
+        else if (option === "off") colorsEnabled = false;
+        else if (!option) colorsEnabled = !colorsEnabled;
+        else {
+          ctx.ui?.notify("Usage: /uwu [on|off|prompt|rewrite|colors [on|off]]", "info");
+          return;
+        }
+        try {
+          await saveState();
+        } catch {
+          ctx.ui?.notify("uwu setting could not be saved; it may reset next session", "info");
+        }
+        syncColors(ctx);
+        updateBadge(ctx);
+        ctx.ui?.notify(`kawaii chat colors ${colorsEnabled ? "on" : "off"}${colorsEnabled && !enabled ? " (active when uwu mode is on)" : ""}`, "info");
+        return;
+      }
       if (arg === "prompt" || arg === "rewrite") {
         style = arg;
         enabled = true;
@@ -54,17 +100,29 @@ export default function uwuExtension(pi: ExtensionAPI) {
       else if (arg === "off") enabled = false;
       else if (!arg) enabled = !enabled;
       else {
-        ctx.ui?.notify("Usage: /uwu [on|off|prompt|rewrite]", "info");
+        ctx.ui?.notify("Usage: /uwu [on|off|prompt|rewrite|colors [on|off]]", "info");
         return;
       }
+      try {
+        await saveState();
+      } catch {
+        ctx.ui?.notify("uwu setting could not be saved; it may reset next session", "info");
+      }
+      syncColors(ctx);
       updateBadge(ctx);
       ctx.ui?.notify(enabled ? `uwu mode ${style === "rewrite" ? "rewrite" : "prompt"}! (◕ᴗ◕✿)` : "uwu mode off", "info");
     },
   });
 
-  pi.on("session_start", (_event, rawCtx) => updateBadge(rawCtx as UiContext));
+  pi.on("session_start", async (_event, rawCtx) => {
+    await loadState();
+    const ctx = rawCtx as UiContext;
+    syncColors(ctx);
+    updateBadge(ctx);
+  });
 
   pi.on("before_agent_start", async (event, rawCtx) => {
+    await loadState();
     promptAddedThisTurn = false;
     const ctx = rawCtx as UiContext;
     if (ctx.agent?.kind === "sub") return undefined;
@@ -76,6 +134,7 @@ export default function uwuExtension(pi: ExtensionAPI) {
     return { systemPrompt: [...event.systemPrompt, UWU_PROMPT] };
   });
 
+
   // The hook is newer than the bundled 18.4.3 types, so register structurally.
   // ExtensionAPI.on stores event names as strings; old hosts simply never emit it.
   (pi.on as unknown as (name: string, handler: (event: RewriteEvent) => unknown) => void)(

package/src/kawaii.ts

@@ -0,0 +1,94 @@
+import type * as PiCodingAgent from "@oh-my-pi/pi-coding-agent";
+import type { Theme, ThemeBg, ThemeColor } from "@oh-my-pi/pi-coding-agent";
+
+export type ThemeRuntime = Pick<typeof PiCodingAgent, "Theme" | "setThemeInstance">;
+
+const FG_KEYS: ThemeColor[] = [
+  "accent", "border", "borderAccent", "borderMuted", "success", "error", "warning", "muted", "dim", "text",
+  "thinkingText", "userMessageText", "customMessageText", "customMessageLabel", "toolTitle", "toolOutput",
+  "mdHeading", "mdLink", "mdLinkUrl", "mdCode", "mdCodeBlock", "mdCodeBlockBorder", "mdQuote", "mdQuoteBorder",
+  "mdHr", "mdListBullet", "toolDiffAdded", "toolDiffRemoved", "toolDiffContext", "syntaxComment", "syntaxKeyword",
+  "syntaxFunction", "syntaxVariable", "syntaxString", "syntaxNumber", "syntaxType", "syntaxOperator", "syntaxPunctuation",
+  "thinkingOff", "thinkingMinimal", "thinkingLow", "thinkingMedium", "thinkingHigh", "thinkingXhigh", "thinkingMax",
+  "bashMode", "pythonMode", "statusLineSep", "statusLineModel", "statusLinePath", "statusLineGitClean", "statusLineGitDirty",
+  "statusLineContext", "statusLineSpend", "statusLineStaged", "statusLineDirty", "statusLineUntracked", "statusLineOutput",
+  "statusLineCost", "statusLineSubagents",
+];
+
+const BG_KEYS: ThemeBg[] = ["selectedBg", "userMessageBg", "customMessageBg", "toolPendingBg", "toolSuccessBg", "toolErrorBg", "statusLineBg"];
+const FG_DEFAULT = "\x1b[39m";
+const BG_DEFAULT = "\x1b[49m";
+
+const DARK_PALETTE: Partial<Record<ThemeColor | ThemeBg, string>> = {
+  accent: "#ff8fcf", border: "#b99aca", borderAccent: "#ff8fcf", thinkingText: "#cbb8ef",
+  mdHeading: "#ff9ed8", mdLink: "#b5a8ff", mdLinkUrl: "#a08fd8", mdCode: "#8fe8c3",
+  mdCodeBlockBorder: "#b99aca", mdQuote: "#ffc4e1", mdQuoteBorder: "#ff8fcf", mdHr: "#b99aca",
+  mdListBullet: "#ffb38a", userMessageBg: "#3a2433",
+};
+
+const LIGHT_PALETTE: Partial<Record<ThemeColor | ThemeBg, string>> = {
+  accent: "#c52f80", border: "#9c75b6", borderAccent: "#c52f80", thinkingText: "#735f9e",
+  mdHeading: "#b32470", mdLink: "#624ac6", mdLinkUrl: "#7663b1", mdCode: "#087b5c",
+  mdCodeBlockBorder: "#9c75b6", mdQuote: "#99416f", mdQuoteBorder: "#c52f80", mdHr: "#9c75b6",
+  mdListBullet: "#c35424", userMessageBg: "#fde8f3",
+};
+
+/** Build a pastel copy of a host theme, retaining its colors outside the palette. */
+export function buildKawaiiTheme(ThemeClass: ThemeRuntime["Theme"], base: Theme): Theme {
+  const palette = base.isLight ? LIGHT_PALETTE : DARK_PALETTE;
+  const fg: Partial<Record<ThemeColor, string>> = {};
+  const bg: Partial<Record<ThemeBg, string>> = {};
+
+  for (const key of FG_KEYS) {
+    try {
+      const ansi = base.getFgAnsi(key);
+      fg[key] = palette[key] ?? (ansi === FG_DEFAULT ? "" : base.getColorHex(key));
+    } catch {
+      // A host newer than the plugin's type dependency may expose extra colors.
+    }
+  }
+  for (const key of BG_KEYS) {
+    try {
+      const ansi = base.getBgAnsi(key);
+      bg[key] = palette[key] ?? (ansi === BG_DEFAULT ? "" : base.getBgHex(key));
+    } catch {
+      // Missing theme colors are left to the host defaults.
+    }
+  }
+
+  const preset = base.getSymbolPreset();
+  return new ThemeClass(
+    fg as Record<ThemeColor, string>,
+    bg as Record<ThemeBg, string>,
+    base.getColorMode(),
+    preset,
+    {},
+    { status: base.getSpinnerFrames("status"), activity: base.getSpinnerFrames("activity") },
+  );
+}
+
+/** A reversible, ephemeral kawaii theme; never edits the user's theme file. */
+export class KawaiiTheme {
+  #runtime: ThemeRuntime;
+  #base: Theme | undefined;
+  #overlay: Theme | undefined;
+
+  constructor(runtime: ThemeRuntime) {
+    this.#runtime = runtime;
+  }
+
+
+  enable(current: Theme): void {
+    if (this.#overlay && current === this.#overlay) return;
+    this.#base = current;
+    this.#overlay = buildKawaiiTheme(this.#runtime.Theme, current);
+    this.#runtime.setThemeInstance(this.#overlay);
+  }
+
+  disable(current: Theme | undefined): void {
+    if (!this.#overlay) return;
+    if (current === this.#overlay && this.#base) this.#runtime.setThemeInstance(this.#base);
+    this.#base = undefined;
+    this.#overlay = undefined;
+  }
+}