chajia/lib/axel/session.ts

import fs from "node:fs";
import { DEFAULT_RPA_STORAGE_STATE_PATH } from "@/lib/rpa/env";

export type PlaywrightStorageState = {
  cookies: Array<{
    name: string;
    value: string;
    domain?: string;
  }>;
};

export function resolveStorageStatePath(): string {
  return process.env.RPA_STORAGE_STATE_PATH?.trim() || DEFAULT_RPA_STORAGE_STATE_PATH;
}

export function loadStorageState(
  path = resolveStorageStatePath(),
): PlaywrightStorageState {
  if (!fs.existsSync(path)) {
    throw new Error(
      `缺少 MotherShip 会话文件 ${path}；系统将尝试自动获取，或运行 npm run record:mothership`,
    );
  }
  return JSON.parse(fs.readFileSync(path, "utf8")) as PlaywrightStorageState;
}

/** 缺会话时自动 bootstrap，再读取 storageState */
export async function loadStorageStateAsync(
  path = resolveStorageStatePath(),
): Promise<PlaywrightStorageState> {
  const { ensureMothershipStorageState } = await import(
    "@/lib/rpa/ensure-storage-state"
  );
  await ensureMothershipStorageState();
  return loadStorageState(path);
}

export function buildCookieHeader(state: PlaywrightStorageState): string {
  return state.cookies.map((c) => `${c.name}=${c.value}`).join("; ");
}