vite-plugin-simple-macro 0.9.1 → 1.0.0

package/dist/index.d.ts

@@ -1,6 +1,7 @@
 import { FilterExpression } from '@rolldown/pluginutils';
 import { ModuleType } from 'rolldown';
 import { NodePath } from '@babel/traverse';
+import { NodePath as NodePath_2 } from '@babel/core';
 import { Plugin } from 'rolldown';
 import { types } from '@babel/core';
 
@@ -12,6 +13,10 @@ export declare interface Context {
     ssr?: boolean;
 }
 
+export declare function ensureGlobal(path: NodePath_2<types.CallExpression>, name: string): boolean;
+
+export declare function ensureImport(path: NodePath_2<types.CallExpression>, module: string, name?: string): boolean;
+
 export declare interface Macro {
     filter?: FilterExpression;
     transform(this: Context, path: NodePath<types.CallExpression>, ctx: Context, t: typeof types): boolean;

package/dist/index.mjs

@@ -1,7 +1,7 @@
 import { and, exprInterpreter, id, include, makeIdFiltersToMatchWithQuery, or } from "@rolldown/pluginutils";
 import { parseAsync, traverse, types } from "@babel/core";
 import { generate } from "@babel/generator";
-//#region src/index.ts
+//#region src/util.ts
 var PREREQUISITE = id(makeIdFiltersToMatchWithQuery(/\.[mc]?[tj]sx?$/));
 var VISITOR = { CallExpression(path, ctx) {
 	for (const macro of ctx.macro) {
@@ -10,6 +10,50 @@ var VISITOR = { CallExpression(path, ctx) {
 		break;
 	}
 } };
+function addSpecificFilter(macro, inner) {
+	const out = [];
+	for (const { filter } of macro) if (filter) out.push(filter);
+	else return inner;
+	return and(inner, or(...out));
+}
+//#endregion
+//#region src/filter.ts
+var DEFAULT_IMPORT_NAME = "default";
+function ensureGlobal(path, name) {
+	const callee = path.get("callee");
+	if (!callee.isIdentifier()) return false;
+	if (callee.node.name !== name) return false;
+	return !path.scope.getBinding(name);
+}
+function ensureImport(path, module, name = DEFAULT_IMPORT_NAME) {
+	const callee = path.get("callee");
+	if (callee.isMemberExpression()) return ensureImportNamespace(callee, module, name);
+	if (!callee.isIdentifier()) return false;
+	const specifier = callee.scope.getBinding(callee.node.name)?.path;
+	if (!specifier) return false;
+	if (specifier.isImportDefaultSpecifier()) return name === DEFAULT_IMPORT_NAME && ensureModule(specifier, module);
+	if (!specifier.isImportSpecifier()) return false;
+	if (!ensureContent(specifier.get("imported"), name)) return false;
+	return ensureModule(specifier, module);
+}
+function ensureImportNamespace(path, module, name) {
+	if (!ensureContent(path.get("property"), name)) return false;
+	const obj = path.get("object");
+	if (!obj.isIdentifier()) return false;
+	const specifier = path.scope.getBinding(obj.node.name)?.path;
+	if (!specifier?.isImportNamespaceSpecifier()) return false;
+	return ensureModule(specifier, module);
+}
+function ensureModule(path, module) {
+	const declaration = path.parentPath;
+	if (!declaration.isImportDeclaration()) throw declaration.buildCodeFrameError("The parent of an import specifier should always be an import declaration");
+	return declaration.get("source").node.value === module;
+}
+function ensureContent(path, content) {
+	return path.isIdentifier() && path.node.name === content || path.isStringLiteral() && path.node.value === content;
+}
+//#endregion
+//#region src/index.ts
 function macroPlugin(opts) {
 	const { filter, macro } = opts;
 	const inner = filter ? and(PREREQUISITE, filter) : PREREQUISITE;
@@ -46,11 +90,5 @@ function macroPlugin(opts) {
 		}
 	};
 }
-function addSpecificFilter(macro, inner) {
-	const out = [];
-	for (const { filter } of macro) if (filter) out.push(filter);
-	else return inner;
-	return and(inner, or(...out));
-}
 //#endregion
-export { macroPlugin as default };
+export { macroPlugin as default, ensureGlobal, ensureImport };

package/package.json

@@ -1,23 +1,19 @@
 {
   "name": "vite-plugin-simple-macro",
-  "version": "0.9.1",
+  "version": "1.0.0",
   "description": "A simple macro plugin for Vite",
   "author": "AFatNiBBa",
   "license": "ISC",
   "types": "./dist/index.d.ts",
-  "exports": {
-    "require": "./dist/index.js",
-    "import": "./dist/index.mjs",
-    "types": "./dist/index.d.ts"
-  },
+  "main": "./dist/index.mjs",
   "scripts": {
     "build": "vite build",
+    "clean": "git clean -Xdf",
     "save": "npm run build && npm publish"
   },
   "devDependencies": {
     "@microsoft/api-extractor": "^7.58.9",
     "@types/babel__core": "^7.20.5",
-    "@types/node": "^20.6.0",
     "@typescript/typescript6": "^6.0.2",
     "unplugin-dts": "^1.0.3",
     "vite": "^8.1.3"
@@ -25,7 +21,9 @@
   "dependencies": {
     "@babel/core": "^7.0.0",
     "@babel/plugin-syntax-typescript": "^7.0.0",
-    "@rolldown/pluginutils": "^1.0.1",
+    "@rolldown/pluginutils": "^1.0.1"
+  },
+  "peerDependencies": {
     "rolldown": "^1.1.4"
   },
   "repository": {

package/dist/index.js

@@ -1,56 +0,0 @@
-let _rolldown_pluginutils = require("@rolldown/pluginutils");
-let _babel_core = require("@babel/core");
-let _babel_generator = require("@babel/generator");
-//#region src/index.ts
-var PREREQUISITE = (0, _rolldown_pluginutils.id)((0, _rolldown_pluginutils.makeIdFiltersToMatchWithQuery)(/\.[mc]?[tj]sx?$/));
-var VISITOR = { CallExpression(path, ctx) {
-	for (const macro of ctx.macro) {
-		if (!macro.transform.call(ctx, path, ctx, _babel_core.types)) continue;
-		ctx.changed++;
-		break;
-	}
-} };
-function macroPlugin(opts) {
-	const { filter, macro } = opts;
-	const inner = filter ? (0, _rolldown_pluginutils.and)(PREREQUISITE, filter) : PREREQUISITE;
-	return {
-		name: "vite-plugin-simple-macro",
-		transform: {
-			order: "pre",
-			filter: [(0, _rolldown_pluginutils.include)(addSpecificFilter(macro, inner))],
-			async handler(code, id, info) {
-				if (!(0, _rolldown_pluginutils.exprInterpreter)(inner, code, id, info.moduleType)) return;
-				const filtered = macro.filter(({ filter }) => !filter || (0, _rolldown_pluginutils.exprInterpreter)(filter, code, id, info.moduleType));
-				if (!filtered.length) return;
-				const ast = await (0, _babel_core.parseAsync)(code, {
-					filename: id,
-					plugins: [["@babel/plugin-syntax-typescript", { isTSX: true }]]
-				});
-				const ctx = {
-					id,
-					macro: filtered,
-					changed: 0,
-					...info
-				};
-				(0, _babel_core.traverse)(ast, VISITOR, void 0, ctx);
-				if (!ctx.changed) return;
-				const out = (0, _babel_generator.generate)(ast, {
-					sourceMaps: true,
-					sourceFileName: id
-				});
-				return {
-					code: out.code,
-					map: out.map
-				};
-			}
-		}
-	};
-}
-function addSpecificFilter(macro, inner) {
-	const out = [];
-	for (const { filter } of macro) if (filter) out.push(filter);
-	else return inner;
-	return (0, _rolldown_pluginutils.and)(inner, (0, _rolldown_pluginutils.or)(...out));
-}
-//#endregion
-module.exports = macroPlugin;