From 7908dbe3211e9e9cb285f7540585267aef3684f1 Mon Sep 17 00:00:00 2001
From: dangered wolf <d@ngeredwolf.me>
Date: Sun, 20 Aug 2023 22:52:15 -0400
Subject: [PATCH] Add translations to IV

---
 src/embed/status.ts       |  2 +-
 src/render/instantview.ts | 27 ++++++++++++++++++++++++++-
 2 files changed, 27 insertions(+), 2 deletions(-)

diff --git a/src/embed/status.ts b/src/embed/status.ts
index deb8730..d25a018 100644
--- a/src/embed/status.ts
+++ b/src/embed/status.ts
@@ -42,7 +42,7 @@ export const handleStatus = async (
     isTelegram /*&& !tweet.possibly_sensitive*/ &&
     !flags?.direct &&
     !flags?.api &&
-    (tweet.media?.mosaic || tweet.is_note_tweet || tweet.quote || flags?.forceInstantView);
+    (tweet.media?.mosaic || tweet.is_note_tweet || tweet.quote || tweet.translation || flags?.forceInstantView);
 
   let ivbody = '';
 
diff --git a/src/render/instantview.ts b/src/render/instantview.ts
index 385574a..c8caf4f 100644
--- a/src/render/instantview.ts
+++ b/src/render/instantview.ts
@@ -1,6 +1,7 @@
 import { Constants } from '../constants';
 import { getSocialTextIV } from '../helpers/author';
 import { sanitizeText } from '../helpers/utils';
+import { Strings } from '../strings';
 
 const populateUserLinks = (tweet: APITweet, text: string): string => {
   /* TODO: Maybe we can add username splices to our API so only genuinely valid users are linked? */
@@ -68,6 +69,28 @@ function paragraphify(text: string, isQuote = false): string {
     .join('\n');
 }
 
+function getTranslatedText(tweet: APITweet, isQuote = false): string | null {
+  if (!tweet.translation) {
+    return null;
+  }
+  let text = paragraphify(sanitizeText(tweet.translation?.text), isQuote);
+  text = htmlifyLinks(text);
+  text = htmlifyHashtags(text);
+  text = populateUserLinks(tweet, text);
+
+  const formatText =
+      tweet.translation.target_lang === 'en'
+        ? Strings.TRANSLATE_TEXT.format({
+            language: tweet.translation.source_lang_en
+          })
+        : Strings.TRANSLATE_TEXT_INTL.format({
+            source: tweet.translation.source_lang.toUpperCase(),
+            destination: tweet.translation.target_lang.toUpperCase()
+          });
+
+  return `<h4>${formatText}</h4>${text}<h4>Original</h4>`;
+}
+
 /* TODO: maybe refactor so all tweets pull from this */
 const generateTweet = (tweet: APITweet, isQuote = false): string => {
   let text = paragraphify(sanitizeText(tweet.text), isQuote);
@@ -75,6 +98,8 @@ const generateTweet = (tweet: APITweet, isQuote = false): string => {
   text = htmlifyHashtags(text);
   text = populateUserLinks(tweet, text);
 
+  const translatedText = getTranslatedText(tweet, isQuote);
+
   return `
   <!-- Embed profile picture, display name, and screen name in table -->
   ${
@@ -95,8 +120,8 @@ const generateTweet = (tweet: APITweet, isQuote = false): string => {
   `
       : ''
   }
-
   <!-- Embed Tweet text -->
+  ${translatedText ? translatedText : ''}
   ${text}
   ${generateTweetMedia(tweet)} 
   ${!isQuote && tweet.quote ? generateTweet(tweet.quote, true) : ''}