fix: farmfe_plugin_virtual options

Author: CCherry07

Cargo.lock

@@ -10,4 +10,6 @@ crate-type = ["cdylib", "rlib"]
 farmfe_toolkit_plugin_types = { workspace = true }
 farmfe_macro_plugin = { workspace = true }
 farmfe_core = { workspace = true }
+farmfe_toolkit = { workspace = true }
 thiserror = "2.0.3"
+serde = { version = "1.0.197", features = ["derive"] }

rust-plugins/virtual/Cargo.toml

@@ -1 +1,29 @@
-export type IPluginOptions = Record<string, string>;
+/**
+ * The value can be a string or an object with raw content and module type.
+ * @example
+ * // As a string
+ * const options: IPluginOptions = {
+ *   'virtual-module.js': "export default 'Hello, world!';"
+ * };
+ *
+ * // As an object
+ * const options: IPluginOptions = {
+ *   'virtual-module.js': {
+ *     raw: "export default 'Hello, world!';",
+ *     moduleType: "js"
+ *   }
+ * };
+ */
+type Value = string | {
+  /**
+   * The raw content of the module.
+   * @example "export default 'Hello, world!';"
+   */
+  raw: string;
+  /**
+   * The type of the module.
+   * @default "js"
+   */
+  moduleType: "js" | "ts" | "jsx" | "tsx" | 'css' | 'html' | 'json' | 'asset';
+};
+export type IPluginOptions = Record<string, Value>;

rust-plugins/virtual/options.d.ts

@@ -9,6 +9,7 @@ use farmfe_core::{
   plugin::{Plugin, PluginLoadHookResult, PluginResolveHookResult},
   serde_json,
 };
+use farmfe_toolkit::script::module_type_from_id;
 use std::{collections::HashMap, path::Path, sync::Arc};
 use thiserror::Error;
 use utils::{normalize_path, path_join};
@@ -30,13 +31,23 @@ pub enum VirtualModuleError {
 #[derive(Debug)]
 #[farm_plugin]
 pub struct FarmPluginVirtualModule {
-  virtual_modules: HashMap<String, String>,
-  resolved_paths: HashMap<String, String>,
+  virtual_modules: HashMap<String, Value>,
+  resolved_paths: HashMap<String, Value>,
+}
+
+#[derive(Debug, Clone, serde::Deserialize)]
+#[serde(untagged)]
+enum Value {
+  Struct {
+    raw: String,
+    module_type: Option<ModuleType>,
+  },
+  Str(String),
 }
 
 impl FarmPluginVirtualModule {
   fn new(_: &Config, options: String) -> Self {
-    let virtual_modules:HashMap<String, String> =
+    let virtual_modules: HashMap<String, Value> =
       serde_json::from_str(&options).expect("Failed to parse virtual module options");
 
     let mut resolved_paths = HashMap::new();
@@ -104,9 +115,17 @@ impl FarmPluginVirtualModule {
       .virtual_modules
       .get(id)
       .or_else(|| self.resolved_paths.get(id))
-      .map(|content| PluginLoadHookResult {
-        content: content.clone(),
-        module_type: ModuleType::Js,
+      .map(|value| PluginLoadHookResult {
+        content: match value {
+          Value::Struct { raw, .. } => raw.clone(),
+          Value::Str(s) => s.clone(),
+        },
+        module_type: match value {
+          Value::Struct { module_type, .. } => module_type
+            .clone()
+            .unwrap_or(module_type_from_id(id).unwrap_or(ModuleType::Js)),
+          Value::Str(_) => module_type_from_id(id).unwrap_or(ModuleType::Js),
+        },
         source_map: None,
       })
   }
@@ -159,7 +178,6 @@ impl Plugin for FarmPluginVirtualModule {
 #[cfg(test)]
 mod tests {
   use super::*;
-  use std::path::PathBuf;
 
   #[test]
   fn test_virtual_module_creation() {

rust-plugins/virtual/src/lib.rs

@@ -82,7 +82,6 @@ pub fn normalize_path<P: AsRef<Path>>(path: P) -> PathBuf {
 #[cfg(test)]
 mod tests {
     use super::*;
-    use std::env;
 
     #[test]
     fn test_normalize_path() {