Merge pull request #2 from liip/multiple-visitors

Allow multiple visitors to transform AST
liip · Oct 11, 2023 · 9ae706b · 9ae706b
2 parents b2907a3 + 3def68a
commit 9ae706b
Show file tree

Hide file tree

Showing 13 changed files with 296 additions and 38 deletions.
diff --git a/packages/class-prefixer-ast-visitor/src/__tests__/parse-class-arguments.spec.ts b/packages/class-prefixer-ast-visitor/src/__tests__/parse-class-arguments.spec.ts
@@ -5,6 +5,7 @@ import type {
   TemplateLiteral,
   ObjectExpression,
   ArrayExpression,
+  CallExpression,
 } from 'estree';
 
 describe('parseClassArguments', () => {
@@ -262,4 +263,125 @@ describe('parseClassArguments', () => {
 
     expect(result).toEqual(expectedNode);
   });
+
+  it('should handle CallExpression node and update required value', () => {
+    const originalNode: CallExpression = {
+      type: 'CallExpression',
+      optional: false,
+      callee: {
+        type: 'Identifier',
+        name: 'calleeName',
+      },
+      arguments: [
+        {
+          type: 'ArrayExpression',
+          elements: [
+            {
+              type: 'Literal',
+              value: 'value second-value',
+              raw: "'value second-value'",
+            },
+            {
+              type: 'ObjectExpression',
+              properties: [
+                {
+                  type: 'Property',
+                  key: {
+                    type: 'Literal',
+                    value: 'third-value',
+                    raw: "'third-value'",
+                  },
+                  value: {
+                    type: 'MemberExpression',
+                    object: { type: 'ThisExpression' },
+                    property: {
+                      type: 'Identifier',
+                      name: 'actionable',
+                    },
+                    computed: false,
+                    optional: false,
+                  },
+                  kind: 'init',
+                  method: false,
+                  shorthand: false,
+                  computed: false,
+                },
+              ],
+            },
+            {
+              type: 'MemberExpression',
+              object: { type: 'ThisExpression' },
+              property: {
+                type: 'Identifier',
+                name: 'rootClass',
+              },
+              computed: false,
+              optional: false,
+            },
+          ],
+        },
+      ],
+    };
+
+    const expectedNode: CallExpression = {
+      type: 'CallExpression',
+      optional: false,
+      callee: {
+        type: 'Identifier',
+        name: 'calleeName',
+      },
+      arguments: [
+        {
+          type: 'ArrayExpression',
+          elements: [
+            {
+              type: 'Literal',
+              value: `${prefix.value}value ${prefix.value}second-value`,
+              raw: `'${prefix.value}value ${prefix.value}second-value'`,
+            },
+            {
+              type: 'ObjectExpression',
+              properties: [
+                {
+                  type: 'Property',
+                  key: {
+                    type: 'Literal',
+                    value: `${prefix.value}third-value`,
+                    raw: `'${prefix.value}third-value'`,
+                  },
+                  value: {
+                    type: 'MemberExpression',
+                    object: { type: 'ThisExpression' },
+                    property: {
+                      type: 'Identifier',
+                      name: 'actionable',
+                    },
+                    computed: false,
+                    optional: false,
+                  },
+                  kind: 'init',
+                  method: false,
+                  shorthand: false,
+                  computed: false,
+                },
+              ],
+            },
+            {
+              type: 'MemberExpression',
+              object: { type: 'ThisExpression' },
+              property: {
+                type: 'Identifier',
+                name: 'rootClass',
+              },
+              computed: false,
+              optional: false,
+            },
+          ],
+        },
+      ],
+    };
+    const result = parseClassArguments(originalNode, { prefix });
+
+    expect(result).toEqual(expectedNode);
+  });
 });
diff --git a/packages/class-prefixer-ast-visitor/src/__tests__/update-literals.spec.ts b/packages/class-prefixer-ast-visitor/src/__tests__/update-literals.spec.ts
@@ -1,3 +1,5 @@
+import { Prefixer } from '@liip/class-prefixer-core';
+
 import { updateLiterals } from '../update-literals';
 
 import type { Literal, TemplateLiteral } from 'estree';
@@ -23,7 +25,7 @@ describe('parseSelectorArguments', () => {
 
     const options = { prefix };
 
-    updateLiterals(originalNode, prefixer, options);
+    updateLiterals(originalNode, prefixer as Prefixer, options);
 
     expect(prefixer).toHaveBeenCalledWith(originalNode.value, options);
   });
@@ -48,7 +50,7 @@ describe('parseSelectorArguments', () => {
 
     const options = { prefix };
 
-    updateLiterals(originalNode, prefixer, options);
+    updateLiterals(originalNode, prefixer as Prefixer, options);
 
     expect(prefixer).toHaveBeenNthCalledWith(
       1,

diff --git a/packages/class-prefixer-ast-visitor/src/create-vue-template-visitor.ts b/packages/class-prefixer-ast-visitor/src/create-vue-template-visitor.ts
@@ -12,7 +12,7 @@ export const createVueTemplateVisitor = (config: PrefixerOptions): Visitor => ({
     if (
       node.type === 'Property' &&
       node.key.type === 'Identifier' &&
-      ['staticClass', 'class'].includes(node.key.name)
+      node.key.name === 'class'
     ) {
       return {
         ...node,

diff --git a/packages/class-prefixer-ast-visitor/src/parse-class-arguments.ts b/packages/class-prefixer-ast-visitor/src/parse-class-arguments.ts
@@ -14,7 +14,14 @@ export function parseClassArguments<N extends Node>(
   node: N,
   options?: PrefixerOptions,
 ): N {
-  if (node.type === 'ArrayExpression' && node.elements) {
+  if (node.type === 'CallExpression' && node.arguments) {
+    return {
+      ...node,
+      arguments: node.arguments.map(
+        (argument) => argument && parseClassArguments(argument, options),
+      ),
+    };
+  } else if (node.type === 'ArrayExpression' && node.elements) {
     return {
       ...node,
       elements: node.elements.map(

diff --git a/packages/esbuild-plugin-ast-vue/src/__tests__/plugin.spec.ts b/packages/esbuild-plugin-ast-vue/src/__tests__/plugin.spec.ts
@@ -12,7 +12,7 @@ describe('astPluginVue', () => {
   const virtualPackage = 'ast-plugin-vue';
 
   const pluginOptions: AstParserVueOptions = {
-    visitor: {
+    visitors: {
       enter(node) {
         if (
           node.type === 'CallExpression' &&
@@ -29,7 +29,7 @@ describe('astPluginVue', () => {
         if (
           node.type === 'Property' &&
           node.key.type === 'Identifier' &&
-          ['staticClass', 'class'].includes(node.key.name)
+          node.key.name === 'class'
         ) {
           return {
             ...node,

diff --git a/packages/esbuild-plugin-ast-vue/src/plugin.ts b/packages/esbuild-plugin-ast-vue/src/plugin.ts
@@ -15,6 +15,8 @@ import { resolveStyle } from './style';
 import { resolveTemplate } from './template';
 import { resolvePath, validateDependency } from './utils';
 
+type ExtractNonArray<T> = T extends Array<any> ? never : T;
+
 export interface AstParserVueOptions extends AstParserOptions {
   templateOptions?: Pick<
     SFCTemplateCompileOptions,
@@ -33,7 +35,7 @@ export interface AstParserVueOptions extends AstParserOptions {
     | 'postcssOptions'
     | 'postcssPlugins'
   >;
-  templateVisitor: AstParserOptions['visitor'];
+  templateVisitor: ExtractNonArray<AstParserOptions['visitors']>;
 }
 
 validateDependency();
@@ -42,7 +44,7 @@ export function astParserVue({
   templateOptions,
   scriptOptions,
   styleOptions,
-  visitor,
+  visitors,
   templateVisitor,
 }: AstParserVueOptions): Plugin {
   return {
@@ -88,7 +90,7 @@ export function astParserVue({
         });
 
         return {
-          contents: parser(code, visitor),
+          contents: parser(code, [visitors, templateVisitor].flat()),
           errors: error,
           resolveDir: dirname,
           loader: isTs ? 'ts' : 'js',

diff --git a/packages/esbuild-plugin-ast-vue/src/script.ts b/packages/esbuild-plugin-ast-vue/src/script.ts
@@ -1,4 +1,8 @@
-import { compileScript } from '@vue/compiler-sfc';
+import {
+  SFCDescriptor,
+  SFCScriptBlock,
+  compileScript,
+} from '@vue/compiler-sfc';
 import { fromObject } from 'convert-source-map';
 import { PartialMessage } from 'esbuild';
 
@@ -7,6 +11,8 @@ import { getTemplateOptions } from './template';
 
 import type { AstParserVueOptions } from './plugin';
 
+export const scriptCache = new WeakMap<SFCDescriptor, SFCScriptBlock | null>();
+
 export function resolveScript({
   filename,
   scriptOptions = {},
@@ -21,6 +27,17 @@ export function resolveScript({
   sourcemap: boolean;
 }) {
   const descriptor = getDescriptorCache(filename);
+
+  const cached = scriptCache.get(descriptor);
+
+  if (cached) {
+    return {
+      code: cached.content,
+      isTs: cached.lang === 'ts',
+      error: [],
+    };
+  }
+
   const error: PartialMessage[] = [];
   const { script, scriptSetup } = descriptor;
   const isTs =
@@ -43,14 +60,22 @@ export function resolveScript({
       inlineTemplate: true,
       babelParserPlugins: scriptOptions.babelParserPlugins,
       templateOptions: descriptor.template
-        ? getTemplateOptions({ descriptor, options: templateOptions, isProd })
+        ? getTemplateOptions({
+            descriptor,
+            options: templateOptions,
+            isProd,
+            scopeId,
+          })
         : {},
     });
 
     code = res.content;
+
     if (res.map) {
       code += fromObject(res.map).toComment();
     }
+
+    scriptCache.set(descriptor, res);
   } catch (e: any) {
     error.push({
       text: e.message,

diff --git a/packages/esbuild-plugin-ast-vue/src/template.ts b/packages/esbuild-plugin-ast-vue/src/template.ts
@@ -7,6 +7,7 @@ import { fromObject } from 'convert-source-map';
 import { PartialMessage } from 'esbuild';
 
 import { getDescriptorCache, getId } from './cache';
+import { scriptCache } from './script';
 
 import type { AstParserVueOptions } from './plugin';
 
@@ -20,10 +21,14 @@ export function resolveTemplate({
   isProd: boolean;
 }) {
   const descriptor = getDescriptorCache(filename);
+  const scopeId = getId(filename);
 
-  let { code, errors, map } = compileTemplate(
-    getTemplateOptions({ descriptor, options, isProd }),
-  );
+  let { code, errors, map } = compileTemplate({
+    ...getTemplateOptions({ descriptor, options, isProd, scopeId }),
+    id: scopeId,
+    source: descriptor.template?.content || '',
+    filename: descriptor.filename,
+  });
 
   if (map) {
     code += fromObject(map).toComment();
@@ -51,27 +56,33 @@ export function getTemplateOptions({
   descriptor,
   options,
   isProd,
+  scopeId,
 }: {
   descriptor: SFCDescriptor;
   options: AstParserVueOptions['templateOptions'];
   isProd: boolean;
-}): SFCTemplateCompileOptions {
-  const filename = descriptor.filename;
-  const scopeId = getId(filename);
+  scopeId: string;
+}): Omit<SFCTemplateCompileOptions, 'source'> | undefined {
+  const block = descriptor.template;
+
+  if (!block) {
+    return;
+  }
+
+  const hasScoped = descriptor.styles.some((s) => s.scoped);
+  const resolvedScript = scriptCache.get(descriptor);
 
   return {
-    source: descriptor.template?.content || '',
-    filename,
     id: scopeId,
-    scoped: descriptor.styles.some((s) => s.scoped),
+    scoped: hasScoped,
     isProd,
-    inMap: descriptor.template?.map,
+    filename: descriptor.filename,
+    inMap: block.src ? undefined : block.map,
     compiler: options?.compiler,
-    preprocessLang: options?.preprocessLang,
-    preprocessOptions: options?.preprocessOptions,
     compilerOptions: {
       ...options?.compilerOptions,
       scopeId,
+      bindingMetadata: resolvedScript ? resolvedScript.bindings : undefined,
     },
     transformAssetUrls: options?.transformAssetUrls,
   };