From 906a2dcd16b9a41d4dd80df37b4b8358bf398b65 Mon Sep 17 00:00:00 2001
From: Andy Newjack <andynewjack@protonmail.com>
Date: Mon, 27 Apr 2020 21:59:15 -0300
Subject: [PATCH] add room password handling

---
 src/ship/location.rs |  2 --
 src/ship/ship.rs     | 21 ++++++++++++++++++++-
 2 files changed, 20 insertions(+), 3 deletions(-)

diff --git a/src/ship/location.rs b/src/ship/location.rs
index bc15201..08af31f 100644
--- a/src/ship/location.rs
+++ b/src/ship/location.rs
@@ -159,7 +159,6 @@ impl ClientLocation {
         Ok(())
     }
 
-
     pub fn add_client_to_next_available_lobby(&mut self, id: ClientId, lobby: LobbyId) -> Result<LobbyId, JoinLobbyError> {
         let l = (0..15)
             .map(|lobby_index| {
@@ -207,7 +206,6 @@ impl ClientLocation {
         Ok(())
     }
 
-
     pub fn get_all_clients_by_client(&self, id: ClientId) -> Result<Vec<AreaClient>, GetNeighborError> {
         let area = self.client_location.get(&id).ok_or(GetNeighborError::InvalidClient)?;
         match area {
diff --git a/src/ship/ship.rs b/src/ship/ship.rs
index 10d2db5..3cf03d2 100644
--- a/src/ship/ship.rs
+++ b/src/ship/ship.rs
@@ -51,6 +51,7 @@ pub enum ShipError {
 pub enum RecvShipPacket {
     Login(Login),
     MenuSelect(MenuSelect),
+    RoomPasswordReq(RoomPasswordReq),
     CharData(CharData),
     Message(Message),
     DirectMessage(DirectMessage),
@@ -70,7 +71,13 @@ impl RecvServerPacket for RecvShipPacket {
     fn from_bytes(data: &[u8]) -> Result<RecvShipPacket, PacketParseError> {
         match u16::from_le_bytes([data[2], data[3]]) {
             0x93 => Ok(RecvShipPacket::Login(Login::from_bytes(data)?)),
-            0x10 => Ok(RecvShipPacket::MenuSelect(MenuSelect::from_bytes(data)?)),
+            0x10 => {
+                match data[0] {
+                    16 => Ok(RecvShipPacket::MenuSelect(MenuSelect::from_bytes(data)?)),
+                    48 => Ok(RecvShipPacket::RoomPasswordReq(RoomPasswordReq::from_bytes(data)?)),
+                    _ => Err(PacketParseError::WrongPacketForServerType(u16::from_le_bytes([data[2], data[3]]), data.to_vec())),
+                }
+            },
             0x61 => Ok(RecvShipPacket::CharData(CharData::from_bytes(data)?)),
             0x60 => Ok(RecvShipPacket::Message(Message::from_bytes(data)?)),
             0x62 => Ok(RecvShipPacket::DirectMessage(DirectMessage::from_bytes(data)?)),
@@ -261,6 +268,18 @@ impl<EG: EntityGateway> ServerState for ShipServerState<EG> {
                     _ => unreachable!(),
                 }
             },
+            RecvShipPacket::RoomPasswordReq(room_password_req) => {
+                if room_password_req.password == self.rooms[room_password_req.item as usize].as_ref().unwrap().password {
+                    let menuselect = MenuSelect {
+                        menu: room_password_req.menu,
+                        item: room_password_req.item,
+                    };
+                    handler::room::join_room(id, &menuselect, &mut self.client_location, &mut self.clients, &self.level_table, &mut self.rooms)?
+                }
+                else {
+                    Box::new(vec![(id, SendShipPacket::SmallDialog(SmallDialog::new("Incorrect password".into())))].into_iter())
+                }
+            },
             RecvShipPacket::CharData(chardata) => {
                 Box::new(handler::lobby::send_player_to_lobby(id, chardata, &mut self.client_location, &self.clients, &self.level_table)?.into_iter())
             },