vite-plugin-enter-dev 0.0.1 → 0.0.3

package/dist/index.d.ts

@@ -1,10 +1,13 @@
-import { PluginOption } from 'vite';
+import { Plugin } from 'vite';
 
-interface Options {
+interface DevOptions {
     /**
      * 是否注入消息监听器脚本（默认：true）
      */
     injectMessageListener?: boolean;
+}
+declare function enterDevPlugin(options?: DevOptions): Plugin[];
+interface ProdOptions {
     /**
      * 是否注入 Google Fonts（默认：true）
      */
@@ -14,6 +17,6 @@ interface Options {
      */
     googleFontsUrl?: string;
 }
-declare function enterDevPlugin(options?: Options): PluginOption[];
+declare function enterProdPlugin(options?: ProdOptions): Plugin[];
 
-export { enterDevPlugin as default };
+export { enterDevPlugin, enterProdPlugin };

package/dist/index.js

@@ -3,10 +3,189 @@ import react from '@vitejs/plugin-react';
 // src/index.ts
 
 // src/sourceInfoInjectPlugin.ts
+var threeFiberElems = /* @__PURE__ */ new Set([
+  "object3D",
+  "audioListener",
+  "positionalAudio",
+  "mesh",
+  "batchedMesh",
+  "instancedMesh",
+  "scene",
+  "sprite",
+  "lOD",
+  "skinnedMesh",
+  "skeleton",
+  "bone",
+  "lineSegments",
+  "lineLoop",
+  "points",
+  "group",
+  "camera",
+  "perspectiveCamera",
+  "orthographicCamera",
+  "cubeCamera",
+  "arrayCamera",
+  "instancedBufferGeometry",
+  "bufferGeometry",
+  "boxBufferGeometry",
+  "circleBufferGeometry",
+  "coneBufferGeometry",
+  "cylinderBufferGeometry",
+  "dodecahedronBufferGeometry",
+  "extrudeBufferGeometry",
+  "icosahedronBufferGeometry",
+  "latheBufferGeometry",
+  "octahedronBufferGeometry",
+  "planeBufferGeometry",
+  "polyhedronBufferGeometry",
+  "ringBufferGeometry",
+  "shapeBufferGeometry",
+  "sphereBufferGeometry",
+  "tetrahedronBufferGeometry",
+  "torusBufferGeometry",
+  "torusKnotBufferGeometry",
+  "tubeBufferGeometry",
+  "wireframeGeometry",
+  "tetrahedronGeometry",
+  "octahedronGeometry",
+  "icosahedronGeometry",
+  "dodecahedronGeometry",
+  "polyhedronGeometry",
+  "tubeGeometry",
+  "torusKnotGeometry",
+  "torusGeometry",
+  "sphereGeometry",
+  "ringGeometry",
+  "planeGeometry",
+  "latheGeometry",
+  "shapeGeometry",
+  "extrudeGeometry",
+  "edgesGeometry",
+  "coneGeometry",
+  "cylinderGeometry",
+  "circleGeometry",
+  "boxGeometry",
+  "capsuleGeometry",
+  "material",
+  "shadowMaterial",
+  "spriteMaterial",
+  "rawShaderMaterial",
+  "shaderMaterial",
+  "pointsMaterial",
+  "meshPhysicalMaterial",
+  "meshStandardMaterial",
+  "meshPhongMaterial",
+  "meshToonMaterial",
+  "meshNormalMaterial",
+  "meshLambertMaterial",
+  "meshDepthMaterial",
+  "meshDistanceMaterial",
+  "meshBasicMaterial",
+  "meshMatcapMaterial",
+  "lineDashedMaterial",
+  "lineBasicMaterial",
+  "primitive",
+  "light",
+  "spotLightShadow",
+  "spotLight",
+  "pointLight",
+  "rectAreaLight",
+  "hemisphereLight",
+  "directionalLightShadow",
+  "directionalLight",
+  "ambientLight",
+  "lightShadow",
+  "ambientLightProbe",
+  "hemisphereLightProbe",
+  "lightProbe",
+  "spotLightHelper",
+  "skeletonHelper",
+  "pointLightHelper",
+  "hemisphereLightHelper",
+  "gridHelper",
+  "polarGridHelper",
+  "directionalLightHelper",
+  "cameraHelper",
+  "boxHelper",
+  "box3Helper",
+  "planeHelper",
+  "arrowHelper",
+  "axesHelper",
+  "texture",
+  "videoTexture",
+  "dataTexture",
+  "dataTexture3D",
+  "compressedTexture",
+  "cubeTexture",
+  "canvasTexture",
+  "depthTexture",
+  "raycaster",
+  "vector2",
+  "vector3",
+  "vector4",
+  "euler",
+  "matrix3",
+  "matrix4",
+  "quaternion",
+  "bufferAttribute",
+  "float16BufferAttribute",
+  "float32BufferAttribute",
+  "float64BufferAttribute",
+  "int8BufferAttribute",
+  "int16BufferAttribute",
+  "int32BufferAttribute",
+  "uint8BufferAttribute",
+  "uint16BufferAttribute",
+  "uint32BufferAttribute",
+  "instancedBufferAttribute",
+  "color",
+  "fog",
+  "fogExp2",
+  "shape",
+  "colorShiftMaterial"
+]);
+function shouldTagElement(elementName, threeDreiImportedElements, threeDreiNamespaces) {
+  if (threeFiberElems.has(elementName)) {
+    return false;
+  }
+  if (threeDreiImportedElements.has(elementName)) {
+    return false;
+  }
+  if (elementName.includes(".")) {
+    const namespace = elementName.split(".")[0] || "";
+    if (threeDreiNamespaces.has(namespace)) {
+      return false;
+    }
+  }
+  return true;
+}
 function injectSourcePlugin({ types: t }) {
   return {
     name: "inject-source-location-enhanced",
     visitor: {
+      Program: {
+        enter(path, state) {
+          state.threeImportedElements = /* @__PURE__ */ new Set();
+          state.threeNamespaces = /* @__PURE__ */ new Set();
+        }
+      },
+      ImportDeclaration(path, state) {
+        const threeLibraries = [
+          "@react-three/fiber",
+          "@react-three/drei",
+          "@react-three/postprocessing",
+          "three"
+        ];
+        if (threeLibraries.includes(path.node.source.value)) {
+          path.node.specifiers.forEach((spec) => {
+            if (spec.type === "ImportSpecifier") {
+              state.threeImportedElements.add(spec.local.name);
+            } else if (spec.type === "ImportNamespaceSpecifier") {
+              state.threeNamespaces.add(spec.local.name);
+            }
+          });
+        }
+      },
       JSXElement(path, state) {
         const { node } = path;
         const filename = state.filename || state.file.opts.filename || "unknown";
@@ -16,6 +195,9 @@ function injectSourcePlugin({ types: t }) {
         const relativePath = filename.replace(/^.*\/src\//, "src/").replace(/\\/g, "/");
         const sourceLocation = `${relativePath}:${start.line}:${start.column + 1}`;
         const elementType = getElementType(node.openingElement.name);
+        if (!shouldTagElement(elementType, state.threeImportedElements || /* @__PURE__ */ new Set(), state.threeNamespaces || /* @__PURE__ */ new Set())) {
+          return;
+        }
         const elementInfo = {
           fileName: relativePath,
           lineNumber: start.line,
@@ -105,9 +287,7 @@ function getParentElementType(parentNode) {
 // src/index.ts
 function enterDevPlugin(options = {}) {
   const {
-    injectMessageListener = true,
-    injectGoogleFonts = true,
-    googleFontsUrl = "https://fonts.googleapis.com/css2?family=Inter:wght@100;200;300;400;500;600;700;800;900;950&family=Poppins:wght@100;200;300;400;500;600;700;800;900;950&family=Roboto:wght@100;200;300;400;500;600;700;800;900;950&display=swap"
+    injectMessageListener = true
   } = options;
   const reactPlugin = react({
     // 确保在开发模式下启用JSX源码映射
@@ -132,12 +312,13 @@ function enterDevPlugin(options = {}) {
     }
   });
   const htmlInjectPlugin = {
-    name: "enter-dev-custom-html-inject",
+    name: "enter-dev-message-listener",
     enforce: "pre",
     transformIndexHtml(html) {
-      let transformedHtml = html;
-      if (injectMessageListener) {
-        const messageListenerScript = `
+      if (!injectMessageListener) {
+        return html;
+      }
+      const messageListenerScript = `
   <script>
     // \u6DFB\u52A0\u6D88\u606F\u76D1\u542C\u5668\uFF0C\u63A5\u6536\u6765\u81EA\u4E3B\u5DE5\u7A0B\u7684\u6CE8\u5165\u811A\u672C
     window.addEventListener('message', (event) => {
@@ -157,26 +338,41 @@ function enterDevPlugin(options = {}) {
     });
 
   </script>`;
-        if (html.includes("</head>")) {
-          transformedHtml = transformedHtml.replace("</head>", `${messageListenerScript}
+      if (html.includes("</head>")) {
+        return html.replace("</head>", `${messageListenerScript}
 </head>`);
-        } else if (html.includes("</title>")) {
-          transformedHtml = transformedHtml.replace("</title>", `</title>${messageListenerScript}`);
-        }
+      } else if (html.includes("</title>")) {
+        return html.replace("</title>", `</title>${messageListenerScript}`);
+      }
+      return html;
+    }
+  };
+  return [...reactPlugin, htmlInjectPlugin];
+}
+function enterProdPlugin(options = {}) {
+  const {
+    injectGoogleFonts = true,
+    googleFontsUrl = "https://fonts.googleapis.com/css2?family=Inter:wght@100;200;300;400;500;600;700;800;900;950&family=Poppins:wght@100;200;300;400;500;600;700;800;900;950&family=Roboto:wght@100;200;300;400;500;600;700;800;900;950&display=swap"
+  } = options;
+  const fontsInjectPlugin = {
+    name: "enter-prod-fonts-inject",
+    enforce: "pre",
+    transformIndexHtml(html) {
+      if (!injectGoogleFonts) {
+        return html;
+      }
+      const googleFontsLink = `<link href="${googleFontsUrl}" rel="stylesheet">`;
+      if (html.includes(googleFontsUrl)) {
+        return html;
       }
-      if (injectGoogleFonts) {
-        const googleFontsLink = `<link href="${googleFontsUrl}" rel="stylesheet">`;
-        if (!transformedHtml.includes(googleFontsUrl)) {
-          if (transformedHtml.includes("</head>")) {
-            transformedHtml = transformedHtml.replace("</head>", `  ${googleFontsLink}
+      if (html.includes("</head>")) {
+        return html.replace("</head>", `  ${googleFontsLink}
 </head>`);
-          }
-        }
       }
-      return transformedHtml;
+      return html;
     }
   };
-  return [...reactPlugin, htmlInjectPlugin];
+  return [fontsInjectPlugin];
 }
 
-export { enterDevPlugin as default };
+export { enterDevPlugin, enterProdPlugin };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "vite-plugin-enter-dev",
-  "version": "0.0.1",
+  "version": "0.0.3",
   "type": "module",
   "private": false,
   "description": "A Vite plugin for development enhancements",
@@ -16,11 +16,6 @@
   "files": [
     "dist"
   ],
-  "scripts": {
-    "build": "tsup",
-    "dev": "tsup --watch",
-    "lint": "eslint . --max-warnings 0"
-  },
   "keywords": [
     "vite",
     "vite-plugin",
@@ -30,19 +25,23 @@
   ],
   "author": "",
   "license": "MIT",
-  "packageManager": "pnpm@10.15.0",
   "devDependencies": {
     "@types/node": "^20.19.9",
-    "@workspace/eslint-config": "workspace:*",
-    "@workspace/typescript-config": "workspace:*",
     "tsup": "^8.5.1",
     "typescript": "^5.9.2",
-    "vite": "^6.0.0"
+    "vite": "^6.0.0",
+    "@workspace/typescript-config": "0.0.0",
+    "@workspace/eslint-config": "0.0.0"
   },
   "peerDependencies": {
     "vite": "^5.0.0 || ^6.0.0"
   },
   "dependencies": {
     "@vitejs/plugin-react": "^5.1.1"
+  },
+  "scripts": {
+    "build": "tsup",
+    "dev": "tsup --watch",
+    "lint": "eslint . --max-warnings 0"
   }
-}
+}