Unnamed repository; edit this file 'description' to name the repository.
Diffstat (limited to 'crates/proc-macro-srv/src/cli.rs')
| -rw-r--r-- | crates/proc-macro-srv/src/cli.rs | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/crates/proc-macro-srv/src/cli.rs b/crates/proc-macro-srv/src/cli.rs new file mode 100644 index 0000000000..f1e131c135 --- /dev/null +++ b/crates/proc-macro-srv/src/cli.rs @@ -0,0 +1,31 @@ +//! Driver for proc macro server +use std::io; + +use proc_macro_api::msg::{self, Message}; + +use crate::ProcMacroSrv; + +pub fn run() -> io::Result<()> { + let mut srv = ProcMacroSrv::default(); + let mut buf = String::new(); + + while let Some(req) = read_request(&mut buf)? { + let res = match req { + msg::Request::ListMacros { dylib_path } => { + msg::Response::ListMacros(srv.list_macros(&dylib_path)) + } + msg::Request::ExpandMacro(task) => msg::Response::ExpandMacro(srv.expand(task)), + }; + write_response(res)? + } + + Ok(()) +} + +fn read_request(buf: &mut String) -> io::Result<Option<msg::Request>> { + msg::Request::read(&mut io::stdin().lock(), buf) +} + +fn write_response(msg: msg::Response) -> io::Result<()> { + msg.write(&mut io::stdout().lock()) +} |