pi-wakeup 0.2.0 → 0.2.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-wakeup",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "schedule_wakeup / cancel_wakeup for the pi coding agent: session-local self-pacing, the pi analog of Claude Code's ScheduleWakeup dynamic mode.",
   "type": "module",
   "license": "MIT",

package/src/index.test.ts

@@ -40,6 +40,7 @@ function harness(
       return idle;
     },
     ui: {
+      theme: { fg: (color: string, text: string) => `<${color}>${text}` },
       notify: (text: string, level: string) => notifies.push({ text, level }),
       setWidget: (key: string, lines: string[] | undefined) => {
         if (opts.widgetThrows) throw new Error("stale widget ctx");
@@ -252,6 +253,7 @@ test("widget mode: scheduling shows a line above the editor and arms a refresh",
   assert.equal(last.lines!.length, 1);
   assert.match(last.lines![0], /2m/); // 120s from now()=1000 -> 2m remaining
   assert.match(last.lines![0], /CI/);
+  assert.match(last.lines![0], /^<warning>⏰ /); // styled like the tmux bar
   assert.equal(h.intervals.length, 1);
   assert.equal(h.intervals[0].cancelled, false);
   // The tmux runner is never touched in widget mode.

package/src/index.ts

@@ -194,6 +194,19 @@ export function createWakeup(pi: any, deps: Deps = {}): void {
     }
   }
 
+  // The widget line matches the tmux bar: a clock and the theme's warning
+  // (yellow) colour, so it stands out above the editor. The theme read is
+  // guarded like every other ctx.ui access; without one the line is plain.
+  function widgetLine(text: string): string {
+    const line = `⏰ ${text}`;
+    try {
+      const theme = ctx?.ui?.theme;
+      return theme && typeof theme.fg === "function" ? theme.fg("warning", line) : line;
+    } catch {
+      return line;
+    }
+  }
+
   // Render the current pending list to the chosen target, or clear it when
   // nothing is pending. Runs inside timer/interval callbacks, so every tmux and
   // ctx.ui call is guarded — a throw here must never escape (best-effort).
@@ -214,7 +227,7 @@ export function createWakeup(pi: any, deps: Deps = {}): void {
       // Widget mode. Guard ctx.ui: a ctx captured at session_start goes stale
       // after /reload or a session switch, and setWidget would then throw.
       if (ctx?.ui?.setWidget) {
-        ctx.ui.setWidget("wakeup", text === undefined ? undefined : [text]);
+        ctx.ui.setWidget("wakeup", text === undefined ? undefined : [widgetLine(text)]);
       }
     } catch (error) {
       log(`display refresh failed: ${String(error)}`);