diff options
Diffstat (limited to 'game_server/src/lobby.rs')
-rw-r--r-- | game_server/src/lobby.rs | 30 |
1 files changed, 28 insertions, 2 deletions
diff --git a/game_server/src/lobby.rs b/game_server/src/lobby.rs index fe3bdee..d03bd45 100644 --- a/game_server/src/lobby.rs +++ b/game_server/src/lobby.rs @@ -1,23 +1,49 @@ use std::collections::HashMap; use super::group::{Group, GroupId}; +use super::scribble_group::ScribbleGroup; + +use super::server::{UserId, GameClient}; pub struct Lobby { groups: HashMap<GroupId, Box<Group>>, } 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()))) + }, + other => 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) { + if !self.groups.contains_key(&group_id) { + let mut group = match Self::generate_group(group_type, group_id, group_name) { + Some(x) => x, + _ => return, + }; + 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() } } } |