1import type { AdapterContext } from "../types";
2import { filterXGraphQlEntries } from "./shared";
3 
4export function getRelevantXThreadEntries(context: AdapterContext) {
5  return filterXGraphQlEntries(context.network.getEntries()).filter(
6    (entry) =>
7      entry.method === "GET" &&
8      entry.finished &&
9      (
10        entry.url.includes("TweetDetail") ||
11        entry.url.includes("TweetResultByRestId") ||
12        entry.url.includes("TweetResultsByRestIds")
13      ),
14  );
15}
16 
17export async function prefetchRelevantXThreadBodies(context: AdapterContext): Promise<void> {
18  const entries = getRelevantXThreadEntries(context).filter((entry) => entry.body === undefined && !entry.bodyError);
19  for (const entry of entries) {
20    await context.network.ensureBody(entry);
21  }
22}
23 
24export async function collectXJsonPayloads(context: AdapterContext): Promise<unknown[]> {
25  await prefetchRelevantXThreadBodies(context);
26  const entries = getRelevantXThreadEntries(context);
27 
28  const payloads: unknown[] = [];
29  for (const entry of entries) {
30    const payload = await context.network.getJsonBody(entry);
31    if (payload) {
32      payloads.push(payload);
33    }
34  }
35  return payloads;
36}
37 
38export async function waitForInitialXPayload(context: AdapterContext): Promise<void> {
39  try {
40    await context.network.waitForResponse(
41      (entry) =>
42        entry.url.includes("/graphql/") &&
43        (entry.url.includes("TweetDetail") || entry.url.includes("TweetResultByRestId")),
44      { timeoutMs: Math.min(context.timeoutMs, 15_000) },
45    );
46    await prefetchRelevantXThreadBodies(context);
47  } catch {
48    context.log.debug("No tweet GraphQL response observed before timeout.");
49  }
50}
51