toornament 2.1.0

use *;

/// Disciplines iterator
pub struct DisciplinesIter<'a> {
    client: &'a Toornament,

    all: bool,
}
impl<'a> DisciplinesIter<'a> {
    /// Creates new disciplines iterator
    pub fn new(client: &'a Toornament) -> DisciplinesIter<'a> {
        DisciplinesIter { client, all: true }
    }

    /// Fetch all disciplines
    pub fn all(mut self) -> Self {
        self.all = true;
        self
    }
}

/// Modifiers
impl<'a> DisciplinesIter<'a> {
    /// Fetch a discipline with id
    pub fn with_id(self, discipline_id: DisciplineId) -> DisciplineIter<'a> {
        DisciplineIter {
            client: self.client,
            discipline_id,
        }
    }
}

/// Terminators
impl<'a> DisciplinesIter<'a> {
    /// Fetch the discipline
    pub fn collect<T: From<Disciplines>>(self) -> Result<T> {
        // TODO check for possible mistake.
        // check `if self.all` ?
        Ok(T::from(self.client.disciplines(None)?))
    }
}

/// Discipline iterator
pub struct DisciplineIter<'a> {
    client: &'a Toornament,

    /// Fetch a discipline with the following id
    discipline_id: DisciplineId,
}

impl<'a> DisciplineIter<'a> {
    /// Creates new discipline iterator
    pub fn new(client: &'a Toornament, discipline_id: DisciplineId) -> DisciplineIter<'a> {
        DisciplineIter {
            client,
            discipline_id,
        }
    }
}

/// Modifiers
impl<'a> DisciplineIter<'a> {
    /// Fetch matches of a discipline
    pub fn matches(self) -> DisciplineMatchesIter<'a> {
        DisciplineMatchesIter::new(self.client, self.discipline_id)
    }
}

/// Terminators
impl<'a> DisciplineIter<'a> {
    /// Fetch the discipline
    pub fn collect<T: From<Discipline>>(self) -> Result<T> {
        match self
            .client
            .disciplines(Some(self.discipline_id.clone()))?
            .0
            .first()
            .take()
        {
            Some(d) => Ok(T::from(d.to_owned())),
            None => Err(Error::Iter(IterError::NoSuchDiscipline(self.discipline_id))),
        }
    }
}