Merge remote-tracking branch 'origin/main' into SDK-client-UA-registration

server/crates/arbiter-proto/build.rs

@@ -1,27 +1,32 @@
+use std::path::PathBuf;
 use tonic_prost_build::configure;
 
 static PROTOBUF_DIR: &str = "../../../protobufs";
 
 fn main() -> Result<(), Box<dyn std::error::Error>> {
-    if std::env::var("PROTOC").is_err() {
-        println!("cargo:warning=PROTOC environment variable not set, using vendored protoc");
-        let protoc = protoc_bin_vendored::protoc_bin_path().unwrap();
-        unsafe { std::env::set_var("PROTOC", protoc) };
+    let manifest_dir = PathBuf::from(std::env::var("CARGO_MANIFEST_DIR")?);
+    let protobuf_dir = manifest_dir.join(PROTOBUF_DIR);
+    let protoc_include = protoc_bin_vendored::include_path()?;
+    let protoc_path = protoc_bin_vendored::protoc_bin_path()?;
+
+    unsafe {
+        std::env::set_var("PROTOC", &protoc_path);
+        std::env::set_var("PROTOC_INCLUDE", &protoc_include);
     }
 
-    println!("cargo::rerun-if-changed={PROTOBUF_DIR}");
+    println!("cargo::rerun-if-changed={}", protobuf_dir.display());
 
     configure()
         .message_attribute(".", "#[derive(::kameo::Reply)]")
+        .compile_well_known_types(true)
         .compile_protos(
             &[
-                format!("{}/arbiter.proto", PROTOBUF_DIR),
-                format!("{}/user_agent.proto", PROTOBUF_DIR),
-                format!("{}/client.proto", PROTOBUF_DIR),
-                format!("{}/evm.proto", PROTOBUF_DIR),
+                protobuf_dir.join("arbiter.proto"),
+                protobuf_dir.join("user_agent.proto"),
+                protobuf_dir.join("client.proto"),
+                protobuf_dir.join("evm.proto"),
             ],
-            &[PROTOBUF_DIR.to_string()],
-        )
-        .unwrap();
+            &[protobuf_dir],
+        )?;
     Ok(())
 }