talos-api-rs 0.2.0

A typed, async, idiomatic Rust client for the Talos Linux gRPC API
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
50
51
52
53
54
55

// SPDX-License-Identifier: MIT OR Apache-2.0

use std::path::PathBuf;

fn main() {
    // Skip code generation on docs.rs - use pre-generated files
    // docs.rs has a read-only filesystem
    if std::env::var("DOCS_RS").is_ok() {
        return;
    }

    let out_dir = PathBuf::from("src/api/generated");
    // Ensure directory exists
    std::fs::create_dir_all(&out_dir).unwrap();

    tonic_prost_build::configure()
        .out_dir(&out_dir)
        .build_server(true)
        .compile_protos(
            &[
                "proto/common/version.proto",
                "proto/common/common.proto",
                "proto/machine/machine.proto",
            ],
            &["proto"],
        )
        .unwrap();

    // Add SPDX header to generated files
    for file_name in &["version.rs", "common.rs", "machine.rs"] {
        let generated_file = out_dir.join(file_name);
        if generated_file.exists() {
            let content = std::fs::read_to_string(&generated_file).unwrap();
            if !content.starts_with("// SPDX-License-Identifier") {
                let new_content = format!(
                    "// SPDX-License-Identifier: MIT OR Apache-2.0\n// DO NOT EDIT\n{}",
                    content
                );
                std::fs::write(&generated_file, new_content).unwrap();
            }

            // Format the generated file
            let _ = std::process::Command::new("rustfmt")
                .arg("--edition")
                .arg("2021")
                .arg(&generated_file)
                .status();
        }
    }

    // Rerun if proto changes
    println!("cargo:rerun-if-changed=proto/common/version.proto");
    println!("cargo:rerun-if-changed=proto/common/common.proto");
    println!("cargo:rerun-if-changed=proto/machine/machine.proto");
}