riimut 1.2.1

use crate::dictionary::DictionaryMapping;

pub fn with_dictionary(content: &str, dictionary: &DictionaryMapping) -> String {
    let mut result: String = String::new();

    for character in content.chars() {
        let lower_case: char = character.to_lowercase().next().unwrap();
       
        match &dictionary {
            DictionaryMapping::LetterDefinitions(mapping) => {
                if let Some(found_character) = mapping.get(&lower_case) {
                    result.push(*found_character);
                } else {
                    result.push(character);
                }
            }
            DictionaryMapping::MultipleLetterDefinitions(mapping) => {
                if let Some(found_string) = mapping.get(&lower_case) {
                    result.push_str(found_string);
                } else {
                    result.push(character);
                }
            }
        }
    }    

    result
}

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

    #[test]
    fn transforms_letters_to_runes() {
        let runemap = younger_futhark::mapping::get_letters_to_runes_map();
        let content = "auk tani karþi kristna";
        let expected = "ᛅᚢᚴ:ᛏᛅᚾᛁ:ᚴᛅᚱᚦᛁ:ᚴᚱᛁᛋᛏᚾᛅ";
        let result = with_dictionary(content, &runemap);

        assert_eq!(result, expected);
    }

    #[test]
    fn transforms_upper_and_lower_to_same() {
        let runemap = younger_futhark::mapping::get_letters_to_runes_map();
        let content = "AUK tani Karþi kriSTnA";
        let expected = "ᛅᚢᚴ:ᛏᛅᚾᛁ:ᚴᛅᚱᚦᛁ:ᚴᚱᛁᛋᛏᚾᛅ";
        let result = with_dictionary(content, &runemap);

        assert_eq!(result, expected);
    }

    #[test]
    fn transforms_char_to_string_mapped_runes() {
        let runemap = futhorc::mapping::get_runes_to_letters_map();
        let content = "ᚪᛒᚳᛞᛖᚠᚷᛠ";
        let expected = "abcdefgea";
        let result = with_dictionary(content, &runemap);

        assert_eq!(result, expected);
    }
}