ai-battery 0.1.4 → 0.1.6

package/README.md

@@ -50,8 +50,8 @@ Codex 86% │ 5h 18:09 │ 7d 82%  ┃  Claude 4% │ 5h 18:10 │ 7d 71%
 | `ai-battery` | 지원 | 지원 | 지원 | 지원 | Node.js 18 이상이 필요합니다. |
 | `ai-battery --watch` | 지원 | 지원 | 지원 | 지원 | 터미널 안에서 주기적으로 갱신합니다. |
 | Claude statusLine | 지원 | 지원 | 지원 | 지원 | Claude Code `statusLine`에 `node <script>` 명령을 저장합니다. |
-| Codex terminal row | 미지원 | 지원 | 지원 | 지원 | `ai-battery-run`이 POSIX PTY와 `python3`를 사용합니다. |
-| `ai-battery setup codex` | 미지원 | 지원 | 지원 | 지원 | `codex` wrapper가 POSIX shell/PTY 기반입니다. |
+| Codex terminal row | 지원 | 지원 | 지원 | 지원 | Windows는 `node-pty`/ConPTY가 있으면 reserved row, 없으면 overlay fallback을 사용합니다. WSL/Linux/macOS는 POSIX PTY와 `python3`를 사용합니다. |
+| `ai-battery setup codex` | 지원 | 지원 | 지원 | 지원 | Windows는 `codex.cmd` wrapper, WSL/Linux/macOS는 POSIX shell wrapper를 설치합니다. |
 | `ai-battery hud` | 지원 | 지원 | 미지원 | 지원 | Windows/WSL은 PowerShell/WinForms HUD, macOS는 menu bar status item입니다. |
 
 실행 중 감지(흰색/회색 표시)는 Linux/WSL은 `/proc`, macOS는 `ps`, Windows는 PowerShell 프로세스 목록을 사용합니다.
@@ -154,7 +154,7 @@ npm uninstall -g ai-battery
 
 ## Setup
 
-`setup`은 한 번만 실행합니다. Claude Code에는 statusLine hook을 설치하고, Codex에는 `codex` wrapper를 설치해서 이후에는 추가 명령 없이 원래처럼 실행하게 합니다.
+`setup`은 한 번만 실행합니다. Claude Code에는 statusLine hook을 설치하고, Codex에는 platform wrapper를 설치해서 이후에는 추가 명령 없이 원래처럼 실행하게 합니다.
 
 ```bash
 ai-battery setup
@@ -169,6 +169,8 @@ ai-battery setup codex
 
 Codex wrapper는 기존 `codex` 명령을 직접 덮어쓰지 않습니다. `~/.local/bin`이 이미 PATH에서 원본 `codex`보다 앞에 있고 `~/.local/bin/codex`가 비어 있거나 AI Battery 관리 파일이면 그 위치에 wrapper를 둬서 바로 잡히게 합니다. 그렇지 않으면 `~/.local/share/ai-battery/bin/codex`에 관리형 wrapper를 만들고, 필요한 경우 셸 설정에 이 디렉터리를 PATH 앞쪽으로 추가합니다. `~/.local/bin/codex` 같은 공용 위치에 이미 다른 파일이 있으면 덮어쓰지 않습니다. 새 터미널부터 `codex`가 자동으로 AI Battery 하단 행과 함께 실행됩니다. 같은 터미널에서 이미 `codex`를 실행한 적이 있으면 셸 캐시 때문에 `hash -r`이 한 번 필요할 수 있고, PATH 추가가 필요한 경우에는 `setup` 출력에 표시되는 `source ...` 명령을 실행하세요.
 
+Windows native `cmd`/PowerShell에서는 `codex.cmd` wrapper가 Windows runner를 실행합니다. `node-pty`가 설치되어 있으면 ConPTY로 Codex 화면을 한 줄 짧게 띄워 하단 row를 예약하고, `node-pty`가 없거나 로드되지 않으면 같은 콘솔의 하단에 overlay row를 다시 그리는 fallback을 사용합니다. Claude statusLine은 일반 `cmd`/PowerShell 프롬프트가 아니라 Claude Code 안에서만 표시됩니다.
+
 Codex 하단 행이 보이지 않으면 진단을 실행합니다.
 
 ```bash
@@ -237,7 +239,7 @@ Claude가 한 번 이상 statusLine payload를 전달해야 Claude의 사용량
 
 ## Desktop HUD
 
-일반 터미널 위에 외부 프로세스가 안전하게 status line을 덧그리는 방식은 안정적이지 않습니다. 그래서 Windows에서는 floating overlay를, macOS에서는 상단 menu bar status item을 제공합니다. Windows native에서는 WSL 없이 PowerShell/WinForms로 바로 실행되고, WSL에서는 `powershell.exe`를 통해 같은 HUD를 띄웁니다. macOS에서는 메뉴바 두께에 맞춰 `Cx 75% | Cl 4%`처럼 한 줄 요약만 표시하고, 클릭하면 자세한 상태를 볼 수 있습니다.
+일반 터미널 위에 외부 프로세스가 안전하게 status line을 덧그리는 방식은 안정적이지 않습니다. 그래서 Windows에서는 floating overlay를, macOS에서는 상단 menu bar status item을 제공합니다. Windows native에서는 WSL 없이 PowerShell/WinForms로 바로 실행되고, WSL에서는 `powershell.exe`를 통해 같은 HUD를 띄웁니다. macOS에서는 투명 배경의 작은 SVG 이미지로 Codex와 Claude 로고, 짧은 미터, 퍼센트만 표시하고, 클릭하면 자세한 상태를 볼 수 있습니다.
 
 ```bash
 ai-battery hud
@@ -303,7 +305,7 @@ Codex는 최근 세션 로그에서 `rate_limits` 이벤트를 찾습니다. Cla
 
 ## Source Environment
 
-기본 CLI는 Node.js 18 이상이 있으면 Windows native, WSL, Linux, macOS에서 실행됩니다. `ai-battery-run`과 Codex terminal row는 Python 3와 POSIX PTY가 필요해서 WSL/Linux/macOS에서 지원됩니다. HUD는 Windows/WSL에서는 PowerShell/WinForms, macOS에서는 내장 `osascript`와 AppleScriptObjC를 사용합니다.
+기본 CLI는 Node.js 18 이상이 있으면 Windows native, WSL, Linux, macOS에서 실행됩니다. Windows native의 Codex terminal row는 Node runner를 사용하며, optional `node-pty`가 있으면 ConPTY reserved row를 사용하고 없으면 overlay fallback을 사용합니다. WSL/Linux/macOS의 `ai-battery-run`은 Python 3와 POSIX PTY를 사용합니다. HUD는 Windows/WSL에서는 PowerShell/WinForms, macOS에서는 내장 `osascript`와 AppleScriptObjC를 사용합니다.
 
 Codex 데이터는 기본적으로 `~/.codex/sessions`를 읽습니다. 다른 위치를 쓰고 있다면 `CODEX_HOME`을 설정하세요.
 

package/bin/ai-battery-hud

@@ -1,89 +1,21 @@
-#!/usr/bin/env bash
-set -euo pipefail
-
-SCRIPT_PATH="$(readlink -f "${BASH_SOURCE[0]}")"
-SCRIPT_DIR="$(cd "$(dirname "$SCRIPT_PATH")" && pwd)"
-HUD_PS1="$SCRIPT_DIR/ai-battery-hud.ps1"
-
-if ! command -v powershell.exe >/dev/null 2>&1; then
-  echo "ai-battery-hud: powershell.exe is required for the Windows always-on-top HUD." >&2
-  exit 1
-fi
-
-FOREGROUND=0
-ONCE=0
-STOP=0
-FILTERED_ARGS=()
-for arg in "$@"; do
-  case "$arg" in
-    -Foreground|--foreground)
-      FOREGROUND=1
-      ;;
-    -Movable|--movable)
-      FILTERED_ARGS+=("-Movable")
-      ;;
-    -Once|--once)
-      ONCE=1
-      FILTERED_ARGS+=("$arg")
-      ;;
-    -Stop|--stop)
-      STOP=1
-      FILTERED_ARGS+=("-StopExisting")
-      ;;
-    *)
-      FILTERED_ARGS+=("$arg")
-      ;;
+#!/usr/bin/env sh
+set -eu
+
+script=$0
+while [ -h "$script" ]; do
+  dir=$(CDPATH= cd -- "$(dirname -- "$script")" && pwd)
+  link=$(readlink "$script")
+  case "$link" in
+    /*) script=$link ;;
+    *) script=$dir/$link ;;
   esac
 done
 
-HUD_PS1_WIN="$(wslpath -w "$HUD_PS1")"
-
-if [[ -n "${CLAUDEX_BATTERY_COMMAND:-}" ]]; then
-  AI_BATTERY_COMMAND="$CLAUDEX_BATTERY_COMMAND"
-fi
+script_dir=$(CDPATH= cd -- "$(dirname -- "$script")" && pwd)
 
-if [[ -z "${AI_BATTERY_COMMAND:-}" ]]; then
-  NODE_BIN="$(command -v node || true)"
-  if [[ -z "$NODE_BIN" ]]; then
-    echo "ai-battery-hud: node was not found in WSL PATH." >&2
-    exit 1
-  fi
-  printf -v WSL_HOME_QUOTED "%q" "$HOME"
-  printf -v NODE_BIN_QUOTED "%q" "$NODE_BIN"
-  printf -v AI_BATTERY_JS_QUOTED "%q" "$SCRIPT_DIR/ai-battery.js"
-  ENV_PREFIX=()
-  for name in AI_BATTERY_STATE_DIR CLAUDEX_BATTERY_STATE_DIR AI_BATTERY_COLUMNS CLAUDEX_BATTERY_COLUMNS AI_BATTERY_COLUMN_GUARD CLAUDEX_BATTERY_COLUMN_GUARD CODEX_HOME; do
-    if [[ -n "${!name:-}" ]]; then
-      printf -v value_quoted "%q" "${!name}"
-      ENV_PREFIX+=("$name=$value_quoted")
-    fi
-  done
-  AI_BATTERY_COMMAND="${ENV_PREFIX[*]} HOME=$WSL_HOME_QUOTED $NODE_BIN_QUOTED $AI_BATTERY_JS_QUOTED --json"
-fi
-
-PS_ARGS=(-NoProfile -ExecutionPolicy Bypass -File "$HUD_PS1_WIN" -BatteryCommand "$AI_BATTERY_COMMAND" "${FILTERED_ARGS[@]}")
-
-if [[ "$FOREGROUND" == "1" || "$ONCE" == "1" || "$STOP" == "1" ]]; then
-  exec powershell.exe "${PS_ARGS[@]}"
+if ! command -v node >/dev/null 2>&1; then
+  echo "ai-battery-hud: node was not found on PATH." >&2
+  exit 1
 fi
 
-ps_quote() {
-  local value="${1//\'/\'\'}"
-  printf "'%s'" "$value"
-}
-
-win_arg_quote() {
-  local value="${1//\"/\\\"}"
-  printf '"%s"' "$value"
-}
-
-PS_CMDLINE=""
-for arg in "${PS_ARGS[@]}"; do
-  if [[ -n "$PS_CMDLINE" ]]; then
-    PS_CMDLINE+=" "
-  fi
-  PS_CMDLINE+="$(win_arg_quote "$arg")"
-done
-
-powershell.exe -NoProfile -ExecutionPolicy Bypass -Command "Start-Process -WindowStyle Hidden -FilePath 'powershell.exe' -ArgumentList $(ps_quote "$PS_CMDLINE")" >/dev/null 2>&1
-echo "AI Battery HUD started or already running. Drag it to place it; right-click and choose Exit to close."
+exec node "$script_dir/ai-battery-hud.js" "$@"

package/bin/ai-battery-hud.js

@@ -78,6 +78,15 @@ function sleepSync(ms) {
   Atomics.wait(new Int32Array(new SharedArrayBuffer(4)), 0, 0, ms);
 }
 
+function waitForFile(filePath, timeoutMs = 3500) {
+  const start = Date.now();
+  while (Date.now() - start < timeoutMs) {
+    if (fs.existsSync(filePath)) return true;
+    sleepSync(100);
+  }
+  return false;
+}
+
 function snapshotHasWeekly(jsonText) {
   try {
     const snapshot = JSON.parse(jsonText);
@@ -188,7 +197,8 @@ function macCommands(options) {
   const providerArgs = macProviderArgs(options.provider);
   const envPrefix = relevantEnvPrefix();
   return {
-    title: `${envPrefix} ${node} ${batteryJs} --menu-bar${providerArgs} 2>/dev/null`,
+    title: `${envPrefix} ${node} ${batteryJs} --menu-bar-image${providerArgs} 2>/dev/null`,
+    detailImage: `${envPrefix} ${node} ${batteryJs} --menu-detail-image${providerArgs} 2>/dev/null`,
     detail: `${envPrefix} ${node} ${batteryJs} --no-color${providerArgs} 2>/dev/null`
   };
 }
@@ -313,14 +323,16 @@ function runMacHud(cliArgs) {
     process.exit(result.status ?? 0);
   }
 
-  if (macHudPids().length) {
-    console.log("AI Battery menu bar is already running.");
-    process.exit(0);
+  const existingPids = macHudPids();
+  if (existingPids.length) {
+    spawnSync("kill", existingPids.map(String), { stdio: "ignore" });
+    sleepSync(400);
   }
 
   const osascriptArgs = [
     macStatusPath,
     commands.title,
+    commands.detailImage,
     commands.detail,
     String(options.interval)
   ];
@@ -335,7 +347,7 @@ function runMacHud(cliArgs) {
     stdio: "ignore"
   });
   child.unref();
-  console.log("AI Battery menu bar started. Click the menu bar item for details; use \"ai-battery hud stop\" to close it.");
+  console.log(`${existingPids.length ? "AI Battery menu bar restarted" : "AI Battery menu bar started"}. Click the menu bar item for details; use "ai-battery hud stop" to close it.`);
   process.exit(0);
 }
 
@@ -511,6 +523,9 @@ if (subcommand === "autostart") {
 }
 
 const initialJson = stop ? null : prefetchInitialJson(batteryCommand, useWsl);
+const readyPath = (!useWsl && process.platform === "win32" && !foreground && !once && !stop)
+  ? path.join(os.tmpdir(), `ai-battery-hud-ready-${process.pid}-${Date.now()}.json`)
+  : null;
 
 const psArgs = [
   "-NoProfile",
@@ -523,6 +538,10 @@ const psArgs = [
   ...filteredArgs
 ];
 
+if (readyPath) {
+  psArgs.push("-ReadyPath", readyPath);
+}
+
 if (initialJson) {
   psArgs.push("-InitialJsonBase64", Buffer.from(initialJson, "utf8").toString("base64"));
 }
@@ -560,4 +579,9 @@ if (useWsl) {
   child.unref();
 }
 
-console.log("AI Battery HUD started or already running. Drag it to place it; right-click and choose Exit to close.");
+if (readyPath && !waitForFile(readyPath)) {
+  console.log("AI Battery HUD start requested, but no visible window was confirmed. Run: ai-battery hud --foreground");
+} else {
+  console.log("AI Battery HUD started. Drag it to place it; right-click and choose Exit to close.");
+}
+if (readyPath) fs.rmSync(readyPath, { force: true });

package/bin/ai-battery-hud.ps1

@@ -12,7 +12,8 @@ param(
   [switch]$ClickThrough,
   [switch]$StopExisting,
   [switch]$UseWsl,
-  [switch]$Once
+  [switch]$Once,
+  [string]$ReadyPath = ""
 )
 
 $ErrorActionPreference = "Stop"
@@ -112,6 +113,22 @@ $script:hudAnchorY = "top"
 $script:codexRowVisible = $true
 $script:claudeRowVisible = $true
 
+function Signal-HudReady {
+  if ([string]::IsNullOrWhiteSpace($ReadyPath)) { return }
+  try {
+    $dir = Split-Path -Parent $ReadyPath
+    if ($dir) { New-Item -ItemType Directory -Force -Path $dir | Out-Null }
+    @{
+      Ready = $true
+      Pid = $PID
+      At = [datetime]::UtcNow.ToString("o")
+      Mode = $Mode
+    } | ConvertTo-Json -Compress | Set-Content -Encoding UTF8 -Path $ReadyPath
+  } catch {
+    # Readiness is diagnostic only.
+  }
+}
+
 function Read-HudPlacement {
   foreach ($path in @((Get-HudStatePath), (Get-LegacyHudStatePath))) {
     try {
@@ -969,6 +986,7 @@ if ($Mode -eq "tray") {
   $timer.add_Tick({ Update-Tray })
   Update-Tray
   $timer.Start()
+  Signal-HudReady
   [System.Windows.Forms.Application]::Run($context)
   Release-SingleInstance
   exit 0
@@ -1807,6 +1825,7 @@ $form.add_SizeChanged({ Sync-HitFormBounds })
 $form.add_Shown({
   Show-HitForm
   Update-HudVisibilityForFullscreen
+  Signal-HudReady
 })
 
 Set-Position

package/bin/ai-battery-macos-status.applescript

@@ -4,9 +4,11 @@ use scripting additions
 
 property statusItem : missing value
 property detailItem : missing value
+property detailView : missing value
 property refreshTimer : missing value
 property titleCommand : ""
 property detailCommand : ""
+property tooltipCommand : ""
 property refreshInterval : 10
 
 on run argv
@@ -14,10 +16,18 @@ on run argv
 
   set titleCommand to item 1 of argv
   set detailCommand to item 2 of argv
-  if (count of argv) is greater than or equal to 3 then
+  set tooltipCommand to detailCommand
+  if (count of argv) is greater than or equal to 4 then
+    set tooltipCommand to item 3 of argv
     try
-      set refreshInterval to (item 3 of argv) as real
+      set refreshInterval to (item 4 of argv) as real
     end try
+  else
+    if (count of argv) is greater than or equal to 3 then
+      try
+        set refreshInterval to (item 3 of argv) as real
+      end try
+    end if
   end if
 
   current application's NSApplication's sharedApplication()
@@ -25,6 +35,7 @@ on run argv
 
   set statusItem to current application's NSStatusBar's systemStatusBar()'s statusItemWithLength:(current application's NSVariableStatusItemLength)
   statusItem's button()'s setTitle:"AI --"
+  statusItem's button()'s setImagePosition:(current application's NSNoImage)
   statusItem's button()'s setToolTip:"AI Battery"
 
   set statusMenu to current application's NSMenu's alloc()'s initWithTitle:"AI Battery"
@@ -46,27 +57,72 @@ on run argv
 
   set refreshTimer to current application's NSTimer's scheduledTimerWithTimeInterval:refreshInterval target:me selector:"refresh:" userInfo:(missing value) repeats:true
   current application's NSRunLoop's currentRunLoop()'s addTimer:refreshTimer forMode:(current application's NSRunLoopCommonModes)
-  current application's NSApp's run()
+  current application's NSApp's |run|()
 end run
 
 on refresh_(sender)
   set titleText to "AI --"
   set detailText to "AI Battery unavailable"
+  set tooltipText to "AI Battery unavailable"
+  set imagePath to ""
+  set menuImage to missing value
+  set detailImage to missing value
 
   try
-    set titleText to do shell script titleCommand
+    set imagePath to (do shell script (titleCommand as text))
   end try
 
   try
-    set detailText to do shell script detailCommand
+    set detailText to (do shell script (detailCommand as text))
   end try
 
-  if titleText is "" then set titleText to "AI --"
+  try
+    set tooltipText to (do shell script (tooltipCommand as text))
+  end try
+
+  if imagePath is "" then set imagePath to "AI --"
   if detailText is "" then set detailText to "AI Battery unavailable"
+  if tooltipText is "" then set tooltipText to detailText
+
+  try
+    set menuImage to current application's NSImage's alloc()'s initWithContentsOfFile:imagePath
+  end try
+
+  try
+    set detailImage to current application's NSImage's alloc()'s initWithContentsOfFile:detailText
+  end try
+
+  if menuImage is not missing value then
+    set imageSize to menuImage's |size|()
+    statusItem's setLength:(width of imageSize)
+    menuImage's setTemplate:false
+    statusItem's button()'s setImage:menuImage
+    statusItem's button()'s setImageScaling:(current application's NSImageScaleProportionallyDown)
+    statusItem's button()'s setImagePosition:(current application's NSImageOnly)
+    statusItem's button()'s setTitle:""
+  else
+    set titleText to imagePath
+    if titleText is "" then set titleText to "AI --"
+    statusItem's setLength:(current application's NSVariableStatusItemLength)
+    statusItem's button()'s setImage:(missing value)
+    statusItem's button()'s setImagePosition:(current application's NSNoImage)
+    statusItem's button()'s setTitle:titleText
+  end if
+
+  if detailImage is not missing value then
+    set detailSize to detailImage's |size|()
+    set detailFrame to current application's NSMakeRect(0, 0, (width of detailSize), (height of detailSize))
+    set detailView to current application's NSImageView's alloc()'s initWithFrame:detailFrame
+    detailView's setImage:detailImage
+    detailView's setImageScaling:(current application's NSImageScaleNone)
+    detailItem's setView:detailView
+    detailItem's setTitle:""
+  else
+    detailItem's setView:(missing value)
+    detailItem's setTitle:detailText
+  end if
 
-  statusItem's button()'s setTitle:titleText
-  statusItem's button()'s setToolTip:detailText
-  detailItem's setTitle:detailText
+  statusItem's button()'s setToolTip:tooltipText
 end refresh_
 
 on quit_(sender)