codex-work-receipt 0.7.4 → 0.8.1

package/src/cli.mjs

@@ -1,45 +1,33 @@
 #!/usr/bin/env node
 
 import { spawn } from "node:child_process";
-import fs from "node:fs";
 import path from "node:path";
 import { fileURLToPath } from "node:url";
 import QRCode from "qrcode";
 
 import { parseArgs, printHelp } from "./core/args.mjs";
-import { buildCanonicalFacts } from "./core/fact-buckets.mjs";
-import { collectMetrics } from "./core/metrics.mjs";
+import {
+  automaticOutputPath,
+  configureManualMode,
+  enableAutomaticMode,
+  getAutomaticStatus,
+  getWorkReceiptHome,
+  readAutoConfig,
+  startAutomaticRun,
+} from "./core/auto-mode.mjs";
+import { generateReceipt } from "./core/generator.mjs";
+import { promptForGenerationMode } from "./core/mode-selector.mjs";
 import { printOpenSourcePrompt } from "./core/open-source.mjs";
 import { getRollingSummaryNotice, getScopeLabel } from "./core/presentation.mjs";
-import { encodeReceiptPayloads } from "./core/qr-payload.mjs";
-import { outputSlugForRange, resolveRange } from "./core/range.mjs";
-import { buildReceiptRecord, persistReceiptRecord } from "./core/receipt-record.mjs";
+import { encodeSingleReceiptQr } from "./core/qr-payload.mjs";
 import { promptForRange } from "./core/selector.mjs";
 import { installCodexSkill } from "./core/skill-installer.mjs";
 import { installCodexPet, uninstallCodexPet } from "./core/pet-installer.mjs";
 import { formatNumber } from "./lib/time.mjs";
-import { listRecentCodexSessions, loadCodexSessions } from "./parsers/codex.mjs";
-import { renderHtml } from "./renderers/html.mjs";
+import { listRecentCodexSessions } from "./parsers/codex.mjs";
 
 const SCRIPT_DIR = path.dirname(fileURLToPath(import.meta.url));
 const PROJECT_DIR = path.dirname(SCRIPT_DIR);
-const DEFAULT_OUTPUT_DIR = path.join(process.cwd(), "codex-work-receipt-output");
-const DEFAULT_MINIPROGRAM_CODE = path.join(PROJECT_DIR, "assets", "miniprogram-code.png");
-
-function mimeTypeForImage(filePath) {
-  const extension = path.extname(filePath).toLowerCase();
-  if (extension === ".png") return "image/png";
-  if (extension === ".jpg" || extension === ".jpeg") return "image/jpeg";
-  if (extension === ".webp") return "image/webp";
-  if (extension === ".svg") return "image/svg+xml";
-  throw new Error(`不支持的小程序码图片格式：${extension || "未知"}`);
-}
-
-function imageAsDataUrl(filePath) {
-  if (!filePath || !fs.existsSync(filePath)) return null;
-  return `data:${mimeTypeForImage(filePath)};base64,${fs.readFileSync(filePath).toString("base64")}`;
-}
-
 function openFile(filePath) {
   const command = process.platform === "darwin" ? "open" : process.platform === "win32" ? "cmd" : "xdg-open";
   const args = process.platform === "win32" ? ["/c", "start", "", filePath] : [filePath];
@@ -47,12 +35,114 @@ function openFile(filePath) {
   child.unref();
 }
 
+function isInteractive() {
+  return Boolean(process.stdin.isTTY && process.stdout.isTTY);
+}
+
+function printAutomaticEnabled(result, locale) {
+  const outputFile = automaticOutputPath(result.config);
+  const importFile = outputFile.replace(/\.html?$/i, ".cwr.json");
+  if (locale === "en") {
+    console.log("Automatic saving enabled.");
+    console.log("Codex will quietly refresh today's receipt and WeChat import file whenever a turn stops.");
+    console.log(`Automatic receipt: ${outputFile}`);
+    console.log(`WeChat import file: ${importFile}`);
+    console.log("Restart Codex. If Codex asks you to review the new hook, open /hooks and trust AI Work Receipt.");
+  } else {
+    console.log("自动保存已启用。");
+    console.log("Codex 每完成一轮工作，都会静默刷新今天的小票和微信导入文件。");
+    console.log(`自动小票：${outputFile}`);
+    console.log(`微信导入文件：${importFile}`);
+    console.log("请重启 Codex；如果 Codex 提示审查新 Hook，请打开 /hooks 并信任 AI 打工小票。");
+  }
+}
+
+function printManualEnabled(result, locale) {
+  if (locale === "en") {
+    console.log("Manual-only mode enabled. No AI Work Receipt hook will run in the background.");
+    if (result.removedHook?.removed) console.log("The AI Work Receipt hook was removed; existing receipts were kept.");
+  } else {
+    console.log("已切换为仅手动模式，AI 打工小票不会在后台自动运行。");
+    if (result.removedHook?.removed) console.log("已移除 AI 打工小票 Hook，历史小票仍然保留。");
+  }
+}
+
+function enableAuto(options) {
+  const result = enableAutomaticMode({
+    projectDir: PROJECT_DIR,
+    dataDir: options.dataDir,
+    locale: options.locale,
+    timezone: options.timezone,
+    theme: options.theme,
+  });
+  startAutomaticRun(result.config);
+  printAutomaticEnabled(result, options.locale);
+  printOpenSourcePrompt("receipt", options.locale);
+  return result;
+}
+
+function enableManual(options, { showOpenSourcePrompt = true } = {}) {
+  const result = configureManualMode({
+    dataDir: options.dataDir,
+    locale: options.locale,
+    timezone: options.timezone,
+    theme: options.theme,
+  });
+  printManualEnabled(result, options.locale);
+  if (showOpenSourcePrompt) printOpenSourcePrompt("receipt", options.locale);
+  return result;
+}
+
+function printAutoStatus(options) {
+  const status = getAutomaticStatus({ dataDir: options.dataDir });
+  if (options.locale === "en") {
+    console.log(`Mode: ${status.mode}`);
+    console.log(`Local data: ${status.workReceiptHome}`);
+    console.log(`Runtime: ${status.runtimeInstalled ? "installed" : "not installed"}`);
+    console.log(`Codex hook: ${status.hookInstalled ? "installed" : "not installed"}`);
+    if (status.state?.last_success_at) console.log(`Last automatic receipt: ${status.state.last_success_at}`);
+    if (status.state?.import_file) console.log(`WeChat import file: ${status.state.import_file}`);
+    if (status.state?.error) console.log(`Last message: ${status.state.error}`);
+  } else {
+    const modeLabel = status.mode === "automatic" ? "自动保存" : status.mode === "manual" ? "仅手动" : "尚未设置";
+    console.log(`模式：${modeLabel}`);
+    console.log(`本地数据：${status.workReceiptHome}`);
+    console.log(`本地执行器：${status.runtimeInstalled ? "已安装" : "未安装"}`);
+    console.log(`Codex Hook：${status.hookInstalled ? "已安装" : "未安装"}`);
+    if (status.state?.last_success_at) console.log(`最近自动生成：${status.state.last_success_at}`);
+    if (status.state?.import_file) console.log(`微信导入文件：${status.state.import_file}`);
+    if (status.state?.error) console.log(`最近状态：${status.state.error}`);
+  }
+}
+
+async function runSetup(options) {
+  if (!isInteractive()) throw new Error("--setup 需要在交互式终端中运行；也可以使用 --enable-auto 或 --disable-auto");
+  const selected = await promptForGenerationMode({ locale: options.locale });
+  return selected === "automatic" ? enableAuto(options) : enableManual(options);
+}
+
 async function main() {
   const options = parseArgs(process.argv.slice(2));
   if (options.help) {
     printHelp(options.locale);
     return;
   }
+  if (options.setup) {
+    await runSetup(options);
+    return;
+  }
+  if (options.enableAuto) {
+    enableAuto(options);
+    return;
+  }
+  if (options.disableAuto) {
+    enableManual(options);
+    return;
+  }
+  if (options.autoStatus) {
+    printAutoStatus(options);
+    return;
+  }
   if (options.installSkill) {
     const installed = installCodexSkill({ projectDir: PROJECT_DIR });
     if (options.locale === "en") {
@@ -95,7 +185,16 @@ async function main() {
     return;
   }
 
-  if (!options.modeExplicit && process.stdin.isTTY && process.stdout.isTTY) {
+  if (!options.modeExplicit && isInteractive()) {
+    const workReceiptHome = getWorkReceiptHome({ dataDir: options.dataDir });
+    if (!readAutoConfig({ workReceiptHome })) {
+      const selectedMode = await promptForGenerationMode({ locale: options.locale });
+      if (selectedMode === "automatic") {
+        enableAuto(options);
+        return;
+      }
+      enableManual(options, { showOpenSourcePrompt: false });
+    }
     const selected = await promptForRange({
       locale: options.locale,
       timezone: options.timezone,
@@ -106,47 +205,44 @@ async function main() {
     options.hours = selected.hours || options.hours;
   }
 
-  const range = resolveRange(options.mode, options.timezone, new Date(), options.sessionId, options.hours);
-  const sessions = loadCodexSessions(range);
-  const metrics = collectMetrics(sessions, range);
-  const observedAt = new Date().toISOString();
-  const canonical = range.scope === "last-hours"
-    ? {}
-    : buildCanonicalFacts(sessions, range, { observedAt });
-  const record = buildReceiptRecord(metrics, options.theme, options.locale, canonical);
-  const qrPayloads = encodeReceiptPayloads(record);
-  const dataQrDataUrls = await Promise.all(qrPayloads.map((payload) => QRCode.toDataURL(payload, {
-    errorCorrectionLevel: "M",
-    margin: 2,
-    width: 360,
-    color: { dark: "#171713", light: "#ffffff" },
-  })));
-
-  const requestedOutput = options.output || path.join(
-    DEFAULT_OUTPUT_DIR,
-    `codex-receipt-${outputSlugForRange(range, record.id)}.html`,
-  );
-  const outputFile = path.resolve(/\.html?$/i.test(requestedOutput) ? requestedOutput : `${requestedOutput}.html`);
-  fs.mkdirSync(path.dirname(outputFile), { recursive: true });
-
-  const miniProgramCodeDataUrl = imageAsDataUrl(DEFAULT_MINIPROGRAM_CODE);
-
-  fs.writeFileSync(
+  const generated = await generateReceipt(options, {
+    projectDir: PROJECT_DIR,
+    createDataQr: async (record) => {
+      const dataQr = encodeSingleReceiptQr(record);
+      if (!dataQr) return null;
+      return {
+        version: dataQr.version,
+        dataUrl: await QRCode.toDataURL(dataQr.payload, {
+          errorCorrectionLevel: "M",
+          margin: 2,
+          width: 360,
+          color: { dark: "#171713", light: "#ffffff" },
+        }),
+      };
+    },
+  });
+  const {
+    record,
     outputFile,
-    renderHtml({ record, dataQrDataUrls, miniProgramCodeDataUrl }),
-    "utf8",
-  );
-  const persisted = persistReceiptRecord(record, outputFile, options.dataDir);
+    persisted,
+    dataQrDataUrl,
+    dataQrVersion,
+    miniProgramCodeDataUrl,
+  } = generated;
 
   if (options.locale === "en") {
     console.log(`Generated HTML: ${outputFile}`);
     console.log(`Structured data: ${persisted.companionPath}`);
+    console.log(`WeChat import file: ${persisted.transferPath}`);
     console.log(`Local history: ${persisted.receiptPath}`);
     console.log(`Range: ${getScopeLabel(record.source.scope, options.locale, record.source.hours)} · ${record.stats.session_count} session(s)`);
     console.log(`Stats: ${record.stats.completed_turns} turns · ${formatNumber(record.stats.tokens.total_tokens, options.locale)} Tokens · ${record.stats.tool_calls} tool calls`);
+    console.log(dataQrDataUrl
+      ? `Data QR: available as one code · QR version ${dataQrVersion}`
+      : "Data QR: not generated because the payload exceeds one code; use the WeChat import file");
     console.log(record.manifest
-      ? `Data QR: ${qrPayloads.length} code(s) · ${record.manifest.fact_count} canonical fact(s) · schema v${record.schema_version}`
-      : `Data QR: ${qrPayloads.length} code(s) · rolling summary · schema v${record.schema_version}`);
+      ? `Import data: ${record.manifest.fact_count} canonical fact(s) · schema v${record.schema_version}`
+      : `Import data: rolling summary · schema v${record.schema_version}`);
     if (record.source.scope === "last-hours") {
       console.log(`Note: ${getRollingSummaryNotice(options.locale, record.source.hours)}`);
     }
@@ -154,12 +250,16 @@ async function main() {
   } else {
     console.log(`已生成网页：${outputFile}`);
     console.log(`结构数据：${persisted.companionPath}`);
+    console.log(`微信导入文件：${persisted.transferPath}`);
     console.log(`本地历史：${persisted.receiptPath}`);
     console.log(`统计范围：${getScopeLabel(record.source.scope, options.locale, record.source.hours)} · ${record.stats.session_count} 个会话`);
     console.log(`统计：${record.stats.completed_turns} 轮 · ${formatNumber(record.stats.tokens.total_tokens, options.locale)} Token · ${record.stats.tool_calls} 次工具调用`);
+    console.log(dataQrDataUrl
+      ? `数据二维码：可用 · 单码 · QR version ${dataQrVersion}`
+      : "数据二维码：数据超过单码容量，已改用微信聊天文件导入");
     console.log(record.manifest
-      ? `数据二维码：${qrPayloads.length} 个 · ${record.manifest.fact_count} 条规范事实 · schema v${record.schema_version}`
-      : `数据二维码：${qrPayloads.length} 个 · 滚动摘要 · schema v${record.schema_version}`);
+      ? `导入数据：${record.manifest.fact_count} 条规范事实 · schema v${record.schema_version}`
+      : `导入数据：滚动摘要 · schema v${record.schema_version}`);
     if (record.source.scope === "last-hours") {
       console.log(`提示：${getRollingSummaryNotice(options.locale, record.source.hours)}`);
     }

package/src/core/args.mjs

@@ -13,6 +13,7 @@ Usage:
   npx codex-work-receipt@latest --range this-week --lang en
   npx codex-work-receipt@latest --install-skill --lang en
   npx codex-work-receipt@latest --install-companion --lang en
+  npx codex-work-receipt@latest --setup --lang en
 
 Options:
   --range <name>              Range: latest, last-hours, today, last-7-days, this-week
@@ -29,6 +30,10 @@ Options:
   --install-pet               Install the Codex pet only
   --uninstall-pet             Remove the installed Codex pet
   --install-companion         Install both the skill and Codex pet
+  --setup                     Choose automatic saving or manual-only mode
+  --enable-auto               Enable automatic daily receipt saving
+  --disable-auto              Switch to manual-only mode
+  --auto-status               Show automatic saving status
   --no-open                   Do not open the browser after generation
   --help                      Show help
 ` : `
@@ -43,6 +48,7 @@ Codex AI 打工小票
   npx codex-work-receipt@latest --range this-week
   npx codex-work-receipt@latest --install-skill
   npx codex-work-receipt@latest --install-companion
+  npx codex-work-receipt@latest --setup
 
 选项：
   --range <name>              统计范围：latest、last-hours、today、last-7-days、this-week
@@ -59,6 +65,10 @@ Codex AI 打工小票
   --install-pet               只安装 Codex 桌宠
   --uninstall-pet             卸载 AI 打工小票 Codex 桌宠
   --install-companion         同时安装 Skill 和 Codex 桌宠
+  --setup                     选择自动保存或仅手动模式
+  --enable-auto               启用自动保存今日小票
+  --disable-auto              切换为仅手动模式
+  --auto-status               查看自动保存状态
   --no-open                   生成后不自动打开浏览器
   --help                      显示帮助
 `);
@@ -79,6 +89,10 @@ export function parseArgs(argv) {
     installPet: false,
     uninstallPet: false,
     installCompanion: false,
+    setup: false,
+    enableAuto: false,
+    disableAuto: false,
+    autoStatus: false,
     open: true,
   };
 
@@ -111,6 +125,10 @@ export function parseArgs(argv) {
       result.mode = scope;
       result.modeExplicit = true;
     } else if (argument === "--install-skill") result.installSkill = true;
+    else if (argument === "--setup") result.setup = true;
+    else if (argument === "--enable-auto") result.enableAuto = true;
+    else if (argument === "--disable-auto") result.disableAuto = true;
+    else if (argument === "--auto-status") result.autoStatus = true;
     else if (argument === "--no-open") result.open = false;
     else if (argument === "--help" || argument === "-h") result.help = true;
     else if (optionsWithValues.has(argument)) {
@@ -134,6 +152,11 @@ export function parseArgs(argv) {
   if (!new Set(["zh-CN", "en"]).has(result.locale)) {
     throw new Error(`不支持的语言：${result.locale}`);
   }
+  try {
+    new Intl.DateTimeFormat("en-US", { timeZone: result.timezone }).format(new Date());
+  } catch {
+    throw new Error(`不支持的时区：${result.timezone}`);
+  }
   if (result.hours !== null) {
     result.hours = Number(result.hours);
     if (!Number.isInteger(result.hours) || result.hours < 1 || result.hours > 168) {
@@ -141,5 +164,16 @@ export function parseArgs(argv) {
     }
   }
   if (result.mode === "last-hours" && result.hours === null) result.hours = 3;
+  const managementActions = [
+    result.setup,
+    result.enableAuto,
+    result.disableAuto,
+    result.autoStatus,
+  ].filter(Boolean).length;
+  if (managementActions > 1) throw new Error("自动保存管理参数不能同时使用");
+  if (managementActions && result.modeExplicit) throw new Error("自动保存管理参数不能与统计范围参数同时使用");
+  if (managementActions && (
+    result.installSkill || result.installPet || result.uninstallPet || result.installCompanion
+  )) throw new Error("自动保存管理参数不能与 Skill 或桌宠管理参数同时使用");
   return result;
 }