app/vendor/trezor/usb-permissions.js


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

/*
Handling messages from usb permissions iframe
*/

const switchToPopupTab = (event) => {

  window.removeEventListener('beforeunload', switchToPopupTab);

  if (!event) {
      // triggered from 'usb-permissions-close' message
      // switch tab to previous index and close current
      chrome.tabs.query({
          currentWindow: true,
          active: true,
      }, (current) => {
          if (current.length < 0) return;
          chrome.tabs.query({
              index: current[0].index - 1
          }, popup => {
              if (popup.length < 0) return;
              chrome.tabs.update(popup[0].id, { active: true });
          })
          chrome.tabs.remove(current[0].id);
      });
      return;
  }

  // triggered from 'beforeunload' event
  // find tab by popup pattern and switch to it
  chrome.tabs.query({
      url: "*://connect.trezor.io/*/popup.html"
  }, (tabs) => {
      if (tabs.length < 0) return;
      chrome.tabs.update(tabs[0].id, { active: true });
  });
}

window.addEventListener('message', event => {
  if (event.data === 'usb-permissions-init') {
      const iframe = document.getElementById('trezor-usb-permissions');
      iframe.contentWindow.postMessage({
          type: 'usb-permissions-init',
          extension: chrome.runtime.id,
      }, '*');
  } else if (event.data === 'usb-permissions-close') {
      switchToPopupTab();
  }
});

window.addEventListener('beforeunload', switchToPopupTab);