feat: add macro context

src/core/index.ts

@@ -16,6 +16,11 @@ import { MagicString, generateTransform } from 'magic-string-ast'
 import { type ImportAttribute, type Node } from '@babel/types'
 import { type ViteNodeRunner } from 'vite-node/client'
 
+export * from './options'
+export interface MacroContext {
+  id: string
+}
+
 interface MacroBase {
   node: Node
   id: string[]
@@ -140,7 +145,10 @@ export async function transformMacros(
 
     let ret: any
     if (macro.type === 'call') {
-      ret = exported(...macro.args)
+      const ctx: MacroContext = {
+        id,
+      }
+      ret = (exported as Function).apply(ctx, macro.args)
     } else {
       ret = exported
     }

src/index.ts

@@ -7,6 +7,8 @@ import { installSourcemapsSupport } from 'vite-node/source-map'
 import { type Options, resolveOptions } from './core/options'
 import { transformMacros } from './core'
 
+export type { Options, MacroContext } from './core'
+
 export default createUnplugin<Options | undefined, false>((rawOptions = {}) => {
   const options = resolveOptions(rawOptions)
   const filter = createFilter(options.include, options.exclude)

tests/__snapshots__/fixtures.test.ts.snap

@@ -20,6 +20,11 @@ const { bar = 1 } = {};
 "
 `;
 
+exports[`fixture > tests/fixtures/ctx.js 1`] = `
+"\\"ctx.js\\" === 'ctx.js';
+"
+`;
+
 exports[`fixture > tests/fixtures/dedent.js 1`] = `
 "const msg = \\"if (true) {\\\\n  const a = 'b'\\\\n}\\";
 "

tests/fixtures/ctx.js

@@ -0,0 +1,3 @@
+import { getCtx } from './macros/ctx' assert { type: 'macro' }
+
+getCtx() === 'ctx.js'

tests/fixtures/macros/ctx.ts

@@ -0,0 +1,6 @@
+import { MacroContext } from '../../../src'
+import path from 'path'
+
+export function getCtx(this: MacroContext) {
+  return path.basename(this.id)
+}

tsconfig.json

@@ -1,8 +1,8 @@
 {
   "compilerOptions": {
-    "target": "es2021",
+    "target": "esnext",
     "module": "esnext",
-    "lib": ["es2021"],
+    "lib": ["es2022"],
     "strict": true,
     "esModuleInterop": true,
     "moduleResolution": "bundler",