Implement From for various types (#33)

worldcoin · Oct 5, 2024 · fcc3d95 · fcc3d95
1 parent 2990ab6
commit fcc3d95
Show file tree

Hide file tree

Showing 3 changed files with 140 additions and 7 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/rust/Cargo.toml b/rust/Cargo.toml
@@ -13,7 +13,7 @@ server = []
 [dependencies]
 prost = "0.13.3"
 prost-types = "0.13.3"
-tonic = "0.12.3"
+tonic = { version = "0.12.3", features = ["tls"] }
 
 [build-dependencies]
 tonic-build = "0.12.2"

diff --git a/rust/src/lib.rs b/rust/src/lib.rs
@@ -1,8 +1,33 @@
 pub use ::prost;
+pub use ::prost_types;
 pub use ::tonic;
 
 pub mod relay {
     tonic::include_proto!("relay");
+
+    impl From<RelayPayload> for RelayConnectRequest {
+        fn from(payload: RelayPayload) -> Self {
+            RelayConnectRequest {
+                msg: Some(relay_connect_request::Msg::Payload(payload)),
+            }
+        }
+    }
+
+    impl From<Heartbeat> for RelayConnectRequest {
+        fn from(heartbeat: Heartbeat) -> Self {
+            RelayConnectRequest {
+                msg: Some(relay_connect_request::Msg::Heartbeat(heartbeat)),
+            }
+        }
+    }
+
+    impl From<ConnectRequest> for RelayConnectRequest {
+        fn from(connect_request: ConnectRequest) -> Self {
+            RelayConnectRequest {
+                msg: Some(relay_connect_request::Msg::ConnectRequest(connect_request)),
+            }
+        }
+    }
 }
 
 pub mod common {