Revert "experiment: preserve modules (#306)"

This reverts commit 6c563bc.

Author: CodyJasonBennett

.github/workflows/main.yml

@@ -28,6 +28,8 @@ jobs:
         run: yarn build
       - name: Check for SSR compat
         run: node dist/index.cjs && node dist/index.js
+      - name: Check tree-shaking
+        run: npx agadoo@latest dist/index.js
       - name: Check NPM config
         run: npx publint@latest dist
   release:

package.json

@@ -36,6 +36,8 @@
   },
   "sideEffects": false,
   "devDependencies": {
+    "@babel/core": "^7.22.20",
+    "@types/babel__core": "^7.20.2",
     "@types/node": "^20.6.3",
     "@types/three": "^0.128.0",
     "copyfiles": "^2.4.1",

vite.config.ts

@@ -1,4 +1,5 @@
 import * as path from 'node:path'
+import * as babel from '@babel/core'
 import { defineConfig } from 'vite'
 
 export default defineConfig({
@@ -13,9 +14,33 @@ export default defineConfig({
     },
     rollupOptions: {
       external: (id: string) => !id.startsWith('.') && !path.isAbsolute(id),
-      output: {
-        preserveModules: true,
-      },
     },
   },
+  plugins: [
+    {
+      name: 'vite-tree-shake',
+      renderChunk: {
+        order: 'post',
+        async handler(code) {
+          function annotate(path: babel.NodePath): void {
+            if (!path.getFunctionParent()) {
+              path.addComment('leading', '@__PURE__')
+            }
+          }
+
+          return babel.transform(code, {
+            sourceMaps: true,
+            plugins: [
+              {
+                visitor: {
+                  CallExpression: annotate,
+                  NewExpression: annotate,
+                },
+              },
+            ],
+          }) as any
+        },
+      },
+    },
+  ],
 })