1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
#[cfg(feature = "iso20022")]
enum Entity {
#[cfg(feature = "iso20022")]
ISO2022,
}
#[cfg(feature = "iso20022")]
impl Entity {
fn protos(&self) -> Vec<&'static str> {
let mut res: Vec<&'static str> = vec![];
#[cfg(feature = "iso20022")]
fn iso20022_protos() -> Vec<&'static str> {
vec![
"proto/iso20022/pacs_008_001_12.proto",
"proto/iso20022/pacs_002_001_12.proto",
]
}
match self {
#[cfg(feature = "iso20022")]
Entity::ISO2022 => {
res.extend(iso20022_protos());
}
}
res
}
}
fn main() -> Result<(), Box<dyn std::error::Error>> {
println!("cargo:rerun-if-changed=../../proto");
#[cfg(feature = "iso20022")]
build_proto("iso20022", Entity::ISO2022)?;
Ok(())
}
#[cfg(feature = "iso20022")]
fn build_proto(package: &str, entity: Entity) -> Result<(), Box<dyn std::error::Error>> {
let out_dir = std::path::PathBuf::from(std::env::var("OUT_DIR").unwrap());
let config = tonic_prost_build::configure()
.server_mod_attribute(
package,
format!("#[cfg(feature = \"rpc-server-{package}\")] #[cfg_attr(docsrs, doc(cfg(feature = \"rpc-server-{package}\")))]"),
)
.client_mod_attribute(
package,
format!("#[cfg(feature = \"rpc-client-{package}\")] #[cfg_attr(docsrs, doc(cfg(feature = \"rpc-client-{package}\")))]"),
);
config
.file_descriptor_set_path(out_dir.join(format!("{package}_descriptor.bin")))
.server_mod_attribute(
package,
format!("#[cfg(feature = \"rpc-server-{package}\")] #[cfg_attr(docsrs, doc(cfg(feature = \"rpc-server-{package}\")))]"),
)
.client_mod_attribute(
package,
format!("#[cfg(feature = \"rpc-client-{package}\")] #[cfg_attr(docsrs, doc(cfg(feature = \"rpc-client-{package}\")))]"),
)
.protoc_arg("-I=../..")
.compile_well_known_types(true)
.compile_protos(
&entity.protos(),
&["../../proto/googleapis", "../../proto"], // specify the root location to search proto dependencies
)?;
Ok(())
}
|
