From ed5043cd84bd4c05290cbbc5be319d48b9f09307 Mon Sep 17 00:00:00 2001
From: Dessalines <tyhou13@gmx.com>
Date: Fri, 30 Aug 2019 13:33:20 -0700
Subject: [PATCH] Fixing crash on unknown op type.

- Fixes #251
---
 server/src/websocket/server.rs | 13 ++++---------
 1 file changed, 4 insertions(+), 9 deletions(-)

diff --git a/server/src/websocket/server.rs b/server/src/websocket/server.rs
index b4cbce3a..900c7b41 100644
--- a/server/src/websocket/server.rs
+++ b/server/src/websocket/server.rs
@@ -233,14 +233,6 @@ impl Handler<Connect> for ChatServer {
       });
     }
 
-    // for (k,v) in &self.rate_limits {
-    //   println!("{}: {:?}", k,v);
-    // }
-
-    // auto join session to Main room
-    // self.rooms.get_mut(&"Main".to_owned()).unwrap().insert(id);
-
-    // send id back
     id
   }
 }
@@ -285,7 +277,10 @@ fn parse_json_message(chat: &mut ChatServer, msg: StandardMessage) -> Result<Str
 
   let json: Value = serde_json::from_str(&msg.msg)?;
   let data = &json["data"].to_string();
-  let op = &json["op"].as_str().unwrap();
+  let op = &json["op"].as_str().ok_or(APIError {
+    op: "Unknown op type".to_string(), 
+    message: format!("Unknown op type"),
+  })?;
 
   let user_operation: UserOperation = UserOperation::from_str(&op)?;
 
-- 
2.44.1