support privacy config for clickToLoad

duckduckgo · Jan 28, 2022 · fd3f151 · fd3f151
1 parent 1675e32
commit fd3f151
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 7 deletions.
diff --git a/DuckDuckGo/BrowserTab/Model/Tab.swift b/DuckDuckGo/BrowserTab/Model/Tab.swift
@@ -337,10 +337,6 @@ final class Tab: NSObject {
         return false
     }
 
-    private func clickToLoadBlockFB() {
-        setFBProtection(enabled: true)
-    }
-
     private func reloadIfNeeded(shouldLoadInBackground: Bool = false) {
         let url: URL
         switch self.content {
@@ -783,6 +779,16 @@ extension Tab: WKNavigationDelegate {
             decisionHandler(.allow)
             return
         }
+
+        let privacyConfigurationManager = ContentBlocking.privacyConfigurationManager
+        let privacyConfiguration = privacyConfigurationManager.privacyConfig
+
+        let featureEnabled = privacyConfiguration.isFeature(.clickToPlay, enabledForDomain: url.host)
+        if featureEnabled {
+            setFBProtection(enabled: true)
+        } else {
+            setFBProtection(enabled: false)
+        }
 
         if navigationAction.shouldDownload {
             // register the navigationAction for legacy _WKDownload to be called back on the Tab
@@ -863,7 +869,6 @@ extension Tab: WKNavigationDelegate {
     }
 
     func webView(_ webView: WKWebView, didStartProvisionalNavigation navigation: WKNavigation!) {
-        self.clickToLoadBlockFB()
         delegate?.tabDidStartNavigation(self)
 
         // Unnecessary assignment triggers publishing

diff --git a/DuckDuckGo/ContentBlocker/ClickToLoadUserScript.swift b/DuckDuckGo/ContentBlocker/ClickToLoadUserScript.swift
@@ -40,12 +40,14 @@ final class ClickToLoadUserScript: NSObject, UserScript, WKScriptMessageHandlerW
     func userContentController(_ userContentController: WKUserContentController, didReceive message: WKScriptMessage,
                                replyHandler: @escaping (Any?, String?) -> Void) {
         if message.name == "initClickToLoad" {
+            let host = message.body as? String
             let controller = userContentController as? UserContentController
             let privacyConfigurationManager = controller!.privacyConfigurationManager
             let privacyConfiguration = privacyConfigurationManager.privacyConfig
 
-            let protected = privacyConfiguration.isProtected(domain: message.body as? String)
-            if protected {
+            let locallyProtected = privacyConfiguration.isProtected(domain: host)
+            let featureEnabled = privacyConfiguration.isFeature(.clickToPlay, enabledForDomain: host)
+            if locallyProtected && featureEnabled {
                 replyHandler(true, nil)
             } else {
                 replyHandler(false, nil)