polychrome/background.js at main · arfct/polychrome · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
chrome.runtime.onInstalled.addListener(function() {
  console.log("Polychrome Installed");
});

chrome.action.onClicked.addListener((tab) => {
  toggleSidebar();
});

chrome.commands.onCommand.addListener((command) => {
  if (command == "search") {
    toggleSearch();
  } else if (command == "sidebar") {
    toggleSidebar();
  }
});


//
// Window Management
//

const SEARCH_WIDTH = 320;
async function toggleSearch() {
  let url = chrome.runtime.getURL("search.html");
  let tabs = await chrome.tabs.query({url:url});
  let win = await chrome.windows.getLastFocused({populate:false, windowTypes:['normal']})
  if (tabs && tabs.length) {
      let sidebarId = tabs[0].windowId;
      chrome.windows.update(sidebarId, {focused:true})
  } else {
    await chrome.windows.create({
      url: url,
      type: "popup",
      width: SEARCH_WIDTH,
      height:28 + 48 + 96,
      left:win.left + 13,
      top:win.top + 7,
    });
  }
}


const DEFAULT_WIDTH = 256;
async function toggleSidebar() {
  let url = chrome.runtime.getURL("sidebar.html");
  let tabs = await chrome.tabs.query({url:url});
  let win = await chrome.windows.getLastFocused({populate:true, windowTypes:['normal']})
  if (tabs.length) {
    let sidebarId = tabs[0].windowId;
    let sidebar = await chrome.windows.get(sidebarId)
    console.log(tabs[0], sidebar);
      if (sidebar.focused) {
        chrome.windows.remove(sidebarId)
        console.log(sidebar.height, win.height, win.left, sidebar.width)
        if (sidebar.height >= win.height && win.left >= sidebar.width) {
          await chrome.windows.update(win.id, {width: win.width + sidebar.width, left:win.left - sidebar.width});
        }
      } else {
        chrome.windows.update(sidebarId, {focused:true})
      }
  } else {
    console.log(win, url);
    let adjustWindow = win.left < DEFAULT_WIDTH;
    await chrome.windows.create({
      url: url,
      type: "popup",
      width:256,
      height:win.width,
      top:win.top,
      left:adjustWindow ? win.left : win.left - DEFAULT_WIDTH
    });
    if (adjustWindow) {
      await chrome.windows.update(win.id, {width: win.width - DEFAULT_WIDTH, left:win.left + DEFAULT_WIDTH});
    }
  }
}

chrome.runtime.onMessage.addListener((message, sender, sendResponse) => {
  let handlers = {
    focusSidebar: focusSidebar,
    'removeDuplicates': removeDuplicates
  }

  if (handlers[message.action]) {
    if (!handlers[message.action](message, sender, sendResponse)) {
      sendResponse();
    };
  } else {
    console.error("handler not found", message)
    sendResponse(undefined)
  }
});

function removeDuplicates() {
  chrome.windows.getAll({populate:true, windowTypes:['normal']})
  .then((windows) => {
    var idsToRemove = []
    var knownURLs = {};
    windows.forEach(w => {
      w.tabs.forEach((tab) => {
        if (!knownURLs[tab.url]) {
          knownURLs[tab.url] = tab.id;
        } else {
          idsToRemove.push(tab.id)
        }
      })
      chrome.tabs.remove(idsToRemove.reverse());
    })
  })
}