gameServer2/src/server/handlers/inroom.rs
changeset 13424 81e0ed105f5d
parent 13421 cdf69667593b
child 13427 5fb27f94fc3b
--- a/gameServer2/src/server/handlers/inroom.rs	Thu Jun 21 23:09:20 2018 +0200
+++ b/gameServer2/src/server/handlers/inroom.rs	Thu Jun 21 17:23:10 2018 -0400
@@ -1,34 +1,36 @@
 use mio;
 
-use server::{
-    server::HWServer,
-    actions::{Action, Action::*}
-};
 use protocol::messages::{
     HWProtocolMessage,
     HWServerMessage::*
 };
+use server::{
+    server::HWServer,
+    client::ClientId,
+    room::HWRoom,
+    actions::{Action, Action::*}
+};
 use utils::is_name_illegal;
 use std::mem::swap;
 
-pub fn handle(server: &mut HWServer, token: usize, message: HWProtocolMessage) {
+pub fn handle(server: &mut HWServer, client_id: ClientId, message: HWProtocolMessage) {
     use protocol::messages::HWProtocolMessage::*;
     match message {
-        Part(None) => server.react(token, vec![
+        Part(None) => server.react(client_id, vec![
             MoveToLobby("part".to_string())]),
-        Part(Some(msg)) => server.react(token, vec![
+        Part(Some(msg)) => server.react(client_id, vec![
             MoveToLobby(format!("part: {}", msg))]),
         Chat(msg) => {
-            let chat_msg;
-            let room_id;
-            {
-                let c = &mut server.clients[token];
-                chat_msg = ChatMsg(c.nick.clone(), msg);
-                room_id = c.room_id;
-            }
-            let client_ids = server.other_clients_in_room(token);
-            server.react(token, vec![
-                SendToSelected(client_ids, chat_msg)]);
+            let actions = {
+                let c = &mut server.clients[client_id];
+                let chat_msg = ChatMsg(c.nick.clone(), msg);
+                if let Some(room_id) = c.room_id {
+                    vec![chat_msg.send_all().in_room(room_id).but_self().action()]
+                } else {
+                    Vec::new()
+                }
+            };
+            server.react(client_id, actions);
         },
         RoomName(new_name) => {
             let actions =
@@ -38,15 +40,130 @@
                     vec![Warn("A room with the same name already exists.".to_string())]
                 } else {
                     let mut old_name = new_name.clone();
-                    if let (c, Some(r)) = server.client_and_room(token) {
+                    if let (c, Some(r)) = server.client_and_room(client_id) {
                         swap(&mut r.name, &mut old_name);
                         vec![SendRoomUpdate(Some(old_name))]
                     } else {
                         Vec::new()
                     }
                 };
-            server.react(token, actions);
+            server.react(client_id, actions);
+        },
+        ToggleReady => {
+            let actions = if let (c, Some(r)) = server.client_and_room(client_id) {
+                let flags = if c.is_ready {
+                    r.ready_players_number -= 1;
+                    "-r"
+                } else {
+                    r.ready_players_number += 1;
+                    "+r"
+                };
+                c.is_ready = !c.is_ready;
+                vec![ClientFlags(flags.to_string(), vec![c.nick.clone()])
+                    .send_all().in_room(r.id).action()]
+            } else {
+                Vec::new()
+            };
+            server.react(client_id, actions);
         }
-        _ => warn!("Unimplemented!"),
+        AddTeam(mut info) => {
+            let mut actions = Vec::new();
+            if let (c, Some(r)) = server.client_and_room(client_id) {
+                let room_id = r.id;
+                if r.teams.len() >= r.team_limit as usize {
+                    actions.push(Warn("Too many teams!".to_string()))
+                } else if r.addable_hedgehogs() == 0 {
+                    actions.push(Warn("Too many hedgehogs!".to_string()))
+                } else if r.find_team(|t| t.name == info.name) != None {
+                    actions.push(Warn("There's already a team with same name in the list.".to_string()))
+                } else if r.game_info != None {
+                    actions.push(Warn("Joining not possible: Round is in progress.".to_string()))
+                } else {
+                    let team = r.add_team(c.id, info);
+                    c.teams_in_game += 1;
+                    c.clan = Some(team.color);
+                    actions.push(TeamAccepted(team.name.clone())
+                        .send_self().action());
+                    actions.push(TeamAdd(HWRoom::team_info(&c, team))
+                        .send_all().in_room(room_id).but_self().action());
+                    actions.push(TeamColor(team.name.clone(), team.color)
+                        .send_all().in_room(room_id).action());
+                    actions.push(HedgehogsNumber(team.name.clone(), team.hedgehogs_number)
+                        .send_all().in_room(room_id).action());
+                    actions.push(SendRoomUpdate(None));
+                }
+            }
+            server.react(client_id, actions);
+        },
+        RemoveTeam(name) => {
+            let mut actions = Vec::new();
+            if let (c, Some(r)) = server.client_and_room(client_id) {
+                match r.find_team_owner(&name) {
+                    None =>
+                        actions.push(Warn("Error: The team you tried to remove does not exist.".to_string())),
+                    Some((id, _)) if id != client_id =>
+                        actions.push(Warn("You can't remove a team you don't own.".to_string())),
+                    Some((_, name)) => {
+                        c.teams_in_game -= 1;
+                        c.clan = r.find_team_color(c.id);
+                        actions.push(Action::RemoveTeam(name.to_string()));
+                    }
+                }
+            };
+            server.react(client_id, actions);
+        },
+        SetHedgehogsNumber(team_name, number) => {
+            let actions = if let (c, Some(r)) = server.client_and_room(client_id) {
+                let room_id = r.id;
+                let addable_hedgehogs = r.addable_hedgehogs();
+                if let Some((_, mut team)) = r.find_team_and_owner_mut(|t| t.name == team_name) {
+                    if !c.is_master {
+                        vec![ProtocolError("You're not the room master!".to_string())]
+                    } else if number < 1 || number > 8
+                           || number > addable_hedgehogs + team.hedgehogs_number {
+                        vec![HedgehogsNumber(team.name.clone(), team.hedgehogs_number)
+                            .send_self().action()]
+                    } else {
+                        team.hedgehogs_number = number;
+                        vec![HedgehogsNumber(team.name.clone(), number)
+                            .send_all().in_room(room_id).but_self().action()]
+                    }
+                } else {
+                    vec![(Warn("No such team.".to_string()))]
+                }
+            } else {
+                Vec::new()
+            };
+            server.react(client_id, actions);
+        },
+        SetTeamColor(team_name, color) => {
+            let mut owner_id = None;
+            let actions = if let (c, Some(r)) = server.client_and_room(client_id) {
+                let room_id = r.id;
+                if let Some((owner, mut team)) = r.find_team_and_owner_mut(|t| t.name == team_name) {
+                    if !c.is_master {
+                        vec![ProtocolError("You're not the room master!".to_string())]
+                    } else if false  {
+                        Vec::new()
+                    } else {
+                        owner_id = Some(owner);
+                        team.color = color;
+                        vec![TeamColor(team.name.clone(), color)
+                            .send_all().in_room(room_id).but_self().action()]
+                    }
+                } else {
+                    vec![(Warn("No such team.".to_string()))]
+                }
+            } else {
+                Vec::new()
+            };
+
+            if let Some(id) = owner_id {
+                server.clients[id].clan = Some(color);
+            }
+
+            server.react(client_id, actions);
+        }
+        _ => warn!("Unimplemented!")
     }
 }