Checking domain instead of url subscript for determining If it is a m…

…ediasite
brave · Dec 16, 2020 · 5f97e02 · 5f97e02
1 parent 690f5f4
commit 5f97e02
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/...nt/Frontend/Browser/BrowserViewController/BrowserViewController+ProductNotification.swift b/...nt/Frontend/Browser/BrowserViewController/BrowserViewController+ProductNotification.swift
@@ -78,12 +78,9 @@ extension BrowserViewController {
         // Step 2: Load a video on a streaming site
         guard !notificationShown else { return }
 
-        let isVideoStreamingSite = (selectedTab.canonicalURL?.absoluteString.contains("youtube") == true
-                                        || selectedTab.canonicalURL?.absoluteString.contains("vimeo") == true)
-
         if isProductNotificationsValid,
            !Preferences.ProductNotificationBenchmarks.videoAdBlockShown.value,
-           isVideoStreamingSite {
+           selectedTab.url?.isMediaSiteURL == true {
 
             notifyVideoAdsBlocked(theme: Theme.of(selectedTab))
             notificationShown = true