ev-types 0.0.1

Protocol buffer generated types for Ev
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83

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

fn main() -> Result<(), Box<dyn std::error::Error>> {
    let manifest_dir_str = env::var("CARGO_MANIFEST_DIR")?;
    let manifest_dir = Path::new(&manifest_dir_str);

    // Create output directories
    let proto_dir = manifest_dir.join("src/proto");
    fs::create_dir_all(&proto_dir)?;

    // Check if generated files already exist
    let messages_file = proto_dir.join("evnode.v1.messages.rs");
    let services_file = proto_dir.join("evnode.v1.services.rs");

    // Check for environment variable to force regeneration
    let force_regen = env::var("EV_TYPES_FORCE_PROTO_GEN").is_ok();

    // If files exist and we're not forcing regeneration, skip generation
    if !force_regen && messages_file.exists() && services_file.exists() {
        println!("cargo:warning=Using pre-generated proto files. Set EV_TYPES_FORCE_PROTO_GEN=1 to regenerate.");
        return Ok(());
    }

    // Make the include dir absolute and resolved (no "..", symlinks, etc.)
    let proto_root = match manifest_dir.join("../../../proto").canonicalize() {
        Ok(path) => path,
        Err(e) => {
            // If proto files don't exist but generated files do, that's ok
            if messages_file.exists() && services_file.exists() {
                println!("cargo:warning=Proto source files not found at ../../../proto, using pre-generated files");
                return Ok(());
            }
            // Otherwise, this is a real error
            return Err(format!(
                "Proto files not found and no pre-generated files available: {e}"
            )
            .into());
        }
    };

    // Collect the .proto files
    let proto_files: Vec<_> = WalkDir::new(&proto_root)
        .into_iter()
        .filter_map(|e| {
            let p = e.ok()?.into_path();
            (p.extension()?.to_str()? == "proto").then_some(p)
        })
        .collect();

    // Always generate both versions and keep them checked in
    // This way users don't need to regenerate based on features

    // 1. Generate pure message types (no tonic dependencies)
    let mut prost_config = prost_build::Config::new();
    prost_config.out_dir(&proto_dir);
    // Important: we need to rename the output to avoid conflicts
    prost_config.compile_protos(&proto_files, &[proto_root.as_path()])?;

    // Rename the generated file to messages.rs
    let generated_file = proto_dir.join("evnode.v1.rs");
    let messages_file = proto_dir.join("evnode.v1.messages.rs");
    if generated_file.exists() {
        fs::rename(&generated_file, &messages_file)?;
    }

    // 2. Generate full code with gRPC services (always generate, conditionally include)
    tonic_build::configure()
        .build_server(true)
        .build_client(true)
        .out_dir(&proto_dir)
        .compile(&proto_files, &[proto_root.as_path()])?;

    // Rename to services.rs
    let generated_file_2 = proto_dir.join("evnode.v1.rs");
    let services_file = proto_dir.join("evnode.v1.services.rs");
    if generated_file_2.exists() {
        fs::rename(&generated_file_2, &services_file)?;
    }

    println!("cargo:rerun-if-changed={}", proto_root.display());
    Ok(())
}