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