Add ResponseMessage::from_request.

whisperfish · Jul 9, 2020 · fd43ed4 · fd43ed4
1 parent a93b46d
commit fd43ed4
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/libsignal-service/src/proto.rs b/libsignal-service/src/proto.rs
@@ -38,3 +38,25 @@ impl WebSocketRequestMessage {
         false
     }
 }
+
+impl WebSocketResponseMessage {
+    /// Equivalent of
+    /// `SignalServiceMessagePipe::isSignalServiceEnvelope(WebSocketMessage)`.
+    pub fn from_request(msg: &WebSocketRequestMessage) -> Self {
+        if msg.is_signal_service_envelope() {
+            WebSocketResponseMessage {
+                id: msg.id,
+                status: Some(200),
+                message: Some("OK".to_string()),
+                ..Default::default()
+            }
+        } else {
+            WebSocketResponseMessage {
+                id: msg.id,
+                status: Some(400),
+                message: Some("Unknown".to_string()),
+                ..Default::default()
+            }
+        }
+    }
+}