feat: init plugin transform

packages/vite-plugin-worker/package.json

@@ -46,12 +46,19 @@
     "test:ci": "vitest --run",
     "typecheck": "tsc --noEmit"
   },
+  "dependencies": {
+    "mlly": "^1.1.0"
+  },
   "devDependencies": {
-    "@types/node": "^18.0.0",
+    "@types/node": "^18.11.18",
+    "@types/sharedworker": "^0.0.91",
     "unbuild": "^1.1.1",
     "vite": "^4.0.4",
     "vitest": "^0.28.3"
   },
+  "peerDependencies": {
+    "vite": "^4.0.4"
+  },
   "engines": {
     "node": ">=v16.16.0"
   }

packages/vite-plugin-worker/src/index.ts

@@ -1,3 +1,47 @@
-export function hello() {
-  return 'world';
-}
+import { type Plugin, normalizePath } from 'vite';
+
+import path from 'path';
+import { findExports } from 'mlly';
+
+export default function SharedWorker(options?: { root?: string }): Plugin {
+  let root = process.cwd();
+  const { root: workerRoot = './worker' } = options ?? {};
+
+  return {
+    name: 'vite-plugin-sharedworker:client',
+    enforce: 'pre',
+    configResolved(config) {
+      root = normalizePath(path.join(config.root, workerRoot));
+    },
+    transform(code, id) {
+      if (id.startsWith(root)) {
+        const isClient = !id.endsWith('worker_file');
+
+        // console.log(id);
+        if (isClient) {
+          const exports = findExports(code);
+
+          return [
+            `const worker = new SharedWorker(new URL(${JSON.stringify(
+              id
+            )}, import.meta.url), { type: 'module', name: 'worker' });`,
+            `worker.port.start()`,
+            ...exports
+              .filter((ex) => ex.name)
+              .map((ex) => `export const ${ex.name} = (...args: any[]) => {}`)
+          ].join('\n');
+        } else {
+          const imports = [
+            `// vite-plugin-sharedworker starts`,
+            `import { defineSharedWorker } from 'vite-plugin-sharedworker/runtime'`,
+            `defineSharedWorker(self)`,
+            `// vite-plugin-sharedworker ends`,
+            ''
+          ];
+
+          return imports.join('\n') + '\n' + code;
+        }
+      }
+    }
+  };
+}

packages/vite-plugin-worker/src/runtime.ts

@@ -1,3 +1,5 @@
+/// <reference lib="WebWorker" />
+
 export function defineSharedWorker(self: SharedWorkerGlobalScope) {
   self.addEventListener('connect', (event) => {
     console.log('connect', event);

packages/vite-plugin-worker/tsconfig.json

@@ -3,7 +3,7 @@
   "compilerOptions": {
     "target": "ESNext",
     "module": "ESNext",
-    "lib": ["ESNext", "WebWorker"],
+    "lib": ["ESNext"],
     "moduleResolution": "Node",
     "esModuleInterop": true,
     "strict": true,

playground/main.ts

@@ -1,17 +1,16 @@
-const worker1 = new SharedWorker(new URL('./worker.ts', import.meta.url), {
-  type: 'module'
-});
-
-worker1.port.start();
+import { add } from './worker';
 
-worker1.port.postMessage('hello 1');
+console.log(add(1, 2));
 
-const worker2 = new SharedWorker(new URL('./worker.ts', import.meta.url), {
-  type: 'module'
+const worker = new SharedWorker(new URL('./worker/index.ts', import.meta.url), {
+  type: 'module',
+  name: 'worker'
 });
 
-worker2.port.start();
+worker.port.start();
+
+worker.port.postMessage('hello 1');
 
-worker2.port.postMessage('hello 2');
+console.log('OK');
 
 export {};

playground/tsconfig.worker.json

@@ -3,6 +3,6 @@
     "lib": ["ESNext", "WebWorker"],
     "types": ["vite/client"]
   },
-  "include": ["./worker.ts"],
+  "include": ["./worker/*.ts"],
   "extends": "./tsconfig.json"
 }

playground/vite.config.ts

@@ -1,7 +1,8 @@
 import { defineConfig } from 'vite';
 
 import BuildInfo from 'vite-plugin-info';
+import SharedWorkerClient from 'vite-plugin-sharedworker';
 
 export default defineConfig({
-  plugins: [BuildInfo()]
+  plugins: [BuildInfo(), SharedWorkerClient()]
 });

playground/worker/index.ts

@@ -0,0 +1,9 @@
+// /// <reference lib="WebWorker" />
+// declare const self: SharedWorkerGlobalScope;
+// import { defineSharedWorker } from 'vite-plugin-sharedworker/runtime';
+// defineSharedWorker(self);
+// export type {};
+
+export function add(a: number, b: number) {
+  return a + b;
+}