ai-notify 0.1.0 → 0.2.0

package/src/cli.mjs

@@ -3,15 +3,31 @@
 // One mute switch for all of them, across every terminal. No daemon.
 
 import { readFileSync } from 'node:fs';
+import { execSync } from 'node:child_process';
 import { providers, byId } from './providers/index.mjs';
 import { emit } from './notify.mjs';
 import { deriveLabel, cliInvocation, isEphemeralInstall } from './util.mjs';
 import { curatedVoices, resolveVoice, previewVoice } from './voices.mjs';
 import * as menubar from './menubar.mjs';
 import { translate } from './translate.mjs';
-import { isMuted, setMuted, toggleMuted, readConfig, writeConfig, paths, DEFAULT_CONFIG } from './state.mjs';
-
-const VERSION = '0.1.0';
+import { diagnose as highlightDiagnose, clearHighlight } from './highlight.mjs';
+import * as voicevox from './voicevox.mjs';
+import {
+  isMuted,
+  setMuted,
+  toggleMuted,
+  readConfig,
+  writeConfig,
+  paths,
+  DEFAULT_CONFIG,
+  readVolume,
+  setVolume,
+  readPanes,
+  readPaneSetting,
+  updatePaneSetting,
+} from './state.mjs';
+
+const VERSION = '0.2.0';
 
 const args = process.argv.slice(2);
 const cmd = args[0];
@@ -43,6 +59,57 @@ const readStdinJson = () => {
   }
 };
 
+// Pull the agent's last assistant text from a Claude Code transcript (JSONL),
+// trimmed to a short summary suitable for a notification / read-out.
+const lastAssistantText = (transcriptPath) => {
+  try {
+    const lines = readFileSync(transcriptPath, 'utf8').split('\n');
+    for (let i = lines.length - 1; i >= 0; i--) {
+      const line = lines[i].trim();
+      if (!line) continue;
+      let obj;
+      try {
+        obj = JSON.parse(line);
+      } catch {
+        continue;
+      }
+      if (obj.type !== 'assistant') continue;
+      const content = obj.message?.content;
+      if (!Array.isArray(content)) continue;
+      const text = content
+        .filter((c) => c?.type === 'text' && c.text)
+        .map((c) => c.text)
+        .join(' ')
+        .replace(/\s+/g, ' ')
+        .trim();
+      if (text) return text.length > 140 ? `${text.slice(0, 140)}…` : text;
+    }
+  } catch {
+    /* unreadable transcript — fall back to the template */
+  }
+  return '';
+};
+
+// Terminals (ttys) currently running a wired agent — so all open panes can be
+// assigned a voice from the menu bar without first firing a notification.
+const livePanes = () => {
+  try {
+    const out = execSync('ps -Ao tty=,command=', { encoding: 'utf8', maxBuffer: 1 << 22 });
+    const ttys = new Set();
+    for (const line of out.split('\n')) {
+      const m = line.match(/^(\S+)\s+(.*)$/);
+      if (!m) continue;
+      const [, tty, cmd] = m;
+      if (tty === '??' || tty === '?') continue;
+      if (/ai-notify|menubar/.test(cmd)) continue; // skip our own hook/agent
+      if (/\bclaude\b|\bcodex\b|\bgemini\b/i.test(cmd)) ttys.add(`/dev/${tty}`);
+    }
+    return [...ttys];
+  } catch {
+    return [];
+  }
+};
+
 const cmds = {
   init() {
     const dryRun = !!opt('dry-run');
@@ -130,6 +197,7 @@ const cmds = {
 
     const setVoice = (name) => {
       config.voice = name; // '' = OS default
+      config.tts = 'say'; // choosing a system voice switches the backend off VOICEVOX
       // Global voice wins only if no per-provider override; clear them so the
       // single switch actually takes effect everywhere.
       for (const k of Object.keys(config.providers || {})) {
@@ -179,6 +247,152 @@ const cmds = {
     log('  Reset:    ai-notify voice default');
   },
 
+  // Speak in VOICEVOX character voices (local engine, free, offline).
+  voicevox() {
+    const sub = positionals[0] || 'status';
+    const config = readConfig();
+    const url = config.voicevox?.url || voicevox.DEFAULT_URL;
+
+    if (sub === 'speakers') {
+      const list = voicevox.listSpeakers(url);
+      if (!list.length) return log(`No speakers (is VOICEVOX running at ${url}?).`);
+      list.forEach((s) => log(`  ${String(s.id).padStart(3)}  ${s.name}`));
+      log(`\nUse one:  ai-notify voicevox on <id>`);
+      return;
+    }
+    if (sub === 'on') {
+      if (!voicevox.isAvailable(url)) {
+        console.error(`VOICEVOX engine not reachable at ${url}. Start the VOICEVOX app first.`);
+        process.exit(1);
+      }
+      const speaker = Number(positionals[1] || config.voicevox?.speaker || 3);
+      config.tts = 'voicevox';
+      config.voicevox = { ...(config.voicevox || {}), url, speaker };
+      writeConfig(config);
+      log(`✓ VOICEVOX on (speaker ${speaker}). Testing…`);
+      voicevox.speak('ボイスボックスで読み上げます。', speaker, url);
+      return;
+    }
+    if (sub === 'off') {
+      config.tts = 'say';
+      writeConfig(config);
+      return log('VOICEVOX off — using the OS voice.');
+    }
+    if (sub === 'test') {
+      const speaker = Number(positionals[1] || config.voicevox?.speaker || 3);
+      const ok = voicevox.speak('これはテスト読み上げです。完了しました。', speaker, url);
+      return log(ok ? `spoke with speaker ${speaker}` : `⚠ failed (is VOICEVOX running at ${url}?)`);
+    }
+    // status
+    log(`VOICEVOX: ${config.tts === 'voicevox' ? `on (speaker ${config.voicevox?.speaker})` : 'off'}`);
+    log(`  engine ${url}: ${voicevox.isAvailable(url) ? '✓ reachable' : '✗ not running'}`);
+    if (config.tts !== 'voicevox') log('\nEnable:  ai-notify voicevox on    (list voices: ai-notify voicevox speakers)');
+  },
+
+  // Output volume 0.0–2.0 (1.0 = normal). Written to a state file the menu bar
+  // slider also drives; $AI_NOTIFY_VOLUME overrides per window.
+  volume() {
+    const arg = positionals[0];
+    if (arg === undefined) {
+      const config = readConfig();
+      const v = readVolume();
+      return log(`volume: ${v != null ? v : typeof config.volume === 'number' ? config.volume : 1}`);
+    }
+    const n = setVolume(arg);
+    log(`🔊 volume → ${n}`);
+  },
+
+  // Assign a voice to a specific pane (by tty), from the menu bar.
+  //   voice-pane <tty> voicevox <id> | say <name> | clear
+  'voice-pane'() {
+    const [tty, kind, ref] = positionals;
+    if (!tty) {
+      console.error('usage: voice-pane <tty> voicevox <id> | say <name> | clear');
+      process.exit(1);
+    }
+    if (!kind || kind === 'clear') {
+      updatePaneSetting(tty, { tts: null, speaker: null, voice: null });
+      return log(`pane ${tty}: voice reset to default`);
+    }
+    if (kind === 'voicevox') updatePaneSetting(tty, { tts: 'voicevox', speaker: Number(ref), voice: null });
+    else if (kind === 'say') updatePaneSetting(tty, { tts: 'say', voice: ref, speaker: null });
+    else {
+      console.error(`unknown kind: ${kind}`);
+      process.exit(1);
+    }
+    log(`pane ${tty}: ${kind} ${ref}`);
+  },
+
+  // Set a specific pane's output volume (0.0–2.0), or `clear` to follow global.
+  //   volume-pane <tty> <0.0-2.0|clear>
+  'volume-pane'() {
+    const [tty, arg] = positionals;
+    if (!tty || arg === undefined) {
+      console.error('usage: volume-pane <tty> <0.0-2.0|clear>');
+      process.exit(1);
+    }
+    if (arg === 'clear') {
+      updatePaneSetting(tty, { volume: null });
+      return log(`pane ${tty}: volume reset to global`);
+    }
+    const v = Math.min(2, Math.max(0, Number(arg)));
+    updatePaneSetting(tty, { volume: v });
+    log(`pane ${tty}: volume ${v}`);
+  },
+
+  // Machine-readable state for the menu bar agent: mute, volume, the selectable
+  // voices, and the recently-active panes (for per-pane assignment). Not human.
+  'menu-json'() {
+    const config = readConfig();
+    const url = config.voicevox?.url || voicevox.DEFAULT_URL;
+    const chars = voicevox.isAvailable(url) ? voicevox.listCharacters(url) : [];
+    const idName = new Map(chars.map((c) => [c.id, c.name]));
+    const voices = [];
+    for (const c of chars)
+      voices.push({
+        section: 'VOICEVOX',
+        label: c.name,
+        kind: 'voicevox',
+        ref: String(c.id),
+        currentGlobal: config.tts === 'voicevox' && Number(config.voicevox?.speaker) === c.id,
+      });
+    for (const n of curatedVoices(10))
+      voices.push({
+        section: 'System',
+        label: n,
+        kind: 'say',
+        ref: n,
+        currentGlobal: config.tts !== 'voicevox' && config.voice === n,
+      });
+    const labelFor = (pv) => {
+      if (!pv) return null;
+      return pv.tts === 'voicevox' ? idName.get(Number(pv.speaker)) || `VOICEVOX ${pv.speaker}` : pv.voice || 'system';
+    };
+    // Panes = live terminals currently running an agent (so they show up before
+    // they ever fire a notification) merged with previously-recorded ones.
+    const globalVol = readVolume() != null ? readVolume() : typeof config.volume === 'number' ? config.volume : 1;
+    const recorded = new Map(readPanes().map((p) => [p.tty, p.label]));
+    const ttys = new Set([...livePanes(), ...recorded.keys()]);
+    const panes = [...ttys].map((tty) => {
+      const s = readPaneSetting(tty);
+      return {
+        tty,
+        label: recorded.get(tty) || tty.replace('/dev/', ''),
+        current: labelFor(s.tts ? s : null),
+        volume: typeof s.volume === 'number' ? s.volume : globalVol,
+        volumeSet: typeof s.volume === 'number',
+      };
+    });
+    log(
+      JSON.stringify({
+        muted: isMuted(),
+        volume: readVolume() != null ? readVolume() : typeof config.volume === 'number' ? config.volume : 1,
+        voices,
+        panes,
+      })
+    );
+  },
+
   // Native menu bar bell (macOS). Self-contained — no Hammerspoon/SwiftBar.
   menubar() {
     const sub = positionals[0] || 'status';
@@ -190,7 +404,7 @@ const cmds = {
       const r = menubar.install();
       log(`  ✓ app:   ${r.app}`);
       log(`  ✓ agent: ${r.plist} (starts at login)`);
-      log('A 🔔 should now be in your menu bar. Left-click toggles, right-click for a menu.');
+      log('A 🔔 is now in your menu bar. Left-click for the menu (volume, voices), right-click to mute.');
       return;
     }
     if (sub === 'uninstall') {
@@ -244,6 +458,24 @@ const cmds = {
     if (!config.translateTo) log('Enable:  ai-notify translate on ja');
   },
 
+  // Diagnose / test the waiting-window highlight. Run it INSIDE the terminal
+  // tab you want to test (not piped) so it has a controlling tty.
+  highlight() {
+    const sub = positionals[0] || 'test';
+    if (sub === 'clear') {
+      clearHighlight();
+      return log('cleared.');
+    }
+    const color = positionals[1] || readConfig().highlightColor || 'yellow';
+    const info = highlightDiagnose(color);
+    log(JSON.stringify(info, null, 2));
+    log('\nThis tab should now be highlighted. Reset it with: ai-notify highlight clear');
+    if (info.appleTerminal && String(info.appleTerminal).startsWith('ERROR')) {
+      log('\n→ AppleScript was blocked. Grant permission in:');
+      log('  System Settings → Privacy & Security → Automation → (your terminal) → Terminal');
+    }
+  },
+
   hook() {
     const source = opt('source', 'default');
     let event = opt('event', 'done');
@@ -263,6 +495,12 @@ const cmds = {
       const data = readStdinJson();
       cwd = data.cwd || '';
       message = data.message || '';
+      // The Stop hook has no message, so "done" would only say "finished".
+      // Pull the agent's last reply from the transcript so the notification
+      // says WHAT was done.
+      if (!message && event === 'done' && data.transcript_path) {
+        message = lastAssistantText(data.transcript_path);
+      }
     }
 
     const label = deriveLabel(cwd);
@@ -284,7 +522,9 @@ Usage:
   ai-notify init [--dry-run] [--only claude,codex]   wire detected agents
   ai-notify uninstall [--only ...]                   remove wiring
   ai-notify toggle | on | off | status               control the mute switch
+  ai-notify volume [0.0-2.0]                          get/set output volume
   ai-notify voice [number|name|preview|default]      pick the spoken voice
+  ai-notify voicevox [on <id>|off|speakers|test]     speak in VOICEVOX character voices
   ai-notify menubar [install|uninstall|status]       native menu bar bell (macOS)
   ai-notify translate [on <lang>|off|test]           speak agent text in your language
   ai-notify doctor                                    check deps & wiring

package/src/highlight.mjs

@@ -0,0 +1,261 @@
+// Visually highlight the terminal window/pane that is waiting for input, so it
+// stands out among many open terminals. Best-effort and terminal-specific:
+//
+//   - tmux              -> color the pane background (select-pane -P)
+//   - Apple Terminal    -> set the tab's background color via AppleScript,
+//                          matched by tty, restoring the original on done
+//   - others (iTerm2,…) -> OSC 11 default-background + a tab-title marker
+//
+// Everything is wrapped so a failure never affects the notification. The tab
+// title marker is the most portable signal and is always emitted.
+
+import { execFileSync } from 'node:child_process';
+import { writeFileSync, readFileSync, existsSync, rmSync, mkdirSync } from 'node:fs';
+import { join } from 'node:path';
+import { stateDir } from './state.mjs';
+
+const isMac = process.platform === 'darwin';
+const BEL = '\x07';
+
+// Controlling terminal of this process (works even when stdio is piped).
+const ttyName = () => {
+  try {
+    const t = execFileSync('ps', ['-o', 'tty=', '-p', String(process.pid)], { encoding: 'utf8' }).trim();
+    if (!t || t === '??' || t === '?') return null;
+    return t.startsWith('/dev/') ? t : `/dev/${t}`;
+  } catch {
+    return null;
+  }
+};
+
+const writeTty = (seq) => {
+  const tty = ttyName() || '/dev/tty';
+  try {
+    writeFileSync(tty, seq);
+  } catch {
+    /* no controlling terminal — ignore */
+  }
+};
+
+// Default-background (OSC 11) + icon/tab title (OSC 1/2). Reset uses OSC 111.
+const oscSet = (hex, title) => `\x1b]11;${hex}${BEL}\x1b]1;${title}${BEL}\x1b]2;${title}${BEL}`;
+const oscReset = `\x1b]111${BEL}\x1b]1;${BEL}\x1b]2;${BEL}`;
+
+// SGR fallback that works even where OSC is ignored (e.g. JetBrains JediTerm):
+// print a bold black-on-yellow bar straight into the pane, plus a BEL so the
+// IDE flags the tab as having activity. Standard ANSI — renders everywhere.
+const sgrBg = { yellow: 103, orange: '48;5;208', red: 101, green: 102 };
+const sgrBar = (label, color) => {
+  const bg = sgrBg[color] || sgrBg.yellow;
+  return `\r\n\x1b[1;30;${bg}m  ⏳ ${label || 'input'}  \x1b[0m${BEL}\r\n`;
+};
+
+const colorHex = (c) => {
+  const map = { yellow: '#FFD400', orange: '#FF9500', red: '#FF3B30', green: '#34C759' };
+  if (!c) return map.yellow;
+  return map[c] || (c.startsWith('#') ? c : map.yellow);
+};
+
+// --- tmux ---
+const tmuxPane = () => (process.env.TMUX && process.env.TMUX_PANE ? process.env.TMUX_PANE : null);
+const tmuxSet = (c) => {
+  const pane = tmuxPane();
+  if (!pane) return;
+  const color = c === 'yellow' || !c ? 'colour220' : c;
+  try {
+    execFileSync('tmux', ['select-pane', '-t', pane, '-P', `bg=${color}`]);
+  } catch {
+    /* ignore */
+  }
+};
+const tmuxReset = () => {
+  const pane = tmuxPane();
+  if (!pane) return;
+  try {
+    execFileSync('tmux', ['select-pane', '-t', pane, '-P', 'bg=default']);
+  } catch {
+    /* ignore */
+  }
+};
+
+// --- Apple Terminal (set the tab bg by tty; store original to restore) ---
+// Some setups don't export TERM_PROGRAM, so detect Apple Terminal robustly:
+// explicit signals first, then — when the terminal is unknown — attempt anyway.
+// The AppleScript matches by tty, so it's a harmless no-op if this isn't really
+// a Terminal.app tab. Known non-Terminal programs (iTerm.app, vscode, …) opt out
+// to avoid needless automation prompts.
+const isAppleTerminal = () => {
+  if (!isMac) return false;
+  const tp = process.env.TERM_PROGRAM;
+  const bundle = process.env.__CFBundleIdentifier || '';
+  if (tp === 'Apple_Terminal' || bundle === 'com.apple.Terminal') return true;
+  if (tp || bundle) return false; // a known other terminal/IDE (iTerm, WebStorm…)
+  return true; // truly unknown -> attempt; the tty match guards it
+};
+
+// JetBrains IDE terminals (WebStorm, IntelliJ, …) — JediTerm/Gen2.
+const isJetBrains = () =>
+  process.env.TERMINAL_EMULATOR === 'JetBrains-JediTerm' ||
+  /jetbrains/i.test(process.env.__CFBundleIdentifier || '');
+const savePath = (tty) => join(stateDir(), `hl-${tty.replace(/[^\w]+/g, '_')}`);
+
+const appleSet = (rgb16) => {
+  const tty = ttyName();
+  if (!tty) return;
+  const script = `tell application "Terminal"
+  repeat with w in windows
+    repeat with t in tabs of w
+      try
+        if (tty of t) is "${tty}" then
+          set c to background color of t
+          set background color of t to {${rgb16}}
+          return ((item 1 of c) & "," & (item 2 of c) & "," & (item 3 of c)) as string
+        end if
+      end try
+    end repeat
+  end repeat
+  return ""
+end tell`;
+  try {
+    const orig = execFileSync('osascript', ['-e', script], { encoding: 'utf8', timeout: 3000 }).trim();
+    if (orig) {
+      mkdirSync(stateDir(), { recursive: true });
+      writeFileSync(savePath(tty), orig);
+    }
+  } catch {
+    /* automation permission not granted / ignore */
+  }
+};
+
+const appleReset = () => {
+  const tty = ttyName();
+  if (!tty) return;
+  const p = savePath(tty);
+  // Only restore if WE highlighted this tab (a saved original exists). Without
+  // this guard a normal 'done' with no prior 'waiting' would blacken the tab.
+  if (!existsSync(p)) return;
+  let rgb16 = '0, 0, 0';
+  try {
+    rgb16 = readFileSync(p, 'utf8').trim() || rgb16;
+  } catch {
+    /* ignore */
+  }
+  const script = `tell application "Terminal"
+  repeat with w in windows
+    repeat with t in tabs of w
+      try
+        if (tty of t) is "${tty}" then set background color of t to {${rgb16}}
+      end try
+    end repeat
+  end repeat
+end tell`;
+  try {
+    execFileSync('osascript', ['-e', script], { timeout: 3000, stdio: 'ignore' });
+  } catch {
+    /* ignore */
+  }
+  try {
+    if (existsSync(p)) rmSync(p);
+  } catch {
+    /* ignore */
+  }
+};
+
+// 16-bit RGB for AppleScript yellow-ish; reuse hex→approx for custom.
+const rgb16From = (c) => {
+  if (c === 'orange') return '65535, 38000, 0';
+  if (c === 'red') return '65535, 15000, 12000';
+  if (c === 'green') return '13000, 51000, 22000';
+  return '65535, 54000, 0'; // yellow
+};
+
+// A per-tty marker so `clear` only touches the terminal when WE highlighted it
+// (a 'done' with no prior 'waiting' must leave the window untouched).
+const markPath = (tty) => join(stateDir(), `hl-on-${tty.replace(/[^\w]+/g, '_')}`);
+
+export const highlightWaiting = (label, color = 'yellow') => {
+  writeTty(oscSet(colorHex(color), `⏳ ${label || 'input'}`));
+  if (isJetBrains()) writeTty(sgrBar(label, color)); // OSC ignored here; SGR works
+  if (isAppleTerminal()) appleSet(rgb16From(color));
+  tmuxSet(color);
+  const tty = ttyName();
+  if (tty) {
+    try {
+      mkdirSync(stateDir(), { recursive: true });
+      writeFileSync(markPath(tty), '');
+    } catch {
+      /* ignore */
+    }
+  }
+};
+
+// Foreground diagnostic: run the highlight and surface what happened (tty,
+// terminal, AppleScript error/permission) instead of swallowing it.
+export const diagnose = (color = 'yellow') => {
+  const info = {
+    platform: process.platform,
+    TERM_PROGRAM: process.env.TERM_PROGRAM || null,
+    __CFBundleIdentifier: process.env.__CFBundleIdentifier || null,
+    isAppleTerminal: isAppleTerminal(),
+    TMUX: process.env.TMUX ? process.env.TMUX_PANE || true : false,
+    tty: ttyName(),
+  };
+  try {
+    writeTty(oscSet(colorHex(color), '⏳ test'));
+    info.osc = `wrote to ${ttyName() || '/dev/tty'}`;
+  } catch (e) {
+    info.osc = `error: ${e.message}`;
+  }
+  info.jetBrains = isJetBrains();
+  if (isJetBrains()) {
+    writeTty(sgrBar('test', color)); // should print a yellow bar right here
+    info.sgrBar = 'printed (look for the yellow bar above)';
+  }
+  if (isAppleTerminal()) {
+    const tty = ttyName();
+    if (!tty) {
+      info.appleTerminal = 'no controlling tty';
+    } else {
+      const script = `tell application "Terminal"
+  repeat with w in windows
+    repeat with t in tabs of w
+      try
+        if (tty of t) is "${tty}" then
+          set c to background color of t
+          set background color of t to {${rgb16From(color)}}
+          return "matched tty, original=" & (c as string)
+        end if
+      end try
+    end repeat
+  end repeat
+  return "no tab matched tty ${tty}"
+end tell`;
+      try {
+        info.appleTerminal = execFileSync('osascript', ['-e', script], {
+          encoding: 'utf8',
+          timeout: 5000,
+        }).trim();
+      } catch (e) {
+        info.appleTerminal = `ERROR: ${(e.stderr || e.message || '').toString().trim()}`;
+      }
+    }
+  } else {
+    info.appleTerminal = 'skipped (detected a non-Terminal program)';
+  }
+  return info;
+};
+
+export const clearHighlight = () => {
+  const tty = ttyName();
+  if (tty && !existsSync(markPath(tty))) return; // we never highlighted this one
+  writeTty(oscReset);
+  if (isAppleTerminal()) appleReset();
+  tmuxReset();
+  if (tty) {
+    try {
+      if (existsSync(markPath(tty))) rmSync(markPath(tty));
+    } catch {
+      /* ignore */
+    }
+  }
+};

package/src/menubar.mjs

@@ -10,8 +10,10 @@
 import { homedir, platform } from 'node:os';
 import { join, dirname } from 'node:path';
 import { fileURLToPath } from 'node:url';
-import { existsSync, mkdirSync, writeFileSync, rmSync } from 'node:fs';
+import { existsSync, mkdirSync, writeFileSync, rmSync, chmodSync } from 'node:fs';
 import { execFileSync, spawnSync } from 'node:child_process';
+import { cliInvocation } from './util.mjs';
+import { stateDir } from './state.mjs';
 
 export const LABEL = 'com.ai-notify.menubar';
 
@@ -81,9 +83,21 @@ const unload = () => {
   return r;
 };
 
+// A tiny launcher the menu bar app shells out to for data/actions, so it works
+// regardless of PATH (embeds the resolved node + cli path).
+const writeCliWrapper = () => {
+  const { node, cliPath } = cliInvocation();
+  const p = join(stateDir(), 'cli');
+  mkdirSync(stateDir(), { recursive: true });
+  writeFileSync(p, `#!/bin/sh\nexec "${node}" "${cliPath}" "$@"\n`);
+  chmodSync(p, 0o755);
+  return p;
+};
+
 export const install = () => {
   if (!isMac()) throw new Error('the menu bar agent is macOS-only');
   if (!isBuilt()) build();
+  writeCliWrapper();
   writePlist();
   load();
   return { app: appPath(), plist: plistPath() };