diff options
Diffstat (limited to 'game_server/src/lobby.rs')
-rw-r--r-- | game_server/src/lobby.rs | 34 |
1 files changed, 31 insertions, 3 deletions
diff --git a/game_server/src/lobby.rs b/game_server/src/lobby.rs index fe3bdee..6d11a5f 100644 --- a/game_server/src/lobby.rs +++ b/game_server/src/lobby.rs @@ -1,27 +1,55 @@ use std::collections::HashMap; -use super::group::{Group, GroupId}; +use crate::group::{Group, GroupId}; +use crate::scribble_group::ScribbleGroup; + +use crate::server::{UserId, GameClient, GameServerError}; pub struct Lobby { groups: HashMap<GroupId, Box<Group>>, } +#[allow(dead_code)] impl Lobby { - pub fn new() -> Lobby { + pub fn new() -> Self { Self { groups: HashMap::new(), } } + fn generate_group(group_type: &str, id: GroupId, name: &str) -> Option<Box<Group>> { + match group_type { + "scribble" => { + Some(Box::new(ScribbleGroup::new(id, name.to_string()))) + }, + _ => None, + } + } + pub fn add_group(&mut self, group: Box<Group>) { self.groups.insert(group.id(), group); } - pub fn iter<'a>(&'a self) -> GroupIterator<'a> { + pub fn add_client(&mut self, group_type: &str, group_id: GroupId, group_name: &str, + user_id: UserId, client: GameClient) -> Result<(), GameServerError> { + if !self.groups.contains_key(&group_id) { + let mut group = match Self::generate_group(group_type, group_id, group_name) { + Some(x) => x, + _ => return Err(GameServerError::GroupCreationError(format!("failed to generate '{}' group", group_type))), + }; + group.run(); + self.groups.insert(group_id, group); + } + let group = self.groups.get_mut(&group_id).unwrap(); + group.add_client(user_id, client) + } + + pub fn iter<'b>(&'b self) -> GroupIterator<'b> { GroupIterator { groups: self.groups.values() } } } +#[allow(dead_code)] pub struct GroupIterator<'a> { groups: std::collections::hash_map::Values<'a, GroupId, Box<Group>> } |