1import { readFile, writeFile, mkdir } from "node:fs/promises";
2import { dirname, join } from "node:path";
3import { resolveChromeProfileDir } from "./profile";
4import type { TargetSession } from "./cdp-client";
5 
6export interface CdpCookie {
7  name: string;
8  value: string;
9  domain: string;
10  path: string;
11  expires: number;
12  size: number;
13  httpOnly: boolean;
14  secure: boolean;
15  session: boolean;
16  sameSite?: string;
17  priority?: string;
18  sameParty?: boolean;
19  sourceScheme?: string;
20  sourcePort?: number;
21  partitionKey?: string;
22}
23 
24interface SidecarData {
25  savedAt: string;
26  cookies: CdpCookie[];
27}
28 
29export interface CookieSidecarConfig {
30  urls: readonly string[];
31  filename: string;
32  requiredCookieNames: readonly string[];
33  filterCookie?: (cookie: CdpCookie) => boolean;
34}
35 
36function sidecarPath(filename: string, profileDir?: string): string {
37  return join(resolveChromeProfileDir(profileDir), filename);
38}
39 
40function hasRequired(cookies: CdpCookie[], names: readonly string[]): boolean {
41  return names.every((name) =>
42    cookies.some((c) => c.name === name && Boolean(c.value)),
43  );
44}
45 
46async function getCookies(session: TargetSession, urls: readonly string[]): Promise<CdpCookie[]> {
47  const { cookies } = await session.send<{ cookies: CdpCookie[] }>(
48    "Network.getCookies",
49    { urls: [...urls] },
50  );
51  return cookies ?? [];
52}
53 
54export async function exportCookies(
55  session: TargetSession,
56  config: CookieSidecarConfig,
57  profileDir?: string,
58): Promise<boolean> {
59  const all = await getCookies(session, config.urls);
60  const filtered = config.filterCookie ? all.filter(config.filterCookie) : all;
61  if (!hasRequired(filtered, config.requiredCookieNames)) return false;
62 
63  const filePath = sidecarPath(config.filename, profileDir);
64  await mkdir(dirname(filePath), { recursive: true });
65  const data: SidecarData = { savedAt: new Date().toISOString(), cookies: filtered };
66  await writeFile(filePath, JSON.stringify(data, null, 2));
67  return true;
68}
69 
70export async function restoreCookies(
71  session: TargetSession,
72  config: CookieSidecarConfig,
73  profileDir?: string,
74): Promise<boolean> {
75  const live = await getCookies(session, config.urls);
76  if (hasRequired(live, config.requiredCookieNames)) return false;
77 
78  const filePath = sidecarPath(config.filename, profileDir);
79  const raw = await readFile(filePath, "utf8");
80  const data = JSON.parse(raw) as SidecarData;
81  if (!data.cookies?.length) return false;
82 
83  const now = Date.now() / 1000;
84  const valid = data.cookies.filter((c) => c.session || !c.expires || c.expires > now);
85  if (!hasRequired(valid, config.requiredCookieNames)) return false;
86 
87  await session.send("Network.setCookies", {
88    cookies: valid.map((c) => ({
89      name: c.name,
90      value: c.value,
91      domain: c.domain,
92      path: c.path,
93      httpOnly: c.httpOnly,
94      secure: c.secure,
95      sameSite: c.sameSite,
96      expires: c.expires,
97    })),
98  });
99  return true;
100}
101