ScriptControllerIconView: Update to reflect policy type

App/Browser View/BrowserViewController.swift

@@ -513,17 +513,20 @@ class BrowserViewController: UIViewController
             numBlockedScripts += 1
         }
         
-        var scriptsAllowedForHost = false
-        if let url = webView.url, let host = url.host, policyManager.allowedOriginsForScriptResources().contains(host) {
-            scriptsAllowedForHost = true
+        var policy: ScriptPolicy? = nil
+        if let url = webView.url, let host = url.host {
+            policy = policyManager.scriptPolicy(forOrigin: host)
         }
         
         let iconView = toolbarController.scriptControllerIconView
         iconView.shieldsDown = tab.javaScriptEnabled
-        iconView.someScriptsAllowed = scriptsAllowedForHost
-        iconView.setBlockedScriptsNumber(numBlockedScripts)
+        // iconView.setBlockedScriptsNumber(numBlockedScripts)
         
-        // iconView.isEnabled = (webView.url != nil)
+        if let policy = policy {
+            iconView.currentPolicy = policy
+        }
+        
+        iconView.isEnabled = (webView.url != nil)
     }
     
     public func createNewTab(withURL url: URL?) {