feat: refactor and support lock core-js path

Author: fz6m

.gitignore

@@ -8,3 +8,4 @@ node_modules
 
 /swc_plugin_auto_css_modules.wasm
 /example/.swc
+/debug

Cargo.lock

@@ -1,24 +1,24 @@
 [package]
 name = "swc_plugin_auto_css_modules"
-version = "1.0.0"
+version = "1.1.0"
 edition = "2021"
 authors = ["fz6m"]
 description = "Auto css modules plugin for swc"
 license = "MIT"
+repository = "https://github.com/xn-sakina/swc-plugin-auto-css-modules.git"
 publish = false
 
 [lib]
-crate-type = ["cdylib"]
+crate-type = ["cdylib", "rlib"]
 
 [profile.release]
 lto = true
 
 [dependencies]
-serde = "1"
+serde_json = "1.0.79"
+auto_css_modules = { version = "1.0.1", path = "./transform" }
 swc_core = { version = "0.48.29", features = [
   "ecma_plugin_transform",
-  "ecma_utils",
   "ecma_visit",
   "ecma_ast",
-  "common",
 ] }

Cargo.toml

@@ -0,0 +1,39 @@
+import { transformSync } from '@swc/core'
+import type { ISwcPluginAutoCssModulesConfig } from '../'
+
+export const transform = (
+  code: string,
+  opts: ISwcPluginAutoCssModulesConfig = {}
+) => {
+  return transformSync(code, {
+    module: {
+      type: 'es6',
+      ignoreDynamic: true,
+    },
+    jsc: {
+      parser: {
+        syntax: 'typescript',
+        dynamicImport: true,
+        tsx: true,
+      },
+      target: 'es5',
+      experimental: {
+        plugins: [
+          [
+            require.resolve(
+              '../target/wasm32-wasi/release/swc_plugin_auto_css_modules.wasm'
+            ),
+            opts,
+          ],
+        ],
+      },
+      minify: {
+        mangle: false,
+        compress: false,
+      },
+    },
+    minify: false,
+    filename: 'index.ts',
+    sourceMaps: false,
+  })
+}

example/_do.ts

@@ -0,0 +1,44 @@
+import assert from 'assert'
+import { transform } from './_do'
+
+const result = transform(
+  `
+import a from 'core-js';
+import b from 'core-js/cc';
+import c from 'core-js/cc/dd';
+
+import 'core-js';
+import 'core-js/cc';
+import 'core-js/cc/dd';
+
+import d from '../core-js';
+import e from './core-js';
+import { f } from 'foo';
+import * as g from 'foo';
+
+a, b, c, d, e, f, g;
+  `,
+  {
+    lock_core_js_pkg_path: '/core-js/absolute-path',
+    //   dirname(
+    //     require.resolve('core-js/package.json')
+    //   )
+  }
+)
+
+assert(
+  result.code.trim() ===
+    `
+import a from "core-js";
+import b from "/core-js/absolute-path/cc";
+import c from "/core-js/absolute-path/cc/dd";
+import "core-js";
+import "/core-js/absolute-path/cc";
+import "/core-js/absolute-path/cc/dd";
+import d from "../core-js";
+import e from "./core-js";
+import { f } from "foo";
+import * as g from "foo";
+a, b, c, d, e, f, g;
+  `.trim()
+)

example/corejs.ts

@@ -1,8 +1,7 @@
-import { transformSync } from '@swc/core'
 import assert from 'assert'
+import { transform } from './_do'
 
-const result = transformSync(
-  `
+const result = transform(`
 import a from 'foo.less';
 import b from 'foo.scss';
 import c from 'foo.sass';
@@ -11,32 +10,7 @@ import e from './foo.styl';
 import { f } from 'foo';
 import * as g from 'foo';
 a, b, c, d, e, f, g;
-`,
-  {
-    module: {
-      type: 'es6',
-      ignoreDynamic: true,
-    },
-    jsc: {
-      parser: {
-        syntax: 'typescript',
-        dynamicImport: true,
-        tsx: true,
-      },
-      target: 'es5',
-      experimental: {
-        plugins: [[require.resolve('../swc_plugin_auto_css_modules.wasm'), {}]],
-      },
-      minify: {
-        mangle: false,
-        compress: false,
-      },
-    },
-    minify: false,
-    filename: 'index.ts',
-    sourceMaps: false,
-  }
-)
+`)
 
 assert(
   result.code.trim() ===

example/index.ts

@@ -2,13 +2,16 @@
   "name": "example",
   "private": true,
   "scripts": {
-    "test": "echo \"Error: no test specified\" && exit 1"
+    "test": "pnpm test:corejs && pnpm test:index",
+    "test:corejs": "tsx ./corejs.ts",
+    "test:index": "tsx ./index.ts"
   },
   "devDependencies": {
     "@swc/core": "^1.3.24",
     "@types/node": "^18.11.17",
     "@xn-sakina/mental": "^2.0.0",
     "tsx": "^3.12.1",
-    "typescript": "^4.9.4"
+    "typescript": "^4.9.4",
+    "core-js": "^3.27.1"
   }
 }

example/package.json

@@ -0,0 +1,15 @@
+export interface ISwcPluginAutoCssModulesConfig {
+  /**
+   * lock core-js package root path
+   * @default undefined
+   * @example dirname(require.resolve('core-js/package.json'))
+   */
+  lock_core_js_pkg_path?: string
+
+  /**
+   * auto add style file suffix
+   * @default '?modules'
+   * @example '?module-mark'
+   */
+  style_file_suffix?: string
+}

index.d.ts

@@ -9,6 +9,7 @@
     "type": "git",
     "url": "https://github.com/xn-sakina/swc-plugin-auto-css-modules"
   },
+  "types": "index.d.ts",
   "keywords": [
     "swc-plugin",
     "auto-css-modules"
@@ -18,7 +19,7 @@
     "build": "cargo build-wasi --release",
     "move": "cp ./target/wasm32-wasi/release/swc_plugin_auto_css_modules.wasm .",
     "test:cargo": "cargo test",
-    "test:node": "cd ./example && pnpm tsx ./index.ts",
+    "test:node": "cd ./example && pnpm test",
     "test": "pnpm test:cargo && pnpm test:node",
     "prepublishOnly": "pnpm build && pnpm move && pnpm test",
     "push": "npm publish --access public --registry https://registry.npmjs.com/"

package.json

@@ -1,59 +1,17 @@
+#![allow(clippy::not_unsafe_ptr_arg_deref)]
 use swc_core::{
-    ecma::{
-        ast::{ImportDecl, ImportSpecifier, Program},
-        transforms::testing::test,
-        visit::{as_folder, FoldWith, VisitMut, VisitMutWith},
-    },
+    ecma::{ast::Program, visit::FoldWith},
     plugin::{plugin_transform, proxies::TransformPluginProgramMetadata},
 };
 
-pub struct TransformVisitor;
-
-const CSS_EXTS: [&str; 5] = [".css", ".less", ".scss", ".sass", ".styl"];
-
-impl VisitMut for TransformVisitor {
-    fn visit_mut_import_decl(&mut self, n: &mut ImportDecl) {
-        n.visit_mut_children_with(self);
-
-        if n.specifiers.len() == 1 {
-            if let ImportSpecifier::Default(_default_specifier) = &n.specifiers[0] {
-                let source = n.src.value.to_string();
-                for ext in CSS_EXTS {
-                    if source.ends_with(ext) {
-                        n.src = Box::new(format!("{}?modules", source).into());
-                        break;
-                    }
-                }
-            }
-        }
-    }
-}
-
 #[plugin_transform]
-pub fn auto_css_modules(program: Program, _metadata: TransformPluginProgramMetadata) -> Program {
-    program.fold_with(&mut as_folder(TransformVisitor))
-}
+pub fn auto_css_modules(program: Program, data: TransformPluginProgramMetadata) -> Program {
+    let config = serde_json::from_str(
+        &data
+            .get_transform_plugin_config()
+            .expect("failed to get plugin config for auto_css_modules"),
+    )
+    .expect("invalid packages");
 
-test!(
-    Default::default(),
-    |_| as_folder(TransformVisitor),
-    boo,
-    r#"
-import a from 'foo.less';
-import b from 'foo.scss';
-import c from 'foo.sass';
-import d from '../foo.css';
-import e from './foo.styl';
-import { f } from 'foo';
-import * as g from 'foo';
-"#,
-    r#"
-import a from "foo.less?modules";
-import b from "foo.scss?modules";
-import c from "foo.sass?modules";
-import d from "../foo.css?modules";
-import e from "./foo.styl?modules";
-import { f } from 'foo';
-import * as g from 'foo';
-"#
-);
+    program.fold_with(&mut auto_css_modules::auto_css_modules(config))
+}