1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119

#[allow(clippy::too_many_lines)]
mod item_def;
mod loc_def;
mod map_def;
mod npc_def;
#[allow(clippy::too_many_lines)]
mod obj_def;

pub use item_def::*;
pub use loc_def::*;
pub use map_def::*;
pub use npc_def::*;
pub use obj_def::*;

use std::collections::HashMap;

use crate::{
    archive::{Archive, ArchiveFileGroup},
    codec, Cache, REFERENCE_TABLE,
};

/// Marker trait for definitions.
pub trait Definition: Sized {
    fn new(id: u16, buffer: &[u8]) -> crate::Result<Self>;
}

/// Adds definition fetching from the cache to every struct that implements `Definition`.
///
/// The main difference between `fetch_from_index` and `fetch_from_archive`:
/// - `fetch_from_index` will get only 1 definition from each archive making it a 1:1 relation.
/// - `fetch_from_archive` will get multiple definitions from each archive making it a N:1 relation
/// where N is atleast 1.
pub trait FetchDefinition: Definition {
    // TODO: finish documentation with example.
    /// Fetches multiple definitions from every archive in the index.
    ///
    /// Note: every archive contains only one definition. (1:1)
    ///
    /// # Errors
    ///
    /// Can return multiple errors: if reading, decoding or parsing definition buffers fail.
    #[inline]
    fn fetch_from_index<D>(cache: &Cache, index_id: u8) -> crate::Result<HashMap<u16, D>>
    where
        D: Definition,
    {
        let buffer = cache.read(REFERENCE_TABLE, index_id as u32)?;
        let buffer = codec::decode(&buffer)?;
        let archives = Archive::parse(&buffer)?;
        let mut definitions = HashMap::new();
        for archive in &archives {
            let buffer = cache.read(index_id, archive.id)?;
            let buffer = codec::decode(&buffer)?;

            definitions.insert(archive.id as u16, D::new(archive.id as u16, &buffer)?);
        }

        Ok(definitions)
    }

    /// Fetches multiple definitions from a single archive.
    ///
    /// Note: every archive contains multiple definitions. (N:1)
    ///
    /// # Errors
    ///
    /// Can return multiple errors: if reading, decoding or parsing definition buffers fail.
    ///
    /// # Examples
    ///
    /// ```
    /// # use std::collections::HashMap;
    /// # use rscache::Cache;
    /// use rscache::definition::osrs::{
    ///     FetchDefinition,
    ///     ItemDefinition,
    /// };
    ///
    /// # fn main() -> rscache::Result<()> {
    /// # let cache = Cache::new("./data/osrs_cache")?;
    /// let index_id = 2; // Config index.
    /// let archive_id = 10; // Archive containing item definitions.
    ///
    /// let item_defs: HashMap<u16, ItemDefinition>
    ///     = ItemDefinition::fetch_from_archive(&cache, index_id, archive_id)?;
    /// # Ok(())
    /// # }
    /// ```
    #[inline]
    fn fetch_from_archive<D>(
        cache: &Cache,
        index_id: u8,
        archive_id: u32,
    ) -> crate::Result<HashMap<u16, D>>
    where
        D: Definition,
    {
        let buffer = cache.read(REFERENCE_TABLE, index_id as u32)?;
        let buffer = codec::decode(&buffer)?;
        let archives = Archive::parse(&buffer)?;
        let entry_count = archives[archive_id as usize - 1].entry_count;
        let buffer = cache.read(index_id, archive_id)?;
        let buffer = codec::decode(&buffer)?;

        let archive_group = ArchiveFileGroup::parse(&buffer, entry_count)?;

        let mut definitions = HashMap::new();
        for archive_file in archive_group {
            definitions.insert(
                archive_file.id as u16,
                D::new(archive_file.id as u16, &archive_file.data)?,
            );
        }

        Ok(definitions)
    }
}

impl<D: Definition> FetchDefinition for D {}