support protobuf versioning

build.rs

@@ -27,7 +27,14 @@ 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/core/proxy.proto"], &["proto/core"])?;
+        .compile_protos(
+            &[
+                "proto/v2/proxy.proto",
+                "proto/v2/common.proto",
+                "proto/common/client_types.proto",
+            ],
+            &["proto"],
+        )?;
 
     println!("cargo:rerun-if-changed=proto");
     Ok(())

src/enterprise/handlers/desktop_client_mfa.rs

@@ -80,8 +80,6 @@ pub(super) async fn mfa_auth_callback(
         code: payload.code,
         nonce,
         state: payload.state,
-        #[allow(deprecated)]
-        callback_url: String::new(),
     };
 
     debug!("Sending MFA OIDC authenticate request to core service");

src/enterprise/handlers/openid_login.rs

@@ -164,8 +164,6 @@ async fn auth_callback(
     let request = AuthCallbackRequest {
         code: payload.code,
         nonce,
-        #[allow(deprecated)]
-        callback_url: String::new(),
     };
 
     let rx = state

src/lib.rs

@@ -16,8 +16,30 @@ pub mod http;
 pub mod logging;
 mod setup;
 
+pub(crate) mod generated {
+    pub(crate) mod defguard {
+        pub(crate) mod proxy {
+            pub(crate) mod v2 {
+                tonic::include_proto!("defguard.proxy.v2");
+            }
+        }
+
+        pub(crate) mod common {
+            pub(crate) mod v2 {
+                tonic::include_proto!("defguard.common.v2");
+            }
+        }
+
+        pub mod client_types {
+            tonic::include_proto!("defguard.client_types");
+        }
+    }
+}
+
 pub(crate) mod proto {
-    tonic::include_proto!("defguard.proxy");
+    pub(crate) use crate::generated::defguard::client_types::*;
+    pub(crate) use crate::generated::defguard::common::v2::*;
+    pub(crate) use crate::generated::defguard::proxy::v2::*;
 }
 
 #[macro_use]