wehub-resource-sync
93 lines
5.2 KiB
Rust
93 lines
5.2 KiB
Rust
// ┏━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┓
|
|
// ┃ ██████ ██████ ██████ █ █ █ █ █ █▄ ▀███ █ ┃
|
|
// ┃ ▄▄▄▄▄█ █▄▄▄▄▄ ▄▄▄▄▄█ ▀▀▀▀▀█▀▀▀▀▀ █ ▀▀▀▀▀█ ████████▌▐███ ███▄ ▀█ █ ▀▀▀▀▀ ┃
|
|
// ┃ █▀▀▀▀▀ █▀▀▀▀▀ █▀██▀▀ ▄▄▄▄▄ █ ▄▄▄▄▄█ ▄▄▄▄▄█ ████████▌▐███ █████▄ █ ▄▄▄▄▄ ┃
|
|
// ┃ █ ██████ █ ▀█▄ █ ██████ █ ███▌▐███ ███████▄ █ ┃
|
|
// ┣━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┫
|
|
// ┃ Copyright (c) 2017, the Perspective Authors. ┃
|
|
// ┃ ╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌ ┃
|
|
// ┃ This file is part of the Perspective library, distributed under the terms ┃
|
|
// ┃ of the [Apache License 2.0](https://www.apache.org/licenses/LICENSE-2.0). ┃
|
|
// ┗━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┛
|
|
|
|
use std::io::Result;
|
|
use std::path::Path;
|
|
|
|
fn prost_build() -> Result<()> {
|
|
println!(
|
|
"cargo::metadata=PROTO_PATH={}",
|
|
std::env::var("CARGO_MANIFEST_DIR").unwrap()
|
|
);
|
|
|
|
// This source file is included at `publish` time, but not `build` time
|
|
// because it is initially generated from the `perspective.proto` definition
|
|
// in the source.
|
|
if std::env::var("CARGO_FEATURE_GENERATE_PROTO").is_ok() {
|
|
println!("cargo:warning=MESSAGE Building in development mode");
|
|
// let root_dir_env = std::env::var("PSP_ROOT_DIR").expect("Must set
|
|
// PSP_ROOT_DIR"); let root_dir = Path::new(root_dir_env.as_str());
|
|
// let proto_file = Path::join(root_dir, "cpp/protos/perspective.proto");
|
|
let proto_file = Path::new("./perspective.proto");
|
|
let include_path = proto_file
|
|
.parent()
|
|
.expect("Couldn't determine parent directory of proto_file")
|
|
.to_path_buf();
|
|
|
|
println!("cargo:rerun-if-changed={}", proto_file.to_str().unwrap());
|
|
|
|
// prost_build reads PROTOC from the environment. When the `protobuf-src`
|
|
// feature is enabled, the build script sets PROTOC to the one built by
|
|
// that crate. When protobuf-src is disabled, builders must set PROTOC
|
|
// in the environment to a protocol buffer compiler.
|
|
#[cfg(feature = "protobuf-src")]
|
|
unsafe {
|
|
std::env::set_var("PROTOC", protobuf_src::protoc())
|
|
};
|
|
#[cfg(not(feature = "protobuf-src"))]
|
|
if std::env::var("PROTOC").is_err() {
|
|
panic!(
|
|
"generate-proto is enabled and protobuf-src is disabled. PROTOC must be set in \
|
|
the environment to the path of a protocol buffer compiler"
|
|
)
|
|
}
|
|
|
|
prost_build::Config::new()
|
|
// .bytes(["ViewToArrowResp.arrow", "from_arrow"])
|
|
.type_attribute("ViewOnUpdateResp", "#[derive(ts_rs::TS)]")
|
|
.field_attribute("ViewOnUpdateResp.delta", "#[ts(as = \"Vec::<u8>\")]")
|
|
.field_attribute("ViewOnUpdateResp.delta", "#[serde(with = \"serde_bytes\")]")
|
|
.type_attribute("ColumnType", "#[derive(ts_rs::TS)]")
|
|
.type_attribute(
|
|
"JoinType",
|
|
"#[derive(serde::Deserialize, ts_rs::TS)] #[serde(rename_all = \"snake_case\")]",
|
|
)
|
|
.field_attribute("ViewToArrowResp.arrow", "#[serde(skip)]")
|
|
.field_attribute("from_arrow", "#[serde(skip)]")
|
|
.type_attribute(".", "#[derive(serde::Serialize)]")
|
|
.type_attribute("ViewPort", "#[derive(serde::Deserialize)]")
|
|
.type_attribute("ViewDimensionsResp", "#[derive(serde::Deserialize)]")
|
|
.type_attribute("TableValidateExprResp", "#[derive(serde::Deserialize)]")
|
|
.type_attribute(
|
|
"ColumnType",
|
|
"#[derive(serde::Deserialize)] #[serde(rename_all = \"snake_case\")]",
|
|
)
|
|
.type_attribute("ExprValidationError", "#[derive(serde::Deserialize)]")
|
|
.compile_protos(&[proto_file], &[include_path])
|
|
.unwrap();
|
|
|
|
std::fs::copy(
|
|
std::env::var("OUT_DIR").unwrap() + "/perspective.proto.rs",
|
|
"src/rust/proto.rs",
|
|
)?;
|
|
|
|
std::fs::remove_file(std::env::var("OUT_DIR").unwrap() + "/perspective.proto.rs")?;
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
fn main() -> Result<()> {
|
|
prost_build()?;
|
|
Ok(())
|
|
}
|