wazabin-jstd 0.2.1

use rustc_hash::FxHashSet as HashSet;

/// Interns strings, returning 'static lifetime strings
#[derive(Default)]
pub struct StringPool(HashSet<&'static str>);

impl StringPool {
    /// Move a string into the pool
    pub fn intern(&mut self, s: &str) -> &'static str {
        if let Some(existing) = self.0.get(s) {
            return existing;
        }

        let boxed = s.to_owned().into_boxed_str();
        let leaked: &'static str = Box::leak(boxed);

        self.0.insert(leaked);
        leaked
    }
}

/// Moves a string into the pool, giving it a 'static lifetime
pub trait Intern {
    type Static;

    fn intern(self, pool: &mut StringPool) -> Self::Static;
}

impl Intern for &str {
    type Static = &'static str;

    fn intern(self, pool: &mut StringPool) -> Self::Static {
        pool.intern(self)
    }
}

impl<T> Intern for Vec<T>
where
    T: Intern,
{
    type Static = Vec<T::Static>;

    fn intern(self, pool: &mut StringPool) -> Self::Static {
        self.into_iter().map(|e| Intern::intern(e, pool)).collect()
    }
}

impl<T> Intern for Option<T>
where
    T: Intern,
{
    type Static = Option<T::Static>;

    fn intern(self, pool: &mut StringPool) -> Self::Static {
        self.map(|e| Intern::intern(e, pool))
    }
}

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

    #[test]
    fn pool_deduplicates_strings_and_interns_nested_values() {
        let mut pool = StringPool::default();
        let first = pool.intern("shared");
        let second = pool.intern("shared");
        assert!(std::ptr::eq(first, second));

        let values = vec![Some("shared"), None, Some("unique")].intern(&mut pool);
        assert!(std::ptr::eq(values[0].unwrap(), first));
        assert_eq!(values[1], None);
        assert_eq!(values[2], Some("unique"));
    }

    #[test]
    fn registry_interning_preserves_ids_and_values() {
        let mut pool = StringPool::default();
        let source: Registry<usize, &str> = ["left", "right"].into_iter().collect();
        let interned = source.intern(&mut pool);

        assert_eq!(interned.len(), 2);
        assert_eq!(interned[0].to_string(), "left");
        assert_eq!(interned[1].to_string(), "right");
    }
}