feat: try to resolve array & object expression

closes #22
unplugin · Mar 4, 2024 · bddf800 · bddf800
1 parent ebd99cf
commit bddf800
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/src/core/index.ts b/src/core/index.ts
@@ -169,7 +169,13 @@ export async function transformMacros({
           if (!imports.has(id[0]) || scope.contains(id[0])) return
           const args = node.arguments.map((arg) => {
             if (isLiteralType(arg)) return resolveLiteral(arg)
-            throw new Error('Macro arguments must be literals.')
+            try {
+              if (isTypeOf(arg, ['ObjectExpression', 'ArrayExpression']))
+                return new Function(
+                  `return (${source.slice(arg.start!, arg.end!)})`,
+                )()
+            } catch {}
+            throw new Error('Macro arguments cannot be resolved.')
           })
 
           macros.push({

diff --git a/tests/__snapshots__/fixtures.test.ts.snap b/tests/__snapshots__/fixtures.test.ts.snap
@@ -6,6 +6,8 @@ exports[`fixture > tests/fixtures/args.js 1`] = `
 false === false;
 null === null;
 undefined === undefined;
+console.log({"a":"b"});
+console.log([1,false,"hello"]);
 "
 `;
 

diff --git a/tests/fixtures/args.js b/tests/fixtures/args.js
@@ -5,3 +5,5 @@ arg('1') === '1'
 arg(false) === false
 arg(null) === null
 arg() === undefined
+console.log(arg({ a: 'b' }))
+console.log(arg([1, false, 'hello']))