Merge pull request #41 from Tim-Zhang/move-ttrpc-rs

Move generated file ttrpc.rs to OUT_DIR

Author: Hui Zhu

build.rs

@@ -1,6 +1,14 @@
+use std::env;
+use std::fs;
+use std::path::PathBuf;
+
 fn main() {
+    let out_dir = env::var("OUT_DIR").unwrap();
+    let path: PathBuf = [out_dir.clone(), "mod.rs".to_string()].iter().collect();
+    fs::write(path, "pub mod ttrpc;").unwrap();
+
     protobuf_codegen_pure::Codegen::new()
-        .out_dir("src")
+        .out_dir(out_dir)
         .inputs(&["src/ttrpc.proto"])
         .include("src")
         .run()

src/lib.rs

@@ -20,7 +20,10 @@ pub mod error;
 #[macro_use]
 pub mod common;
 #[allow(clippy::type_complexity, clippy::too_many_arguments)]
-pub mod ttrpc;
+pub mod compiled {
+    include!(concat!(env!("OUT_DIR"), "/mod.rs"));
+}
+pub use compiled::ttrpc;
 
 pub use crate::common::MessageHeader;
 pub use crate::error::{get_status, Error, Result};