feat: added parse options

Author: Nejc

package.json

@@ -1,7 +1,7 @@
 {
   "name": "@nejcm/rollup-plugin-size-snapshot-vite",
   "private": false,
-  "version": "0.0.7",
+  "version": "0.0.8",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "files": [

src/index.ts

@@ -1,3 +1,4 @@
+import { Options as AcornOptions } from 'acorn';
 import bytes from 'bytes';
 import chalk from 'chalk';
 import gzipSize from 'gzip-size';
@@ -7,12 +8,11 @@ import * as snapshot from './snapshot';
 import { treeshakeWithRollup } from './treeshakeWithRollup';
 
 type Options = {
-  minify?: boolean;
-  snapshot?: string;
   snapshotPath?: string;
   matchSnapshot?: boolean;
   threshold?: number;
   printInfo?: boolean;
+  parseOptions?: AcornOptions;
 };
 
 type OutputOptions = {
@@ -42,6 +42,7 @@ const validateOptions = (options: any) => {
     'matchSnapshot',
     'threshold',
     'printInfo',
+    'parseOptions',
   ];
 
   const invalidKeys = Object.keys(options).filter(
@@ -86,7 +87,7 @@ export const sizeSnapshot = (options: Options = {}): Plugin => {
 
       const minified = (await minify(source)).code || '';
       const treeshakeSize = (code: string) =>
-        Promise.all([treeshakeWithRollup(code)]);
+        Promise.all([treeshakeWithRollup(code, options.parseOptions)]);
 
       return Promise.all([
         gzipSize(minified),

src/treeshakeWithRollup.ts

@@ -1,5 +1,5 @@
 import replace from '@rollup/plugin-replace';
-import { parse } from 'acorn';
+import { Options, parse } from 'acorn';
 import { rollup } from 'rollup';
 import { minify } from 'terser';
 
@@ -32,7 +32,10 @@ const resolvePlugin = ({ code }) => ({
   },
 });
 
-export const treeshakeWithRollup = (code: string): Promise<Output> => {
+export const treeshakeWithRollup = (
+  code: string,
+  parseOptions?: Options,
+): Promise<Output> => {
   const config = {
     name: 'rollup-plugin-size-snapshot-vite',
     input: `/${inputName}`,
@@ -54,21 +57,23 @@ export const treeshakeWithRollup = (code: string): Promise<Output> => {
         },
       ),
     )
-    .then(
-      (result): Output => {
-        const { code = '' } = result;
-        const ast = parse(code, { sourceType: 'module' }) as any;
-        const import_statements = ast.body
-          // collect all toplevel import statements
-          .filter((node) => node.type === 'ImportDeclaration')
-          // endpos is the next character after node -> substract 1
-          .map((node) => node.end - node.start)
-          .reduce((acc, size) => acc + size, 0);
+    .then((result): Output => {
+      const { code = '' } = result;
+      const ast = parse(code, {
+        sourceType: 'module',
+        ecmaVersion: 11,
+        ...parseOptions,
+      }) as any;
+      const import_statements = ast.body
+        // collect all toplevel import statements
+        .filter((node) => node.type === 'ImportDeclaration')
+        // endpos is the next character after node -> substract 1
+        .map((node) => node.end - node.start)
+        .reduce((acc, size) => acc + size, 0);
 
-        return {
-          code: code.length,
-          import_statements,
-        };
-      },
-    );
+      return {
+        code: code.length,
+        import_statements,
+      };
+    });
 };