ltk_ritobin 0.8.1

//! Hash provider traits and implementations for ritobin writing.
//!
//! When writing ritobin files, hashes can be written as hex values (e.g., `0xdeadbeef`)
//! or as human-readable strings (e.g., `"Characters/Aatrox/Skins/Skin0"`).
//!
//! This module provides traits and implementations for looking up hash values
//! to convert them back to their original strings.

use std::io::{BufRead, BufReader};
use std::path::Path;
use std::{borrow::Cow, collections::HashMap};
use std::{fs::File, str::FromStr};

use ltk_hash::{BinHash, WadHash};

/// Trait for looking up hash values to get their original string representation.
///
/// Implement this trait to provide custom hash lookup behavior when writing ritobin files.
#[allow(unused)]
pub trait HashProvider {
    /// Look up a bin entry path hash (root object paths like "Characters/Aatrox/Skins/Skin0").
    fn lookup_entry(&self, hash: BinHash) -> Option<Cow<'_, str>> {
        None
    }

    /// Look up a bin field/property name hash.
    fn lookup_field(&self, hash: BinHash) -> Option<Cow<'_, str>> {
        None
    }

    /// Look up a bin hash value (hash property type values).
    fn lookup_hash(&self, hash: BinHash) -> Option<Cow<'_, str>> {
        None
    }

    /// Look up a bin type/class name hash (for objects, structs, embeds).
    fn lookup_type(&self, hash: BinHash) -> Option<Cow<'_, str>> {
        None
    }

    /// Look up a wad path hash (for WadChunkLink)
    fn lookup_wad(&self, hash: WadHash) -> Option<Cow<'_, str>> {
        None
    }
}

impl HashProvider for () {}

/// A hash provider backed by HashMaps for each hash category.
///
/// This is the primary implementation for looking up hashes from loaded hash tables.
#[derive(Debug, Clone, Default)]
pub struct HashMapProvider {
    /// Hashes for bin entry paths (root object paths).
    pub entries: HashMap<BinHash, String>,
    /// Hashes for bin field/property names.
    pub fields: HashMap<BinHash, String>,
    /// Hashes for bin hash property values.
    pub hashes: HashMap<BinHash, String>,
    /// Hashes for bin type/class names.
    pub types: HashMap<BinHash, String>,
    /// Hashes for wad paths
    pub wads: HashMap<WadHash, String>,
}

impl HashMapProvider {
    pub fn new() -> Self {
        Self::default()
    }

    /// Load hash entries from a file with format "{hex_hash} {string}".
    ///
    /// Hash values are expected to be raw hex without "0x" prefix (e.g., "deadbeef SomeName").
    /// Lines starting with '#' are treated as comments and skipped.
    fn load_file<H: ltk_hash::Hash + FromStr>(
        path: impl AsRef<Path>,
    ) -> std::io::Result<HashMap<H, String>> {
        let file = File::open(path)?;
        let reader = BufReader::new(file);
        let mut map = HashMap::new();

        for line in reader.lines() {
            let line = line?;
            let line = line.trim();

            // Skip empty lines and comments
            if line.is_empty() || line.starts_with('#') {
                continue;
            }

            // Parse "{hex} {string}" format (hex without 0x prefix)
            if let Some((hash_str, value)) = line.split_once(' ') {
                if let Ok(hash) = H::from_str(hash_str) {
                    map.insert(hash, value.to_string());
                }
            }
        }

        Ok(map)
    }

    /// Load bin entry hashes from a file (hashes.binentries.txt).
    pub fn load_entries(&mut self, path: impl AsRef<Path>) -> std::io::Result<&mut Self> {
        self.entries = Self::load_file(path)?;
        Ok(self)
    }

    /// Load bin field hashes from a file (hashes.binfields.txt).
    pub fn load_fields(&mut self, path: impl AsRef<Path>) -> std::io::Result<&mut Self> {
        self.fields = Self::load_file(path)?;
        Ok(self)
    }

    /// Load bin hash value hashes from a file (hashes.binhashes.txt).
    pub fn load_hashes(&mut self, path: impl AsRef<Path>) -> std::io::Result<&mut Self> {
        self.hashes = Self::load_file(path)?;
        Ok(self)
    }

    /// Load bin type/class hashes from a file (hashes.bintypes.txt).
    pub fn load_types(&mut self, path: impl AsRef<Path>) -> std::io::Result<&mut Self> {
        self.types = Self::load_file(path)?;
        Ok(self)
    }

    /// Load wad hashes from a file (hashes.game.txt).
    pub fn load_wads(&mut self, path: impl AsRef<Path>) -> std::io::Result<&mut Self> {
        self.wads = Self::load_file(path)?;
        Ok(self)
    }

    /// Load all hash files from a directory.
    ///
    /// Expects files named:
    /// - `hashes.binentries.txt`
    /// - `hashes.binfields.txt`
    /// - `hashes.binhashes.txt`
    /// - `hashes.bintypes.txt`
    ///
    /// Missing files are silently ignored.
    pub fn load_from_directory(&mut self, dir: impl AsRef<Path>) -> &mut Self {
        let dir = dir.as_ref();

        let _ = self.load_entries(dir.join("hashes.binentries.txt"));
        let _ = self.load_fields(dir.join("hashes.binfields.txt"));
        let _ = self.load_hashes(dir.join("hashes.binhashes.txt"));
        let _ = self.load_types(dir.join("hashes.bintypes.txt"));
        let _ = self.load_wads(dir.join("hashes.game.txt"));

        self
    }

    pub fn insert_entry(
        &mut self,
        hash: impl Into<BinHash>,
        value: impl Into<String>,
    ) -> &mut Self {
        self.entries.insert(hash.into(), value.into());
        self
    }

    pub fn insert_field(
        &mut self,
        hash: impl Into<BinHash>,
        value: impl Into<String>,
    ) -> &mut Self {
        self.fields.insert(hash.into(), value.into());
        self
    }

    pub fn insert_hash(&mut self, hash: impl Into<BinHash>, value: impl Into<String>) -> &mut Self {
        self.hashes.insert(hash.into(), value.into());
        self
    }

    pub fn insert_type(&mut self, hash: impl Into<BinHash>, value: impl Into<String>) -> &mut Self {
        self.types.insert(hash.into(), value.into());
        self
    }

    pub fn insert_wad(&mut self, hash: impl Into<WadHash>, value: impl Into<String>) -> &mut Self {
        self.wads.insert(hash.into(), value.into());
        self
    }

    /// The total number of loaded hashes across all categories.
    pub fn total_count(&self) -> usize {
        self.entries.len() + self.fields.len() + self.hashes.len() + self.types.len()
    }
}

impl HashProvider for HashMapProvider {
    fn lookup_entry(&self, hash: BinHash) -> Option<Cow<'_, str>> {
        self.entries.get(&hash).map(|s| s.as_str().into())
    }

    fn lookup_field(&self, hash: BinHash) -> Option<Cow<'_, str>> {
        self.fields.get(&hash).map(|s| s.as_str().into())
    }

    fn lookup_hash(&self, hash: BinHash) -> Option<Cow<'_, str>> {
        self.hashes.get(&hash).map(|s| s.as_str().into())
    }

    fn lookup_type(&self, hash: BinHash) -> Option<Cow<'_, str>> {
        self.types.get(&hash).map(|s| s.as_str().into())
    }
    fn lookup_wad(&self, hash: WadHash) -> Option<Cow<'_, str>> {
        self.wads.get(&hash).map(|s| s.as_str().into())
    }
}

impl<T: HashProvider + ?Sized> HashProvider for &T {
    fn lookup_entry(&self, hash: BinHash) -> Option<Cow<'_, str>> {
        (*self).lookup_entry(hash)
    }

    fn lookup_field(&self, hash: BinHash) -> Option<Cow<'_, str>> {
        (*self).lookup_field(hash)
    }

    fn lookup_hash(&self, hash: BinHash) -> Option<Cow<'_, str>> {
        (*self).lookup_hash(hash)
    }

    fn lookup_type(&self, hash: BinHash) -> Option<Cow<'_, str>> {
        (*self).lookup_type(hash)
    }
    fn lookup_wad(&self, hash: WadHash) -> Option<Cow<'_, str>> {
        (*self).lookup_wad(hash)
    }
}

impl HashProvider for Box<dyn HashProvider> {
    fn lookup_entry(&self, hash: BinHash) -> Option<Cow<'_, str>> {
        self.as_ref().lookup_entry(hash)
    }

    fn lookup_field(&self, hash: BinHash) -> Option<Cow<'_, str>> {
        self.as_ref().lookup_field(hash)
    }

    fn lookup_hash(&self, hash: BinHash) -> Option<Cow<'_, str>> {
        self.as_ref().lookup_hash(hash)
    }

    fn lookup_type(&self, hash: BinHash) -> Option<Cow<'_, str>> {
        self.as_ref().lookup_type(hash)
    }

    fn lookup_wad(&self, hash: WadHash) -> Option<Cow<'_, str>> {
        self.as_ref().lookup_wad(hash)
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_empty_provider() {
        let provider = ();
        assert_eq!(provider.lookup_entry(0x12345678.into()), None);
        assert_eq!(provider.lookup_field(0x12345678.into()), None);
        assert_eq!(provider.lookup_hash(0x12345678.into()), None);
        assert_eq!(provider.lookup_type(0x12345678.into()), None);
        assert_eq!(provider.lookup_wad(0x12345678.into()), None);
    }

    #[test]
    fn test_hashmap_provider() {
        let mut provider = HashMapProvider::new();
        provider.insert_entry(0x12345678, "Characters/Test/Skin0");
        provider.insert_field(0xdeadbeef, "skinName");
        provider.insert_hash(0xcafebabe, "some/path");
        provider.insert_type(0xfeedface, "SkinData");
        provider.insert_wad(0xfeedface, "SkinDataWad");

        assert_eq!(
            provider.lookup_entry(0x12345678.into()),
            Some("Characters/Test/Skin0".into())
        );
        assert_eq!(
            provider.lookup_field(0xdeadbeef.into()),
            Some("skinName".into())
        );
        assert_eq!(
            provider.lookup_hash(0xcafebabe.into()),
            Some("some/path".into())
        );
        assert_eq!(
            provider.lookup_type(0xfeedface.into()),
            Some("SkinData".into())
        );
        assert_eq!(
            provider.lookup_wad(0xfeedface.into()),
            Some("SkinDataWad".into())
        );

        // Unknown hashes return None
        assert_eq!(provider.lookup_entry(0x11111111.into()), None);
    }
}