From ea662c5d1d9002df1c9548629c84b67274f2466a Mon Sep 17 00:00:00 2001
From: jokemanfire <hu.dingyang@zte.com.cn>
Date: Mon, 4 Nov 2024 19:50:06 +0800
Subject: [PATCH] Add default out_dir support

If not set out_dir,it will use env 'OUT_DIR',to generate proto rs file.
You can use 'include!' marco to include env 'OUT_DIR' proto rs .

Signed-off-by: jokemanfire <hu.dingyang@zte.com.cn>
---
 ttrpc-codegen/src/lib.rs | 23 +++++++++++++++++------
 1 file changed, 17 insertions(+), 6 deletions(-)

diff --git a/ttrpc-codegen/src/lib.rs b/ttrpc-codegen/src/lib.rs
index 559e48b8..2f82b56e 100644
--- a/ttrpc-codegen/src/lib.rs
+++ b/ttrpc-codegen/src/lib.rs
@@ -23,6 +23,10 @@
 //!        .run()
 //!        .expect("Gen async code failed.");
 //! }
+//! ```
+//! If there's no out_dir and use 'gen_mod' feature
+//! You can use the following method to include the target file
+//! include!(concat!(env!("OUT_DIR"), "/mod.rs"));
 
 pub use protobuf_codegen::{
     Customize as ProtobufCustomize, CustomizeCallback as ProtobufCustomizeCallback,
@@ -45,8 +49,8 @@ mod str_lit;
 /// Invoke pure rust codegen.
 #[derive(Debug, Default)]
 pub struct Codegen {
-    /// --lang_out= param
-    out_dir: PathBuf,
+    /// --lang_out= param ,if out_dir is none ,will use env 'OUT_DIR' path
+    out_dir: Option<PathBuf>,
     /// -I args
     includes: Vec<PathBuf>,
     /// List of .proto files to compile
@@ -65,9 +69,9 @@ impl Codegen {
         Self::default()
     }
 
-    /// Set the output directory for codegen.
+    /// Set the output directory for codegen. Support None out_dir
     pub fn out_dir(&mut self, out_dir: impl AsRef<Path>) -> &mut Self {
-        self.out_dir = out_dir.as_ref().to_owned();
+        self.out_dir = Some(out_dir.as_ref().to_owned());
         self
     }
 
@@ -132,11 +136,18 @@ impl Codegen {
         let includes: Vec<&Path> = self.includes.iter().map(|p| p.as_path()).collect();
         let inputs: Vec<&Path> = self.inputs.iter().map(|p| p.as_path()).collect();
         let p = parse_and_typecheck(&includes, &inputs)?;
+        // Add default path from env OUT_DIR, if no OUT_DIR env ,that's will be current path
+        let path_dir = std::env::var("OUT_DIR").map_or_else(
+            |_| std::env::current_dir().unwrap_or_default(),
+            PathBuf::from,
+        );
+        // If out_dir is none ,dst_path will be setting in path_dir
+        let dst_path = self.out_dir.clone().unwrap_or(path_dir);
 
         if self.rust_protobuf {
             self.rust_protobuf_codegen
                 .pure()
-                .out_dir(&self.out_dir)
+                .out_dir(&dst_path)
                 .inputs(&self.inputs)
                 .includes(&self.includes)
                 .run()
@@ -146,7 +157,7 @@ impl Codegen {
         ttrpc_compiler::codegen::gen_and_write(
             &p.file_descriptors,
             &p.relative_paths,
-            &self.out_dir,
+            &dst_path,
             &self.customize,
         )
     }