Restore indication of tracking protection, if extesion was disabled

(#196)
Infocatcher · Dec 7, 2015 · 2d0d619 · 2d0d619
1 parent 6e215dc
commit 2d0d619
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/bootstrap.js b/bootstrap.js
@@ -549,6 +549,16 @@ var privateTab = {
 		if("TrackingProtection" in window) { // Firefox 42+
 			var identityPopup = document.getElementById("identity-popup");
 			identityPopup && identityPopup.removeEventListener("popupshowing", this, true);
+			if(reason != WINDOW_CLOSED) try {
+				window.TrackingProtection.updateEnabled();
+				if(!window.TrackingProtection.enabled)
+					window.TrackingProtection.icon.removeAttribute("state");
+				if(window.XULBrowserWindow && "_state" in window.XULBrowserWindow)
+					window.TrackingProtection.onSecurityChange(window.XULBrowserWindow._state, true /*aIsSimulated*/);
+			}
+			catch(e) {
+				Components.utils.reportError(e);
+			}
 		}
 		this.setupListAllTabs(window, false);
 		this.setupUndoCloseTabs(window, false);