From 472cef2382f3320bcae84ca6e591cbd37d0e000e Mon Sep 17 00:00:00 2001
From: JustAnotherArchivist <JustAnotherArchivist@users.noreply.github.com>
Date: Thu, 1 Oct 2020 18:51:14 +0000
Subject: [PATCH] Add support for link previews

---
 snscrape/modules/telegram.py | 28 +++++++++++++++++++++++++++-
 1 file changed, 27 insertions(+), 1 deletion(-)

diff --git a/snscrape/modules/telegram.py b/snscrape/modules/telegram.py
index 9524cc3..a8cdbad 100644
--- a/snscrape/modules/telegram.py
+++ b/snscrape/modules/telegram.py
@@ -9,12 +9,21 @@ import urllib.parse
 logger = logging.getLogger(__name__)
 
 
+class LinkPreview(typing.NamedTuple):
+	href: str
+	siteName: str
+	title: str
+	description: str
+	image: typing.Optional[str] = None
+
+
 class TelegramPost(typing.NamedTuple, snscrape.base.Item):
 	url: str
 	date: datetime.datetime
 	content: str
 	outlinks: list
 	outlinksss: str
+	linkPreview: typing.Optional[LinkPreview] = None
 
 	def __str__(self):
 		return self.url
@@ -84,7 +93,24 @@ class TelegramChannelScraper(snscrape.base.Scraper):
 				content = None
 				outlinks = []
 				outlinksss = ''
-			yield TelegramPost(url = f'https://t.me/s/{post["data-post"]}', date = date, content = content, outlinks = outlinks, outlinksss = outlinksss)
+			linkPreview = None
+			linkPreviewA = post.find('a', class_ = 'tgme_widget_message_link_preview')
+			if linkPreviewA:
+				image = None
+				imageI = linkPreviewA.find('i', class_ = 'link_preview_image')
+				if imageI:
+					if imageI['style'].startswith("background-image:url('"):
+						image = imageI['style'][22 : imageI['style'].index("'", 22)]
+					else:
+						self.logger.warning(f'Could not process link preview image on https://t.me/s/{post["data-post"]}')
+				linkPreview = LinkPreview(
+					href = urllib.parse.urljoin(pageUrl, linkPreviewA['href']),
+					siteName = linkPreviewA.find('div', class_ = 'link_preview_site_name').text,
+					title = linkPreviewA.find('div', class_ = 'link_preview_title').text,
+					description = linkPreviewA.find('div', class_ = 'link_preview_description').text,
+					image = image,
+				)
+			yield TelegramPost(url = f'https://t.me/s/{post["data-post"]}', date = date, content = content, outlinks = outlinks, outlinksss = outlinksss, linkPreview = linkPreview)
 
 	def get_items(self):
 		r, soup = self._initial_page()