tasker-shared 0.1.2

Shared components for tasker orchestration and worker systems
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
84
85
86
87
88
89
90
91
92
93
94
95
96

//! Build script for tasker-shared crate
//!
//! Compiles Protocol Buffer definitions for gRPC services when the `grpc-api` feature is enabled.
//! Generated code is output to `$OUT_DIR/tasker.v1.rs` and included via `include!` macro in
//! `src/proto/mod.rs`.
//!
//! Proto files are located in `../proto/tasker/v1/` relative to this crate.
//!
//! # Protocol Buffer Compiler
//!
//! This build script requires the `protoc` compiler to be installed on the system.
//! On macOS, install via: `brew install protobuf`
//!
//! If `protoc` is not found, the build will fail with an error message explaining
//! how to install it.

fn main() -> Result<(), Box<dyn std::error::Error>> {
    // Only compile protos when grpc-api feature is enabled
    #[cfg(feature = "grpc-api")]
    {
        use std::path::PathBuf;

        // Find proto directory: check crate-local first (published crate),
        // then workspace root (development builds)
        let manifest_dir = PathBuf::from(std::env::var("CARGO_MANIFEST_DIR")?);
        let local_proto = manifest_dir.join("proto");
        let workspace_proto = manifest_dir
            .parent()
            .map(|p| p.join("proto"))
            .unwrap_or_default();

        let proto_root = if local_proto.join("tasker/v1").exists() {
            local_proto
        } else if workspace_proto.join("tasker/v1").exists() {
            workspace_proto
        } else {
            panic!(
                "Proto directory not found. Checked {:?} and {:?}. Expected proto files at proto/tasker/v1/",
                local_proto, workspace_proto
            );
        };

        // List of proto files to compile
        let proto_files = [
            "tasker/v1/common.proto",
            "tasker/v1/tasks.proto",
            "tasker/v1/steps.proto",
            "tasker/v1/templates.proto",
            "tasker/v1/analytics.proto",
            "tasker/v1/dlq.proto",
            "tasker/v1/health.proto",
            "tasker/v1/config.proto",
            "tasker/v1/worker.proto", // TAS-177: Worker-specific health, config, templates
        ];

        // Convert to full paths and verify each exists
        let proto_paths: Vec<PathBuf> = proto_files
            .iter()
            .map(|f| {
                let path = proto_root.join(f);
                if !path.exists() {
                    panic!("Proto file not found: {:?}", path);
                }
                path
            })
            .collect();

        // Configure tonic-prost-build (tonic 0.14 API)
        tonic_prost_build::configure()
            // Generate server code
            .build_server(true)
            // Generate client code
            .build_client(true)
            // Generate transport implementations
            .build_transport(true)
            // Include file descriptor set for reflection
            .file_descriptor_set_path(
                PathBuf::from(std::env::var("OUT_DIR")?).join("tasker_descriptor.bin"),
            )
            // Emit rerun-if-changed directives
            .emit_rerun_if_changed(true)
            // Compile with proto_path set to proto root
            .compile_protos(&proto_paths, std::slice::from_ref(&proto_root))?;

        // Emit rerun-if-changed for the proto directory
        println!("cargo:rerun-if-changed={}", proto_root.display());
        for proto in &proto_files {
            println!(
                "cargo:rerun-if-changed={}",
                proto_root.join(proto).display()
            );
        }
    }

    Ok(())
}