Add ExpandDerive handler

commit-id:20ad0a61

scarb/src/ops/proc_macro_server/methods/expand_derive.rs

@@ -0,0 +1,41 @@
+use crate::{
+    compiler::plugin::proc_macro::{Expansion, ExpansionKind, ProcMacroHost},
+    ops::proc_macro_server::json_rpc::Handler,
+};
+use anyhow::Result;
+use cairo_lang_macro::TokenStream;
+use convert_case::{Case, Casing};
+use scarb_proc_macro_server_types::methods::{expand::ExpandDerive, ProcMacroResult};
+use std::sync::Arc;
+
+impl Handler for ExpandDerive {
+    fn handle(proc_macros: Arc<ProcMacroHost>, params: Self::Params) -> Result<Self::Response> {
+        let mut derived_code = String::new();
+        let mut all_diagnostics = vec![];
+
+        for derive in params.derives {
+            let expansion = Expansion::new(derive.to_case(Case::Snake), ExpansionKind::Derive);
+            let instance = proc_macros
+                .macros()
+                .into_iter()
+                .find(|e| e.get_expansions().contains(&expansion))
+                .unwrap();
+
+            let result = instance.generate_code(
+                expansion.name.clone(),
+                TokenStream::empty(),
+                params.item.clone(),
+            );
+
+            // Register diagnostics.
+            all_diagnostics.extend(result.diagnostics);
+            // Add generated code.
+            derived_code.push_str(&result.token_stream.to_string());
+        }
+
+        Ok(ProcMacroResult {
+            token_stream: TokenStream::new(derived_code),
+            diagnostics: all_diagnostics,
+        })
+    }
+}

scarb/src/ops/proc_macro_server/methods/mod.rs

@@ -1,2 +1,3 @@
 pub mod defined_macros;
 pub mod expand_attribute;
+pub mod expand_derive;

scarb/src/ops/proc_macro_server/mod.rs

@@ -8,7 +8,7 @@ use crossbeam_channel::{Receiver, Sender};
 use json_rpc::Handler;
 use scarb_proc_macro_server_types::jsonrpc::{ResponseError, RpcRequest, RpcResponse};
 use scarb_proc_macro_server_types::methods::defined_macros::DefinedMacros;
-use scarb_proc_macro_server_types::methods::expand::ExpandAttribute;
+use scarb_proc_macro_server_types::methods::expand::{ExpandAttribute, ExpandDerive};
 use scarb_proc_macro_server_types::methods::Method;
 use serde_json::Value;
 
@@ -75,6 +75,7 @@ fn route_request(proc_macros: Arc<ProcMacroHost>, request: RpcRequest) -> Result
         ExpandAttribute::METHOD => {
             run_handler::<ExpandAttribute>(proc_macros.clone(), request.value)
         }
+        ExpandDerive::METHOD => run_handler::<ExpandDerive>(proc_macros.clone(), request.value),
         _ => Err(anyhow!("method not found")),
     }
 }