From 1fe0296d73c394e3dfcefd67312bc58c6b9eeb7c Mon Sep 17 00:00:00 2001
From: Michael Chung <6252325+ArEnSc@users.noreply.github.com>
Date: Wed, 18 Sep 2024 13:44:29 -0400
Subject: [PATCH] Ok Modified the worker it should catch via vite.

---
 src/KonvaApp/RenderingPrimitives/RenderEngine.ts | 16 ++++------------
 .../Diffusion/DiffusionSharedWorkerClient.ts     | 15 ++++++++++++---
 2 files changed, 16 insertions(+), 15 deletions(-)

diff --git a/src/KonvaApp/RenderingPrimitives/RenderEngine.ts b/src/KonvaApp/RenderingPrimitives/RenderEngine.ts
index b69586dd50..7bb27ebf83 100644
--- a/src/KonvaApp/RenderingPrimitives/RenderEngine.ts
+++ b/src/KonvaApp/RenderingPrimitives/RenderEngine.ts
@@ -217,18 +217,10 @@ export class RenderEngine {
     if (typeof SharedWorker !== "undefined") {
       console.log("Shared Workers are supported in this browser.");
       // Debug chrome://inspect/#workers
-      if (import.meta.env.DEV) {
-        this.diffusionWorker = new DiffusionSharedWorkerClient(
-          "src\\KonvaApp\\SharedWorkers\\Diffusion\\DiffusionSharedWorker.ts",
-          this.onRenderingSystemMessageRecieved,
-        );
-      } else {
-        console.log("Initializing Shared Worker");
-        this.diffusionWorker = new DiffusionSharedWorkerClient(
-          "workers/DiffusionSharedWorker.ts",
-          this.onRenderingSystemMessageRecieved,
-        );
-      }
+      this.diffusionWorker = new DiffusionSharedWorkerClient(
+        "src\\KonvaApp\\SharedWorkers\\Diffusion\\DiffusionSharedWorker.ts",
+        this.onRenderingSystemMessageRecieved,
+      );
 
       this.canUseSharedWorker = true;
     } else {
diff --git a/src/KonvaApp/SharedWorkers/Diffusion/DiffusionSharedWorkerClient.ts b/src/KonvaApp/SharedWorkers/Diffusion/DiffusionSharedWorkerClient.ts
index d0001dab09..9ec489a361 100644
--- a/src/KonvaApp/SharedWorkers/Diffusion/DiffusionSharedWorkerClient.ts
+++ b/src/KonvaApp/SharedWorkers/Diffusion/DiffusionSharedWorkerClient.ts
@@ -27,9 +27,18 @@ export class DiffusionSharedWorkerClient<
     ) => void,
   ) {
     this.messageReceived = messageReceived;
-    this.sharedWorker = new SharedWorker(workerPath, {
-      type: "module",
-    });
+    if (import.meta.env.DEV) {
+      this.sharedWorker = new SharedWorker(workerPath, {
+        type: "module",
+      });
+    } else {
+      this.sharedWorker = new SharedWorker(
+        new URL(workerPath, import.meta.url),
+        {
+          type: "module",
+        },
+      );
+    }
     this.port = this.sharedWorker.port;
     this.port.onmessage = this.onMessage.bind(this);
     this.port.start();