From 4edeba2775f7d6b17c8da4f5974139d86e00df47 Mon Sep 17 00:00:00 2001
From: Nikhil Kothari <nik.kothari22@live.com>
Date: Fri, 10 May 2024 19:52:27 +0530
Subject: [PATCH] fix: show twitter link previews

---
 .../feature/chat/ChatMessage/MessageItem.tsx         |  4 ++--
 .../ChatMessage/Renderers/TiptapRenderer/Link.tsx    |  8 ++++----
 raven/api/preview_links.py                           | 12 ++++++++++--
 3 files changed, 16 insertions(+), 8 deletions(-)

diff --git a/raven-app/src/components/feature/chat/ChatMessage/MessageItem.tsx b/raven-app/src/components/feature/chat/ChatMessage/MessageItem.tsx
index 890f3aace..2f6869f5b 100644
--- a/raven-app/src/components/feature/chat/ChatMessage/MessageItem.tsx
+++ b/raven-app/src/components/feature/chat/ChatMessage/MessageItem.tsx
@@ -87,9 +87,9 @@ export const MessageItem = ({ message, setDeleteMessage, isHighlighted, onReplyM
                             transition-colors
                             p-2
                             rounded-md`, isHighlighted ? 'bg-yellow-50 hover:bg-yellow-50 dark:bg-yellow-300/20 dark:hover:bg-yellow-300/20' : '')}>
-                    <Flex gap='3' >
+                    <Flex gap='3'>
                         <MessageLeftElement message={message} user={user} isActive={isActive} />
-                        <Flex direction='column' className='gap-0.5' justify='center'>
+                        <Flex direction='column' className='gap-0.5' justify='center' width='100%'>
                             {!is_continuation ? <Flex align='center' gap='2' mt='-1'>
                                 <UserHoverCard
                                     user={user}
diff --git a/raven-app/src/components/feature/chat/ChatMessage/Renderers/TiptapRenderer/Link.tsx b/raven-app/src/components/feature/chat/ChatMessage/Renderers/TiptapRenderer/Link.tsx
index 189241be5..8f9ec0567 100644
--- a/raven-app/src/components/feature/chat/ChatMessage/Renderers/TiptapRenderer/Link.tsx
+++ b/raven-app/src/components/feature/chat/ChatMessage/Renderers/TiptapRenderer/Link.tsx
@@ -98,7 +98,7 @@ export const LinkPreview = memo(({ messageID }: { messageID: string }) => {
     const linkPreview = data?.message?.[0]
 
     return <Flex direction='column' gap='2' py='2'>
-        {linkPreview ? linkPreview.site_name && linkPreview.description ? <Flex gap='4'>
+        {linkPreview ? linkPreview.site_name && linkPreview.description ? <Flex gap='4' width='100%'>
             {(linkPreview.absolute_image || linkPreview.image) &&
                 <a href={href} target='_blank'>
                     <Box className='relative min-w-64 min-h-32 w-64 h-32'>
@@ -116,10 +116,10 @@ export const LinkPreview = memo(({ messageID }: { messageID: string }) => {
                     </Box>
                 </a>
             }
-            <Flex className='group pr-2'>
-                <a href={href} target='_blank'>
+            <Flex className='group pr-2' width='100%' gap='2'>
+                <a href={href} target='_blank' className='block w-full'>
 
-                    <Flex direction='column' gap='1' py='1' className='w-84'>
+                    <Flex direction='column' gap='1' py='1'>
                         <Flex gap='1' direction='column'>
                             <Text as='span' weight='bold' size='3'>{linkPreview.title}</Text>
                             <Text as='span' color='gray' size='2' weight='medium'>{linkPreview.site_name}</Text>
diff --git a/raven/api/preview_links.py b/raven/api/preview_links.py
index 30997f040..f018e7214 100644
--- a/raven/api/preview_links.py
+++ b/raven/api/preview_links.py
@@ -2,7 +2,7 @@
 import re
 
 import frappe
-from linkpreview import link_preview
+from linkpreview import Link, LinkGrabber, LinkPreview, link_preview
 
 
 @frappe.whitelist(methods=["GET"])
@@ -39,7 +39,15 @@ def get_preview_link(urls):
 				else:
 					preview = None
 					try:
-						preview = link_preview(url)
+						# If this is a Twitter/X URL, then we need to fetch the preview from the Twitter API
+						# This is because the linkpreview library doesn't support Twitter previews with the default bot
+						if "twitter.com" in url or "x.com" in url:
+							grabber = LinkGrabber()
+							content, url_fetched = grabber.get_content(url, headers="imessagebot")
+							link = Link(url_fetched, content)
+							preview = LinkPreview(link)
+						else:
+							preview = link_preview(url)
 					except Exception:
 						preview = None
 						pass