From ca97e3c375c0fdf5c06e3b4bcdddbafb9ea90e6f Mon Sep 17 00:00:00 2001
From: John Preston <johnprestonmail@gmail.com>
Date: Wed, 3 Jun 2020 12:44:46 +0400
Subject: [PATCH] Add more warnings for suspicious urls.

---
 Telegram/SourceFiles/overview/overview_layout.cpp | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/Telegram/SourceFiles/overview/overview_layout.cpp b/Telegram/SourceFiles/overview/overview_layout.cpp
index f456de3fe..171ed4e8a 100644
--- a/Telegram/SourceFiles/overview/overview_layout.cpp
+++ b/Telegram/SourceFiles/overview/overview_layout.cpp
@@ -1433,6 +1433,11 @@ Link::Link(
 		}
 	}
 
+	const auto createHandler = [](const QString &url) {
+		return UrlClickHandler::IsSuspicious(url)
+			? std::make_shared<HiddenUrlClickHandler>(url)
+			: std::make_shared<UrlClickHandler>(url);
+	};
 	_page = media ? media->webpage() : nullptr;
 	if (_page) {
 		mainUrl = _page->url;
@@ -1442,7 +1447,7 @@ Link::Link(
 				parent->fullId());
 		} else if (_page->photo) {
 			if (_page->type == WebPageType::Profile || _page->type == WebPageType::Video) {
-				_photol = std::make_shared<UrlClickHandler>(_page->url);
+				_photol = createHandler(_page->url);
 			} else if (_page->type == WebPageType::Photo
 				|| _page->siteName == qstr("Twitter")
 				|| _page->siteName == qstr("Facebook")) {
@@ -1450,13 +1455,13 @@ Link::Link(
 					_page->photo,
 					parent->fullId());
 			} else {
-				_photol = std::make_shared<UrlClickHandler>(_page->url);
+				_photol = createHandler(_page->url);
 			}
 		} else {
-			_photol = std::make_shared<UrlClickHandler>(_page->url);
+			_photol = createHandler(_page->url);
 		}
 	} else if (!mainUrl.isEmpty()) {
-		_photol = std::make_shared<UrlClickHandler>(mainUrl);
+		_photol = createHandler(mainUrl);
 	}
 	if (from >= till && _page) {
 		text = _page->description.text;