feat: 实现 Sub2API 每日抽奖与排行榜

scripts/src/cli.ts

@@ -0,0 +1,136 @@
+import { AdminHttpClient } from "../../src/admin-http-client";
+import { createEmbeddedContext } from "../../src/bootstrap";
+import { loadConfig, type EmbeddedCliTarget, type HttpCliTarget } from "../../src/config";
+
+interface Parsed {
+  configPath: string;
+  targetId: string | null;
+  command: string[];
+  confirm: boolean;
+  includeRecords: boolean;
+  id: string | null;
+  limit: number | null;
+  draws: number | null;
+}
+
+function value(args: string[], name: string): string | null {
+  const index = args.indexOf(name);
+  if (index < 0) return null;
+  const result = args[index + 1];
+  if (!result || result.startsWith("--")) throw new Error(`${name} requires a value`);
+  return result;
+}
+
+function parseArgs(args: string[]): Parsed {
+  const configPath = value(args, "--config");
+  if (!configPath) throw new Error("--config is required");
+  const optionNames = new Set(["--config", "--target", "--id", "--limit", "--draws"]);
+  const flags = new Set(["--confirm", "--include-records"]);
+  const command: string[] = [];
+  for (let index = 0; index < args.length; index += 1) {
+    const item = args[index]!;
+    if (optionNames.has(item)) { index += 1; continue; }
+    if (flags.has(item)) continue;
+    if (item.startsWith("--")) throw new Error(`unknown option ${item}`);
+    command.push(item);
+  }
+  const integer = (name: string): number | null => {
+    const raw = value(args, name);
+    if (raw === null) return null;
+    const parsed = Number(raw);
+    if (!Number.isInteger(parsed) || parsed < 0) throw new Error(`${name} must be a non-negative integer`);
+    return parsed;
+  };
+  return {
+    configPath,
+    targetId: value(args, "--target"),
+    command,
+    confirm: args.includes("--confirm"),
+    includeRecords: args.includes("--include-records"),
+    id: value(args, "--id"),
+    limit: integer("--limit"),
+    draws: integer("--draws"),
+  };
+}
+
+function help(): Record<string, unknown> {
+  return {
+    ok: true,
+    usage: "bun scripts/sub2rank-cli.ts --config config/sub2rank.yaml [--target local|production] <command>",
+    commands: [
+      "config validate",
+      "backend check",
+      "lottery status",
+      "lottery draw [--confirm]",
+      "lottery reset [--draws N] [--include-records] --confirm",
+      "records list [--limit N]",
+      "records delete --id <record-id> --confirm",
+      "credit test [--confirm]",
+    ],
+  };
+}
+
+async function embedded(parsed: Parsed, config: ReturnType<typeof loadConfig>, target: EmbeddedCliTarget): Promise<unknown> {
+  const context = createEmbeddedContext(config, target);
+  try {
+    const [group, action] = parsed.command;
+    if (group === "backend" && action === "check") return await context.service.status(true);
+    if (group === "lottery" && action === "status") return await context.service.status(false);
+    if (group === "lottery" && action === "draw") return parsed.confirm ? { ok: true, record: await context.service.draw() } : { ok: true, mutation: false, action: "lottery-draw", hint: "add --confirm to execute" };
+    if (group === "lottery" && action === "reset") {
+      const draws = parsed.draws ?? config.lottery.initialDrawCount;
+      return parsed.confirm ? context.service.reset(draws, parsed.includeRecords) : { ok: true, mutation: false, action: "lottery-reset", draws, includeRecords: parsed.includeRecords, hint: "add --confirm to execute" };
+    }
+    if (group === "records" && action === "list") return { ok: true, records: context.service.listRecords(parsed.limit ?? config.records.publicLimit) };
+    if (group === "records" && action === "delete") {
+      if (!parsed.id) throw new Error("records delete requires --id");
+      return parsed.confirm ? context.service.deleteRecord(parsed.id) : { ok: true, mutation: false, action: "record-delete", id: parsed.id, hint: "add --confirm to execute" };
+    }
+    if (group === "credit" && action === "test") return await context.service.creditTest(parsed.confirm);
+    throw new Error(`unknown command: ${parsed.command.join(" ")}`);
+  } finally {
+    context.close();
+  }
+}
+
+async function remote(parsed: Parsed, config: ReturnType<typeof loadConfig>, target: HttpCliTarget): Promise<unknown> {
+  const client = new AdminHttpClient(config, target);
+  const [group, action] = parsed.command;
+  if (group === "backend" && action === "check") return await client.backendCheck();
+  if (group === "lottery" && action === "status") return await client.status();
+  if (group === "lottery" && action === "draw") return parsed.confirm ? await client.draw() : { ok: true, mutation: false, action: "lottery-draw", hint: "add --confirm to execute" };
+  if (group === "lottery" && action === "reset") {
+    const draws = parsed.draws ?? config.lottery.initialDrawCount;
+    return parsed.confirm ? await client.reset(draws, parsed.includeRecords) : { ok: true, mutation: false, action: "lottery-reset", draws, includeRecords: parsed.includeRecords, hint: "add --confirm to execute" };
+  }
+  if (group === "records" && action === "list") return await client.records(parsed.limit ?? config.records.publicLimit);
+  if (group === "records" && action === "delete") {
+    if (!parsed.id) throw new Error("records delete requires --id");
+    return parsed.confirm ? await client.deleteRecord(parsed.id) : { ok: true, mutation: false, action: "record-delete", id: parsed.id, hint: "add --confirm to execute" };
+  }
+  if (group === "credit" && action === "test") return await client.creditTest(parsed.confirm);
+  throw new Error(`unknown command: ${parsed.command.join(" ")}`);
+}
+
+export async function runCli(args: string[]): Promise<void> {
+  try {
+    if (args.includes("--help") || args.length === 0) {
+      console.log(JSON.stringify(help(), null, 2));
+      return;
+    }
+    const parsed = parseArgs(args);
+    const config = loadConfig(parsed.configPath);
+    if (parsed.command.join(" ") === "config validate") {
+      console.log(JSON.stringify({ ok: true, configPath: config.configPath, kind: config.kind, automaticCreditEnabled: config.lottery.automaticCredit.enabled, excludedIdentities: config.lottery.eligibility.excludedIdentities }, null, 2));
+      return;
+    }
+    const targetId = parsed.targetId ?? config.runtime.defaultCliTarget;
+    const target = config.runtime.cliTargets[targetId];
+    if (!target) throw new Error(`runtime.cliTargets.${targetId} does not exist`);
+    const result = target.mode === "embedded" ? await embedded(parsed, config, target) : await remote(parsed, config, target);
+    console.log(JSON.stringify({ target: targetId, ...result as Record<string, unknown> }, null, 2));
+  } catch (error) {
+    console.error(JSON.stringify({ ok: false, error: error instanceof Error ? error.message : String(error) }, null, 2));
+    process.exitCode = 1;
+  }
+}

scripts/sub2rank-cli.ts

@@ -0,0 +1,4 @@
+#!/usr/bin/env bun
+import { runCli } from "./src/cli";
+
+await runCli(process.argv.slice(2));