Fix tonic_build OSS build

Reviewed By: capickett Differential Revision: D44324686 fbshipit-source-id: 930a442bed79bcee3f17cd7e984963fa78ea27c4
facebookresearch · Mar 23, 2023 · d4eba22 · d4eba22
1 parent 28cf26d
commit d4eba22
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/protocol-rpc/build.rs b/protocol-rpc/build.rs
@@ -12,9 +12,11 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
         "pjc.proto",
         "suidcreate.proto",
     ];
+    let out_env = if cfg!(fbcode_build) { "OUT" } else { "OUT_DIR" };
+    let out_dir = std::env::var_os(out_env).unwrap_or_else(|| panic!("env `{out_env}` not set"));
 
     tonic_build::configure()
-        .out_dir(std::env::var("OUT").unwrap())
+        .out_dir(&out_dir)
         .compile(
             proto_files,
             // HACK: we need '.' directory for build with Buck