charge-connect 0.1.0 → 0.1.2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/cli.js CHANGED
@@ -16,6 +16,8 @@ const HOME = process.env.HOME ?? process.env.USERPROFILE;
16
16
  const CONF_DIR = process.env.CHARGE_HOME ?? path.join(HOME, ".charge");
17
17
  const CONF = path.join(CONF_DIR, "config.json");
18
18
  const WIN = process.platform === "win32";
19
+ const LINUX = process.platform === "linux";
20
+ const INSTALLER = WIN ? "install.ps1" : LINUX ? "install.linux.sh" : "install.sh";
19
21
 
20
22
  // 백엔드 주소 — collector/cloud.json (출시 시 charge 전용 프로젝트로 교체, npm 패키지에 포함)
21
23
  const CLOUD = (() => {
@@ -57,12 +59,15 @@ async function pair(code) {
57
59
  function installRuntime() {
58
60
  const dest = path.join(CONF_DIR, "app");
59
61
  fs.mkdirSync(dest, { recursive: true, mode: 0o700 });
60
- for (const f of ["cli.js", "collect.js", "install.sh", "install.ps1", "cloud.json", "package.json"]) {
62
+ const files = ["cli.js", "collect.js", "install.sh", "install.linux.sh", "install.ps1", "cloud.json", "package.json"];
63
+ for (const f of files) {
61
64
  const src = path.join(__dirname, f);
62
65
  if (fs.existsSync(src)) fs.copyFileSync(src, path.join(dest, f));
63
66
  }
64
67
  if (!WIN) {
65
- try { fs.chmodSync(path.join(dest, "install.sh"), 0o755); } catch {}
68
+ for (const f of ["install.sh", "install.linux.sh"]) {
69
+ try { fs.chmodSync(path.join(dest, f), 0o755); } catch {}
70
+ }
66
71
  }
67
72
  return dest;
68
73
  }
@@ -71,9 +76,8 @@ function collect(dir = __dirname) {
71
76
  execFileSync(process.execPath, [path.join(dir, "collect.js")], { stdio: "inherit" });
72
77
  }
73
78
 
74
- function installSchedule() {
75
- const dir = installRuntime();
76
- const p = path.join(dir, WIN ? "install.ps1" : "install.sh");
79
+ function installSchedule(dir) {
80
+ const p = path.join(dir, INSTALLER);
77
81
  if (WIN) {
78
82
  execFileSync("powershell.exe", ["-ExecutionPolicy", "Bypass", "-File", p], { stdio: "inherit" });
79
83
  } else {
@@ -96,12 +100,47 @@ function installSchedule() {
96
100
  console.log("✓ 페어링 해제 (스케줄 해제는 install.sh/install.ps1 안내 참고)");
97
101
  return;
98
102
  }
99
- // 페어링 코드로 간주: 페어링 → 첫 수집 → 스케줄 등록
103
+ // 페어링 코드로 간주: 페어링 → 스케줄 등록 → 첫 수집.
104
+ // 스케줄을 먼저 등록한다 — 첫 수집이 실패해도(네트워크가 잠깐 끊기는 등) 자동 수집은
105
+ // 살아 있어야 한다. 페어링 코드는 이미 소모돼서 같은 코드로 재시도할 수 없기 때문이다.
100
106
  await pair(arg);
101
- console.log("첫 수집을 실행합니다…");
102
- collect();
107
+ const dir = installRuntime();
108
+
103
109
  console.log("5분 간격 자동 수집을 등록합니다…");
104
- installSchedule();
110
+ let scheduled = true;
111
+ let guided = false; // 설치 스크립트가 수동 등록 방법을 이미 출력했다 (종료 코드 3)
112
+ try {
113
+ installSchedule(dir);
114
+ } catch (e) {
115
+ scheduled = false;
116
+ if (e.status === 3) guided = true;
117
+ else console.error(`자동 수집 등록 실패: ${e.message ?? e}`);
118
+ }
119
+
120
+ console.log("첫 수집을 실행합니다…");
121
+ let collected = true;
122
+ try {
123
+ collect(dir);
124
+ } catch {
125
+ collected = false;
126
+ console.error("첫 수집에 실패했습니다 (네트워크 문제일 수 있습니다).");
127
+ }
128
+
129
+ if (!scheduled) {
130
+ if (guided) {
131
+ console.error("페어링은 끝났습니다. 위 안내대로 자동 수집을 등록하면 완료됩니다.");
132
+ } else {
133
+ const p = path.join(dir, INSTALLER);
134
+ console.error("페어링은 끝났습니다. 자동 수집 등록만 직접 마무리해주세요.");
135
+ console.error(` 다시 시도: ${WIN ? `powershell -ExecutionPolicy Bypass -File "${p}"` : `bash "${p}"`}`);
136
+ }
137
+ process.exit(1);
138
+ }
139
+ // 첫 수집이 실패했으면 앱에 아직 데이터가 없다 — 됐다고 말하지 않는다
140
+ if (!collected) {
141
+ console.log("✓ 페어링과 자동 수집 등록은 끝났습니다. 5분 뒤 수집을 다시 시도합니다.");
142
+ return;
143
+ }
105
144
  console.log("✓ 설정 끝! 이제 앱에서 데이터가 보입니다.");
106
145
  })().catch((e) => {
107
146
  console.error(e.message ?? e);
package/collect.js CHANGED
@@ -1,40 +1,92 @@
1
1
  #!/usr/bin/env node
2
2
  // Charge 수집기 — ccusage로 토큰 사용량을 뽑아 Charge 백엔드에 업로드한다.
3
- // 사용법: node collect.js [--dry-run] [--days N]
3
+ // 사용법: node collect.js [--dry-run] [--days N] [--log 경로]
4
4
  // 인증: `npx charge-connect <페어링코드>`가 저장한 ~/.charge/config.json의 디바이스 토큰
5
5
 
6
- const { execFileSync } = require("node:child_process");
6
+ const { execFile } = require("node:child_process");
7
+ const { promisify } = require("node:util");
7
8
  const fs = require("node:fs");
8
9
  const path = require("node:path");
9
10
 
11
+ const execFileAsync = promisify(execFile);
12
+
10
13
  const DRY_RUN = process.argv.includes("--dry-run");
11
14
  const daysArg = process.argv.indexOf("--days");
12
15
  const parsedDays = daysArg > -1 ? parseInt(process.argv[daysArg + 1], 10) : NaN;
13
16
  const DAYS = Number.isFinite(parsedDays) && parsedDays > 0 ? parsedDays : 60;
14
17
  const CACHE_FILE = path.join(__dirname, ".last-payload.json");
15
18
  const HOME = process.env.HOME ?? process.env.USERPROFILE;
19
+
20
+ // 스케줄러가 부를 때는 볼 콘솔이 없으니 출력을 직접 파일로 받는다.
21
+ // 셸 리다이렉션(cmd `>`)에 맡기면 경로의 %VAR% 확장·로그 잠금 같은 문제를 떠안게 된다.
22
+ // 5분마다 도는 작업이라 그냥 이어 붙이면 끝없이 자라므로 5MB에서 .old로 민다.
23
+ const logArg = process.argv.indexOf("--log");
24
+ const LOG_FILE = logArg > -1 ? process.argv[logArg + 1] : null;
25
+ if (logArg > -1 && (!LOG_FILE || LOG_FILE.startsWith("-"))) {
26
+ // 값을 안 주면 조용히 로그가 꺼지고, 다음 옵션을 삼키면 "--dry-run"이라는 파일에 쓰게 된다
27
+ console.error("--log 뒤에는 파일 경로가 필요합니다.");
28
+ process.exit(1);
29
+ }
30
+ if (LOG_FILE) {
31
+ try {
32
+ if (fs.statSync(LOG_FILE).size > 5 * 1024 * 1024) fs.renameSync(LOG_FILE, `${LOG_FILE}.old`);
33
+ } catch {}
34
+ // 비동기 스트림은 process.exit에서 버퍼를 잃을 수 있어 동기로 쓴다 (줄 수가 적다)
35
+ const write = (...args) => {
36
+ try {
37
+ fs.appendFileSync(LOG_FILE, `${args.map(String).join(" ")}\n`);
38
+ } catch {}
39
+ };
40
+ console.log = write;
41
+ console.error = write;
42
+ // 스케줄러가 돌릴 때는 Node 기본 stderr가 어디에도 안 남는다. 잡지 않으면
43
+ // 크래시했을 때 로그에 시작 줄만 덩그러니 남아 원인을 알 수 없다.
44
+ const fatal = (label) => (err) => {
45
+ write(`[${new Date().toISOString()}] ${label}: ${err?.stack ?? err}`);
46
+ process.exit(1);
47
+ };
48
+ process.on("uncaughtException", fatal("치명적 오류"));
49
+ process.on("unhandledRejection", fatal("처리되지 않은 거부"));
50
+ // 시작 줄을 바로 남긴다 — '언제 마지막으로 돌았나'를 알 수 있고,
51
+ // install.ps1이 스케줄 작업이 실제로 떴는지 판정하는 근거이기도 하다.
52
+ write(`[${new Date().toISOString()}] 수집 시작`);
53
+ }
16
54
  const WIN = process.platform === "win32";
17
- const EXTRA_PATH = WIN ? "" : ":/opt/homebrew/bin:/usr/local/bin";
55
+ // 스케줄러(작업 스케줄러/launchd)가 만드는 환경에는 셸 프로필에서 붙는 PATH가 없다.
56
+ // Windows에서 nvm-windows·fnm·volta로 Node를 깐 사용자는 그래서 npx/ccusage를 못 찾는다.
57
+ // node가 자기 위치는 알고 있으므로, 그 디렉터리와 npm 전역 bin을 직접 얹어준다.
58
+ const EXTRA_PATH = WIN
59
+ ? [path.dirname(process.execPath), process.env.APPDATA && path.join(process.env.APPDATA, "npm")]
60
+ .filter(Boolean)
61
+ .map((p) => `;${p}`)
62
+ .join("")
63
+ : ":/opt/homebrew/bin:/usr/local/bin";
18
64
  const DEFAULT_CODEXBAR_CLI = "/Applications/CodexBar.app/Contents/Helpers/CodexBarCLI";
19
65
 
20
- function run(cmd, args, timeout = 120_000) {
21
- return execFileSync(cmd, args, {
22
- encoding: "utf8",
23
- maxBuffer: 64 * 1024 * 1024,
24
- timeout,
25
- env: { ...process.env, PATH: `${process.env.PATH}${EXTRA_PATH}` },
26
- // Windows에서 npx/ccusage는 .cmd 셔틀이라 셸을 거쳐야 실행된다 (인자는 모두 고정 문자열)
27
- shell: WIN,
28
- });
66
+ // 모든 외부 명령은 비동기로 실행한다 — 동기(execFileSync)는 이벤트 루프를 세워
67
+ // 진행 중인 fetch(AbortSignal 타이머 포함)를 전부 타임아웃시킨다.
68
+ // Windows에서 npx/ccusage는 .cmd 셔틀이라 셸을 거쳐야 실행된다 (인자는 모두 고정 문자열).
69
+ const execOpts = (timeout) => ({
70
+ encoding: "utf8",
71
+ maxBuffer: 64 * 1024 * 1024,
72
+ timeout,
73
+ env: { ...process.env, PATH: `${process.env.PATH}${EXTRA_PATH}` },
74
+ shell: WIN,
75
+ });
76
+
77
+ async function runAsync(cmd, args, timeout = 120_000) {
78
+ // promisify된 execFile은 실패 시에도 err.stdout/stderr를 붙여준다
79
+ return (await execFileAsync(cmd, args, execOpts(timeout))).stdout;
29
80
  }
30
81
 
31
- function runCcusage(args) {
82
+ async function runCcusage(args) {
32
83
  // 전역 설치본이 있으면 사용(빠름), 없으면 npx로 대체
33
84
  try {
34
- return JSON.parse(run("ccusage", args));
85
+ return JSON.parse(await runAsync("ccusage", args));
35
86
  } catch (e) {
36
- console.error(`ccusage 직접 실행 실패(${e.code ?? e.message}), npx로 재시도`);
37
- return JSON.parse(run("npx", ["-y", "ccusage@latest", ...args], 300_000));
87
+ console.error(`ccusage 직접 실행 실패(${e.code ?? e.message}), npx로 재시도 — 'npm i -g ccusage'를 해두면 빨라집니다`);
88
+ // --prefer-offline: npx 캐시가 있으면 레지스트리 조회 없이 재사용 (5분마다 재다운로드 방지)
89
+ return JSON.parse(await runAsync("npx", ["-y", "--prefer-offline", "ccusage@latest", ...args], 300_000));
38
90
  }
39
91
  }
40
92
 
@@ -51,6 +103,21 @@ function dropExpired(w) {
51
103
  return w;
52
104
  }
53
105
 
106
+ // 캐시에서 복원한 프로바이더도 신선 수집과 같은 만료 규칙을 적용한다.
107
+ // 안 하면 수집이 실패하는 동안 리셋이 지난 창이 무한 재업로드된다 (유령 게이지).
108
+ function sanitizeCachedProvider(prev) {
109
+ if (!prev || typeof prev !== "object") return null;
110
+ const extras = (prev.extras ?? [])
111
+ .map((extra) => ({ ...extra, window: dropExpired(extra.window) }))
112
+ .filter((extra) => extra.window);
113
+ return {
114
+ ...prev,
115
+ session: dropExpired(prev.session),
116
+ weekly: dropExpired(prev.weekly),
117
+ extras: extras.length ? extras : null,
118
+ };
119
+ }
120
+
54
121
  const PROVIDER_NAMES = {
55
122
  "azure-openai": "Azure OpenAI",
56
123
  "openai": "OpenAI",
@@ -137,8 +204,8 @@ function sinceStr(days) {
137
204
  return `${d.getFullYear()}${String(d.getMonth() + 1).padStart(2, "0")}${String(d.getDate()).padStart(2, "0")}`;
138
205
  }
139
206
 
140
- function collectDaily() {
141
- const data = runCcusage(["daily", "--json", "--since", sinceStr(DAYS)]);
207
+ async function collectDaily() {
208
+ const data = await runCcusage(["daily", "--json", "--since", sinceStr(DAYS)]);
142
209
  let rows = data.daily ?? [];
143
210
  // 에이전트별 행과 "all" 집계 행이 섞여 있으면 "all"만 사용해 중복 합산을 방지
144
211
  if (rows.some((r) => r.agent === "all")) rows = rows.filter((r) => r.agent === "all");
@@ -161,15 +228,15 @@ function collectDaily() {
161
228
  return [...byDay.values()].sort((a, b) => a.period.localeCompare(b.period));
162
229
  }
163
230
 
164
- function collectLive() {
165
- const data = runCcusage(["blocks", "--active", "--json"]);
231
+ async function collectLive() {
232
+ const data = await runCcusage(["blocks", "--active", "--json"]);
166
233
  return (data.blocks ?? []).find((b) => b.isActive) ?? null;
167
234
  }
168
235
 
169
236
  // Claude Code 자격증명: macOS는 Keychain, Windows/Linux는 ~/.claude/.credentials.json
170
- function claudeCredentials() {
237
+ async function claudeCredentials() {
171
238
  try {
172
- return JSON.parse(run("security", ["find-generic-password", "-s", "Claude Code-credentials", "-w"]));
239
+ return JSON.parse(await runAsync("security", ["find-generic-password", "-s", "Claude Code-credentials", "-w"]));
173
240
  } catch {
174
241
  return JSON.parse(fs.readFileSync(path.join(HOME, ".claude", ".credentials.json"), "utf8"));
175
242
  }
@@ -178,7 +245,7 @@ function claudeCredentials() {
178
245
  // Claude: Claude Code OAuth 세션을 재사용해 공식 usage API 조회
179
246
  async function claudeProvider() {
180
247
  try {
181
- const cred = claudeCredentials();
248
+ const cred = await claudeCredentials();
182
249
  const oauth = cred.claudeAiOauth ?? cred;
183
250
  const token = oauth.accessToken;
184
251
  // rateLimitTier "default_claude_max_20x" → "Max 20x", 없으면 subscriptionType "max" → "Max"
@@ -192,7 +259,11 @@ async function claudeProvider() {
192
259
  headers: { Authorization: `Bearer ${token}`, "anthropic-beta": "oauth-2025-04-20" },
193
260
  signal: AbortSignal.timeout(10_000),
194
261
  });
195
- if (!res.ok) return null;
262
+ if (!res.ok) {
263
+ // 401 = 토큰 만료 (Claude Code를 한 번 실행하면 갱신됨), 그 외는 일시 장애일 가능성
264
+ console.error(`claude usage API ${res.status}${res.status === 401 ? " — 토큰 만료, Claude Code를 실행하면 갱신됩니다" : ""}`);
265
+ return null;
266
+ }
196
267
  const d = await res.json();
197
268
  // 계정 UUID → 해시 (프로필 조회 실패 시 null — 캐시 폴백이 채워준다)
198
269
  let account = null;
@@ -298,6 +369,7 @@ async function codexLiveWindows(auth) {
298
369
  // 마지막으로 Codex를 실제 사용한 시점의 값이라 실시간 조회가 실패했을 때만 쓴다
299
370
  function codexSnapshotWindows() {
300
371
  const dir = path.join(HOME, ".codex", "sessions");
372
+ if (!fs.existsSync(dir)) return null; // Codex 미설치
301
373
  let latest = null;
302
374
  (function walk(d) {
303
375
  for (const e of fs.readdirSync(d, { withFileTypes: true })) {
@@ -460,7 +532,7 @@ function parseCodexBarJSON(raw) {
460
532
  }
461
533
  }
462
534
 
463
- function collectCodexBarProviders() {
535
+ async function collectCodexBarProviders() {
464
536
  if (process.env.CHARGE_DISABLE_CODEXBAR === "1") return { providers: [], complete: false };
465
537
  const cli = process.env.CHARGE_CODEXBAR_CLI
466
538
  ?? (!WIN && fs.existsSync(DEFAULT_CODEXBAR_CLI) ? DEFAULT_CODEXBAR_CLI : null);
@@ -469,7 +541,9 @@ function collectCodexBarProviders() {
469
541
  let raw = "";
470
542
  let commandSucceeded = true;
471
543
  try {
472
- raw = run(cli, ["usage", "--format", "json"], 90_000);
544
+ // CodexBar CLI는 실측 20초 이상 걸릴 수 있다 — 동기로 돌리면 그동안 이벤트 루프가
545
+ // 멈춰 Claude/Codex fetch의 abort 타이머가 전부 발화하므로 반드시 비동기로.
546
+ raw = await runAsync(cli, ["usage", "--format", "json"], 90_000);
473
547
  } catch (e) {
474
548
  commandSucceeded = false;
475
549
  raw = e.stdout?.toString?.() ?? "";
@@ -506,16 +580,12 @@ async function statusOf(url) {
506
580
  }
507
581
 
508
582
  async function collectProviders() {
509
- const claudePromise = claudeProvider();
510
- const claudeStatusPromise = statusOf("https://status.anthropic.com/api/v2/status.json");
511
- const codexStatusPromise = statusOf("https://status.openai.com/api/v2/status.json");
512
- const codexPromise = codexProvider();
513
- const codexBar = collectCodexBarProviders();
514
- const [claude, codex, claudeStatus, codexStatus] = await Promise.all([
515
- claudePromise,
516
- codexPromise,
517
- claudeStatusPromise,
518
- codexStatusPromise,
583
+ const [claude, codex, claudeStatus, codexStatus, codexBar] = await Promise.all([
584
+ claudeProvider(),
585
+ codexProvider(),
586
+ statusOf("https://status.anthropic.com/api/v2/status.json"),
587
+ statusOf("https://status.openai.com/api/v2/status.json"),
588
+ collectCodexBarProviders(),
519
589
  ]);
520
590
  if (claude) claude.status = claudeStatus;
521
591
  if (codex) codex.status = codexStatus;
@@ -560,24 +630,32 @@ async function main() {
560
630
  cache = JSON.parse(fs.readFileSync(CACHE_FILE, "utf8"));
561
631
  } catch {}
562
632
 
633
+ // ccusage 두 갈래와 프로바이더 수집(네트워크+CodexBar CLI)은 서로 독립 —
634
+ // 병렬로 돌려 전체 수집 시간을 max(ccusage, network)로 줄인다.
635
+ // allSettled라 한 갈래가 실패해도 나머지는 살고, 실패분만 이전 값으로 채운다.
636
+ const [dailyR, liveR, providersR] = await Promise.allSettled([
637
+ collectDaily(),
638
+ collectLive(),
639
+ collectProviders(),
640
+ ]);
563
641
  let daily, live, providers;
564
- try {
565
- daily = collectDaily();
566
- } catch (e) {
567
- console.error(`daily 수집 실패, 이전 값 유지: ${e.message ?? e}`);
642
+ if (dailyR.status === "fulfilled") {
643
+ daily = dailyR.value;
644
+ } else {
645
+ console.error(`daily 수집 실패, 이전 값 유지: ${dailyR.reason?.message ?? dailyR.reason}`);
568
646
  daily = cache.daily ?? [];
569
647
  }
570
- try {
571
- live = collectLive();
572
- } catch (e) {
573
- console.error(`live 수집 실패, 이전 값 유지: ${e.message ?? e}`);
648
+ if (liveR.status === "fulfilled") {
649
+ live = liveR.value;
650
+ } else {
651
+ console.error(`live 수집 실패, 이전 값 유지: ${liveR.reason?.message ?? liveR.reason}`);
574
652
  live = cache.live ?? null;
575
653
  }
576
654
  let providerCollection;
577
- try {
578
- providerCollection = await collectProviders();
579
- } catch (e) {
580
- console.error(`프로바이더 수집 실패, 이전 값 유지: ${e.message ?? e}`);
655
+ if (providersR.status === "fulfilled") {
656
+ providerCollection = providersR.value;
657
+ } else {
658
+ console.error(`프로바이더 수집 실패, 이전 값 유지: ${providersR.reason?.message ?? providersR.reason}`);
581
659
  providerCollection = { providers: [], codexBarComplete: false };
582
660
  }
583
661
  providers = providerCollection.providers;
@@ -586,7 +664,8 @@ async function main() {
586
664
  if (providers.some((p) => p.id === prev.id)) continue;
587
665
  // CodexBar가 정상 완료된 경우, 거기서 더 이상 반환하지 않는 항목은 사용자가 끈 것으로 본다.
588
666
  if (providerCollection.codexBarComplete && prev.collector_source === "codexbar") continue;
589
- providers.push(prev);
667
+ const sanitized = sanitizeCachedProvider(prev);
668
+ if (sanitized) providers.push(sanitized);
590
669
  }
591
670
  // 계정 식별 실패 시 마지막으로 알려진 계정 해시 유지 (계정 미상('')과 실제 계정 행이 갈라지는 것 방지)
592
671
  for (const p of providers) {
@@ -629,6 +708,7 @@ module.exports = {
629
708
  normalizeResetAt,
630
709
  parseCodexBarJSON,
631
710
  resolveMode,
711
+ sanitizeCachedProvider,
632
712
  titleCaseProvider,
633
713
  usefulPlan,
634
714
  };
@@ -0,0 +1,60 @@
1
+ #!/bin/bash
2
+ # Linux/WSL에 수집기를 등록한다 (5분 간격).
3
+ # systemd 사용자 세션이 있으면 타이머로 등록하고, 없으면(WSL에서 흔하다) cron 한 줄을 안내한다.
4
+ set -euo pipefail
5
+
6
+ SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
7
+ LOG_DIR="${CHARGE_HOME:-$HOME/.charge}"
8
+ LOG="$LOG_DIR/collector.log"
9
+
10
+ NODE="$(command -v node || true)"
11
+ if [ -z "$NODE" ]; then
12
+ echo "Node.js를 찾을 수 없습니다."
13
+ echo "배포판 패키지나 https://nodejs.org 로 설치한 뒤 다시 실행해주세요."
14
+ exit 1
15
+ fi
16
+
17
+ mkdir -p "$LOG_DIR"
18
+
19
+ # systemctl 이 있어도 사용자 세션이 안 뜬 경우가 많다 (WSL, 컨테이너, SSH) — 실제로 물어본다
20
+ if command -v systemctl >/dev/null 2>&1 && systemctl --user show-environment >/dev/null 2>&1; then
21
+ UNIT_DIR="$HOME/.config/systemd/user"
22
+ mkdir -p "$UNIT_DIR"
23
+
24
+ cat > "$UNIT_DIR/charge-connect.service" <<EOF
25
+ [Unit]
26
+ Description=Charge usage collector
27
+
28
+ [Service]
29
+ Type=oneshot
30
+ ExecStart="$NODE" "$SCRIPT_DIR/collect.js" --log "$LOG"
31
+ EOF
32
+
33
+ cat > "$UNIT_DIR/charge-connect.timer" <<EOF
34
+ [Unit]
35
+ Description=Charge usage collector (5분 간격)
36
+
37
+ [Timer]
38
+ OnBootSec=1min
39
+ OnUnitActiveSec=5min
40
+ Persistent=true
41
+
42
+ [Install]
43
+ WantedBy=timers.target
44
+ EOF
45
+
46
+ systemctl --user daemon-reload
47
+ systemctl --user enable --now charge-connect.timer
48
+ echo "등록 완료: charge-connect.timer (5분 간격). 로그: $LOG"
49
+ echo "해제하려면: systemctl --user disable --now charge-connect.timer"
50
+ exit 0
51
+ fi
52
+
53
+ # 여기까지 왔으면 자동 등록은 못 한다. 종료 코드 3으로 알려 cli.js가 안내를 덧붙이게 한다.
54
+ echo "systemd 사용자 세션을 찾지 못했습니다 (WSL에서는 흔한 일입니다)."
55
+ echo "아래 한 줄을 crontab에 넣으면 5분마다 수집합니다:"
56
+ echo
57
+ echo " */5 * * * * '$NODE' '$SCRIPT_DIR/collect.js' --log '$LOG'"
58
+ echo
59
+ echo "등록: crontab -e (cron이 안 돌고 있으면: sudo service cron start)"
60
+ exit 3
package/install.ps1 CHANGED
@@ -1,18 +1,36 @@
1
- # Charge 수집기를 Windows 작업 스케줄러에 등록한다 (5분 간격).
1
+ # Charge 수집기를 Windows 작업 스케줄러에 등록한다 (5분 간격).
2
2
  # 실행: PowerShell에서 .\install.ps1 (실행 정책 오류 시: powershell -ExecutionPolicy Bypass -File install.ps1)
3
+ #
4
+ # 이 파일은 UTF-8 BOM으로 저장해야 한다 — cli.js가 부르는 Windows PowerShell 5.1은
5
+ # BOM이 없으면 시스템 ANSI 코드페이지로 읽어서 아래 한글 안내가 전부 깨진다.
3
6
  $ErrorActionPreference = "Stop"
4
7
 
5
8
  $taskName = "ChargeConnect"
6
- $scriptDir = Split-Path -Parent $MyInvocation.MyCommand.Path
7
- $node = (Get-Command node).Source
8
- $log = Join-Path $env:USERPROFILE "charge-connect.log"
9
+ $scriptDir = if ($PSScriptRoot) { $PSScriptRoot } else { Split-Path -Parent $MyInvocation.MyCommand.Path }
9
10
 
10
- # 예전 이름(ChargeCollector)으로 등록된 잔재가 있으면 정리
11
+ # Node가 없을 때 PowerShell 원문 에러 대신 사람이 읽을 수 있는 안내를 낸다
12
+ $nodeCmd = Get-Command node -ErrorAction SilentlyContinue
13
+ if (-not $nodeCmd) {
14
+ Write-Host "Node.js를 찾을 수 없습니다."
15
+ Write-Host "https://nodejs.org 에서 LTS 버전을 설치한 뒤 다시 실행해주세요."
16
+ exit 1
17
+ }
18
+ $node = $nodeCmd.Source
19
+
20
+ $logDir = if ($env:CHARGE_HOME) { $env:CHARGE_HOME } else { Join-Path $env:USERPROFILE ".charge" }
21
+ New-Item -ItemType Directory -Force -Path $logDir | Out-Null
22
+ $log = Join-Path $logDir "collector.log"
23
+ $logOld = $log + ".old"
24
+
25
+ # 예전 이름(ChargeCollector)·예전 로그 위치의 잔재가 있으면 정리
11
26
  Unregister-ScheduledTask -TaskName "ChargeCollector" -Confirm:$false -ErrorAction SilentlyContinue
27
+ Remove-Item -Force -ErrorAction SilentlyContinue (Join-Path $env:USERPROFILE "charge-connect.log")
12
28
 
13
- # cmd /c 로 감싸 로그 리다이렉션 (작업 스케줄러는 자체 리다이렉션이 없음)
14
- $arg = "/c `"`"$node`" `"$scriptDir\collect.js`" >> `"$log`" 2>&1`""
15
- $action = New-ScheduledTaskAction -Execute "cmd.exe" -Argument $arg
29
+ # node 를 직접 실행하고 로그는 collect.js 가 --log 로 직접 쓴다.
30
+ # cmd /c 로 감싸 리다이렉션하면 경로 안의 %VAR% 가 확장되고, move 실패가 조용히 묻히고,
31
+ # 실행이 겹칠 때 로그 잠금으로 출력이 통째로 사라진다. 셸을 안 거치면 그 셋이 모두 없어진다.
32
+ $arg = "`"$scriptDir\collect.js`" --log `"$log`""
33
+ $action = New-ScheduledTaskAction -Execute $node -Argument $arg -WorkingDirectory $scriptDir
16
34
  $trigger = New-ScheduledTaskTrigger -Once -At (Get-Date) `
17
35
  -RepetitionInterval (New-TimeSpan -Minutes 5) `
18
36
  -RepetitionDuration (New-TimeSpan -Days 3650)
@@ -20,8 +38,74 @@ $settings = New-ScheduledTaskSettingsSet -StartWhenAvailable `
20
38
  -AllowStartIfOnBatteries -DontStopIfGoingOnBatteries `
21
39
  -ExecutionTimeLimit (New-TimeSpan -Minutes 10)
22
40
 
23
- Register-ScheduledTask -TaskName $taskName -Action $action -Trigger $trigger -Settings $settings -Force | Out-Null
24
- Start-ScheduledTask -TaskName $taskName
41
+ function Register-ChargeTask([switch]$Background) {
42
+ $p = @{ TaskName = $taskName; Action = $action; Trigger = $trigger; Settings = $settings; Force = $true }
43
+ if ($Background) {
44
+ # 기본값(LogonType Interactive)으로 등록하면 콘솔 창이 5분마다 화면에 뜬다.
45
+ # S4U는 대화형 세션 없이 실행돼 창이 뜨지 않고, 로그오프 상태에서도 수집이 돈다.
46
+ $p.Principal = New-ScheduledTaskPrincipal -UserId "$env:USERDOMAIN\$env:USERNAME" `
47
+ -LogonType S4U -RunLevel Limited
48
+ }
49
+ Register-ScheduledTask @p | Out-Null
50
+ }
51
+
52
+ function Get-LogState {
53
+ $s = ""
54
+ foreach ($f in @($log, $logOld)) {
55
+ # Test-Path 로 확인하고 Get-Item 을 따로 부르면, 그 사이 수집기가 로그를 밀었을 때
56
+ # ErrorActionPreference='Stop' 에 걸려 설치가 통째로 죽는다. 한 번에 가져온다.
57
+ $i = Get-Item -LiteralPath $f -ErrorAction SilentlyContinue
58
+ if ($i) { $s += "$($i.Length):$($i.LastWriteTime.Ticks);" } else { $s += "-;" }
59
+ }
60
+ return $s
61
+ }
62
+
63
+ # 등록에 성공해도 실행은 실패할 수 있다. S4U 로그온에는 '일괄 작업으로 로그온'
64
+ # (SeBatchLogonRight) 권한이 필요한데, 비관리자 계정이나 GPO로 제한된 PC에서는
65
+ # 등록만 통과하고 실행이 매번 0x80070569로 죽는다. 그때는 프로세스 자체가 안 떠서
66
+ # 로그도 안 남기 때문에, 확인하지 않으면 수집이 조용히 영영 멈춘다.
67
+ # 그래서 실제로 한 번 돌려보고 프로세스가 떴다는 증거(로그 파일 변화)를 확인한다.
68
+ function Test-ChargeTaskRuns {
69
+ # 예전 정의로 돌던 인스턴스가 남아 있으면 그쪽 출력을 새 작업의 성공으로 오인한다
70
+ Stop-ScheduledTask -TaskName $taskName -ErrorAction SilentlyContinue
71
+ Start-Sleep -Milliseconds 500
72
+ $before = Get-LogState
73
+ Start-ScheduledTask -TaskName $taskName
74
+ $deadline = (Get-Date).AddSeconds(60)
75
+ while ((Get-Date) -lt $deadline) {
76
+ # collect.js 는 뜨자마자 로그에 시작 줄을 남긴다 — 프로세스가 떴다는 직접 증거.
77
+ # '실행 중(0x41301)' 상태만으로 성공 처리하면 남아 있던 인스턴스에 속을 수 있다.
78
+ if ((Get-LogState) -ne $before) { return $true }
79
+ $r = (Get-ScheduledTaskInfo -TaskName $taskName).LastTaskResult
80
+ # 0x80070569 요청한 로그온 유형이 부여되지 않음 / 0x8007052E 로그온 실패
81
+ if ($r -eq 2147943785 -or $r -eq 2147943726) { return $false }
82
+ Start-Sleep -Milliseconds 500
83
+ }
84
+ return $false
85
+ }
86
+
87
+ $background = $true
88
+ try {
89
+ Register-ChargeTask -Background
90
+ } catch {
91
+ Write-Host "백그라운드 실행 등록 실패: $($_.Exception.Message)"
92
+ $background = $false
93
+ }
94
+ if ($background -and -not (Test-ChargeTaskRuns)) {
95
+ Write-Host "백그라운드 실행 권한이 없어 대화형 모드로 되돌립니다."
96
+ $background = $false
97
+ }
98
+ if (-not $background) {
99
+ try {
100
+ Register-ChargeTask
101
+ Start-ScheduledTask -TaskName $taskName
102
+ } catch {
103
+ Write-Host "작업 스케줄러 등록에 실패했습니다: $($_.Exception.Message)"
104
+ Write-Host "수동으로 수집하려면: node `"$scriptDir\collect.js`""
105
+ exit 1
106
+ }
107
+ Write-Host "참고: 5분마다 콘솔 창이 잠깐 보일 수 있습니다 (백그라운드 실행 권한이 없는 환경)."
108
+ }
25
109
 
26
110
  Write-Host "등록 완료: $taskName (5분 간격). 로그: $log"
27
111
  Write-Host "해제하려면: Unregister-ScheduledTask -TaskName $taskName -Confirm:`$false"
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "charge-connect",
3
- "version": "0.1.0",
3
+ "version": "0.1.2",
4
4
  "description": "Connects your computer to the Charge iPhone app and uploads AI coding usage",
5
5
  "bin": {
6
6
  "charge-connect": "cli.js"
@@ -12,6 +12,7 @@
12
12
  "cli.js",
13
13
  "collect.js",
14
14
  "install.sh",
15
+ "install.linux.sh",
15
16
  "install.ps1",
16
17
  "cloud.json"
17
18
  ],