Reload script before page reload when changing protection status

Author: CrisBarreiro

app/src/main/java/com/duckduckgo/app/browser/BrowserTabFragment.kt

@@ -1935,11 +1935,17 @@ class BrowserTabFragment :
             is Command.ShowFireproofWebSiteConfirmation -> fireproofWebsiteConfirmation(it.fireproofWebsiteEntity)
             is Command.DeleteFireproofConfirmation -> removeFireproofWebsiteConfirmation(it.fireproofWebsiteEntity)
             is Command.RefreshAndShowPrivacyProtectionEnabledConfirmation -> {
+                webView?.let {
+                    webViewClient.configureWebView(it, null)
+                }
                 refresh()
                 privacyProtectionEnabledConfirmation(it.domain)
             }
 
             is Command.RefreshAndShowPrivacyProtectionDisabledConfirmation -> {
+                webView?.let {
+                    webViewClient.configureWebView(it, null)
+                }
                 refresh()
                 privacyProtectionDisabledConfirmation(it.domain)
             }

app/src/main/java/com/duckduckgo/app/browser/BrowserWebViewClient.kt

@@ -469,15 +469,17 @@ class BrowserWebViewClient @Inject constructor(
         webView.settings.mediaPlaybackRequiresUserGesture = mediaPlayback.doesMediaPlaybackRequireUserGestureForUrl(url)
     }
 
-    fun configureWebView(webView: DuckDuckGoWebView, callback: WebViewCompatMessageCallback) {
+    fun configureWebView(webView: DuckDuckGoWebView, callback: WebViewCompatMessageCallback?) {
         appCoroutineScope.launch {
             val activeExperiments = contentScopeExperiments.getActiveExperiments()
             addDocumentStartJavascriptPlugins.getPlugins().forEach { plugin ->
                 plugin.addDocumentStartJavaScript(activeExperiments, webView)
             }
 
-            webMessagingPlugins.getPlugins().forEach { plugin ->
-                plugin.register(callback, webView)
+            callback?.let {
+                webMessagingPlugins.getPlugins().forEach { plugin ->
+                    plugin.register(callback, webView)
+                }
             }
         }
     }