use std::collections::HashMap;

use super::group::{Group, GroupId};

pub struct Lobby {
    groups: HashMap<GroupId, Group>,
}

impl Lobby {
    pub fn new() -> Lobby {
        Self {
            groups: HashMap::new(),
        }
    }

    pub fn add_group(&mut self, group: Group) {
        self.groups.insert(group.get_id(), group);
    }

    pub fn iter<'a>(&'a self) -> GroupIterator<'a> {
        GroupIterator { groups: self.groups.values() }
    }
}

pub struct GroupIterator<'a> {
    groups: std::collections::hash_map::Values<'a, GroupId, Group>
}

impl<'a> Iterator for GroupIterator<'a> {
    type Item = &'a Group;

    fn next(&mut self) -> Option<Self::Item> {
        self.groups.next()
    }
}