yaohao 1.0.0

package/src/commands/init.js

@@ -0,0 +1,102 @@
+import { select, confirm } from '@inquirer/prompts';
+import { isInitialized, updateUser } from '../lib/config-manager.js';
+import { listImplemented, getCityLabel, listPlanned } from '../source/index.js';
+import { DEFAULT_CITY } from '../constants.js';
+import { output, success, error } from '../output.js';
+
+export function registerInitCommand(program) {
+  program
+    .command('init')
+    .description('初始化（设置默认城市 / 指标类型 / 申请人类型 / 通知渠道）')
+    .option('--city <city>', '默认城市（非交互）')
+    .option('--reg-type <type>', '指标类型 PTC / XNY（非交互）')
+    .option('--apply-type <type>', '申请人类型 person / family（非交互）')
+    .option('--notify <url>', '通知渠道 URL（可多次指定）', (val, acc) => {
+      acc.push(val); return acc;
+    }, [])
+    .option('-f, --force', '已有配置时强制覆盖')
+    .action(async (opts) => {
+      try {
+        const alreadyInit = isInitialized();
+        const nonInteractive = opts.city || opts.regType || opts.applyType || opts.notify.length > 0;
+
+        if (alreadyInit && nonInteractive && !opts.force) {
+          output(error('已有配置，加 -f 强制覆盖，或用 `yaohao set <key> <value>` / `yaohao notify` 修改单项'));
+          process.exitCode = 1;
+          return;
+        }
+
+        let city, regType, applyType, notifyUrls;
+
+        if (nonInteractive) {
+          city = opts.city || DEFAULT_CITY;
+          regType = opts.regType || 'PTC';
+          applyType = opts.applyType || 'person';
+          notifyUrls = opts.notify;
+        } else {
+          if (alreadyInit) {
+            const ok = await confirm({ message: '已有配置，是否覆盖？', default: false });
+            if (!ok) {
+              output(success(null, '已取消'));
+              return;
+            }
+          }
+          const implemented = listImplemented();
+          const planned = listPlanned();
+          city = await select({
+            message: '默认城市:',
+            choices: [
+              ...implemented.map((c) => ({ name: getCityLabel(c), value: c })),
+              ...planned.map((c) => ({
+                name: `${getCityLabel(c)}（v1 未实现，敬请期待）`,
+                value: c,
+                disabled: '（暂不可选）',
+              })),
+            ],
+            default: DEFAULT_CITY,
+          });
+          regType = await select({
+            message: '默认关注的指标类型:',
+            choices: [
+              { name: '普通指标', value: 'PTC' },
+              { name: '新能源指标', value: 'XNY' },
+            ],
+            default: 'PTC',
+          });
+          applyType = await select({
+            message: '申请人类型:',
+            choices: [
+              { name: '个人', value: 'person' },
+              { name: '家庭', value: 'family' },
+            ],
+            default: 'person',
+          });
+          notifyUrls = [];
+        }
+
+        updateUser({
+          default_city: city,
+          reg_type: regType,
+          apply_type: applyType,
+          notify_urls: notifyUrls,
+        });
+        output(success(
+          { default_city: city, reg_type: regType, apply_type: applyType, notify_count: notifyUrls.length },
+          [
+            '初始化完成。',
+            `  默认城市: ${getCityLabel(city)}`,
+            `  指标类型: ${regType === 'PTC' ? '普通指标' : '新能源指标'}`,
+            `  申请人类型: ${applyType === 'person' ? '个人' : '家庭'}`,
+            `  通知渠道: ${notifyUrls.length} 个${notifyUrls.length === 0 ? '（运行 `yaohao notify add <url>` 添加）' : ''}`,
+          ].join('\n'),
+        ));
+      } catch (err) {
+        if (err && err.name === 'ExitPromptError') {
+          output(success(null, '已取消'));
+          return;
+        }
+        output(error(`初始化失败: ${err.message}`));
+        process.exitCode = 1;
+      }
+    });
+}

package/src/commands/market.js

@@ -0,0 +1,40 @@
+import { getSource, listImplemented } from '../source/index.js';
+import { getUser } from '../lib/config-manager.js';
+import { DEFAULT_CITY } from '../constants.js';
+import { output, success, error } from '../output.js';
+
+function resolveCity(opts) {
+  if (opts.city) return opts.city;
+  const user = getUser();
+  return user?.default_city || DEFAULT_CITY;
+}
+
+export function registerMarketCommand(program) {
+  program
+    .command('market')
+    .description('当前形势：申请数、配置数、中签率')
+    .option('--city <city>', `城市 (${listImplemented().join('|')})`)
+    .option('--no-pdf', '跳过 PDF 解析（更快，但拿不到中签率）')
+    .option('--no-cache', '禁用本地缓存')
+    .option('--json', '原始 JSON 输出')
+    .action(async (opts) => {
+      try {
+        const city = resolveCity(opts);
+        const source = getSource(city);
+        const result = await source.extractMarketMetrics(opts);
+        if (!result.ok) {
+          output(error(result.error));
+          process.exitCode = 1;
+          return;
+        }
+        if (opts.json) {
+          console.log(JSON.stringify({ city, ...result.data }, null, 2));
+          return;
+        }
+        output(success({ city, ...result.data }, result.lines.join('\n')));
+      } catch (err) {
+        output(error(err.message));
+        process.exitCode = 1;
+      }
+    });
+}

package/src/commands/notify.js

@@ -0,0 +1,92 @@
+import { getUser, isInitialized, updateUser } from '../lib/config-manager.js';
+import { testNotify } from '../lib/notifier.js';
+import { output, success, error } from '../output.js';
+
+export function registerNotifyCommand(program) {
+  const notify = program.command('notify').description('通知渠道管理');
+
+  notify
+    .command('add <url>')
+    .description('添加通知渠道')
+    .action(async (url) => {
+      try {
+        if (!isInitialized()) {
+          output(error('尚未初始化，请先运行 yaohao init'));
+          process.exitCode = 1;
+          return;
+        }
+
+        const user = getUser();
+        const urls = user.notify_urls || [];
+
+        if (urls.includes(url)) {
+          output(success({ notify_urls: urls }, '该通知渠道已存在'));
+          return;
+        }
+
+        urls.push(url);
+        updateUser({ notify_urls: urls });
+        output(success({ notify_urls: urls }, `通知渠道已添加: ${url}`));
+      } catch (err) {
+        output(error(`添加通知渠道失败: ${err.message}`));
+        process.exitCode = 1;
+      }
+    });
+
+  notify
+    .command('remove <url>')
+    .description('移除通知渠道')
+    .action(async (url) => {
+      try {
+        if (!isInitialized()) {
+          output(error('尚未初始化，请先运行 yaohao init'));
+          process.exitCode = 1;
+          return;
+        }
+
+        const user = getUser();
+        const urls = (user.notify_urls || []).filter((u) => u !== url);
+        updateUser({ notify_urls: urls });
+        output(success({ notify_urls: urls }, `通知渠道已移除: ${url}`));
+      } catch (err) {
+        output(error(`移除通知渠道失败: ${err.message}`));
+        process.exitCode = 1;
+      }
+    });
+
+  notify
+    .command('test')
+    .description('发送测试通知')
+    .action(async () => {
+      try {
+        if (!isInitialized()) {
+          output(error('尚未初始化，请先运行 yaohao init'));
+          process.exitCode = 1;
+          return;
+        }
+
+        const user = getUser();
+        const urls = user.notify_urls || [];
+
+        if (urls.length === 0) {
+          output(error('未配置通知渠道，请先运行 yaohao notify add <url>'));
+          process.exitCode = 1;
+          return;
+        }
+
+        const results = await testNotify(urls);
+        const succeeded = results.filter((r) => r.status === 'fulfilled').length;
+        const failed = results.filter((r) => r.status === 'rejected').length;
+
+        output(
+          success(
+            { succeeded, failed, total: urls.length },
+            `测试通知已发送: ${succeeded} 成功, ${failed} 失败 (共 ${urls.length} 个渠道)`,
+          ),
+        );
+      } catch (err) {
+        output(error(`测试通知失败: ${err.message}`));
+        process.exitCode = 1;
+      }
+    });
+}

package/src/commands/result.js

@@ -0,0 +1,18 @@
+import { output, success } from '../output.js';
+
+// v1 范围：本人是否中签的查询请登录官网。开奖结果公告的订阅见 `yaohao watch result`。
+export function registerResultCommand(program) {
+  program
+    .command('result')
+    .description('（v1 不支持）查询本人是否中签')
+    .action(async () => {
+      output(success(
+        { supported: false },
+        [
+          '本人中签结果查询 v1 不支持（基于隐私和合规考虑）。',
+          '官方开奖公告订阅请用 `yaohao watch result`。',
+          '查个人结果请直接登录官网：https://apply.jtw.beijing.gov.cn/apply/',
+        ].join('\n'),
+      ));
+    });
+}

package/src/commands/set.js

@@ -0,0 +1,53 @@
+import { isInitialized, updateUser } from '../lib/config-manager.js';
+import { listImplemented } from '../source/index.js';
+import { output, success, error } from '../output.js';
+
+const VALID_KEYS = {
+  'default-city': {
+    configKey: 'default_city',
+    validValues: listImplemented(),
+    description: '默认城市',
+  },
+  'reg-type': {
+    configKey: 'reg_type',
+    validValues: ['PTC', 'XNY'],
+    description: '指标类型（PTC=普通指标，XNY=新能源指标）',
+  },
+  'apply-type': {
+    configKey: 'apply_type',
+    validValues: ['person', 'family'],
+    description: '申请人类型（person=个人，family=家庭）',
+  },
+};
+
+export function registerSetCommand(program) {
+  program
+    .command('set <key> <value>')
+    .description(`修改配置项（支持: ${Object.keys(VALID_KEYS).join(', ')}）`)
+    .action(async (key, value) => {
+      try {
+        if (!isInitialized()) {
+          output(error('尚未初始化，请先运行 yaohao init'));
+          process.exitCode = 1;
+          return;
+        }
+        const keyDef = VALID_KEYS[key];
+        if (!keyDef) {
+          const validKeys = Object.keys(VALID_KEYS).join(', ');
+          output(error(`不支持的配置项: ${key}，支持: ${validKeys}`));
+          process.exitCode = 1;
+          return;
+        }
+        if (!keyDef.validValues.includes(value)) {
+          output(error(`无效的值: ${value}，${keyDef.description}必须是: ${keyDef.validValues.join(' / ')}`));
+          process.exitCode = 1;
+          return;
+        }
+        updateUser({ [keyDef.configKey]: value });
+        output(success({ key, value }, `${keyDef.description}已设置为: ${value}`));
+      } catch (err) {
+        output(error(`设置失败: ${err.message}`));
+        process.exitCode = 1;
+      }
+    });
+}

package/src/commands/status.js

@@ -0,0 +1,17 @@
+import { output, success } from '../output.js';
+
+// v1 范围：仅做公开数据查询和提醒。本人账号状态查询请登录官网。
+export function registerStatusCommand(program) {
+  program
+    .command('status')
+    .description('（v1 不支持）查看本人申请编码状态、阶梯倍率')
+    .action(async () => {
+      output(success(
+        { supported: false },
+        [
+          '本人账号状态查询 v1 不支持（基于隐私和合规考虑，本工具不持有用户密码）。',
+          '请直接登录官网查看：https://apply.jtw.beijing.gov.cn/apply/',
+        ].join('\n'),
+      ));
+    });
+}

package/src/commands/waitlist.js

@@ -0,0 +1,17 @@
+import { output, success } from '../output.js';
+
+// v1 范围：本人新能源轮候位置查询请登录官网。新能源相关公告的订阅见 `yaohao watch result/policy`。
+export function registerWaitlistCommand(program) {
+  program
+    .command('waitlist')
+    .description('（v1 不支持）新能源轮候位置（前面还有多少人）')
+    .action(async () => {
+      output(success(
+        { supported: false },
+        [
+          '本人新能源轮候位置 v1 不支持（基于隐私和合规考虑）。',
+          '请直接登录官网查看：https://apply.jtw.beijing.gov.cn/apply/',
+        ].join('\n'),
+      ));
+    });
+}

package/src/commands/watch.js

@@ -0,0 +1,159 @@
+import fs from 'node:fs';
+import path from 'node:path';
+import { homedir } from 'node:os';
+import { getSource, listImplemented } from '../source/index.js';
+import { getUser, isInitialized } from '../lib/config-manager.js';
+import { notify } from '../lib/notifier.js';
+import { DEFAULT_CITY, CONFIG_DIR } from '../constants.js';
+import { output, success, error } from '../output.js';
+
+const SEEN_FILE = path.join(homedir(), CONFIG_DIR, 'cache', 'watch-seen.json');
+
+const UNSUPPORTED_TARGETS = {
+  renewal: '年度延期确认提醒',
+  expiry: '申请编码到期提醒',
+  ranking: '家庭排名变化通知',
+};
+
+function resolveCity(opts) {
+  if (opts.city) return opts.city;
+  const user = getUser();
+  return user?.default_city || DEFAULT_CITY;
+}
+
+function loadSeen() {
+  try {
+    return JSON.parse(fs.readFileSync(SEEN_FILE, 'utf-8'));
+  } catch {
+    return {};
+  }
+}
+function saveSeen(seen) {
+  fs.mkdirSync(path.dirname(SEEN_FILE), { recursive: true });
+  fs.writeFileSync(SEEN_FILE, JSON.stringify(seen, null, 2));
+}
+function fingerprint(item) {
+  return `${item.url}|${item.date}`;
+}
+
+async function runWatch(city, targetName, opts) {
+  if (UNSUPPORTED_TARGETS[targetName]) {
+    output(success(
+      { supported: false, target: targetName },
+      [
+        `${UNSUPPORTED_TARGETS[targetName]} v1 不支持（依赖本人账号登录态，基于隐私和合规考虑）。`,
+        '请直接登录官网查看本人状态。',
+      ].join('\n'),
+    ));
+    return;
+  }
+
+  const source = getSource(city);
+  const target = source.watchTargets?.[targetName];
+  if (!target) {
+    output(error(`${city} 暂不支持 watch ${targetName}`));
+    process.exitCode = 1;
+    return;
+  }
+
+  const seen = loadSeen();
+  const seenKey = `${city}_${targetName}`;
+  const seenForTarget = new Set(seen[seenKey] || []);
+  const isFirstRun = seenForTarget.size === 0;
+
+  let items;
+  try {
+    items = await target.fetch(opts);
+  } catch (err) {
+    output(error(`抓取失败: ${err.message}`));
+    process.exitCode = 1;
+    return;
+  }
+
+  const newItems = [];
+  for (const item of items) {
+    const fp = fingerprint(item);
+    if (seenForTarget.has(fp)) continue;
+    seenForTarget.add(fp);
+    newItems.push(item);
+  }
+
+  seen[seenKey] = Array.from(seenForTarget).slice(-200);
+  saveSeen(seen);
+
+  const shouldNotify = !isFirstRun && newItems.length > 0;
+
+  if (opts.json) {
+    console.log(JSON.stringify({ city, target: targetName, isFirstRun, newCount: newItems.length, items: newItems }, null, 2));
+    return;
+  }
+
+  const lines = [];
+  if (isFirstRun) {
+    lines.push(`[${city} ${target.desc}] 首次运行，已记录当前 ${items.length} 条历史（不推送），下次运行后开始检测新增`);
+  } else if (newItems.length === 0) {
+    lines.push(`[${city} ${target.desc}] 无新增内容`);
+  } else {
+    lines.push(`[${city} ${target.desc}] 检测到 ${newItems.length} 条新增：`);
+    for (const item of newItems) {
+      lines.push(`  ${item.date}  ${item.kind ? `[${item.kind}] ` : ''}${item.title}`);
+      lines.push(`    ${item.url}`);
+    }
+  }
+
+  let notified = 0;
+  if (shouldNotify && opts.notify !== false && isInitialized()) {
+    const user = getUser();
+    const notifyUrls = user.notify_urls || [];
+    if (notifyUrls.length > 0) {
+      const title = `[yaohao] ${city} ${target.desc}：${newItems.length} 条新增`;
+      const body = newItems.map((i) => `${i.date} ${i.title}\n${i.url}`).join('\n\n');
+      try {
+        await notify(notifyUrls, title, body);
+        notified = notifyUrls.length;
+        lines.push('');
+        lines.push(`已推送到 ${notified} 个通知渠道`);
+      } catch (err) {
+        lines.push('');
+        lines.push(`(推送失败: ${err.message})`);
+      }
+    }
+  }
+
+  output(success(
+    { city, target: targetName, isFirstRun, newCount: newItems.length, items: newItems, notified },
+    lines.join('\n'),
+  ));
+}
+
+export function registerWatchCommand(program) {
+  const watch = program.command('watch').description('订阅/提醒类命令');
+
+  const supportedTargets = [
+    { name: 'result', desc: '开奖结果' },
+    { name: 'policy', desc: '政策变化' },
+    { name: 'window', desc: '申请窗口开放' },
+  ];
+
+  for (const t of supportedTargets) {
+    watch
+      .command(t.name)
+      .description(t.desc)
+      .option('--city <city>', `城市 (${listImplemented().join('|')})`)
+      .option('--no-cache', '禁用缓存，强制重新抓取')
+      .option('--no-notify', '不推送通知，只输出到 stdout')
+      .option('--json', '原始 JSON 输出')
+      .action(async (opts) => {
+        await runWatch(resolveCity(opts), t.name, opts);
+      });
+  }
+
+  for (const [name, desc] of Object.entries(UNSUPPORTED_TARGETS)) {
+    watch
+      .command(name)
+      .description(`（v1 不支持）${desc}`)
+      .action(async () => {
+        await runWatch(null, name, {});
+      });
+  }
+}

package/src/constants.js

@@ -0,0 +1,18 @@
+// 全局通用常量（非城市特定）。城市特定的 URL / 规则放在 src/source/{city}/ 下。
+
+export const CONFIG_DIR = '.yaohao';
+export const CONFIG_FILE = 'config.json';
+
+export const DEFAULT_CITY = 'beijing';
+
+// 指标类型（适用于多数城市，少数城市可能有不同分类，由 source 自己处理）
+export const REG_TYPE_MAP = {
+  PTC: '普通指标',
+  XNY: '新能源指标',
+};
+
+// 申请人类型
+export const APPLY_TYPE_MAP = {
+  person: '个人',
+  family: '家庭',
+};

package/src/lib/config-manager.js

@@ -0,0 +1,67 @@
+import { existsSync, mkdirSync, readFileSync, writeFileSync } from 'node:fs';
+import { homedir } from 'node:os';
+import { join } from 'node:path';
+import { CONFIG_DIR, CONFIG_FILE, DEFAULT_CITY } from '../constants.js';
+
+export function getConfigDir() {
+  return join(homedir(), CONFIG_DIR);
+}
+
+export function getConfigPath() {
+  return join(getConfigDir(), CONFIG_FILE);
+}
+
+export function ensureConfigDir() {
+  const dir = getConfigDir();
+  if (!existsSync(dir)) {
+    mkdirSync(dir, { recursive: true });
+  }
+}
+
+export function loadConfig() {
+  const configPath = getConfigPath();
+  if (!existsSync(configPath)) {
+    return null;
+  }
+  const raw = readFileSync(configPath, 'utf-8');
+  return JSON.parse(raw);
+}
+
+export function saveConfig(config) {
+  ensureConfigDir();
+  const configPath = getConfigPath();
+  writeFileSync(configPath, JSON.stringify(config, null, 4), 'utf-8');
+}
+
+function defaultUser() {
+  return {
+    default_city: DEFAULT_CITY,
+    reg_type: 'PTC',
+    apply_type: 'person',
+    notify_urls: [],
+  };
+}
+
+export function getUser() {
+  const config = loadConfig();
+  if (!config || !Array.isArray(config.users) || config.users.length === 0) {
+    return null;
+  }
+  return config.users[0];
+}
+
+export function updateUser(updates) {
+  let config = loadConfig();
+  if (!config) {
+    config = { users: [defaultUser()] };
+  }
+  if (!Array.isArray(config.users) || config.users.length === 0) {
+    config.users = [defaultUser()];
+  }
+  Object.assign(config.users[0], updates);
+  saveConfig(config);
+}
+
+export function isInitialized() {
+  return loadConfig() !== null;
+}