pogo-masterfile 0.1.5

//! Generated from Pokémon GO masterfile — accessor for "codeGateProto".

use std::collections::HashMap;

use pogo_masterfile_types::{
    MasterfileEntry,
    code_gate_proto::{CodeGateProtoEntry, CodeGateProtoTemplateId},
};

pub struct CodeGateProtoAccessor<'a> {
    pub(crate) entries: &'a [MasterfileEntry],
    pub(crate) index: &'a HashMap<CodeGateProtoTemplateId, usize>,
    pub(crate) order: &'a [usize],
}

impl<'a> CodeGateProtoAccessor<'a> {
    /// Look up an entry by its templateId. Accepts either the typed
    /// `CodeGateProtoTemplateId` enum variant (compile-time validated) or
    /// `&str` (runtime-parsed via `FromStr`). Returns `None` if the string
    /// fails to parse OR no entry exists for the ID.
    pub fn get<I>(&self, id: I) -> Option<&'a CodeGateProtoEntry>
    where
        I: TryInto<CodeGateProtoTemplateId>,
    {
        let typed = id.try_into().ok()?;
        let idx = *self.index.get(&typed)?;
        match &self.entries[idx] {
            MasterfileEntry::CodeGateProto(e) => Some(e),
            _ => None,
        }
    }

    pub fn has<I>(&self, id: I) -> bool
    where
        I: TryInto<CodeGateProtoTemplateId>,
    {
        self.get(id).is_some()
    }

    pub fn iter(&self) -> impl Iterator<Item = &'a CodeGateProtoEntry> + '_ {
        self.order
            .iter()
            .filter_map(|&idx| match &self.entries[idx] {
                MasterfileEntry::CodeGateProto(e) => Some(e),
                _ => None,
            })
    }

    pub fn template_ids(&self) -> impl Iterator<Item = CodeGateProtoTemplateId> + '_ {
        self.index.keys().copied()
    }

    pub fn len(&self) -> usize {
        self.order.len()
    }

    pub fn is_empty(&self) -> bool {
        self.order.is_empty()
    }
}