43 lines
1.7 KiB
JavaScript
43 lines
1.7 KiB
JavaScript
#!/usr/bin/env node
|
|
/*
|
|
* 从远程 D1 的 raw_mails 拉取 manifest 里列出的真实邮件, 存成 test/fixtures/<name>.eml。
|
|
* .eml 含真实地址/令牌, 已 gitignore — 只存在于本地, 供 test:mail / test:eval 使用。
|
|
*
|
|
* 用法: pnpm run fixtures:pull (远程 D1, 需要 wrangler 登录)
|
|
* pnpm run fixtures:pull --local (本地 D1)
|
|
*/
|
|
import { execFileSync } from "node:child_process";
|
|
import { readFileSync, writeFileSync } from "node:fs";
|
|
import { dirname, join } from "node:path";
|
|
import { fileURLToPath } from "node:url";
|
|
|
|
const root = join(dirname(fileURLToPath(import.meta.url)), "..");
|
|
const fixturesDir = join(root, "test", "fixtures");
|
|
const manifest = JSON.parse(readFileSync(join(fixturesDir, "manifest.json"), "utf8"));
|
|
|
|
const where = process.argv.includes("--local") ? "--local" : "--remote";
|
|
const ids = manifest.fixtures.map((f) => f.rawMailId);
|
|
const sql = `SELECT id, raw FROM raw_mails WHERE id IN (${ids.join(",")})`;
|
|
|
|
console.log(`Fetching ${ids.length} fixtures from ${where} D1 ...`);
|
|
const out = execFileSync(
|
|
"npx",
|
|
["wrangler", "d1", "execute", "inbox-d1", where, "--json", "--command", sql],
|
|
{ cwd: root, encoding: "utf8", maxBuffer: 64 * 1024 * 1024 },
|
|
);
|
|
const rows = JSON.parse(out)[0].results;
|
|
const byId = new Map(rows.map((r) => [r.id, r.raw]));
|
|
|
|
let missing = 0;
|
|
for (const fixture of manifest.fixtures) {
|
|
const raw = byId.get(fixture.rawMailId);
|
|
if (!raw) {
|
|
console.warn(`✗ ${fixture.name}: raw_mails id ${fixture.rawMailId} not found`);
|
|
missing++;
|
|
continue;
|
|
}
|
|
writeFileSync(join(fixturesDir, `${fixture.name}.eml`), raw);
|
|
console.log(`✓ ${fixture.name}.eml (${raw.length} bytes)`);
|
|
}
|
|
process.exit(missing ? 1 : 0);
|