morethantext/src/data/record.rs

use crate::data::id::ID;
use std::{collections::HashMap, fmt};

#[derive(Clone)]
pub enum Field {
    ID(ID),
}

impl Field {
    fn new(field_type: &str) -> Field {
        ID::new_field()
    }
}

pub trait FieldRecord {
    fn new_field() -> Field;
    fn get_type() -> String;
}

pub struct Record {
    data: HashMap<String, Field>,
}

impl Record {
    fn new(data: HashMap<String, Field>) -> Self {
        Self { data: data }
    }

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

    fn get(&self, name: &str) -> Field {
        match self.data.get(name) {
            Some(data) => data.clone(),
            None => unreachable!(),
        }
    }
}

impl fmt::Display for Field {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            Field::ID(id) => write!(f, "{}", id),
        }
    }
}

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

    #[test]
    fn creaAte_new_id() {
        match Field::new("id") {
            Field::ID(_) => {}
            _ => unreachable!("Fould should be an ID type."),
        }
    }
}

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

    #[test]
    fn new_record() {
        let data: HashMap<String, Field> = HashMap::new();
        let rec = Record::new(data);
        assert_eq!(rec.len(), 0);
    }

    #[test]
    fn new_record_more_data() {
        let a = ID::new_field();
        let b = ID::new_field();
        let c = ID::new_field();
        let mut data: HashMap<String, Field> = HashMap::new();
        data.insert("a".to_string(), a.clone());
        data.insert("b".to_string(), b.clone());
        data.insert("c".to_string(), c.clone());
        let rec = Record::new(data);
        assert_eq!(rec.len(), 3);
        assert_eq!(rec.get("a").to_string(), a.to_string(), "record a");
        assert_eq!(rec.get("b").to_string(), b.to_string(), "record b");
        assert_eq!(rec.get("c").to_string(), c.to_string(), "record c");
    }
}