1import type { AdapterContext } from "../types";
2 
3const X_SESSION_URLS = ["https://x.com/", "https://twitter.com/"] as const;
4const REQUIRED_X_SESSION_COOKIES = ["auth_token", "ct0"] as const;
5 
6interface CookieLike {
7  name?: string;
8  value?: string | null;
9}
10 
11interface NetworkGetCookiesResult {
12  cookies?: CookieLike[];
13}
14 
15export function buildXSessionCookieMap(cookies: readonly CookieLike[]): Record<string, string> {
16  const cookieMap: Record<string, string> = {};
17  for (const cookie of cookies) {
18    const name = cookie.name?.trim();
19    const value = cookie.value?.trim();
20    if (!name || !value) {
21      continue;
22    }
23    cookieMap[name] = value;
24  }
25  return cookieMap;
26}
27 
28export function hasRequiredXSessionCookies(cookieMap: Record<string, string>): boolean {
29  return REQUIRED_X_SESSION_COOKIES.every((name) => Boolean(cookieMap[name]));
30}
31 
32export async function readXSessionCookieMap(
33  context: Pick<AdapterContext, "browser">,
34): Promise<Record<string, string>> {
35  const { cookies } = await context.browser.targetSession.send<NetworkGetCookiesResult>(
36    "Network.getCookies",
37    { urls: [...X_SESSION_URLS] },
38  );
39  return buildXSessionCookieMap(cookies ?? []);
40}
41 
42export async function isXSessionReady(
43  context: Pick<AdapterContext, "browser">,
44): Promise<boolean> {
45  const cookieMap = await readXSessionCookieMap(context);
46  return hasRequiredXSessionCookies(cookieMap);
47}
48