arcbox-grpc 0.4.17

gRPC service clients and servers for ArcBox (tonic + prost)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

//! Build script for gRPC service code generation.
//!
//! This generates Rust client and server code for gRPC services
//! defined in the proto files using tonic-build.
//!
//! Message types are imported from arcbox-protocol (prost-generated).
//! All protos use the unified `arcbox.v1` package namespace.

fn main() {
    // Use proto files from arcbox-protocol
    let proto_dir = "../arcbox-protocol/proto";

    let protos = [
        "../arcbox-protocol/proto/machine.proto",
        "../arcbox-protocol/proto/container.proto",
        "../arcbox-protocol/proto/image.proto",
        "../arcbox-protocol/proto/agent.proto",
        "../arcbox-protocol/proto/api.proto",
        "../arcbox-protocol/proto/kubernetes.proto",
        "../arcbox-protocol/proto/sandbox.proto",
    ];

    let descriptor_path =
        std::path::PathBuf::from(std::env::var("OUT_DIR").expect("OUT_DIR set by cargo"))
            .join("arcbox_descriptor.bin");

    // Configure tonic-build
    tonic_build::configure()
        // Emit the file descriptor set for gRPC server reflection.
        .file_descriptor_set_path(&descriptor_path)
        // Map arcbox.v1 package to arcbox_protocol::v1 types
        .extern_path(".arcbox.v1", "::arcbox_protocol::v1")
        // Map sandbox.v1 package to arcbox_protocol::sandbox_v1 types
        .extern_path(".sandbox.v1", "::arcbox_protocol::sandbox_v1")
        // Generate client code
        .build_client(true)
        // Generate server code
        .build_server(true)
        // Compile protos from arcbox-protocol
        .compile_protos(&protos, &[proto_dir])
        .expect("Failed to compile protos");

    // Tell cargo to recompile if any proto file changes
    for proto in &protos {
        println!("cargo:rerun-if-changed={proto}");
    }
    println!("cargo:rerun-if-changed={proto_dir}/common.proto");
}