use anyhow::Result;
use std::process::Command;
use crate::config::StandardConfig;
use crate::graph::Product;
use crate::processors::{check_command_output, ensure_output_dir, run_command};
use crate::processors::{DiscoverMode, SimpleGenerator, SimpleGeneratorParams};
fn execute_protobuf(
ctx: &crate::build_context::BuildContext,
config: &StandardConfig,
product: &Product,
) -> Result<()> {
let input = product.primary_input();
let output = product.primary_output();
let output_dir = crate::processors::parent_dir(output);
ensure_output_dir(output)?;
let command = config.require_command("protobuf")?;
let mut cmd = Command::new(command);
if let Some(parent) = input.parent() {
cmd.arg(format!("--proto_path={}", parent.display()));
}
cmd.arg(format!("--cpp_out={}", output_dir.display()));
for arg in &config.args {
cmd.arg(arg);
}
cmd.arg(input);
let out = run_command(ctx, &cmd)?;
check_command_output(&out, format_args!("protoc {}", input.display()))
}
fn create_protobuf(toml: &toml::Value) -> anyhow::Result<Box<dyn crate::processors::Processor>> {
crate::registries::deserialize_and_create(toml, |cfg| {
Box::new(SimpleGenerator::new(
cfg,
SimpleGeneratorParams {
extra_tools: &[],
extra_tools_fn: None,
discover_mode: DiscoverMode::SingleFormat("pb.cc"),
execute_fn: execute_protobuf,
is_native: false,
},
))
})
}
inventory::submit! { crate::registries::ProcessorPlugin {
version: 1,
name: "protobuf", processor_type: crate::processors::ProcessorType::Generator, create: create_protobuf,
fields: &[],
omit_standard_fields: &[],
scan_defaults: Some(crate::config::ScanDefaultsData { src_dirs: &[], src_extensions: &[".proto"], src_exclude_dirs: &[] }),
defaults: Some(crate::config::ProcessorDefaults { output_dir: "out/protobuf", command: "protoc", ..crate::config::ProcessorDefaults::EMPTY }),
defconfig_json: crate::registries::default_config_json::<crate::config::StandardConfig>,
keywords: &["protobuf", "proto", "generator", "grpc", "serialization"],
description: "Compile Protocol Buffer definitions",
is_native: false,
can_fix: false,
supports_batch: false,
max_jobs_cap: None,
} }