charge-connect 0.1.1 → 0.1.2

package/cli.js

@@ -16,6 +16,8 @@ const HOME = process.env.HOME ?? process.env.USERPROFILE;
 const CONF_DIR = process.env.CHARGE_HOME ?? path.join(HOME, ".charge");
 const CONF = path.join(CONF_DIR, "config.json");
 const WIN = process.platform === "win32";
+const LINUX = process.platform === "linux";
+const INSTALLER = WIN ? "install.ps1" : LINUX ? "install.linux.sh" : "install.sh";
 
 // 백엔드 주소 — collector/cloud.json (출시 시 charge 전용 프로젝트로 교체, npm 패키지에 포함)
 const CLOUD = (() => {
@@ -57,12 +59,15 @@ async function pair(code) {
 function installRuntime() {
   const dest = path.join(CONF_DIR, "app");
   fs.mkdirSync(dest, { recursive: true, mode: 0o700 });
-  for (const f of ["cli.js", "collect.js", "install.sh", "install.ps1", "cloud.json", "package.json"]) {
+  const files = ["cli.js", "collect.js", "install.sh", "install.linux.sh", "install.ps1", "cloud.json", "package.json"];
+  for (const f of files) {
     const src = path.join(__dirname, f);
     if (fs.existsSync(src)) fs.copyFileSync(src, path.join(dest, f));
   }
   if (!WIN) {
-    try { fs.chmodSync(path.join(dest, "install.sh"), 0o755); } catch {}
+    for (const f of ["install.sh", "install.linux.sh"]) {
+      try { fs.chmodSync(path.join(dest, f), 0o755); } catch {}
+    }
   }
   return dest;
 }
@@ -71,9 +76,8 @@ function collect(dir = __dirname) {
   execFileSync(process.execPath, [path.join(dir, "collect.js")], { stdio: "inherit" });
 }
 
-function installSchedule() {
-  const dir = installRuntime();
-  const p = path.join(dir, WIN ? "install.ps1" : "install.sh");
+function installSchedule(dir) {
+  const p = path.join(dir, INSTALLER);
   if (WIN) {
     execFileSync("powershell.exe", ["-ExecutionPolicy", "Bypass", "-File", p], { stdio: "inherit" });
   } else {
@@ -96,12 +100,47 @@ function installSchedule() {
     console.log("✓ 페어링 해제 (스케줄 해제는 install.sh/install.ps1 안내 참고)");
     return;
   }
-  // 페어링 코드로 간주: 페어링 → 첫 수집 → 스케줄 등록
+  // 페어링 코드로 간주: 페어링 → 스케줄 등록 → 첫 수집.
+  // 스케줄을 먼저 등록한다 — 첫 수집이 실패해도(네트워크가 잠깐 끊기는 등) 자동 수집은
+  // 살아 있어야 한다. 페어링 코드는 이미 소모돼서 같은 코드로 재시도할 수 없기 때문이다.
   await pair(arg);
-  console.log("첫 수집을 실행합니다…");
-  collect();
+  const dir = installRuntime();
+
   console.log("5분 간격 자동 수집을 등록합니다…");
-  installSchedule();
+  let scheduled = true;
+  let guided = false; // 설치 스크립트가 수동 등록 방법을 이미 출력했다 (종료 코드 3)
+  try {
+    installSchedule(dir);
+  } catch (e) {
+    scheduled = false;
+    if (e.status === 3) guided = true;
+    else console.error(`자동 수집 등록 실패: ${e.message ?? e}`);
+  }
+
+  console.log("첫 수집을 실행합니다…");
+  let collected = true;
+  try {
+    collect(dir);
+  } catch {
+    collected = false;
+    console.error("첫 수집에 실패했습니다 (네트워크 문제일 수 있습니다).");
+  }
+
+  if (!scheduled) {
+    if (guided) {
+      console.error("페어링은 끝났습니다. 위 안내대로 자동 수집을 등록하면 완료됩니다.");
+    } else {
+      const p = path.join(dir, INSTALLER);
+      console.error("페어링은 끝났습니다. 자동 수집 등록만 직접 마무리해주세요.");
+      console.error(`  다시 시도: ${WIN ? `powershell -ExecutionPolicy Bypass -File "${p}"` : `bash "${p}"`}`);
+    }
+    process.exit(1);
+  }
+  // 첫 수집이 실패했으면 앱에 아직 데이터가 없다 — 됐다고 말하지 않는다
+  if (!collected) {
+    console.log("✓ 페어링과 자동 수집 등록은 끝났습니다. 5분 뒤 수집을 다시 시도합니다.");
+    return;
+  }
   console.log("✓ 설정 끝! 이제 앱에서 데이터가 보입니다.");
 })().catch((e) => {
   console.error(e.message ?? e);

package/collect.js

@@ -1,6 +1,6 @@
 #!/usr/bin/env node
 // Charge 수집기 — ccusage로 토큰 사용량을 뽑아 Charge 백엔드에 업로드한다.
-// 사용법: node collect.js [--dry-run] [--days N]
+// 사용법: node collect.js [--dry-run] [--days N] [--log 경로]
 // 인증: `npx charge-connect <페어링코드>`가 저장한 ~/.charge/config.json의 디바이스 토큰
 
 const { execFile } = require("node:child_process");
@@ -16,8 +16,51 @@ const parsedDays = daysArg > -1 ? parseInt(process.argv[daysArg + 1], 10) : NaN;
 const DAYS = Number.isFinite(parsedDays) && parsedDays > 0 ? parsedDays : 60;
 const CACHE_FILE = path.join(__dirname, ".last-payload.json");
 const HOME = process.env.HOME ?? process.env.USERPROFILE;
+
+// 스케줄러가 부를 때는 볼 콘솔이 없으니 출력을 직접 파일로 받는다.
+// 셸 리다이렉션(cmd `>`)에 맡기면 경로의 %VAR% 확장·로그 잠금 같은 문제를 떠안게 된다.
+// 5분마다 도는 작업이라 그냥 이어 붙이면 끝없이 자라므로 5MB에서 .old로 민다.
+const logArg = process.argv.indexOf("--log");
+const LOG_FILE = logArg > -1 ? process.argv[logArg + 1] : null;
+if (logArg > -1 && (!LOG_FILE || LOG_FILE.startsWith("-"))) {
+  // 값을 안 주면 조용히 로그가 꺼지고, 다음 옵션을 삼키면 "--dry-run"이라는 파일에 쓰게 된다
+  console.error("--log 뒤에는 파일 경로가 필요합니다.");
+  process.exit(1);
+}
+if (LOG_FILE) {
+  try {
+    if (fs.statSync(LOG_FILE).size > 5 * 1024 * 1024) fs.renameSync(LOG_FILE, `${LOG_FILE}.old`);
+  } catch {}
+  // 비동기 스트림은 process.exit에서 버퍼를 잃을 수 있어 동기로 쓴다 (줄 수가 적다)
+  const write = (...args) => {
+    try {
+      fs.appendFileSync(LOG_FILE, `${args.map(String).join(" ")}\n`);
+    } catch {}
+  };
+  console.log = write;
+  console.error = write;
+  // 스케줄러가 돌릴 때는 Node 기본 stderr가 어디에도 안 남는다. 잡지 않으면
+  // 크래시했을 때 로그에 시작 줄만 덩그러니 남아 원인을 알 수 없다.
+  const fatal = (label) => (err) => {
+    write(`[${new Date().toISOString()}] ${label}: ${err?.stack ?? err}`);
+    process.exit(1);
+  };
+  process.on("uncaughtException", fatal("치명적 오류"));
+  process.on("unhandledRejection", fatal("처리되지 않은 거부"));
+  // 시작 줄을 바로 남긴다 — '언제 마지막으로 돌았나'를 알 수 있고,
+  // install.ps1이 스케줄 작업이 실제로 떴는지 판정하는 근거이기도 하다.
+  write(`[${new Date().toISOString()}] 수집 시작`);
+}
 const WIN = process.platform === "win32";
-const EXTRA_PATH = WIN ? "" : ":/opt/homebrew/bin:/usr/local/bin";
+// 스케줄러(작업 스케줄러/launchd)가 만드는 환경에는 셸 프로필에서 붙는 PATH가 없다.
+// Windows에서 nvm-windows·fnm·volta로 Node를 깐 사용자는 그래서 npx/ccusage를 못 찾는다.
+// node가 자기 위치는 알고 있으므로, 그 디렉터리와 npm 전역 bin을 직접 얹어준다.
+const EXTRA_PATH = WIN
+  ? [path.dirname(process.execPath), process.env.APPDATA && path.join(process.env.APPDATA, "npm")]
+      .filter(Boolean)
+      .map((p) => `;${p}`)
+      .join("")
+  : ":/opt/homebrew/bin:/usr/local/bin";
 const DEFAULT_CODEXBAR_CLI = "/Applications/CodexBar.app/Contents/Helpers/CodexBarCLI";
 
 // 모든 외부 명령은 비동기로 실행한다 — 동기(execFileSync)는 이벤트 루프를 세워
@@ -326,6 +369,7 @@ async function codexLiveWindows(auth) {
 // 마지막으로 Codex를 실제 사용한 시점의 값이라 실시간 조회가 실패했을 때만 쓴다
 function codexSnapshotWindows() {
   const dir = path.join(HOME, ".codex", "sessions");
+  if (!fs.existsSync(dir)) return null; // Codex 미설치
   let latest = null;
   (function walk(d) {
     for (const e of fs.readdirSync(d, { withFileTypes: true })) {

package/install.linux.sh

@@ -0,0 +1,60 @@
+#!/bin/bash
+# Linux/WSL에 수집기를 등록한다 (5분 간격).
+# systemd 사용자 세션이 있으면 타이머로 등록하고, 없으면(WSL에서 흔하다) cron 한 줄을 안내한다.
+set -euo pipefail
+
+SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
+LOG_DIR="${CHARGE_HOME:-$HOME/.charge}"
+LOG="$LOG_DIR/collector.log"
+
+NODE="$(command -v node || true)"
+if [ -z "$NODE" ]; then
+  echo "Node.js를 찾을 수 없습니다."
+  echo "배포판 패키지나 https://nodejs.org 로 설치한 뒤 다시 실행해주세요."
+  exit 1
+fi
+
+mkdir -p "$LOG_DIR"
+
+# systemctl 이 있어도 사용자 세션이 안 뜬 경우가 많다 (WSL, 컨테이너, SSH) — 실제로 물어본다
+if command -v systemctl >/dev/null 2>&1 && systemctl --user show-environment >/dev/null 2>&1; then
+  UNIT_DIR="$HOME/.config/systemd/user"
+  mkdir -p "$UNIT_DIR"
+
+  cat > "$UNIT_DIR/charge-connect.service" <<EOF
+[Unit]
+Description=Charge usage collector
+
+[Service]
+Type=oneshot
+ExecStart="$NODE" "$SCRIPT_DIR/collect.js" --log "$LOG"
+EOF
+
+  cat > "$UNIT_DIR/charge-connect.timer" <<EOF
+[Unit]
+Description=Charge usage collector (5분 간격)
+
+[Timer]
+OnBootSec=1min
+OnUnitActiveSec=5min
+Persistent=true
+
+[Install]
+WantedBy=timers.target
+EOF
+
+  systemctl --user daemon-reload
+  systemctl --user enable --now charge-connect.timer
+  echo "등록 완료: charge-connect.timer (5분 간격). 로그: $LOG"
+  echo "해제하려면: systemctl --user disable --now charge-connect.timer"
+  exit 0
+fi
+
+# 여기까지 왔으면 자동 등록은 못 한다. 종료 코드 3으로 알려 cli.js가 안내를 덧붙이게 한다.
+echo "systemd 사용자 세션을 찾지 못했습니다 (WSL에서는 흔한 일입니다)."
+echo "아래 한 줄을 crontab에 넣으면 5분마다 수집합니다:"
+echo
+echo "  */5 * * * * '$NODE' '$SCRIPT_DIR/collect.js' --log '$LOG'"
+echo
+echo "등록: crontab -e   (cron이 안 돌고 있으면: sudo service cron start)"
+exit 3

package/install.ps1

@@ -1,18 +1,36 @@
-# Charge 수집기를 Windows 작업 스케줄러에 등록한다 (5분 간격).
+# Charge 수집기를 Windows 작업 스케줄러에 등록한다 (5분 간격).
 # 실행: PowerShell에서 .\install.ps1  (실행 정책 오류 시: powershell -ExecutionPolicy Bypass -File install.ps1)
+#
+# 이 파일은 UTF-8 BOM으로 저장해야 한다 — cli.js가 부르는 Windows PowerShell 5.1은
+# BOM이 없으면 시스템 ANSI 코드페이지로 읽어서 아래 한글 안내가 전부 깨진다.
 $ErrorActionPreference = "Stop"
 
 $taskName = "ChargeConnect"
-$scriptDir = Split-Path -Parent $MyInvocation.MyCommand.Path
-$node = (Get-Command node).Source
-$log = Join-Path $env:USERPROFILE "charge-connect.log"
+$scriptDir = if ($PSScriptRoot) { $PSScriptRoot } else { Split-Path -Parent $MyInvocation.MyCommand.Path }
 
-# 예전 이름(ChargeCollector)으로 등록된 잔재가 있으면 정리
+# Node가 없을 때 PowerShell 원문 에러 대신 사람이 읽을 수 있는 안내를 낸다
+$nodeCmd = Get-Command node -ErrorAction SilentlyContinue
+if (-not $nodeCmd) {
+    Write-Host "Node.js를 찾을 수 없습니다."
+    Write-Host "https://nodejs.org 에서 LTS 버전을 설치한 뒤 다시 실행해주세요."
+    exit 1
+}
+$node = $nodeCmd.Source
+
+$logDir = if ($env:CHARGE_HOME) { $env:CHARGE_HOME } else { Join-Path $env:USERPROFILE ".charge" }
+New-Item -ItemType Directory -Force -Path $logDir | Out-Null
+$log = Join-Path $logDir "collector.log"
+$logOld = $log + ".old"
+
+# 예전 이름(ChargeCollector)·예전 로그 위치의 잔재가 있으면 정리
 Unregister-ScheduledTask -TaskName "ChargeCollector" -Confirm:$false -ErrorAction SilentlyContinue
+Remove-Item -Force -ErrorAction SilentlyContinue (Join-Path $env:USERPROFILE "charge-connect.log")
 
-# cmd /c 로 감싸 로그 리다이렉션 (작업 스케줄러는 자체 리다이렉션이 없음)
-$arg = "/c `"`"$node`" `"$scriptDir\collect.js`" >> `"$log`" 2>&1`""
-$action = New-ScheduledTaskAction -Execute "cmd.exe" -Argument $arg
+# node 를 직접 실행하고 로그는 collect.js 가 --log 로 직접 쓴다.
+# cmd /c 로 감싸 리다이렉션하면 경로 안의 %VAR% 가 확장되고, move 실패가 조용히 묻히고,
+# 실행이 겹칠 때 로그 잠금으로 출력이 통째로 사라진다. 셸을 안 거치면 그 셋이 모두 없어진다.
+$arg = "`"$scriptDir\collect.js`" --log `"$log`""
+$action = New-ScheduledTaskAction -Execute $node -Argument $arg -WorkingDirectory $scriptDir
 $trigger = New-ScheduledTaskTrigger -Once -At (Get-Date) `
     -RepetitionInterval (New-TimeSpan -Minutes 5) `
     -RepetitionDuration (New-TimeSpan -Days 3650)
@@ -20,8 +38,74 @@ $settings = New-ScheduledTaskSettingsSet -StartWhenAvailable `
     -AllowStartIfOnBatteries -DontStopIfGoingOnBatteries `
     -ExecutionTimeLimit (New-TimeSpan -Minutes 10)
 
-Register-ScheduledTask -TaskName $taskName -Action $action -Trigger $trigger -Settings $settings -Force | Out-Null
-Start-ScheduledTask -TaskName $taskName
+function Register-ChargeTask([switch]$Background) {
+    $p = @{ TaskName = $taskName; Action = $action; Trigger = $trigger; Settings = $settings; Force = $true }
+    if ($Background) {
+        # 기본값(LogonType Interactive)으로 등록하면 콘솔 창이 5분마다 화면에 뜬다.
+        # S4U는 대화형 세션 없이 실행돼 창이 뜨지 않고, 로그오프 상태에서도 수집이 돈다.
+        $p.Principal = New-ScheduledTaskPrincipal -UserId "$env:USERDOMAIN\$env:USERNAME" `
+            -LogonType S4U -RunLevel Limited
+    }
+    Register-ScheduledTask @p | Out-Null
+}
+
+function Get-LogState {
+    $s = ""
+    foreach ($f in @($log, $logOld)) {
+        # Test-Path 로 확인하고 Get-Item 을 따로 부르면, 그 사이 수집기가 로그를 밀었을 때
+        # ErrorActionPreference='Stop' 에 걸려 설치가 통째로 죽는다. 한 번에 가져온다.
+        $i = Get-Item -LiteralPath $f -ErrorAction SilentlyContinue
+        if ($i) { $s += "$($i.Length):$($i.LastWriteTime.Ticks);" } else { $s += "-;" }
+    }
+    return $s
+}
+
+# 등록에 성공해도 실행은 실패할 수 있다. S4U 로그온에는 '일괄 작업으로 로그온'
+# (SeBatchLogonRight) 권한이 필요한데, 비관리자 계정이나 GPO로 제한된 PC에서는
+# 등록만 통과하고 실행이 매번 0x80070569로 죽는다. 그때는 프로세스 자체가 안 떠서
+# 로그도 안 남기 때문에, 확인하지 않으면 수집이 조용히 영영 멈춘다.
+# 그래서 실제로 한 번 돌려보고 프로세스가 떴다는 증거(로그 파일 변화)를 확인한다.
+function Test-ChargeTaskRuns {
+    # 예전 정의로 돌던 인스턴스가 남아 있으면 그쪽 출력을 새 작업의 성공으로 오인한다
+    Stop-ScheduledTask -TaskName $taskName -ErrorAction SilentlyContinue
+    Start-Sleep -Milliseconds 500
+    $before = Get-LogState
+    Start-ScheduledTask -TaskName $taskName
+    $deadline = (Get-Date).AddSeconds(60)
+    while ((Get-Date) -lt $deadline) {
+        # collect.js 는 뜨자마자 로그에 시작 줄을 남긴다 — 프로세스가 떴다는 직접 증거.
+        # '실행 중(0x41301)' 상태만으로 성공 처리하면 남아 있던 인스턴스에 속을 수 있다.
+        if ((Get-LogState) -ne $before) { return $true }
+        $r = (Get-ScheduledTaskInfo -TaskName $taskName).LastTaskResult
+        # 0x80070569 요청한 로그온 유형이 부여되지 않음 / 0x8007052E 로그온 실패
+        if ($r -eq 2147943785 -or $r -eq 2147943726) { return $false }
+        Start-Sleep -Milliseconds 500
+    }
+    return $false
+}
+
+$background = $true
+try {
+    Register-ChargeTask -Background
+} catch {
+    Write-Host "백그라운드 실행 등록 실패: $($_.Exception.Message)"
+    $background = $false
+}
+if ($background -and -not (Test-ChargeTaskRuns)) {
+    Write-Host "백그라운드 실행 권한이 없어 대화형 모드로 되돌립니다."
+    $background = $false
+}
+if (-not $background) {
+    try {
+        Register-ChargeTask
+        Start-ScheduledTask -TaskName $taskName
+    } catch {
+        Write-Host "작업 스케줄러 등록에 실패했습니다: $($_.Exception.Message)"
+        Write-Host "수동으로 수집하려면: node `"$scriptDir\collect.js`""
+        exit 1
+    }
+    Write-Host "참고: 5분마다 콘솔 창이 잠깐 보일 수 있습니다 (백그라운드 실행 권한이 없는 환경)."
+}
 
 Write-Host "등록 완료: $taskName (5분 간격). 로그: $log"
 Write-Host "해제하려면: Unregister-ScheduledTask -TaskName $taskName -Confirm:`$false"

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "charge-connect",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "description": "Connects your computer to the Charge iPhone app and uploads AI coding usage",
   "bin": {
     "charge-connect": "cli.js"
@@ -12,6 +12,7 @@
     "cli.js",
     "collect.js",
     "install.sh",
+    "install.linux.sh",
     "install.ps1",
     "cloud.json"
   ],