chore: add ping command type

revoltchat · Jan 12, 2024 · 378c836 · 378c836
1 parent 488ba0d
commit 378c836
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/ws/mod.rs b/src/ws/mod.rs
@@ -77,6 +77,7 @@ async fn handle(
                             )
                             .await?;
                         break (room, id);
+                    } else if let WSCommandType::Ping = out.command_type {
                     } else {
                         return Err(WSCloseType::InvalidState);
                     }
@@ -96,6 +97,7 @@ async fn handle(
                 if let Ok(text) = message.to_str() {
                     let out: WSCommand = serde_json::from_str(text)?;
                     match out.command_type {
+                        WSCommandType::Ping => {}
                         WSCommandType::InitializeTransports { init_data } => {
                             let router = room.router().ok_or(WSCloseType::RoomClosed)?;
                             let rtc_state = RtcState::initialize(router, init_data)
@@ -151,6 +153,7 @@ async fn event_loop(
                     if let Ok(text) = message.to_str() {
                         let out: WSCommand = serde_json::from_str(text)?;
                         match &out.command_type {
+                            WSCommandType::Ping => {},
                             WSCommandType::ConnectTransport { connect_data } => {
                                 let result = rtc_state.connect_transport(connect_data).await;
                                 if let Ok(_) = result {

diff --git a/src/ws/types.rs b/src/ws/types.rs
@@ -19,6 +19,8 @@ pub struct WSCommand {
 #[derive(Deserialize, IntoStaticStr)]
 #[serde(tag = "type", content = "data")]
 pub enum WSCommandType {
+    Ping,
+
     #[serde(rename_all = "camelCase")]
     Authenticate {
         room_id: String,