Set current working directory for procedural macros

crates/proc_macro_api/src/lib.rs

@@ -156,12 +156,18 @@ impl ProcMacro {
         attr: Option<&Subtree>,
         env: Vec<(String, String)>,
     ) -> Result<Result<Subtree, PanicMessage>, ServerError> {
+        let current_dir = env
+            .iter()
+            .find(|(name, _)| name == "CARGO_MANIFEST_DIR")
+            .map(|(_, value)| value.clone());
+
         let task = ExpandMacro {
             macro_body: FlatTree::new(subtree),
             macro_name: self.name.to_string(),
             attributes: attr.map(FlatTree::new),
             lib: self.dylib_path.to_path_buf().into(),
             env,
+            current_dir,
         };
 
         let request = msg::Request::ExpandMacro(task);

crates/proc_macro_api/src/msg.rs

@@ -48,6 +48,8 @@ pub struct ExpandMacro {
 
     /// Environment variables to set during macro expansion.
     pub env: Vec<(String, String)>,
+
+    pub current_dir: Option<String>,
 }
 
 pub trait Message: Serialize + DeserializeOwned {
@@ -143,6 +145,7 @@ mod tests {
             attributes: None,
             lib: std::env::current_dir().unwrap(),
             env: Default::default(),
+            current_dir: Default::default(),
         };
 
         let json = serde_json::to_string(&task).unwrap();