//
//  Copyright 2021 Google, Inc.
//
//  Licensed under the Apache License, Version 2.0 (the "License");
//  you may not use this file except in compliance with the License.
//  You may obtain a copy of the License at:
//
//  http://www.apache.org/licenses/LICENSE-2.0
//
//  Unless required by applicable law or agreed to in writing, software
//  distributed under the License is distributed on an "AS IS" BASIS,
//  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//  See the License for the specific language governing permissions and
//  limitations under the License.

extern crate protobuf_codegen;
extern crate protoc_rust;

use protoc_rust::Customize;
use std::env;
use std::path::{Path, PathBuf};

fn paths_to_strs<P: AsRef<Path>>(paths: &[P]) -> Vec<&str> {
    paths.iter().map(|p| p.as_ref().as_os_str().to_str().unwrap()).collect()
}

fn main() {
    // Proto root is //tools/netsim/src/proto
    let proto_root = match env::var("PLATFORM_SUBDIR") {
        Ok(dir) => PathBuf::from(dir).join("tools/netsim"),
        // Currently at //tools/netsim/rust/backend-proto
        Err(_) => PathBuf::from(env::current_dir().unwrap()).join("../..").canonicalize().unwrap(),
    };

    let proto_out_dir = proto_root.join("rust/backend-proto/src");

    // Generate protobuf output
    let proto_dir = proto_root.join("src/proto");
    let proto_input_files = [
        proto_dir.join("packet_streamer.proto"),
        proto_dir.join("hci_packet.proto"),
        proto_dir.join("startup.proto"),
        proto_dir.join("common.proto"),
    ];
    let proto_include_dirs = [proto_dir.clone()];

    println!("cargo:warning=proto_outdir={:?}", proto_out_dir);

    protoc_rust::Codegen::new()
        .out_dir(proto_out_dir.as_os_str().to_str().unwrap())
        .inputs(&paths_to_strs(&proto_input_files))
        .includes(&paths_to_strs(&proto_include_dirs))
        .customize(Customize { serde_derive: Some(true), ..Default::default() })
        .run()
        .expect("protoc");
}