From 90e9c3a1eb359d3e44e13f28158de21098e90b9b Mon Sep 17 00:00:00 2001
From: dangered wolf <d@ngeredwolf.me>
Date: Sun, 14 Jan 2024 19:37:43 -0500
Subject: [PATCH] Implement thread unrolling in Instant View

---
 src/embed/status.ts                   |  8 +++++---
 src/providers/twitter/conversation.ts | 22 ++++++++++++++++++----
 src/providers/twitter/processor.ts    |  4 +++-
 src/render/instantview.ts             | 27 ++++++++++++++++-----------
 src/types/types.d.ts                  |  3 ++-
 5 files changed, 44 insertions(+), 20 deletions(-)

diff --git a/src/embed/status.ts b/src/embed/status.ts
index caa9cd9..113252d 100644
--- a/src/embed/status.ts
+++ b/src/embed/status.ts
@@ -35,9 +35,8 @@ export const handleStatus = async (
 
   let fetchWithThreads = false;
 
-  /* TODO: Enable actually pulling threads once we can actually do something with them */
-  if (c.req.header('user-agent')?.includes('Telegram') && !flags?.direct) {
-    fetchWithThreads = false;
+  if (c.req.header('user-agent')?.includes('Telegram') && !flags?.direct && flags.instantViewUnrollThreads) {
+    fetchWithThreads = true;
   }
 
   const thread = await constructTwitterThread(
@@ -48,6 +47,8 @@ export const handleStatus = async (
     flags?.api ?? false
   );
 
+  console.log('twitterThread', thread);
+
   const status = thread?.status as APITwitterStatus;
 
   const api = {
@@ -191,6 +192,7 @@ export const handleStatus = async (
     try {
       const instructions = renderInstantView({
         status: status,
+        thread: thread,
         text: newText,
         flags: flags
       });
diff --git a/src/providers/twitter/conversation.ts b/src/providers/twitter/conversation.ts
index 0a5a111..917e7c9 100644
--- a/src/providers/twitter/conversation.ts
+++ b/src/providers/twitter/conversation.ts
@@ -494,11 +494,25 @@ export const constructTwitterThread = async (
     author: author,
     code: 200
   };
+  
+  await Promise.all(threadStatuses.map(async status => {
+    console.log('Processing status for', status)
+    const builtStatus = await buildAPITwitterStatus(c, status, undefined, true, false) as APITwitterStatus;
+    console.log('builtStatus', builtStatus);
+    socialThread.thread?.push(builtStatus);
+  }));
 
-  threadStatuses.forEach(async status => {
-    socialThread.thread?.push(
-      (await buildAPITwitterStatus(c, status, undefined, true, false)) as APITwitterStatus
-    );
+  // Sort socialThread.thread by id converted to bigint
+  socialThread.thread?.sort((a, b) => {
+    const aId = BigInt(a.id);
+    const bId = BigInt(b.id);
+    if (aId < bId) {
+      return -1;
+    }
+    if (aId > bId) {
+      return 1;
+    }
+    return 0;
   });
 
   return socialThread;
diff --git a/src/providers/twitter/processor.ts b/src/providers/twitter/processor.ts
index 7db0ebf..90b1775 100644
--- a/src/providers/twitter/processor.ts
+++ b/src/providers/twitter/processor.ts
@@ -94,7 +94,9 @@ export const buildAPITwitterStatus = async (
     delete apiStatus.author.global_screen_name;
   } else {
     apiStatus.reposts = status.legacy.retweet_count;
-    apiStatus.author.global_screen_name = apiUser.global_screen_name;
+    if (!threadPiece) {
+      apiStatus.author.global_screen_name = apiUser.global_screen_name;
+    }
   }
   apiStatus.likes = status.legacy.favorite_count;
   apiStatus.embed_card = 'tweet';
diff --git a/src/render/instantview.ts b/src/render/instantview.ts
index 15058a2..96b5617 100644
--- a/src/render/instantview.ts
+++ b/src/render/instantview.ts
@@ -68,7 +68,7 @@ const htmlifyHashtags = (input: string): string => {
   const hashtagPattern = /#([a-zA-Z_]\w*)/g;
   return input.replace(hashtagPattern, (match, hashtag) => {
     const encodedHashtag = encodeURIComponent(hashtag);
-    return `<a href="${Constants.TWITTER_ROOT}/hashtag/${encodedHashtag}?src=hashtag_click">${match}</a>`;
+    return `  <a href="${Constants.TWITTER_ROOT}/hashtag/${encodedHashtag}?src=hashtag_click">${match}</a>  `;
   });
 };
 
@@ -117,9 +117,7 @@ const truncateSocialCount = (count: number): string => {
   }
 };
 
-const generateStatusFooter = (status: APIStatus, isQuote = false): string => {
-  const { author } = status;
-
+const generateStatusFooter = (status: APIStatus, isQuote = false, author: APIUser): string => {
   let description = author.description;
   description = htmlifyLinks(description);
   description = htmlifyHashtags(description);
@@ -163,7 +161,7 @@ const generateStatusFooter = (status: APIStatus, isQuote = false): string => {
   });
 };
 
-const generateStatus = (status: APIStatus, isQuote = false): string => {
+const generateStatus = (status: APIStatus, author: APIUser, isQuote = false): string => {
   let text = paragraphify(sanitizeText(status.text), isQuote);
   text = htmlifyLinks(text);
   text = htmlifyHashtags(text);
@@ -180,20 +178,23 @@ const generateStatus = (status: APIStatus, isQuote = false): string => {
   <!-- Embed Status text -->
   ${text}
   <!-- Embedded quote status -->
-  ${!isQuote && status.quote ? generateStatus(status.quote, true) : notApplicableComment}
-  ${!isQuote ? generateStatusFooter(status) : ''}
-  <br>${!isQuote ? `<a href="${status.url}">View original post</a>` : notApplicableComment}
+  ${!isQuote && status.quote ? generateStatus(status.quote, author, true) : notApplicableComment}
   `.format({
     quoteHeader: isQuote
-      ? `<h4><a href="${status.url}">Quoting</a> ${status.author.name} (<a href="${Constants.TWITTER_ROOT}/${status.author.screen_name}">@${status.author.screen_name}</a>)</h4>`
+      ? `<h4><a href="${status.url}">Quoting</a> ${author.name} (<a href="${Constants.TWITTER_ROOT}/${author.screen_name}">@${author.screen_name}</a>)</h4>`
       : ''
   });
 };
 
 export const renderInstantView = (properties: RenderProperties): ResponseInstructions => {
   console.log('Generating Instant View...');
-  const { status, flags } = properties;
+  const { status, thread, flags } = properties;
   const instructions: ResponseInstructions = { addHeaders: [] };
+
+  if (!status) {
+    throw new Error('Status is undefined');
+  }
+
   /* Use ISO date for Medium template */
   const statusDate = new Date(status.created_at).toISOString();
 
@@ -210,6 +211,8 @@ export const renderInstantView = (properties: RenderProperties): ResponseInstruc
       : ``
   ];
 
+  console.log('thread', thread?.thread)
+
   instructions.text = `
     <section class="section-backgroundImage">
       <figure class="graf--layoutFillWidth"></figure>
@@ -224,7 +227,9 @@ export const renderInstantView = (properties: RenderProperties): ResponseInstruc
     <sub><a href="${status.url}">View original</a></sub>
     <h1>${status.author.name} (@${status.author.screen_name})</h1>
 
-    ${generateStatus(status)}
+    ${thread?.thread?.map(status => generateStatus(status, thread?.author ?? status.author, false)).join('')}
+    ${generateStatusFooter(status, false, thread?.author ?? status.author)}
+    <br>${`<a href="${status.url}">View original post</a>`}
   </article>`;
 
   return instructions;
diff --git a/src/types/types.d.ts b/src/types/types.d.ts
index 3c66776..b690539 100644
--- a/src/types/types.d.ts
+++ b/src/types/types.d.ts
@@ -28,7 +28,8 @@ interface ResponseInstructions {
 }
 
 interface RenderProperties {
-  status: APITwitterStatus;
+  status?: APITwitterStatus;
+  thread?: SocialThread;
   siteText?: string;
   authorText?: string;
   engagementText?: string;