57 lines
2.0 KiB
Rust
57 lines
2.0 KiB
Rust
//
|
|
// 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");
|
|
}
|