diff -r 6843c4551cde -r 06672690d71b gameServer2/src/server/room.rs --- a/gameServer2/src/server/room.rs Mon Dec 10 21:23:52 2018 +0100 +++ /dev/null Thu Jan 01 00:00:00 1970 +0000 @@ -1,391 +0,0 @@ -use std::{ - iter, collections::HashMap -}; -use crate::server::{ - coretypes::{ - ClientId, RoomId, TeamInfo, GameCfg, GameCfg::*, Voting, - MAX_HEDGEHOGS_PER_TEAM - }, - client::{HWClient} -}; -use bitflags::*; -use serde::{Serialize, Deserialize}; -use serde_derive::{Serialize, Deserialize}; -use serde_yaml; - -const MAX_TEAMS_IN_ROOM: u8 = 8; -const MAX_HEDGEHOGS_IN_ROOM: u8 = - MAX_HEDGEHOGS_PER_TEAM * MAX_HEDGEHOGS_PER_TEAM; - -#[derive(Clone, Serialize, Deserialize)] -struct Ammo { - name: String, - settings: Option -} - -#[derive(Clone, Serialize, Deserialize)] -struct Scheme { - name: String, - settings: Vec -} - -#[derive(Clone, Serialize, Deserialize)] -struct RoomConfig { - feature_size: u32, - map_type: String, - map_generator: u32, - maze_size: u32, - seed: String, - template: u32, - - ammo: Ammo, - scheme: Scheme, - script: String, - theme: String, - drawn_map: Option -} - -impl RoomConfig { - fn new() -> RoomConfig { - RoomConfig { - feature_size: 12, - map_type: "+rnd+".to_string(), - map_generator: 0, - maze_size: 0, - seed: "seed".to_string(), - template: 0, - - ammo: Ammo {name: "Default".to_string(), settings: None }, - scheme: Scheme {name: "Default".to_string(), settings: Vec::new() }, - script: "Normal".to_string(), - theme: "\u{1f994}".to_string(), - drawn_map: None - } - } -} - -fn client_teams_impl(teams: &[(ClientId, TeamInfo)], client_id: ClientId) - -> impl Iterator + Clone -{ - teams.iter().filter(move |(id, _)| *id == client_id).map(|(_, t)| t) -} - -fn map_config_from(c: &RoomConfig) -> Vec { - vec![c.feature_size.to_string(), c.map_type.to_string(), - c.map_generator.to_string(), c.maze_size.to_string(), - c.seed.to_string(), c.template.to_string()] -} - -fn game_config_from(c: &RoomConfig) -> Vec { - use crate::server::coretypes::GameCfg::*; - let mut v = vec![ - Ammo(c.ammo.name.to_string(), c.ammo.settings.clone()), - Scheme(c.scheme.name.to_string(), c.scheme.settings.clone()), - Script(c.script.to_string()), - Theme(c.theme.to_string())]; - if let Some(ref m) = c.drawn_map { - v.push(DrawnMap(m.to_string())) - } - v -} - -pub struct GameInfo { - pub teams_in_game: u8, - pub teams_at_start: Vec<(ClientId, TeamInfo)>, - pub left_teams: Vec, - pub msg_log: Vec, - pub sync_msg: Option, - pub is_paused: bool, - config: RoomConfig -} - -impl GameInfo { - fn new(teams: Vec<(ClientId, TeamInfo)>, config: RoomConfig) -> GameInfo { - GameInfo { - left_teams: Vec::new(), - msg_log: Vec::new(), - sync_msg: None, - is_paused: false, - teams_in_game: teams.len() as u8, - teams_at_start: teams, - config - } - } - - pub fn client_teams(&self, client_id: ClientId) -> impl Iterator + Clone { - client_teams_impl(&self.teams_at_start, client_id) - } -} - -#[derive(Serialize, Deserialize)] -pub struct RoomSave { - pub location: String, - config: RoomConfig -} - -bitflags!{ - pub struct RoomFlags: u8 { - const FIXED = 0b0000_0001; - const RESTRICTED_JOIN = 0b0000_0010; - const RESTRICTED_TEAM_ADD = 0b0000_0100; - const RESTRICTED_UNREGISTERED_PLAYERS = 0b0000_1000; - } -} - -pub struct HWRoom { - pub id: RoomId, - pub master_id: Option, - pub name: String, - pub password: Option, - pub greeting: String, - pub protocol_number: u16, - pub flags: RoomFlags, - - pub players_number: u8, - pub default_hedgehog_number: u8, - pub team_limit: u8, - pub ready_players_number: u8, - pub teams: Vec<(ClientId, TeamInfo)>, - config: RoomConfig, - pub voting: Option, - pub saves: HashMap, - pub game_info: Option -} - -impl HWRoom { - pub fn new(id: RoomId) -> HWRoom { - HWRoom { - id, - master_id: None, - name: String::new(), - password: None, - greeting: "".to_string(), - flags: RoomFlags::empty(), - protocol_number: 0, - players_number: 0, - default_hedgehog_number: 4, - team_limit: MAX_TEAMS_IN_ROOM, - ready_players_number: 0, - teams: Vec::new(), - config: RoomConfig::new(), - voting: None, - saves: HashMap::new(), - game_info: None - } - } - - pub fn hedgehogs_number(&self) -> u8 { - self.teams.iter().map(|(_, t)| t.hedgehogs_number).sum() - } - - pub fn addable_hedgehogs(&self) -> u8 { - MAX_HEDGEHOGS_IN_ROOM - self.hedgehogs_number() - } - - pub fn add_team(&mut self, owner_id: ClientId, mut team: TeamInfo, preserve_color: bool) -> &TeamInfo { - if !preserve_color { - team.color = iter::repeat(()).enumerate() - .map(|(i, _)| i as u8).take(u8::max_value() as usize + 1) - .find(|i| self.teams.iter().all(|(_, t)| t.color != *i)) - .unwrap_or(0u8) - }; - team.hedgehogs_number = if self.teams.is_empty() { - self.default_hedgehog_number - } else { - self.teams[0].1.hedgehogs_number.min(self.addable_hedgehogs()) - }; - self.teams.push((owner_id, team)); - &self.teams.last().unwrap().1 - } - - pub fn remove_team(&mut self, name: &str) { - if let Some(index) = self.teams.iter().position(|(_, t)| t.name == name) { - self.teams.remove(index); - } - } - - pub fn set_hedgehogs_number(&mut self, n: u8) -> Vec { - let mut names = Vec::new(); - let teams = match self.game_info { - Some(ref mut info) => &mut info.teams_at_start, - None => &mut self.teams - }; - - if teams.len() as u8 * n <= MAX_HEDGEHOGS_IN_ROOM { - for (_, team) in teams.iter_mut() { - team.hedgehogs_number = n; - names.push(team.name.clone()) - }; - self.default_hedgehog_number = n; - } - names - } - - pub fn find_team_and_owner_mut(&mut self, f: F) -> Option<(ClientId, &mut TeamInfo)> - where F: Fn(&TeamInfo) -> bool { - self.teams.iter_mut().find(|(_, t)| f(t)).map(|(id, t)| (*id, t)) - } - - pub fn find_team(&self, f: F) -> Option<&TeamInfo> - where F: Fn(&TeamInfo) -> bool { - self.teams.iter().find_map(|(_, t)| Some(t).filter(|t| f(&t))) - } - - pub fn client_teams(&self, client_id: ClientId) -> impl Iterator { - client_teams_impl(&self.teams, client_id) - } - - pub fn client_team_indices(&self, client_id: ClientId) -> Vec { - self.teams.iter().enumerate() - .filter(move |(_, (id, _))| *id == client_id) - .map(|(i, _)| i as u8).collect() - } - - pub fn find_team_owner(&self, team_name: &str) -> Option<(ClientId, &str)> { - self.teams.iter().find(|(_, t)| t.name == team_name) - .map(|(id, t)| (*id, &t.name[..])) - } - - pub fn find_team_color(&self, owner_id: ClientId) -> Option { - self.client_teams(owner_id).nth(0).map(|t| t.color) - } - - pub fn has_multiple_clans(&self) -> bool { - self.teams.iter().min_by_key(|(_, t)| t.color) != - self.teams.iter().max_by_key(|(_, t)| t.color) - } - - pub fn set_config(&mut self, cfg: GameCfg) { - let c = &mut self.config; - match cfg { - FeatureSize(s) => c.feature_size = s, - MapType(t) => c.map_type = t, - MapGenerator(g) => c.map_generator = g, - MazeSize(s) => c.maze_size = s, - Seed(s) => c.seed = s, - Template(t) => c.template = t, - - Ammo(n, s) => c.ammo = Ammo {name: n, settings: s}, - Scheme(n, s) => c.scheme = Scheme {name: n, settings: s}, - Script(s) => c.script = s, - Theme(t) => c.theme = t, - DrawnMap(m) => c.drawn_map = Some(m) - }; - } - - pub fn start_round(&mut self) { - if self.game_info.is_none() { - self.game_info = Some(GameInfo::new( - self.teams.clone(), self.config.clone())); - } - } - - pub fn is_fixed(&self) -> bool { - self.flags.contains(RoomFlags::FIXED) - } - pub fn is_join_restricted(&self) -> bool { - self.flags.contains(RoomFlags::RESTRICTED_JOIN) - } - pub fn is_team_add_restricted(&self) -> bool { - self.flags.contains(RoomFlags::RESTRICTED_TEAM_ADD) - } - pub fn are_unregistered_players_restricted(&self) -> bool { - self.flags.contains(RoomFlags::RESTRICTED_UNREGISTERED_PLAYERS) - } - - pub fn set_is_fixed(&mut self, value: bool) { - self.flags.set(RoomFlags::FIXED, value) - } - pub fn set_join_restriction(&mut self, value: bool) { - self.flags.set(RoomFlags::RESTRICTED_JOIN, value) - } - pub fn set_team_add_restriction(&mut self, value: bool) { - self.flags.set(RoomFlags::RESTRICTED_TEAM_ADD, value) - } - pub fn set_unregistered_players_restriction(&mut self, value: bool) { - self.flags.set(RoomFlags::RESTRICTED_UNREGISTERED_PLAYERS, value) - } - - fn flags_string(&self) -> String { - let mut result = "-".to_string(); - if self.game_info.is_some() { result += "g" } - if self.password.is_some() { result += "p" } - if self.is_join_restricted() { result += "j" } - if self.are_unregistered_players_restricted() { - result += "r" - } - result - } - - pub fn info(&self, master: Option<&HWClient>) -> Vec { - let c = &self.config; - vec![ - self.flags_string(), - self.name.clone(), - self.players_number.to_string(), - self.teams.len().to_string(), - master.map_or("[]", |c| &c.nick).to_string(), - c.map_type.to_string(), - c.script.to_string(), - c.scheme.name.to_string(), - c.ammo.name.to_string() - ] - } - - pub fn map_config(&self) -> Vec { - match self.game_info { - Some(ref info) => map_config_from(&info.config), - None => map_config_from(&self.config) - } - } - - pub fn game_config(&self) -> Vec { - match self.game_info { - Some(ref info) => game_config_from(&info.config), - None => game_config_from(&self.config) - } - } - - pub fn save_config(&mut self, name: String, location: String) { - self.saves.insert(name, RoomSave { location, config: self.config.clone() }); - } - - pub fn load_config(&mut self, name: &str) -> Option<&str> { - if let Some(save) = self.saves.get(name) { - self.config = save.config.clone(); - Some(&save.location[..]) - } else { - None - } - } - - pub fn delete_config(&mut self, name: &str) -> bool { - self.saves.remove(name).is_some() - } - - pub fn get_saves(&self) -> Result { - serde_yaml::to_string(&(&self.greeting, &self.saves)) - } - - pub fn set_saves(&mut self, text: &str) -> Result<(), serde_yaml::Error> { - serde_yaml::from_str::<(String, HashMap)>(text).map(|(greeting, saves)| { - self.greeting = greeting; - self.saves = saves; - }) - } - - pub fn team_info(owner: &HWClient, team: &TeamInfo) -> Vec { - let mut info = vec![ - team.name.clone(), - team.grave.clone(), - team.fort.clone(), - team.voice_pack.clone(), - team.flag.clone(), - owner.nick.clone(), - team.difficulty.to_string()]; - let hogs = team.hedgehogs.iter().flat_map(|h| - iter::once(h.name.clone()).chain(iter::once(h.hat.clone()))); - info.extend(hogs); - info - } -} \ No newline at end of file