Merge pull request #4 from isaurssaurav/issue/#2

fix(#2): refresh the page after extension is reloaded
isaurssaurav · Oct 28, 2023 · 09d3f73 · 09d3f73
2 parents 6c6dae9 + 020d4ca
commit 09d3f73
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/src/main.ts b/src/main.ts
@@ -10,8 +10,6 @@ export type hotReloadExtensionOptions = {
   log?: boolean;
 };
 
-let IS_TRANSFORMED = false;
-
 const hotReloadExtension = (options: hotReloadExtensionOptions): Plugin => {
   const { log, backgroundPath } = options;
   let ws: WebSocket | null = null;
@@ -28,13 +26,11 @@ const hotReloadExtension = (options: hotReloadExtensionOptions): Plugin => {
   return {
     name: PLUGIN_NAME,
     async transform(code: string, id: string) {
-      if (!isDev || IS_TRANSFORMED) {
+      if (!isDev) {
         return;
       }
 
       if (id.includes(backgroundPath)) {
-        IS_TRANSFORMED = true;
-
         const buffer = fs.readFileSync(resolve(__dirname, 'scripts/background-reload.js'));
         return {
           code: code + buffer.toString()
@@ -50,7 +46,6 @@ const hotReloadExtension = (options: hotReloadExtensionOptions): Plugin => {
         chalkLogger.red('Load extension to browser...');
         return;
       }
-
       setTimeout(() => {
         ws?.send(Message.FILE_CHANGE);
         if (log) chalkLogger.green('Extension Reloaded...');

diff --git a/src/scripts/background-reload.ts b/src/scripts/background-reload.ts
@@ -4,9 +4,14 @@ import { HOT_RELOAD_EXTENSION_VITE_PORT, Message } from '../utils';
  */
 const socket = new WebSocket(`ws://localhost:${HOT_RELOAD_EXTENSION_VITE_PORT}`);
 
+chrome.runtime.onInstalled.addListener((details) => {
+  if (details.reason === chrome.runtime.OnInstalledReason.UPDATE) {
+    chrome.tabs.reload();
+  }
+});
+
 socket.addEventListener('message', (event) => {
   if (event.data === Message.FILE_CHANGE) {
     chrome.runtime.reload();
-    chrome.tabs.reload();
   }
 });