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

// Copyright Materialize, Inc. and contributors. All rights reserved.
//
// Use of this software is governed by the Business Source License
// included in the LICENSE file.
//
// As of the Change Date specified in that file, in accordance with
// the Business Source License, use of this software will be governed
// by the Apache License, Version 2.0.

use std::path::PathBuf;

use mz_protoc::Protoc;

/// Compile protocol buffers.
#[derive(clap::Parser)]
// Use unusual underscores in option names to match Google's protoc.
#[clap(rename_all = "snake")]
struct Args {
    /// Import search directory.
    #[clap(short = 'I', long, value_name = "PATH")]
    proto_path: Vec<String>,
    /// Generate Rust source code into OUT_DIR.
    #[clap(long, required = true, value_name = "OUT_DIR")]
    rust_out: PathBuf,
    /// Input protobuf schemas.
    #[clap(required = true, value_name = "PROTO_FILES")]
    proto_files: Vec<String>,
}

fn main() -> anyhow::Result<()> {
    let args: Args = ore::cli::parse_args();
    let mut protoc = Protoc::new();
    for path in args.proto_path {
        protoc.include(path);
    }
    for path in &args.proto_files {
        protoc.input(path);
    }
    protoc.compile_into(&args.rust_out)
}