spark-protos 0.1.8

Autogenerated Protobuf definitions for Spark
Documentation 
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
49

use std::{env, fs, path::PathBuf};

use prost_wkt_build::{FileDescriptorSet, Message as _};

fn main() -> Result<(), Box<dyn std::error::Error>> {
    let out_dir = PathBuf::from(env::var("OUT_DIR").unwrap());

    // Compile with tonic_build for gRPC
    let spark_descriptor_file = out_dir.clone().join("spark_descriptor.bin");
    tonic_build::configure()
        .server_mod_attribute(".", "#[cfg(feature = \"server\")]")
        .client_mod_attribute(".", "#[cfg(feature = \"client\")]")
        .type_attribute(".", "#[derive(serde::Serialize, serde::Deserialize)]")
        .type_attribute(".", "#[derive(utoipa::ToSchema, utoipa::ToResponse)]")
        .type_attribute(".", "#[serde(rename_all = \"camelCase\")]")
        .extern_path(".google.protobuf.Any", "::prost_wkt_types::Any")
        .extern_path(".google.protobuf.Timestamp", "::prost_wkt_types::Timestamp")
        .extern_path(".google.protobuf.Value", "::prost_wkt_types::Value")
        .protoc_arg("--experimental_allow_proto3_optional")
        .file_descriptor_set_path(&spark_descriptor_file)
        .compile_protos(
            &[
                "v1/common/types.proto",
                "v1/frost/service.proto",
                "v1/spark/service.proto",
                "v1/spark_authn/service.proto",
                "v1/spark_tree/service.proto",
                "v1/validate/api.proto",
            ],
            &["protos/spark"],
        )
        .unwrap();

    let spark_descriptor_bytes = std::fs::read(&spark_descriptor_file).unwrap();
    let spark_descriptor = FileDescriptorSet::decode(&spark_descriptor_bytes[..]).unwrap();
    prost_wkt_build::add_serde(out_dir.clone(), spark_descriptor);

    // Create a lib.rs shim with the attribute
    let mut lib_rs_path = out_dir.clone();
    lib_rs_path.push("lib.rs");
    fs::write(
        lib_rs_path,
        "#![allow(non_camel_case_types)]\n\ninclude!(concat!(env!(\"OUT_DIR\"), \"/spark.rs\"));\n",
    )?;

    println!("cargo:rerun-if-changed=protos");

    Ok(())
}