mirror of
https://github.com/NVIDIA/OpenShell.git
synced 2026-10-21 06:11:31 +08:00
Polite_realism
64c5bcdc46
The core, server, and sandbox-backend build scripts always used the vendored protoc binary and ignored PROTOC, so a distro build could not use its system compiler. When PROTOC is set, use it and let prost resolve the includes (PROTOC_INCLUDE still applies); otherwise behave exactly as before. Contributes to #3510. Signed-off-by: politerealism <burdcat17@gmail.com>
78 lines
3.0 KiB
Rust
78 lines
3.0 KiB
Rust
// SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
use std::env;
|
|
use std::path::PathBuf;
|
|
|
|
fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|
// A caller-supplied PROTOC (for example a distro package's system
|
|
// compiler) is used as-is and prost resolves its own includes. Without it,
|
|
// use the vendored compiler and include tree.
|
|
println!("cargo:rerun-if-env-changed=PROTOC");
|
|
println!("cargo:rerun-if-env-changed=PROTOC_INCLUDE");
|
|
let use_vendored_protoc = env::var_os("PROTOC").is_none();
|
|
|
|
// In vendored mode, preserve the vendored include-tree guarantee even when
|
|
// the caller has PROTOC_INCLUDE set: prost validates that ambient path
|
|
// independently of explicit includes. Re-enter with a child-only
|
|
// environment override.
|
|
if use_vendored_protoc && env::var_os("PROTOC_INCLUDE").is_some() {
|
|
let status = std::process::Command::new(env::current_exe()?)
|
|
.env_remove("PROTOC_INCLUDE")
|
|
.status()?;
|
|
if !status.success() {
|
|
return Err(format!("vendored protobuf build failed: {status}").into());
|
|
}
|
|
return Ok(());
|
|
}
|
|
|
|
let manifest_dir = PathBuf::from(env::var("CARGO_MANIFEST_DIR")?);
|
|
let storage_proto_dir = manifest_dir.join("proto");
|
|
let public_proto_dir = manifest_dir.join("../../proto");
|
|
let storage_proto = storage_proto_dir.join("storage.proto");
|
|
|
|
println!("cargo:rerun-if-changed={}", storage_proto.display());
|
|
for imported_proto in [
|
|
"datamodel.proto",
|
|
"openshell.proto",
|
|
"options.proto",
|
|
"sandbox.proto",
|
|
] {
|
|
println!(
|
|
"cargo:rerun-if-changed={}",
|
|
public_proto_dir.join(imported_proto).display()
|
|
);
|
|
}
|
|
|
|
// Configure the vendored compiler and well-known includes without changing
|
|
// the build process environment. With a caller-supplied PROTOC, leave both
|
|
// to prost.
|
|
let mut proto_config = tonic_prost_build::Config::new();
|
|
let proto_include = if use_vendored_protoc {
|
|
proto_config.protoc_executable(protoc_bin_vendored::protoc_bin_path()?);
|
|
Some(protoc_bin_vendored::include_path()?)
|
|
} else {
|
|
None
|
|
};
|
|
|
|
let descriptor_path = PathBuf::from(env::var("OUT_DIR")?).join("storage_descriptor.bin");
|
|
let mut include_dirs = vec![storage_proto_dir, public_proto_dir];
|
|
include_dirs.extend(proto_include);
|
|
tonic_prost_build::configure()
|
|
.build_server(false)
|
|
.build_client(false)
|
|
.extern_path(".openshell.v1", "::openshell_core::proto")
|
|
.extern_path(
|
|
".openshell.datamodel.v1",
|
|
"::openshell_core::proto::datamodel::v1",
|
|
)
|
|
.extern_path(
|
|
".openshell.sandbox.v1",
|
|
"::openshell_core::proto::sandbox::v1",
|
|
)
|
|
.file_descriptor_set_path(&descriptor_path)
|
|
.compile_with_config(proto_config, &[storage_proto], &include_dirs)?;
|
|
|
|
Ok(())
|
|
}
|