Always run cleanup routine when URI scheme changed

JustOff · Apr 17, 2018 · e62b020 · e62b020
1 parent 597bcda
commit e62b020
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/content/tabs.js b/content/tabs.js
@@ -8,17 +8,20 @@ let Tabs = function(Cleaner, Buttons, Utils) {
 		}
 	};
 
-	this.onTabProgress = {	
+	this.onTabProgress = {
 		onLocationChange: function(aBrowser, aWebProgress, aRequest, aURI, aFlag) {
 			if (aFlag & Components.interfaces.nsIWebProgressListener.LOCATION_CHANGE_SAME_DOCUMENT) {
 				return;
 			}
 			try {
-				if (aURI.host && aURI.scheme && (aURI.scheme == "http" || aURI.scheme == "https")) {
+				if (aURI.scheme == "http" || aURI.scheme == "https") {
 					if (aBrowser.previousURI && aBrowser.previousURI.host != aURI.host) {
 						Cleaner.trackTabs(aBrowser.previousURI);
 					}
 					aBrowser["previousURI"] = aURI;
+				} else if (aBrowser.previousURI) {
+					Cleaner.trackTabs(aBrowser.previousURI);
+					aBrowser["previousURI"] = null;
 				}
 			} catch(e) {}
 		}