feat(css/plugin): Support Wasm plugin (#6568)

crates/swc_core/src/lib.rs

@@ -12,8 +12,8 @@ pub mod quote;
 pub extern crate swc_ecma_quote_macros;
 
 // Plugins
-#[cfg(feature = "__plugin_transform")]
-#[cfg_attr(docsrs, doc(cfg(feature = "__plugin_transform")))]
+#[cfg(feature = "__common_plugin_transform")]
+#[cfg_attr(docsrs, doc(cfg(feature = "__common_plugin_transform")))]
 pub mod plugin;
 
 #[cfg(feature = "__ecma")]

crates/swc_core/src/plugin.rs

@@ -1,20 +1,20 @@
 // #[plugin_transform] macro
-#[cfg(any(docsrs, feature = "__plugin_transform"))]
-#[cfg_attr(docsrs, doc(cfg(feature = "__plugin_transform")))]
-pub use swc_plugin_macro::plugin_transform;
+#[cfg(any(docsrs, feature = "__common_plugin_transform"))]
+#[cfg_attr(docsrs, doc(cfg(feature = "__common_plugin_transform")))]
+pub use swc_plugin_macro::{css_plugin_transform, plugin_transform};
 
 /// exported __alloc / __free fn for the guest (plugin)
 /// allows to allocate memory from the host side.
 /// This should not be directly referenced.
-#[cfg(all(feature = "__plugin_transform", target_arch = "wasm32"))]
+#[cfg(all(feature = "__common_plugin_transform", target_arch = "wasm32"))]
 pub mod memory {
     pub use swc_plugin::allocation::*;
 }
 
 /// Global HANDLER implementation for the plugin
 /// for error reporting.
-#[cfg(any(docsrs, feature = "__plugin_transform"))]
-#[cfg_attr(docsrs, doc(cfg(feature = "__plugin_transform")))]
+#[cfg(any(docsrs, feature = "__common_plugin_transform"))]
+#[cfg_attr(docsrs, doc(cfg(feature = "__common_plugin_transform")))]
 pub mod errors {
     /// global context HANDLER in plugin's transform function.
     pub static HANDLER: swc_plugin::pseudo_scoped_key::PseudoScopedKey<
@@ -25,16 +25,16 @@ pub mod errors {
 }
 
 /// Plugin's environment metadata context.
-#[cfg(any(docsrs, feature = "__plugin_transform"))]
-#[cfg_attr(docsrs, doc(cfg(feature = "__plugin_transform")))]
+#[cfg(any(docsrs, feature = "__common_plugin_transform"))]
+#[cfg_attr(docsrs, doc(cfg(feature = "__common_plugin_transform")))]
 pub mod metadata {
     pub use swc_common::plugin::metadata::TransformPluginMetadataContextKind;
     pub use swc_plugin_proxy::TransformPluginProgramMetadata;
 }
 
 /// Proxy to the host's data not attached to the AST, like sourcemap / comments.
-#[cfg(any(docsrs, feature = "__plugin_transform"))]
-#[cfg_attr(docsrs, doc(cfg(feature = "__plugin_transform")))]
+#[cfg(any(docsrs, feature = "__common_plugin_transform"))]
+#[cfg_attr(docsrs, doc(cfg(feature = "__common_plugin_transform")))]
 pub mod proxies {
     pub use swc_plugin_proxy::*;
 }

crates/swc_plugin_macro/src/lib.rs

@@ -12,12 +12,25 @@ pub fn plugin_transform(
     let token = proc_macro2::TokenStream::from(input);
     let parsed_results = syn::parse2::<SynItem>(token).expect("Failed to parse tokens");
     match parsed_results {
-        SynItem::Fn(func) => handle_func(func),
+        SynItem::Fn(func) => handle_func(func, Ident::new("Program", Span::call_site())),
         _ => panic!("Please confirm if plugin macro is specified for the function"),
     }
 }
 
-fn handle_func(func: ItemFn) -> TokenStream {
+#[proc_macro_attribute]
+pub fn css_plugin_transform(
+    _args: proc_macro::TokenStream,
+    input: proc_macro::TokenStream,
+) -> proc_macro::TokenStream {
+    let token = proc_macro2::TokenStream::from(input);
+    let parsed_results = syn::parse2::<SynItem>(token).expect("Failed to parse tokens");
+    match parsed_results {
+        SynItem::Fn(func) => handle_func(func, Ident::new("Stylesheet", Span::call_site())),
+        _ => panic!("Please confirm if plugin macro is specified for the function"),
+    }
+}
+
+fn handle_func(func: ItemFn, ast_type: Ident) -> TokenStream {
     let ident = func.sig.ident.clone();
     let transform_process_impl_ident =
         Ident::new("__transform_plugin_process_impl", Span::call_site());
@@ -80,6 +93,7 @@ fn handle_func(func: ItemFn) -> TokenStream {
         }
 
         #[no_mangle]
+        #[allow(clippy::not_unsafe_ptr_arg_deref)]
         pub fn #transform_core_pkg_diag_ident() -> u32 {
             let schema_version = swc_core::common::plugin::PLUGIN_TRANSFORM_AST_SCHEMA_VERSION;
             let core_pkg_diag = swc_core::diagnostics::get_core_engine_diagnostics();
@@ -109,6 +123,7 @@ fn handle_func(func: ItemFn) -> TokenStream {
         // There are some cases error won't be wrapped up however - for example, we expect
         // serialization of PluginError itself should succeed.
         #[no_mangle]
+        #[allow(clippy::not_unsafe_ptr_arg_deref)]
         pub fn #transform_process_impl_ident(
             ast_ptr: *const u8, ast_ptr_len: u32,
             unresolved_mark: u32, should_enable_comments_proxy: i32) -> u32 {
@@ -119,7 +134,7 @@ fn handle_func(func: ItemFn) -> TokenStream {
                 let err = swc_core::common::plugin::serialized::PluginError::Deserialize("Failed to deserialize program received from host".to_string());
                 return construct_error_ptr(err);
             }
-            let program: Program = program.expect("Should be a program");
+            let program: #ast_type = program.expect("Should be a program");
 
             // Create a handler wired with plugin's diagnostic emitter, set it for global context.
             let handler = swc_core::common::errors::Handler::with_emitter(

crates/swc_plugin_runner/Cargo.toml

@@ -58,17 +58,20 @@ wasmer-compiler-cranelift = { version = "2.3.0" }
 wasmer-engine-universal   = { version = "2.3.0" }
 
 [dev-dependencies]
-criterion       = "0.3"
-swc_atoms       = { version = "0.4.25", path = '../swc_atoms' }
-swc_ecma_loader = { version = "0.41.18", path = "../swc_ecma_loader" }
-swc_ecma_parser = { version = "0.123.2", path = "../swc_ecma_parser" }
-swc_ecma_visit  = { version = "0.81.1", path = "../swc_ecma_visit" }
-swc_node_base   = { version = "0.5.8", path = "../swc_node_base" }
-testing         = { version = "0.31.18", path = "../testing" }
-# This allows we can run non-wasm32 target build command while some pkg select features for wasmer/js omits its transitive deps
-swc_ecma_ast = { version = "0.95.1", path = "../swc_ecma_ast", feature = [
+criterion = "0.3"
+swc_atoms = { version = "0.4.25", path = '../swc_atoms' }
+swc_css_ast = { version = "0.128.1", path = "../swc_css_ast", features = [
+  "rkyv-impl",
+] }
+swc_css_parser = { version = "0.137.3", path = "../swc_css_parser" }
+swc_ecma_ast = { version = "0.95.1", path = "../swc_ecma_ast", features = [
   "rkyv-impl",
 ] }
+swc_ecma_loader = { version = "0.41.18", path = "../swc_ecma_loader" }
+swc_ecma_parser = { version = "0.123.2", path = "../swc_ecma_parser" }
+swc_ecma_visit = { version = "0.81.1", path = "../swc_ecma_visit" }
+swc_node_base = { version = "0.5.8", path = "../swc_node_base" }
+testing = { version = "0.31.18", path = "../testing" }
 wasmer = "2.3.0"
 wasmer-wasi = "2.3.0"