spiffe 0.6.5

Rust client library implementation for SPIFFE
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

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

use anyhow::{ensure, Context as _};

fn main() -> anyhow::Result<()> {
    println!("cargo::rerun-if-changed=src/proto");
    println!("cargo::rerun-if-env-changed=DOCS_RS");

    // Check if this is a docs.rs build
    if env::var_os("DOCS_RS").is_some() {
        println!("cargo:warning=Skipping protobuf code generation on docs.rs.");
        return Ok(());
    }
    let out_dir = env::var_os("OUT_DIR").context("failed to lookup `OUT_DIR`")?;
    let out_dir = Path::new(&out_dir);

    let mut proto_config = prost_build::Config::new();
    proto_config.bytes(["."]);

    let file_descriptors = protox::compile(["src/proto/workload.proto"], ["src/proto"])
        .context("failed to compile protocol buffer file set")?;

    tonic_build::configure()
        .build_client(true)
        .out_dir(out_dir)
        .compile_fds_with_config(proto_config, file_descriptors)
        .context("failed to compile protocol buffers")?;

    let workload = out_dir.join("_.rs");
    ensure!(
        workload.exists(),
        "expected generated file `_.rs` not found in `{}`",
        out_dir.display()
    );
    fs::rename(&workload, out_dir.join("workload.rs"))
        .with_context(|| format!("failed to rename `{}` to `workload.rs`", workload.display()))?;

    Ok(())
}