1import fs from 'node:fs';
2import path from 'node:path';
3import { mkdir, readFile, writeFile } from 'node:fs/promises';
4 
5import { resolveGeminiWebCookiePath } from './paths.js';
6 
7export type CookieMap = Record<string, string>;
8 
9export type CookieFileData =
10  | {
11      cookies: CookieMap;
12      updated_at: number;
13      source?: string;
14    }
15  | {
16      version: number;
17      updatedAt: string;
18      cookieMap: CookieMap;
19      source?: string;
20    };
21 
22export async function read_cookie_file(p: string = resolveGeminiWebCookiePath()): Promise<CookieMap | null> {
23  try {
24    if (!fs.existsSync(p) || !fs.statSync(p).isFile()) return null;
25    const raw = await readFile(p, 'utf8');
26    const data = JSON.parse(raw) as unknown;
27 
28    if (data && typeof data === 'object' && 'cookies' in (data as any)) {
29      const cookies = (data as any).cookies as unknown;
30      if (cookies && typeof cookies === 'object') {
31        const out: CookieMap = {};
32        for (const [k, v] of Object.entries(cookies as Record<string, unknown>)) {
33          if (typeof v === 'string') out[k] = v;
34        }
35        return out;
36      }
37    }
38 
39    if (data && typeof data === 'object' && 'cookieMap' in (data as any)) {
40      const cookies = (data as any).cookieMap as unknown;
41      if (cookies && typeof cookies === 'object') {
42        const out: CookieMap = {};
43        for (const [k, v] of Object.entries(cookies as Record<string, unknown>)) {
44          if (typeof v === 'string') out[k] = v;
45        }
46        return Object.keys(out).length > 0 ? out : null;
47      }
48    }
49 
50    if (data && typeof data === 'object') {
51      const out: CookieMap = {};
52      for (const [k, v] of Object.entries(data as Record<string, unknown>)) {
53        if (typeof v === 'string') out[k] = v;
54      }
55      return Object.keys(out).length > 0 ? out : null;
56    }
57 
58    return null;
59  } catch {
60    return null;
61  }
62}
63 
64export async function write_cookie_file(
65  cookies: CookieMap,
66  p: string = resolveGeminiWebCookiePath(),
67  source?: string,
68): Promise<void> {
69  const dir = path.dirname(p);
70  await mkdir(dir, { recursive: true });
71 
72  const payload: CookieFileData = {
73    version: 1,
74    updatedAt: new Date().toISOString(),
75    cookieMap: cookies,
76    source,
77  };
78  await writeFile(p, JSON.stringify(payload, null, 2), 'utf8');
79}
80 
81export const readCookieFile = read_cookie_file;
82export const writeCookieFile = write_cookie_file;
83