handle new proto structure

Author: wojcik91

build.rs

@@ -27,7 +27,10 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
         // Make all messages serde-serializable.
         .type_attribute(".", "#[derive(serde::Serialize,serde::Deserialize)]")
         // Compiling protos using path on build time.
-        .compile_protos(&["proto/v2/core/proxy.proto"], &["proto/v2/core"])?;
+        .compile_protos(
+            &["proto/v2/proxy.proto", "proto/common/client_types.proto"],
+            &["proto"],
+        )?;
 
     println!("cargo:rerun-if-changed=proto");
     Ok(())

src/lib.rs

@@ -23,10 +23,15 @@ pub(crate) mod generated {
                 tonic::include_proto!("defguard.proxy.v2");
             }
         }
+
+        pub mod client_types {
+            tonic::include_proto!("defguard.client_types");
+        }
     }
 }
 
 pub(crate) mod proto {
+    pub(crate) use crate::generated::defguard::client_types::*;
     pub(crate) use crate::generated::defguard::proxy::v2::*;
 }