v0.0.6

src/bg.js

@@ -1,5 +1,6 @@
 const kTST_ID = 'treestyletab@piro.sakura.ne.jp';
 const MY_EXTENSION_NAME = 'tab-groupcolor';
+let tabIcons = {};
 
 var changeIcon = function(color, currentIcon, options) {
 	var canvas = document.createElement('canvas');
@@ -23,11 +24,16 @@ var changeIcon = function(color, currentIcon, options) {
 			ctx.fillRect(0, 0, 16, options.globals.width);
 		}
 
-
 		var link = document.createElement('link');
 		link.type = 'image/x-icon';
 		link.rel = 'shortcut icon';
 		link.href = canvas.toDataURL("image/x-icon");
+		link.id = 'tab-groupcolor' + '-favicon';
+		var oldicon = document.getElementById('tab-groupcolor' + '-favicon');
+		if (oldicon) {
+			oldicon.outerHTML = '';
+			delete oldicon;
+		}
 		document.getElementsByTagName('head')[0].appendChild(link);
 	}
 };
@@ -58,21 +64,21 @@ var registerToTST = async function() {
 };
 
 //returns the index of the tabs array object that contains the tab with the passed id
-var getTabParent = async (tabs, id, currentTopLevelTabPos) => {
+var getTabParent = (tabs, id, currentTopLevelTabPos) => {
 	for (let i = 0; i < tabs.length; i++) {
 
 		//on a root node because there is no opener
-		if (typeof(tabs[i].openerTabId) === 'undefined' || tabs[i].openerTabId === tabs[i].id) {
+		if (!tabs[i].openerTabId || typeof(tabs[i].openerTabId) === 'undefined' || tabs[i].openerTabId === tabs[i].id) {
 			currentTopLevelTabPos = i;
 		}
 
 		if (tabs[i].id === id) {
 
 			//get original favicon for this tab
-			let orginalFavIconUrl = await browser.sessions.getTabValue(id, 'orginalFavIconUrl');
+			let orginalFavIconUrl = tabIcons[id];
 			if (!orginalFavIconUrl) {
 				orginalFavIconUrl = tabs[i].favIconUrl;
-				await browser.sessions.setTabValue(id, 'orginalFavIconUrl', orginalFavIconUrl);
+				tabIcons[id] = orginalFavIconUrl;
 			}
 
 			return {
@@ -81,7 +87,7 @@ var getTabParent = async (tabs, id, currentTopLevelTabPos) => {
 			};
 		} else {
 			if (tabs[i].children && tabs[i].children.length > 0) {
-				let ret = await getTabParent(tabs[i].children, id, currentTopLevelTabPos);
+				let ret = getTabParent(tabs[i].children, id, currentTopLevelTabPos);
 				if (ret) {
 					return ret;
 				}
@@ -95,7 +101,7 @@ var updateAllColorsOnAllTabs = async () => {
 	allWindows.forEach(async w => {
 		var tstTabs = await browser.runtime.sendMessage(kTST_ID, {
 			type: 'get-tree',
-			window:  w.id
+			window: w.id
 		});
 
 		var changeTabs = async (tabs) => {
@@ -103,13 +109,12 @@ var updateAllColorsOnAllTabs = async () => {
 			for (let i = 0; i < tabs.length; i++) {
 
 				if (tabs[i].status === 'complete') {
-					let t = await getTabParent(tstTabs, tabs[i].id);
+					let t = getTabParent(tstTabs, tabs[i].id);
+					let options = await loadOptions();
 
-					loadOptions().then((options) => {
-						browser.tabs.executeScript(tabs[i].id, {
-							code: '(' + changeIcon.toString() + ')' +
-							'("' + generateRandomColor(t.parentIndex) + '", "' + t.faviconUrl + '", ' + JSON.stringify(options) + ')'
-						});
+					browser.tabs.executeScript(tabs[i].id, {
+						code: '(' + changeIcon.toString() + ')' +
+						'("' + generateRandomColor(t.parentIndex) + '", "' + t.faviconUrl + '", ' + JSON.stringify(options) + ')'
 					});
 				}
 
@@ -138,13 +143,20 @@ browser.runtime.onMessageExternal.addListener((aMessage, aSender) => {
 });
 
 //when something happens with a tab
-let tabEvents = ['onUpdated', 'onAttached', 'onDetached', 'onMoved', 'onRemoved']
+let tabEvents = ['onAttached', 'onDetached', 'onMoved', 'onRemoved']
 tabEvents.forEach((ev) => {
-	browser.tabs[ev].addListener(async (tabId, changeInfo, tab) => {
+	browser.tabs[ev].addListener((tabId, changeInfo, tab) => {
 		updateAllColorsOnAllTabs();
 	});
 });
 
+//when a tab reloads
+browser.webNavigation.onCompleted.addListener((details) => {
+	delete tabIcons[details.tabId];
+	updateAllColorsOnAllTabs();
+});
+
+
 //when the options are updated
 browser.storage.onChanged.addListener((changes) => {
 	updateAllColorsOnAllTabs();

src/manifest.json

@@ -2,7 +2,7 @@
   "description": "Color tabs based on their parent.",
   "manifest_version": 2,
   "name": "Color Tab Group",
-  "version": "0.0.4",
+  "version": "0.0.6",
   "homepage_url": "https://git.e.tern.al/s2/tab-groupcolor",
   "icons": {
     "48": "img/icon-48.png"
@@ -18,7 +18,7 @@
   },
   "permissions": [
     "storage",
-    "sessions",
+    "webNavigation",
     "<all_urls>"
   ],
   "options_ui": {