1import { fetchXTweet } from "./graphql.js";
2import {
3  extractReferencedTweetIds,
4  type ReferencedTweetInfo,
5} from "./markdown.js";
6 
7type ResolveReferencedTweetsOptions = {
8  log?: (message: string) => void;
9};
10 
11function extractReferencedTweetInfo(tweet: any, fallbackTweetId: string): ReferencedTweetInfo {
12  const userCore = tweet?.core?.user_results?.result?.core;
13  const userLegacy = tweet?.core?.user_results?.result?.legacy;
14 
15  const authorName =
16    typeof userCore?.name === "string"
17      ? userCore.name
18      : typeof userLegacy?.name === "string"
19        ? userLegacy.name
20        : undefined;
21 
22  const authorUsername =
23    typeof userCore?.screen_name === "string"
24      ? userCore.screen_name
25      : typeof userLegacy?.screen_name === "string"
26        ? userLegacy.screen_name
27        : undefined;
28 
29  const text =
30    tweet?.note_tweet?.note_tweet_results?.result?.text ??
31    tweet?.legacy?.full_text ??
32    tweet?.legacy?.text ??
33    undefined;
34 
35  const tweetId =
36    typeof tweet?.rest_id === "string" && tweet.rest_id.length > 0
37      ? tweet.rest_id
38      : fallbackTweetId;
39 
40  const url = authorUsername
41    ? `https://x.com/${authorUsername}/status/${tweetId}`
42    : `https://x.com/i/web/status/${tweetId}`;
43 
44  return {
45    id: tweetId,
46    url,
47    authorName,
48    authorUsername,
49    text: typeof text === "string" ? text : undefined,
50  };
51}
52 
53export async function resolveReferencedTweetsFromArticle(
54  article: unknown,
55  cookieMap: Record<string, string>,
56  options: ResolveReferencedTweetsOptions = {}
57): Promise<Map<string, ReferencedTweetInfo>> {
58  const log = options.log ?? (() => {});
59  const ids = extractReferencedTweetIds(article);
60  const referencedTweets = new Map<string, ReferencedTweetInfo>();
61 
62  for (const id of ids) {
63    try {
64      const tweet = await fetchXTweet(id, cookieMap, false);
65      const info = extractReferencedTweetInfo(tweet, id);
66      referencedTweets.set(id, info);
67    } catch (error) {
68      log(
69        `[x-to-markdown] Failed to fetch referenced tweet ${id}: ${
70          error instanceof Error ? error.message : String(error)
71        }`
72      );
73      referencedTweets.set(id, {
74        id,
75        url: `https://x.com/i/web/status/${id}`,
76      });
77    }
78  }
79 
80  return referencedTweets;
81}
82