s2/tab-groupcolor
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix updateAllColorsOnAllTabs

Browse Source
This commit is contained in:
s2
2017-12-11 22:29:30 +01:00
parent 90f7108eee
commit 724bec321b
1 changed files with 30 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
src/bg.js
View File

@@ -1,24 +1,13 @@
const kTST_ID = 'treestyletab@piro.sakura.ne.jp'; const kTST_ID = 'treestyletab@piro.sakura.ne.jp';
const MY_EXTENSION_NAME = 'tab-groupcolor'; const MY_EXTENSION_NAME = 'tab-groupcolor';
var changeIcon = function(color) { var changeIcon = function(color, currentIcon) {
var getFavicon = function() {
var favicon = undefined;
var nodeList = document.getElementsByTagName("link");
for (var i = 0; i < nodeList.length; i++) {
if ((nodeList[i].getAttribute("rel") == "icon") || (nodeList[i].getAttribute("rel") == "shortcut icon")) {
favicon = nodeList[i].getAttribute("href");
}
}
return favicon;
};
var canvas = document.createElement('canvas'); var canvas = document.createElement('canvas');
canvas.width = 16; canvas.width = 16;
canvas.height = 16; canvas.height = 16;
var ctx = canvas.getContext('2d'); var ctx = canvas.getContext('2d');
var img = new Image(); var img = new Image();
img.src = getFavicon(); img.src = currentIcon;
img.onload = function() { img.onload = function() {
ctx.drawImage(img, 0, 0); ctx.drawImage(img, 0, 0);
ctx.fillStyle = color; ctx.fillStyle = color;
@@ -67,7 +56,10 @@ var getTabParent = (tabs, id, currentTopLevelTabPos) => {
} }
if (tabs[i].id === id) { if (tabs[i].id === id) {
return currentTopLevelTabPos; return {
parentIndex: currentTopLevelTabPos,
faviconUrl: tabs[i].favIconUrl
};
} else { } else {
if (tabs[i].children && tabs[i].children.length > 0) { if (tabs[i].children && tabs[i].children.length > 0) {
let ret = getTabParent(tabs[i].children, id, currentTopLevelTabPos); let ret = getTabParent(tabs[i].children, id, currentTopLevelTabPos);
@@ -87,13 +79,26 @@ var updateAllColorsOnAllTabs = async () => {
window: w.id window: w.id
}); });
for (let i = 0; i < tstTabs.length; i++) { var changeTabs = (tabs) => {
if (tstTabs[i].status === 'complete') {
browser.tabs.executeScript(tstTabs[i].id, { for (let i = 0; i < tabs.length; i++) {
code: '(' + changeIcon.toString() + ')("' + generateRandomColor(getTabParent(tstTabs, tstTabs[i].id)) + '")'
}); if (tabs[i].status === 'complete') {
let t = getTabParent(tstTabs, tabs[i].id);
browser.tabs.executeScript(tabs[i].id, {
code: '(' + changeIcon.toString() + ')' +
'("' + generateRandomColor(t.parentIndex) + '", "' + t.faviconUrl + '")'
});
}
if (tabs[i].children && tabs[i].children.length > 0) {
changeTabs(tabs[i].children);
}
} }
} };
changeTabs(tstTabs);
}); });
} }
@@ -112,14 +117,17 @@ browser.runtime.onMessageExternal.addListener((aMessage, aSender) => {
//when a tab is created //when a tab is created
browser.tabs.onUpdated.addListener(async (tabId, changeInfo, tab) => { browser.tabs.onUpdated.addListener(async (tabId, changeInfo, tab) => {
var tstTab = await browser.runtime.sendMessage(kTST_ID, { var tstTabs = await browser.runtime.sendMessage(kTST_ID, {
type: 'get-tree', type: 'get-tree',
window: tab.windowId window: tab.windowId
}); });
if (tab.status === 'complete') { if (tab.status === 'complete') {
var t = getTabParent(tstTabs, tab.id);
browser.tabs.executeScript(tab.id, { browser.tabs.executeScript(tab.id, {
code: '(' + changeIcon.toString() + ')("' + generateRandomColor(getTabParent(tstTab, tab.id)) + '")' code: '(' + changeIcon.toString() + ')' +
'("' + generateRandomColor(t.parentIndex) + '", "' + t.faviconUrl + '")'
}); });
} }
}); });