1 files changed, 65 insertions, 5 deletions
diff --git a/lib/warden-core/build.rs b/lib/warden-core/build.rs
index 024e9bb..be4f8f2 100644
--- a/lib/warden-core/build.rs
+++ b/lib/warden-core/build.rs
@@ -1,11 +1,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>> {
-    tonic_prost_build::configure()
-        .build_server(false)
-        //.out_dir("src/google")  // you can change the generated code's location
+    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(
-            &["proto/googleapis/google/pubsub/v1/pubsub.proto"],
-            &["../../proto/googleapis"], // specify the root location to search proto dependencies
+            &entity.protos(),
+            &["../../proto/googleapis", "../../proto"], // specify the root location to search proto dependencies
         )?;
+
     Ok(())
 }