|
|
/**
|
|
|
* Playwright codegen headed 录制 Flock Freight
|
|
|
* 匿名 get-a-quote + Step 12.78 登录后查价（--logged-in）
|
|
|
*
|
|
|
* 用法:
|
|
|
* npm run record:flock
|
|
|
* npm run record:flock -- list
|
|
|
* npm run record:flock -- --logged-in
|
|
|
* npm run record:flock -- --logged-in https://app.flockfreight.com/login
|
|
|
*/
|
|
|
import { spawn } from "node:child_process";
|
|
|
import fs from "node:fs";
|
|
|
import path from "node:path";
|
|
|
import {
|
|
|
DEFAULT_FLOCK_DEMO,
|
|
|
DEFAULT_FLOCK_QUOTE_URL,
|
|
|
FLOCK_CANONICAL_PATHS,
|
|
|
formatFlockRecordingChecklist,
|
|
|
} from "@/workers/rpa/flock/demo-input";
|
|
|
|
|
|
const ROOT = process.cwd();
|
|
|
const OUTPUT_DIR = path.join(ROOT, ".rpa", "recordings");
|
|
|
|
|
|
export type FlockRecordMode = "anonymous" | "logged-in";
|
|
|
|
|
|
export type ParsedFlockRecordArgs = {
|
|
|
mode: FlockRecordMode;
|
|
|
startUrl?: string;
|
|
|
pathId?: string;
|
|
|
help: boolean;
|
|
|
list: boolean;
|
|
|
fresh: boolean;
|
|
|
reuseStorage: boolean;
|
|
|
};
|
|
|
|
|
|
export const ANON_FLOCK_STORAGE_DEFAULT = path.join(
|
|
|
ROOT,
|
|
|
".rpa",
|
|
|
"flock-storage.json",
|
|
|
);
|
|
|
export const LOGGED_IN_FLOCK_STORAGE_DEFAULT = path.join(
|
|
|
ROOT,
|
|
|
".rpa",
|
|
|
"flock-logged-in-storage.json",
|
|
|
);
|
|
|
|
|
|
export const DEFAULT_FLOCK_LOGIN_URL =
|
|
|
process.env.FLOCK_LOGIN_URL?.trim() ||
|
|
|
"https://app.flockfreight.com/login";
|
|
|
|
|
|
export function parseFlockRecordArgs(argv: string[]): ParsedFlockRecordArgs {
|
|
|
const tokens = argv.slice(2).filter(Boolean);
|
|
|
let mode: FlockRecordMode = "anonymous";
|
|
|
let help = false;
|
|
|
let list = false;
|
|
|
let fresh = false;
|
|
|
let reuseStorage = false;
|
|
|
let startUrl: string | undefined;
|
|
|
let pathId: string | undefined;
|
|
|
|
|
|
for (const token of tokens) {
|
|
|
if (token === "-h" || token === "--help") {
|
|
|
help = true;
|
|
|
continue;
|
|
|
}
|
|
|
if (token === "list") {
|
|
|
list = true;
|
|
|
continue;
|
|
|
}
|
|
|
if (token === "--logged-in" || token === "--login") {
|
|
|
mode = "logged-in";
|
|
|
continue;
|
|
|
}
|
|
|
if (token === "--fresh") {
|
|
|
fresh = true;
|
|
|
continue;
|
|
|
}
|
|
|
if (token === "--reuse-storage") {
|
|
|
reuseStorage = true;
|
|
|
continue;
|
|
|
}
|
|
|
if (token.startsWith("http://") || token.startsWith("https://")) {
|
|
|
startUrl = token;
|
|
|
continue;
|
|
|
}
|
|
|
if (!token.startsWith("-") && !pathId) {
|
|
|
pathId = token;
|
|
|
continue;
|
|
|
}
|
|
|
}
|
|
|
|
|
|
if (process.env.FLOCK_RECORD_FRESH === "true") {
|
|
|
fresh = true;
|
|
|
}
|
|
|
if (process.env.FLOCK_RECORD_REUSE_STORAGE === "true") {
|
|
|
reuseStorage = true;
|
|
|
}
|
|
|
|
|
|
return { mode, startUrl, pathId, help, list, fresh, reuseStorage };
|
|
|
}
|
|
|
|
|
|
export function resolveFlockStoragePath(mode: FlockRecordMode): string {
|
|
|
if (mode === "logged-in") {
|
|
|
return (
|
|
|
process.env.FLOCK_LOGGED_IN_STORAGE_STATE_PATH?.trim() ||
|
|
|
LOGGED_IN_FLOCK_STORAGE_DEFAULT
|
|
|
);
|
|
|
}
|
|
|
return (
|
|
|
process.env.FLOCK_STORAGE_STATE_PATH?.trim() || ANON_FLOCK_STORAGE_DEFAULT
|
|
|
);
|
|
|
}
|
|
|
|
|
|
export function resolveFlockOutputBasename(
|
|
|
mode: FlockRecordMode,
|
|
|
ts: string,
|
|
|
pathId?: string,
|
|
|
): string {
|
|
|
if (mode === "logged-in") {
|
|
|
return `flock-logged-in-${ts}.js`;
|
|
|
}
|
|
|
const suffix = pathId ?? "manual";
|
|
|
return `flock-${suffix}-${ts}.js`;
|
|
|
}
|
|
|
|
|
|
export function formatTimestamp(d = new Date()): string {
|
|
|
const pad = (n: number) => String(n).padStart(2, "0");
|
|
|
return `${d.getFullYear()}${pad(d.getMonth() + 1)}${pad(d.getDate())}-${pad(d.getHours())}${pad(d.getMinutes())}${pad(d.getSeconds())}`;
|
|
|
}
|
|
|
|
|
|
export function formatFlockLoggedInRecordingChecklist(): string {
|
|
|
const demo = DEFAULT_FLOCK_DEMO;
|
|
|
return [
|
|
|
"【Flock 登录后查价录制清单】",
|
|
|
"1. 打开登录页，手动粘贴 FLOCK_LOGIN_EMAIL / FLOCK_LOGIN_PASSWORD 并提交",
|
|
|
"2. 进入登录后查价入口（记录最终 URL；常见 Let's build a quote / Quick）",
|
|
|
`3. 标准样例：Pickup ZIP ${demo.pickupZip} → Delivery ZIP ${demo.deliveryZip}`,
|
|
|
`4. 托盘/重量/尺寸：${demo.palletCount} pallets / ${demo.totalWeightLb} lb / ${demo.lengthIn}×${demo.widthIn}×${demo.heightIn} in`,
|
|
|
"5. 填 Type / Packaging / Freight Class / Description 等登录后必填项",
|
|
|
"6. 点 Generate my quote（或等价提交），等到报价结果出现",
|
|
|
"7. 关闭 codegen，确认 flock-logged-in-*.js + flock-logged-in-storage.json 落盘",
|
|
|
].join("\n");
|
|
|
}
|
|
|
|
|
|
function hasFlockEnvCredentials(): boolean {
|
|
|
const email = process.env.FLOCK_LOGIN_EMAIL?.trim() ?? "";
|
|
|
const password = process.env.FLOCK_LOGIN_PASSWORD?.trim() ?? "";
|
|
|
return Boolean(email && password);
|
|
|
}
|
|
|
|
|
|
function printHelp(): void {
|
|
|
console.log(`
|
|
|
用法:
|
|
|
npm run record:flock
|
|
|
npm run record:flock -- list
|
|
|
npm run record:flock -- <path-id>
|
|
|
npm run record:flock -- <起始URL>
|
|
|
npm run record:flock -- --logged-in [起始URL]
|
|
|
npm run record:flock -- --login [起始URL]
|
|
|
|
|
|
模式:
|
|
|
匿名（默认） 录制 marketing get-a-quote；可用 --fresh 无痕
|
|
|
--logged-in 录制账号登录后查价完整路径；storage 与匿名隔离
|
|
|
|
|
|
默认入口:
|
|
|
匿名: ${DEFAULT_FLOCK_QUOTE_URL}
|
|
|
登录后: ${DEFAULT_FLOCK_LOGIN_URL}
|
|
|
|
|
|
产出（匿名）:
|
|
|
.rpa/recordings/flock-<path|manual>-<时间戳>.js
|
|
|
.rpa/flock-storage.json
|
|
|
|
|
|
产出（登录后）:
|
|
|
.rpa/recordings/flock-logged-in-<时间戳>.js
|
|
|
.rpa/flock-logged-in-storage.json
|
|
|
|
|
|
登录后录制说明:
|
|
|
- 请在 codegen 浏览器中手动粘贴账密（脚本不自动填）
|
|
|
- SOP: docs/rpa-flock-logged-in-sop.md
|
|
|
|
|
|
标准样例（匿名）:
|
|
|
${formatFlockRecordingChecklist(DEFAULT_FLOCK_DEMO)}
|
|
|
|
|
|
canonical 路径:
|
|
|
${FLOCK_CANONICAL_PATHS.map((p) => ` - ${p.id}: ${p.title}`).join("\n")}
|
|
|
`);
|
|
|
}
|
|
|
|
|
|
function resolveAnonymousStart(
|
|
|
arg: string | undefined,
|
|
|
): { startUrl: string; pathId?: string; demo: typeof DEFAULT_FLOCK_DEMO } {
|
|
|
if (!arg) {
|
|
|
return { startUrl: DEFAULT_FLOCK_QUOTE_URL, demo: DEFAULT_FLOCK_DEMO };
|
|
|
}
|
|
|
if (arg.startsWith("http://") || arg.startsWith("https://")) {
|
|
|
return { startUrl: arg, demo: DEFAULT_FLOCK_DEMO };
|
|
|
}
|
|
|
const canonical = FLOCK_CANONICAL_PATHS.find((p) => p.id === arg);
|
|
|
if (canonical) {
|
|
|
console.log(`\n路径: ${canonical.id} — ${canonical.title}`);
|
|
|
console.log(`说明: ${canonical.notes}\n`);
|
|
|
console.log(formatFlockRecordingChecklist(canonical.input));
|
|
|
console.log("");
|
|
|
return {
|
|
|
startUrl: DEFAULT_FLOCK_QUOTE_URL,
|
|
|
pathId: canonical.id,
|
|
|
demo: canonical.input,
|
|
|
};
|
|
|
}
|
|
|
console.warn(`未知 path-id「${arg}」，使用默认样例`);
|
|
|
return { startUrl: DEFAULT_FLOCK_QUOTE_URL, demo: DEFAULT_FLOCK_DEMO };
|
|
|
}
|
|
|
|
|
|
function main(): void {
|
|
|
const parsed = parseFlockRecordArgs(process.argv);
|
|
|
|
|
|
if (parsed.help) {
|
|
|
printHelp();
|
|
|
process.exit(0);
|
|
|
}
|
|
|
|
|
|
if (parsed.list) {
|
|
|
console.log("Flock canonical 录制路径:\n");
|
|
|
for (const p of FLOCK_CANONICAL_PATHS) {
|
|
|
console.log(` ${p.id}`);
|
|
|
console.log(` ${p.title}`);
|
|
|
console.log(` ${p.notes}\n`);
|
|
|
}
|
|
|
process.exit(0);
|
|
|
}
|
|
|
|
|
|
const mode = parsed.mode;
|
|
|
const storagePath = resolveFlockStoragePath(mode);
|
|
|
const ts = formatTimestamp();
|
|
|
|
|
|
fs.mkdirSync(OUTPUT_DIR, { recursive: true });
|
|
|
fs.mkdirSync(path.dirname(storagePath), { recursive: true });
|
|
|
|
|
|
let startUrl: string;
|
|
|
let pathId: string | undefined;
|
|
|
let demo = DEFAULT_FLOCK_DEMO;
|
|
|
let checklistBody: string;
|
|
|
|
|
|
if (mode === "logged-in") {
|
|
|
startUrl = parsed.startUrl ?? DEFAULT_FLOCK_LOGIN_URL;
|
|
|
checklistBody = formatFlockLoggedInRecordingChecklist();
|
|
|
} else {
|
|
|
const resolved = resolveAnonymousStart(parsed.startUrl ?? parsed.pathId);
|
|
|
startUrl = resolved.startUrl;
|
|
|
pathId = resolved.pathId;
|
|
|
demo = resolved.demo;
|
|
|
checklistBody = formatFlockRecordingChecklist(demo);
|
|
|
}
|
|
|
|
|
|
const outputFile = path.join(
|
|
|
OUTPUT_DIR,
|
|
|
resolveFlockOutputBasename(mode, ts, pathId),
|
|
|
);
|
|
|
const checklistFile = outputFile.replace(/\.js$/, ".checklist.txt");
|
|
|
fs.writeFileSync(checklistFile, checklistBody, "utf8");
|
|
|
|
|
|
const args = [
|
|
|
"playwright",
|
|
|
"codegen",
|
|
|
"--target=javascript",
|
|
|
"-o",
|
|
|
outputFile,
|
|
|
];
|
|
|
|
|
|
if (mode === "logged-in") {
|
|
|
// 登录态必须持久 session，禁止 --fresh 跳过 save
|
|
|
args.push(`--save-storage=${storagePath}`);
|
|
|
if (fs.existsSync(storagePath)) {
|
|
|
args.push(`--load-storage=${storagePath}`);
|
|
|
console.log(`已加载已有 storageState: ${storagePath}`);
|
|
|
}
|
|
|
} else if (parsed.fresh) {
|
|
|
console.log("【无痕录制 --fresh】不加载/保存 storage");
|
|
|
} else if (parsed.reuseStorage) {
|
|
|
args.push(`--save-storage=${storagePath}`);
|
|
|
if (fs.existsSync(storagePath)) {
|
|
|
args.push(`--load-storage=${storagePath}`);
|
|
|
console.log(`已加载已有 storageState: ${storagePath}`);
|
|
|
}
|
|
|
} else {
|
|
|
args.push(`--save-storage=${storagePath}`);
|
|
|
console.log(`录制结束将保存 storage: ${storagePath}`);
|
|
|
console.log("若已达报价上限，请使用: npm run record:flock -- --fresh");
|
|
|
}
|
|
|
|
|
|
args.push(startUrl);
|
|
|
|
|
|
if (mode === "logged-in") {
|
|
|
console.log("=== Flock Freight 登录后查价录制（headed）===");
|
|
|
console.log("RPA_HEADLESS=false");
|
|
|
console.log("模式: --logged-in（storage 与匿名 flock-storage.json 隔离）");
|
|
|
if (hasFlockEnvCredentials()) {
|
|
|
console.log(
|
|
|
"已检测到 .env 中 FLOCK_LOGIN_EMAIL/PASSWORD：请在浏览器中手动粘贴登录（脚本不自动填）",
|
|
|
);
|
|
|
} else {
|
|
|
console.log(
|
|
|
"未检测到 FLOCK_LOGIN_EMAIL/PASSWORD：请用管理端客户账密或准备官网账号后手动登录",
|
|
|
);
|
|
|
}
|
|
|
console.log(`起始 URL: ${startUrl}`);
|
|
|
console.log(`脚本产出: ${outputFile}`);
|
|
|
console.log(`清单文件: ${checklistFile}`);
|
|
|
console.log(`storageState: ${storagePath}`);
|
|
|
console.log("SOP: docs/rpa-flock-logged-in-sop.md");
|
|
|
} else {
|
|
|
console.log("=== Flock Freight RPA 录制（headed / 匿名）===");
|
|
|
console.log("RPA_HEADLESS=false");
|
|
|
console.log(`起始 URL: ${startUrl}`);
|
|
|
console.log(`脚本产出: ${outputFile}`);
|
|
|
console.log(`清单文件: ${checklistFile}`);
|
|
|
console.log(`storageState: ${storagePath}`);
|
|
|
console.log("登录后请用: npm run record:flock -- --logged-in");
|
|
|
}
|
|
|
console.log("");
|
|
|
console.log(checklistBody);
|
|
|
console.log("");
|
|
|
|
|
|
const child = spawn("npx", args, {
|
|
|
stdio: "inherit",
|
|
|
shell: true,
|
|
|
cwd: ROOT,
|
|
|
env: { ...process.env, RPA_HEADLESS: "false" },
|
|
|
});
|
|
|
|
|
|
child.on("error", (error) => {
|
|
|
console.error("启动 playwright codegen 失败:", error);
|
|
|
process.exit(1);
|
|
|
});
|
|
|
|
|
|
child.on("exit", (code) => {
|
|
|
if (code === 0) {
|
|
|
if (mode === "logged-in") {
|
|
|
console.log(
|
|
|
"\n录制结束。请回填 docs/rpa-flock-logged-in-sop.md 附录 A；可运行: npx tsx scripts/finalize-flock-logged-in-recording.ts",
|
|
|
);
|
|
|
} else {
|
|
|
console.log(
|
|
|
"\n录制结束。请将 flock-*.js 中的 selector 填入 .env FLOCK_SELECTOR_*",
|
|
|
);
|
|
|
}
|
|
|
}
|
|
|
process.exit(code ?? 0);
|
|
|
});
|
|
|
}
|
|
|
|
|
|
// vitest 导入本模块时勿启动 codegen
|
|
|
const isRecordEntrypoint = /record-flock\.(ts|js|mjs|cjs)$/.test(
|
|
|
process.argv[1] ?? "",
|
|
|
);
|
|
|
if (isRecordEntrypoint && !process.env.VITEST && !process.env.VITEST_WORKER_ID) {
|
|
|
main();
|
|
|
}
|
