Merge pull request #15 from fireblocks-labs/fix/wakep-service-worker

fix: Wakeup service worker on content script launch

Author: rdvorkin

src/common.ts

@@ -6,6 +6,7 @@ export const Actions = {
   headerUpdate: 'headerUpdate',
   verifyBalances: 'verifyBalances',
   configUpdate: 'configUpdate',
+  wakeUp: 'wakeUp',
 };
 
 export const configStorageName = 'config';

src/pages/background/index.ts

@@ -42,7 +42,9 @@ let globalLightClientVerifier: LightClientVerifier;
 
 chrome.runtime.onMessage.addListener((message, sender, sendResponse) => {
   console.log('Got message', message);
-  if (message.action === Actions.verifyBalances) {
+  if (message.action === Actions.wakeUp) {
+    console.log('Got wakeUp message');
+  } else if (message.action === Actions.verifyBalances) {
     if (globalLightClientVerifier) {
       globalLightClientVerifier
         .verifyBalances(message.accountsToVerify, message.ethRoundingDigits, message.tokenRoundingDigits)

src/pages/content/index.ts

@@ -3,6 +3,7 @@ import { EtherscanTokenholdingsPageHandler } from './sites/etherscan/tokenholdin
 import { EtherscanAddressPageHandler } from './sites/etherscan/address';
 import { MetaMaskPortfolioPageHandler } from './sites/metamask/portfolio';
 import { PageHandler } from './PageHandler';
+import { Actions } from '../../common';
 
 function intializePageHandler<T extends PageHandler>(handler: { new (document: Document, window: Window): T }): T {
   return new handler(globalThis.document, globalThis.window);
@@ -18,6 +19,7 @@ const currentURL = window.location.href;
 (async function () {
   for (const pageHandler of pageHandlers) {
     if (pageHandler.checkMatch(currentURL)) {
+      chrome.runtime.sendMessage({ action: Actions.wakeUp });
       await pageHandler.setup();
       await waitForClientToStart();
       pageHandler.run();