Merge pull request #310 from Dstack-TEE/rpc-gen

kvinwang · web-flow · commit 4b756f6dfa1c · 2025-08-23T18:00:59.000+08:00
prpc: Move generated files to OUT_DIR
diff --git a/gateway/rpc/build.rs b/gateway/rpc/build.rs
@@ -4,7 +4,7 @@
 
 fn main() {
     prpc_build::configure()
-        .out_dir("./src/generated")
+        .out_dir(std::env::var_os("OUT_DIR").unwrap())
         .mod_prefix("super::")
         .build_scale_ext(false)
         .disable_package_emission()
diff --git a/gateway/rpc/src/generated.rs b/gateway/rpc/src/generated.rs
@@ -1,4 +1,3 @@
-pub use gateway::*;
+#![allow(async_fn_in_trait)]
 
-#[allow(async_fn_in_trait)]
-mod gateway;
+include!(concat!(env!("OUT_DIR"), "/gateway.rs"));
diff --git a/guest-agent/rpc/build.rs b/guest-agent/rpc/build.rs
@@ -4,7 +4,7 @@
 
 fn main() {
     prpc_build::configure()
-        .out_dir("./src/generated")
+        .out_dir(std::env::var_os("OUT_DIR").unwrap())
         .mod_prefix("super::")
         .build_scale_ext(false)
         .disable_package_emission()
diff --git a/guest-agent/rpc/src/generated.rs b/guest-agent/rpc/src/generated.rs
@@ -1,4 +1,3 @@
-pub use dstack_guest::*;
+#![allow(async_fn_in_trait)]
 
-#[allow(async_fn_in_trait)]
-mod dstack_guest;
+include!(concat!(env!("OUT_DIR"), "/dstack_guest.rs"));
diff --git a/guest-api/build.rs b/guest-api/build.rs
@@ -4,7 +4,7 @@
 
 fn main() {
     prpc_build::configure()
-        .out_dir("./src/generated")
+        .out_dir(std::env::var_os("OUT_DIR").unwrap())
         .mod_prefix("super::")
         .build_scale_ext(false)
         .disable_service_name_emission()
diff --git a/guest-api/src/generated/mod.rs b/guest-api/src/generated/mod.rs
@@ -1,4 +1,3 @@
-pub use guest_api::*;
+#![allow(async_fn_in_trait)]
 
-#[allow(async_fn_in_trait)]
-mod guest_api;
+include!(concat!(env!("OUT_DIR"), "/guest_api.rs"));
diff --git a/host-api/build.rs b/host-api/build.rs
@@ -4,7 +4,7 @@
 
 fn main() {
     prpc_build::configure()
-        .out_dir("./src/generated")
+        .out_dir(std::env::var_os("OUT_DIR").unwrap())
         .mod_prefix("super::")
         .build_scale_ext(false)
         .disable_service_name_emission()
diff --git a/host-api/src/generated/mod.rs b/host-api/src/generated/mod.rs
@@ -1,4 +1,3 @@
-pub use host_api::*;
+#![allow(async_fn_in_trait)]
 
-#[allow(async_fn_in_trait)]
-mod host_api;
+include!(concat!(env!("OUT_DIR"), "/host_api.rs"));
diff --git a/kms/rpc/build.rs b/kms/rpc/build.rs
@@ -4,7 +4,7 @@
 
 fn main() {
     prpc_build::configure()
-        .out_dir("./src/generated")
+        .out_dir(std::env::var_os("OUT_DIR").unwrap())
         .mod_prefix("super::")
         .build_scale_ext(false)
         .disable_package_emission()
diff --git a/kms/rpc/src/generated.rs b/kms/rpc/src/generated.rs
@@ -1,4 +1,3 @@
-pub use kms::*;
+#![allow(async_fn_in_trait)]
 
-#[allow(async_fn_in_trait)]
-mod kms;
+include!(concat!(env!("OUT_DIR"), "/kms.rs"));
diff --git a/vmm/rpc/build.rs b/vmm/rpc/build.rs
@@ -4,7 +4,7 @@
 
 fn main() {
     prpc_build::configure()
-        .out_dir("./src/generated")
+        .out_dir(std::env::var_os("OUT_DIR").unwrap())
         .mod_prefix("super::")
         .build_scale_ext(false)
         .disable_package_emission()
diff --git a/vmm/rpc/src/generated.rs b/vmm/rpc/src/generated.rs
@@ -1,4 +1,3 @@
-pub use vmm::*;
+#![allow(async_fn_in_trait)]
 
-#[allow(async_fn_in_trait)]
-mod vmm;
+include!(concat!(env!("OUT_DIR"), "/vmm.rs"));
