build.rs

// Copyright The ocicrypt Authors.
// SPDX-License-Identifier: Apache-2.0

fn main() -> Result<(), Box<dyn std::error::Error>> {
    #[cfg(feature = "gen-proto-grpc")]
    tonic_build::configure()
        .build_server(true)
        .out_dir("src/utils/grpc/")
        .compile(&["src/utils/proto/keyprovider.proto"], &["src/utils"])?;

    #[cfg(feature = "gen-proto-ttrpc")]
    {
        use std::fs::File;
        use std::io::{Read, Write};

        fn replace_text_in_file(
            file_name: &str,
            from: &str,
            to: &str,
        ) -> Result<(), std::io::Error> {
            let mut src = File::open(file_name)?;
            let mut contents = String::new();
            src.read_to_string(&mut contents).unwrap();
            drop(src);

            let new_contents = contents.replace(from, to);

            let mut dst = File::create(file_name)?;
            dst.write_all(new_contents.as_bytes())?;

            Ok(())
        }

        ttrpc_codegen::Codegen::new()
            .out_dir("src/utils/ttrpc")
            .inputs(["src/utils/proto/keyprovider.proto"])
            .include("src/utils")
            .rust_protobuf()
            .customize(ttrpc_codegen::Customize {
                async_all: false,
                ..Default::default()
            })
            .rust_protobuf_customize(ttrpc_codegen::ProtobufCustomize::default().gen_mod_rs(false))
            .run()
            .expect("ttrpc gen async code failed.");

        // Fix clippy warnings of code generated from ttrpc_codegen
        replace_text_in_file(
            "src/utils/ttrpc/keyprovider_ttrpc.rs",
            "client: client",
            "client",
        )?;
        replace_text_in_file(
            "src/utils/ttrpc/keyprovider_ttrpc.rs",
            "#![allow(box_pointers)]\n",
            "",
        )?;
    }

    Ok(())
}