prpc: Move generated files to OUT_DIR

kvinwang · kvinwang · commit f148c599591e · 2025-08-23T04:00:08.000Z
diff --git a/gateway/rpc/build.rs b/gateway/rpc/build.rs
@@ -1,6 +1,6 @@
 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 @@
-pub use gateway::*;
-
-#[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
@@ -1,6 +1,6 @@
 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 @@
-pub use dstack_guest::*;
-
-#[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
@@ -1,6 +1,6 @@
 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 @@
-pub use guest_api::*;
-
-#[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
@@ -1,6 +1,6 @@
 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 @@
-pub use host_api::*;
-
-#[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
@@ -1,6 +1,6 @@
 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 @@
-pub use kms::*;
-
-#[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
@@ -1,6 +1,6 @@
 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 @@
-pub use vmm::*;
-
-#[allow(async_fn_in_trait)]
-mod vmm;
+include!(concat!(env!("OUT_DIR"), "/vmm.rs"));
