feat: source-ref support rollup

Author: moonrailgun

package.json

@@ -20,7 +20,8 @@
   "pnpm": {
     "peerDependencyRules": {
       "ignoreMissing": [
-        "acorn"
+        "acorn",
+        "acorn-dynamic-import"
       ]
     }
   },

packages/plugin-declaration-generator/src/index.ts

@@ -3,12 +3,13 @@ import traverse from '@babel/traverse';
 import generate from '@babel/generator';
 import template from '@babel/template';
 import type { Comment } from '@babel/types';
+import { program } from '@babel/types';
 import fs from 'fs-extra';
 import _ from 'lodash';
 
 const babelPlugins: ParserPlugin[] = ['jsx', 'typescript'];
 const buildNamedExport = template('export function %%name%%(): any', {
-  plugins: babelPlugins as any,
+  plugins: babelPlugins,
 });
 
 interface Options {
@@ -26,10 +27,7 @@ export async function generateFunctionDeclare(options: Options) {
     });
   });
 
-  const code = generate({
-    type: 'Program',
-    body: _.flatten(astList),
-  } as any).code;
+  const code = generate(program(_.flatten(astList))).code;
 
   return code;
 }

packages/rollup-plugin-source-ref/README.md

@@ -0,0 +1 @@
+## source-ref for webpack loader

packages/rollup-plugin-source-ref/package.json

@@ -0,0 +1,37 @@
+{
+  "name": "rollup-plugin-source-ref",
+  "version": "1.0.0",
+  "description": "",
+  "main": "dist/index.js",
+  "files": [
+    "dist"
+  ],
+  "scripts": {
+    "build": "tsc",
+    "prepare": "tsc"
+  },
+  "keywords": [
+    "sourcecode",
+    "pointer",
+    "webpack"
+  ],
+  "author": "moonrailgun",
+  "license": "MIT",
+  "devDependencies": {
+    "@types/babel__generator": "^7.6.4",
+    "@types/babel__traverse": "^7.17.1",
+    "@types/node": "^15.12.5",
+    "@types/webpack": "^5.28.0",
+    "rollup": "^2.75.0",
+    "ts-node": "^10.0.0",
+    "typescript": "^4.5.2",
+    "webpack": "^5.72.0",
+    "webpack-test-utils": "^1.1.0"
+  },
+  "dependencies": {
+    "@babel/generator": "^7.17.7",
+    "@babel/parser": "^7.17.7",
+    "@babel/traverse": "^7.17.3",
+    "@babel/types": "^7.17.10"
+  }
+}

packages/rollup-plugin-source-ref/src/index.ts

@@ -0,0 +1,56 @@
+import type { Plugin } from 'rollup';
+import { parse } from '@babel/parser';
+import traverse from '@babel/traverse';
+import generate from '@babel/generator';
+import { jsxAttribute, jsxIdentifier, stringLiteral } from '@babel/types';
+
+const TRACE_ID = 'data-source';
+
+export default function sourceRef(): Plugin {
+  return {
+    name: 'source-ref',
+    transform(code, id) {
+      const filepath = id;
+
+      const ast = parse(code, {
+        sourceType: 'module',
+        plugins: ['jsx', 'typescript'],
+      });
+
+      traverse(ast, {
+        JSXOpeningElement(path) {
+          const location = path.node.loc;
+          if (!location) {
+            return;
+          }
+
+          if (Array.isArray(location)) {
+            return;
+          }
+
+          const line = location.start.line;
+          const col = location.start.column;
+
+          const attrs = path.node.attributes;
+          for (let i = 0; i < attrs.length; i++) {
+            const attr = attrs[i];
+            if (attr.type === 'JSXAttribute' && attr.name.name === TRACE_ID) {
+              // existed
+              return;
+            }
+          }
+
+          const traceId = `${filepath}:${line}:${col}`;
+
+          attrs.push(
+            jsxAttribute(jsxIdentifier(TRACE_ID), stringLiteral(traceId))
+          );
+        },
+      });
+
+      const res = generate(ast);
+
+      return { code: res.code, map: res.map };
+    },
+  };
+}

packages/rollup-plugin-source-ref/tsconfig.json

@@ -0,0 +1,14 @@
+{
+	"compilerOptions": {
+		"outDir": "dist",
+		"module": "commonjs",
+		"target": "ES5",
+		"lib": ["ESNext"],
+		"declaration": true,
+		"noUnusedLocals": true,
+		"noUnusedParameters": true,
+		"esModuleInterop": true,
+		"skipLibCheck": true
+	},
+	"include": ["src"]
+}