any-intern 0.1.5

use super::common::{Interned, RawInterned, UnsafeLock};
use bumpalo::Bump;
use core::str;
use hashbrown::{hash_table::Entry, HashTable};
use std::{
    alloc::Layout,
    fmt::{self, Debug, Display, Write},
    hash::{BuildHasher, Hash, Hasher},
    mem::MaybeUninit,
    ptr::{self, NonNull},
    slice,
};

/// A trait for types that can be stored in a dropless interner.
///
/// The `Dropless` trait provides methods for working with types that can be stored in a
/// [`DroplessInterner`] or [`DroplessInternSet`]. It defines how instances of a type are converted
/// to and from raw byte representations, hashed, and compared for equality. This is useful for
/// interning values without requiring ownership.
///
/// # Safety
///
/// Implementing the `Dropless` trait requires careful attention to alignment and memory safety. The
/// methods in this trait are marked as `unsafe` because they rely on the caller to ensure that the
/// provided byte slices are properly aligned and valid for the type. Misuse of these methods can
/// lead to undefined behavior.
///
/// # Examples
///
/// ```
/// use any_intern::Dropless;
/// use std::alloc::Layout;
/// use std::ptr::NonNull;
/// use std::hash::{Hash, Hasher, BuildHasher};
///
/// #[derive(PartialEq, Eq, Hash, Debug)]
/// struct MyType(u32);
///
/// impl Dropless for MyType {
///     fn as_byte_ptr(&self) -> NonNull<u8> {
///         unsafe { NonNull::new_unchecked(self as *const Self as *mut Self).cast::<u8>() }
///     }
///
///     fn layout(&self) -> Layout {
///         Layout::for_value(self)
///     }
///
///     unsafe fn from_bytes(bytes: &[u8]) -> &Self {
///         let ptr = bytes.as_ptr().cast::<Self>();
///         unsafe { ptr.as_ref().unwrap_unchecked() }
///     }
///
///     unsafe fn hash<S: BuildHasher>(hash_builder: &S, bytes: &[u8]) -> u64 {
///         let this = unsafe { Self::from_bytes(bytes) };
///         let mut hasher = hash_builder.build_hasher();
///         this.hash(&mut hasher);
///         hasher.finish()
///     }
///
///     unsafe fn eq(a: &[u8], b: &[u8]) -> bool {
///         unsafe { Self::from_bytes(a) == Self::from_bytes(b) }
///     }
/// }
/// ```
pub trait Dropless {
    /// Returns pointer to the instance.
    fn as_byte_ptr(&self) -> NonNull<u8> {
        // Safety: A reference is non-null
        unsafe { NonNull::new_unchecked(self as *const Self as *mut Self).cast::<u8>() }
    }

    /// Returns layout of the type.
    fn layout(&self) -> Layout {
        Layout::for_value(self)
    }

    /// Converts a byte slice into a reference to the type.
    ///
    /// The byte slice is guaranteed to be well aligned and correct data for the type.
    ///
    /// # Safety
    ///
    /// Undefined behavior if any conditions below are not met.
    /// * Implementation should interpret the byte slice into the type correctly.
    /// * Caller should give well aligned data for the type.
    unsafe fn from_bytes(bytes: &[u8]) -> &Self;

    /// Computes a hash value for the type using the provided byte slice.
    ///
    /// The byte slice is guaranteed to be well aligned and correct data for the type.
    ///
    /// # Safety
    ///
    /// Undefined behavior if any conditions below are not met.
    /// * Implementation should interpret the byte slice into the type correctly.
    /// * Caller should give well aligned data for the type.
    unsafe fn hash<S: BuildHasher>(hash_builder: &S, bytes: &[u8]) -> u64;

    /// Compares two byte slices for equality as instances of the type.
    ///
    /// The byte slices are guaranteed to be well aligned and correct data for the type.
    ///
    /// # Safety
    ///
    /// Undefined behavior if any conditions below are not met.
    /// * Implementation should interpret the byte slice into the type correctly.
    /// * Caller should give well aligned data for the type.
    unsafe fn eq(a: &[u8], b: &[u8]) -> bool;
}

macro_rules! simple_impl_dropless_fn {
    (from_bytes) => {
        unsafe fn from_bytes(bytes: &[u8]) -> &Self {
            let ptr = bytes.as_ptr().cast::<Self>();
            unsafe { ptr.as_ref().unwrap_unchecked() }
        }
    };
    (hash) => {
        unsafe fn hash<S: BuildHasher>(hash_builder: &S, bytes: &[u8]) -> u64 {
            let this = unsafe { Self::from_bytes(bytes) };
            let mut hasher = hash_builder.build_hasher();
            this.hash(&mut hasher);
            hasher.finish()
        }
    };
    (eq) => {
        unsafe fn eq(a: &[u8], b: &[u8]) -> bool {
            unsafe { Self::from_bytes(a) == Self::from_bytes(b) }
        }
    };
}

macro_rules! impl_dropless {
    ($($ty:ty)*) => {
        $(
            impl Dropless for $ty {
                simple_impl_dropless_fn!(from_bytes);
                simple_impl_dropless_fn!(hash);
                simple_impl_dropless_fn!(eq);
            }
        )*
    };
}

impl_dropless!(i8 i16 i32 i64 i128 isize u8 u16 u32 u64 u128 usize bool char);

impl<T: Dropless + Hash + Eq> Dropless for [T] {
    unsafe fn from_bytes(bytes: &[u8]) -> &Self {
        let ptr = bytes.as_ptr().cast::<T>();
        let stride = Layout::new::<T>().pad_to_align().size();
        let num_elems = bytes.len() / stride;
        unsafe { slice::from_raw_parts(ptr, num_elems) }
    }
    simple_impl_dropless_fn!(hash);
    simple_impl_dropless_fn!(eq);
}

impl<T: Dropless + Hash + Eq, const N: usize> Dropless for [T; N] {
    simple_impl_dropless_fn!(from_bytes);
    simple_impl_dropless_fn!(hash);
    simple_impl_dropless_fn!(eq);
}

impl Dropless for str {
    unsafe fn from_bytes(bytes: &[u8]) -> &Self {
        unsafe { std::str::from_utf8_unchecked(bytes) }
    }
    simple_impl_dropless_fn!(hash);
    simple_impl_dropless_fn!(eq);
}

impl<T: Dropless + Hash + Eq> Dropless for Option<T> {
    simple_impl_dropless_fn!(from_bytes);
    simple_impl_dropless_fn!(hash);
    simple_impl_dropless_fn!(eq);
}

impl<T: Dropless + Hash + Eq, E: Dropless + Hash + Eq> Dropless for Result<T, E> {
    simple_impl_dropless_fn!(from_bytes);
    simple_impl_dropless_fn!(hash);
    simple_impl_dropless_fn!(eq);
}

/// An interner for storing and deduplicating values without requiring ownership.
///
/// The `DroplessInterner` is designed for interning values that implement the [`Dropless`] trait.
/// It allows efficient storage and retrieval of values by ensuring that each unique value is stored
/// only once. Interning with this type always copies the given value into an internal buffer,
/// making it suitable for use cases where ownership is not required.
///
/// # Examples
///
/// ```
/// use any_intern::DroplessInterner;
///
/// let mut interner = DroplessInterner::new();
///
/// // Interning strings
/// let hello = interner.intern("hello");
/// let world = interner.intern("world");
/// let another_hello = interner.intern("hello");
///
/// assert_eq!(hello, another_hello); // Same value, same reference
/// assert_ne!(hello, world); // Different values, different references
///
/// // Checking if a value exists
/// assert!(interner.get("hello").is_some());
/// assert!(interner.get("unknown").is_none());
///
/// // Clearing the interner
/// interner.clear();
/// assert!(interner.is_empty());
/// ```
///
/// # Safety
///
/// The `DroplessInterner` relies on the `Dropless` trait for converting values to and from raw
/// byte representations. It is the responsibility of the `Dropless` implementation to ensure
/// memory safety and alignment when interacting with the interner.
#[derive(Debug)]
pub struct DroplessInterner<S = fxhash::FxBuildHasher> {
    inner: UnsafeLock<DroplessInternSet<S>>,
}

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

impl<S: BuildHasher> DroplessInterner<S> {
    pub fn with_hasher(hash_builder: S) -> Self {
        // Safety: Only one instance
        let inner = unsafe { UnsafeLock::new(DroplessInternSet::with_hasher(hash_builder)) };
        Self { inner }
    }

    /// Returns number of values the interner contains.
    pub fn len(&self) -> usize {
        self.with_inner(|set| set.len())
    }

    /// Returns true if the interner is empty.
    pub fn is_empty(&self) -> bool {
        self.with_inner(|set| set.is_empty())
    }

    /// Stores a value in the interner, returning a reference to the interned value.
    ///
    /// This method inserts the given value into the interner if it does not already exist.  If the
    /// value already exists, a reference to the existing value is returned. The value is copied
    /// into an internal buffer, making it suitable for use cases where ownership is not required.
    ///
    /// # Examples
    ///
    /// ```
    /// use any_intern::DroplessInterner;
    ///
    /// let interner = DroplessInterner::new();
    ///
    /// // Interning strings
    /// let hello = interner.intern("hello");
    /// let world = interner.intern("world");
    /// let another_hello = interner.intern("hello");
    ///
    /// assert_eq!(hello, another_hello); // Same value, same reference
    /// assert_ne!(hello, world); // Different values, different references
    ///
    /// // Interning arrays
    /// let array1 = interner.intern(&[1, 2, 3]);
    /// let array2 = interner.intern(&[1, 2, 3]);
    /// let array3 = interner.intern(&[4, 5, 6]);
    ///
    /// assert_eq!(array1, array2); // Same value, same reference
    /// assert_ne!(array1, array3); // Different values, different references
    /// ```
    pub fn intern<K: Dropless + ?Sized>(&self, value: &K) -> Interned<'_, K> {
        self.with_inner(|set| set.intern(value))
    }

    /// Stores a value in the interner as a formatted string through [`Display`], returning a
    /// reference to the interned value.
    ///
    /// This method provides a buffer for making string. This will be benefit in terms of
    /// performance when you frequently make `String` via something like `to_string()` by exploiting
    /// chunk memory.
    ///
    /// This method first formats the given value using the `Display` trait and stores the resulting
    /// string in the interner's buffer, then compares the string with existing values. If the
    /// formatted string already exists in the interner, formatted string is discarded and reference
    /// to the existing value is returned.
    ///
    /// If you give insufficient `upper_size`, then error is returned.
    ///
    /// # Examples
    ///
    /// ```
    /// use any_intern::DroplessInterner;
    ///
    /// let interner = DroplessInterner::new();
    ///
    /// let value = 42;
    /// let interned = interner.intern_formatted_str(&value, 10).unwrap();
    ///
    /// assert_eq!(&*interned, "42");
    /// ```
    pub fn intern_formatted_str<K: Display + ?Sized>(
        &self,
        value: &K,
        upper_size: usize,
    ) -> Result<Interned<'_, str>, fmt::Error> {
        self.with_inner(|set| set.intern_formatted_str(value, upper_size))
    }

    /// Retrieves a reference to a value in the interner based on the provided key.
    ///
    /// This method checks if a value corresponding to the given key exists in the interner. If it
    /// exists, a reference to the interned value is returned. Otherwise, `None` is returned.
    ///
    /// # Eaxmples
    ///
    /// ```
    /// use any_intern::DroplessInterner;
    ///
    /// let interner = DroplessInterner::new();
    ///
    /// // Interning strings
    /// let hello = interner.intern("hello");
    ///
    /// assert_eq!(interner.get("hello").as_deref(), Some("hello"));
    /// assert!(interner.get("world").is_none());
    /// ```
    pub fn get<K: Dropless + ?Sized>(&self, value: &K) -> Option<Interned<'_, K>> {
        self.with_inner(|set| set.get(value))
    }

    /// Removes all items in the interner.
    ///
    /// Although the interner support interior mutability, clear method requires mutable access
    /// to the interner to invalidate all [`Interned`]s referencing the interner.
    pub fn clear(&mut self) {
        self.with_inner(|set| set.clear())
    }

    fn with_inner<'this, F, R>(&'this self, f: F) -> R
    where
        F: FnOnce(&'this mut DroplessInternSet<S>) -> R,
        R: 'this,
    {
        // Safety: Mutex unlocking is paired with the locking.
        unsafe {
            let set = self.inner.lock().as_mut();
            let ret = f(set);
            self.inner.unlock();
            ret
        }
    }
}

impl<S: Default> Default for DroplessInterner<S> {
    fn default() -> Self {
        // Safety: Only one instance
        let inner = unsafe { UnsafeLock::new(DroplessInternSet::default()) };
        Self { inner }
    }
}

/// A dropless interner.
///
/// Interning on this type always copies the given value into internal buffer.
#[derive(Debug, Default)]
pub struct DroplessInternSet<S = fxhash::FxBuildHasher> {
    bump: Bump,
    str_buf: StringBuffer,
    set: HashTable<DynInternEntry<S>>,
    hash_builder: S,
}

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

impl<S: BuildHasher> DroplessInternSet<S> {
    pub fn with_hasher(hash_builder: S) -> Self {
        Self {
            bump: Bump::new(),
            str_buf: StringBuffer::default(),
            set: HashTable::new(),
            hash_builder,
        }
    }

    pub fn intern<K: Dropless + ?Sized>(&mut self, value: &K) -> Interned<'_, K> {
        let src = value.as_byte_ptr();
        let layout = value.layout();

        unsafe {
            // Allocation for zero sized data should be avoided even if Bump allocator allows it.
            // Plus, we change its address to make it have fixed address for equality test.
            // But this would change nothing actually.
            if layout.size() == 0 {
                // Safety: Making a pointer to ZST and acessing through it is safe.
                let mut ptr = value as *const K;
                let addr_part = &mut ptr as *mut *const K as *mut *const () as *mut usize;
                *addr_part = layout.align(); // Like ptr::dangling()
                return Interned::unique(&*ptr);
            }

            let bytes = slice::from_raw_parts(src.as_ptr(), layout.size());
            let hash = <K as Dropless>::hash(&self.hash_builder, bytes);
            let eq = Self::table_eq::<K>(bytes, layout);
            let hasher = Self::table_hasher(&self.hash_builder);
            let ref_ = match self.set.entry(hash, eq, hasher) {
                Entry::Occupied(entry) => Self::ref_from_entry(entry.get()),
                Entry::Vacant(entry) => {
                    // New allocation & copy
                    let dst = self.bump.alloc_layout(layout);
                    ptr::copy_nonoverlapping(src.as_ptr(), dst.as_ptr(), layout.size());

                    // New set entry
                    let occupied = entry.insert(DynInternEntry {
                        data: RawInterned(dst).cast(),
                        layout,
                        hash: <K as Dropless>::hash,
                    });

                    Self::ref_from_entry(occupied.get())
                }
            };
            Interned::unique(ref_)
        }
    }

    pub fn intern_formatted_str<K: Display + ?Sized>(
        &mut self,
        value: &K,
        upper_size: usize,
    ) -> Result<Interned<'_, str>, fmt::Error> {
        let mut write_buf = self.str_buf.speculative_alloc(upper_size);
        write!(write_buf, "{value}")?;
        let bytes = write_buf.as_bytes();

        unsafe {
            let layout = Layout::from_size_align_unchecked(bytes.len(), 1);

            // We change address to make it have fixed address for equality test.
            // But this would change nothing actually.
            if bytes.is_empty() {
                // Safety: Making a pointer to ZST and acessing through it is safe.
                let dangling_ptr = NonNull::<u8>::dangling().as_ptr().cast_const();
                let empty_slice = slice::from_raw_parts(dangling_ptr, 0);
                let empty_str = std::str::from_utf8_unchecked(empty_slice);
                return Ok(Interned::unique(empty_str));
            }

            let hash = <str as Dropless>::hash(&self.hash_builder, bytes);
            let eq = Self::table_eq::<str>(bytes, layout);
            let hasher = Self::table_hasher(&self.hash_builder);
            let ref_ = match self.set.entry(hash, eq, hasher) {
                Entry::Occupied(entry) => {
                    // Discards the change to the string buffer by just dropping the `write_buf`
                    // because we have the same value in the bump alloator.
                    // drop(write_buf);

                    Self::ref_from_entry(entry.get())
                }
                Entry::Vacant(entry) => {
                    // We keep the change to the string buffer because we're going to return the
                    // reference to the string buffer.
                    let bytes = write_buf.commit();

                    // Safety: Zero size was filtered out above.
                    let ptr = NonNull::new_unchecked(bytes.as_ptr().cast_mut());

                    // New set entry
                    let occupied = entry.insert(DynInternEntry {
                        data: RawInterned(ptr).cast(),
                        layout,
                        hash: <str as Dropless>::hash,
                    });

                    Self::ref_from_entry(occupied.get())
                }
            };
            Ok(Interned::unique(ref_))
        }
    }

    /// # Eaxmples
    ///
    /// ```
    /// use any_intern::DroplessInternSet;
    /// use std::hash::RandomState;
    ///
    /// let mut set = DroplessInternSet::with_hasher(RandomState::new());
    /// set.intern("hello");
    ///
    /// assert_eq!(set.get("hello").as_deref(), Some("hello"));
    /// assert!(set.get("hi").is_none());
    /// ```
    pub fn get<K: Dropless + ?Sized>(&self, value: &K) -> Option<Interned<'_, K>> {
        let ptr = value.as_byte_ptr();
        let layout = value.layout();
        unsafe {
            let bytes = slice::from_raw_parts(ptr.as_ptr(), layout.size());
            let hash = <K as Dropless>::hash(&self.hash_builder, bytes);
            let eq = Self::table_eq::<K>(bytes, layout);
            let entry = self.set.find(hash, eq)?;
            let ref_ = Self::ref_from_entry(entry);
            Some(Interned::unique(ref_))
        }
    }

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

    pub fn is_empty(&self) -> bool {
        self.len() == 0
    }

    pub fn clear(&mut self) {
        self.bump.reset();
        self.set.clear();
    }

    /// Returns `eq` closure that is used for some methods on the [`HashTable`].
    #[inline]
    fn table_eq<'a, K: Dropless + ?Sized>(
        key: &'a [u8],
        layout: Layout,
    ) -> impl FnMut(&DynInternEntry<S>) -> bool + 'a {
        move |entry: &DynInternEntry<S>| unsafe {
            if layout == entry.layout {
                let entry_bytes =
                    slice::from_raw_parts(entry.data.cast::<u8>().as_ptr(), entry.layout.size());
                <K as Dropless>::eq(key, entry_bytes)
            } else {
                false
            }
        }
    }

    /// Returns `hasher` closure that is used for some methods on the [`HashTable`].
    #[inline]
    fn table_hasher<'a>(hash_builder: &'a S) -> impl Fn(&DynInternEntry<S>) -> u64 + 'a {
        |entry: &DynInternEntry<S>| unsafe {
            let entry_bytes =
                slice::from_raw_parts(entry.data.cast::<u8>().as_ptr(), entry.layout.size());
            // Safety: `entry_bytes` is aligned for the entry hash function.
            (entry.hash)(hash_builder, entry_bytes)
        }
    }

    #[inline]
    unsafe fn ref_from_entry<'a, K: Dropless + ?Sized>(entry: &DynInternEntry<S>) -> &'a K {
        unsafe {
            let bytes =
                slice::from_raw_parts(entry.data.cast::<u8>().as_ptr(), entry.layout.size());
            K::from_bytes(bytes)
        }
    }
}

/// A buffer for strings.
#[derive(Debug, Default)]
struct StringBuffer {
    chunks: Vec<Vec<MaybeUninit<u8>>>,
    last_chunk_start: usize,
}

const INIT_CHUNK_SIZE: usize = 1 << 5;
const GROW_MAX_CHUNK_SIZE: usize = 1 << 12;

impl StringBuffer {
    fn speculative_alloc(&mut self, upper_size: usize) -> StringWriteBuffer<'_> {
        match self.chunks.last() {
            None => {
                let chunk_size = INIT_CHUNK_SIZE.max(upper_size.next_power_of_two());
                self.append_new_chunk(chunk_size)
            }
            Some(last_chunk) if last_chunk.len() - self.last_chunk_start < upper_size => {
                let chunk_size = (last_chunk.len() * 2)
                    .min(GROW_MAX_CHUNK_SIZE)
                    .max(upper_size.next_power_of_two());
                self.append_new_chunk(chunk_size);
            }
            _ => {}
        }

        // Safety: We added the last chunk above.
        let buf = unsafe {
            let last_chunk = self.chunks.last_mut().unwrap_unchecked();
            last_chunk.as_mut_ptr().add(self.last_chunk_start)
        };
        StringWriteBuffer {
            buf,
            buf_cap: upper_size,
            last_chuck_start: &mut self.last_chunk_start,
            written: 0,
        }
    }

    #[inline]
    fn append_new_chunk(&mut self, chunk_size: usize) {
        let mut chunk: Vec<MaybeUninit<u8>> = Vec::with_capacity(chunk_size);

        // Safety: We reserved enough amount of memory, also it can contain uninitialized data.
        unsafe { chunk.set_len(chunk_size) };

        self.chunks.push(chunk);
        self.last_chunk_start = 0;
    }
}

struct StringWriteBuffer<'a> {
    buf: *mut MaybeUninit<u8>,
    buf_cap: usize,
    last_chuck_start: &'a mut usize,
    written: usize,
}

impl<'a> StringWriteBuffer<'a> {
    #[inline]
    fn as_bytes(&self) -> &[u8] {
        // Safety: bytes 0..written were initialized by write_str.
        unsafe { slice::from_raw_parts(self.buf.cast::<u8>(), self.written) }
    }

    #[inline]
    fn commit(self) -> &'a [u8] {
        *self.last_chuck_start += self.written;

        // Safety: bytes 0..written were initialized by write_str.
        // The lifetime 'a is valid because buf points into a Vec owned by the StringBuffer that is
        // mutably borrowed for 'a (evidenced by last_chuck_start: &'a mut usize).
        unsafe { slice::from_raw_parts(self.buf.cast::<u8>(), self.written) }
    }
}

impl Write for StringWriteBuffer<'_> {
    fn write_str(&mut self, s: &str) -> std::fmt::Result {
        let size = s.len();

        if self.buf_cap - self.written >= size {
            let src = s.as_ptr();

            // Safety
            // * `src` is valid for reading of `size` bytes
            // * `dst` is valid for writing of `size` bytes
            //   (`written + size <= buf_cap <= chunk len`)
            // * Both alignments are 1
            // * `src` and `dst` are not overlapping (src is a string literal / caller data)
            let dst = unsafe { self.buf.add(self.written).cast::<u8>() };
            unsafe { ptr::copy_nonoverlapping(src, dst, size) };

            self.written += size;
            Ok(())
        } else {
            Err(std::fmt::Error)
        }
    }
}

struct DynInternEntry<S> {
    data: RawInterned,
    layout: Layout,
    /// Input bytes must be well aligned.
    hash: unsafe fn(&S, &[u8]) -> u64,
}

impl<S> Debug for DynInternEntry<S> {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        f.debug_struct("DynInternEntry")
            .field("data", &self.data)
            .field("layout", &self.layout)
            .finish_non_exhaustive()
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::common;
    use std::num::NonZeroU32;

    #[test]
    fn test_dropless_interner() {
        test_dropless_interner_int();
        test_dropless_interner_str();
        test_dropless_interner_bytes();
        test_dropless_interner_mixed();
        test_dropless_interner_many();
        test_dropless_interner_alignment_handling();
        test_dropless_interner_complex_display_type();
        test_dropless_interner_consecutive_formatted_strs();
    }

    fn test_dropless_interner_int() {
        let interner = DroplessInterner::new();

        let a = interner.intern(&0_u32).erased_raw();
        let b = interner.intern(&0_u32).erased_raw();
        let c = interner.intern(&1_u32).erased_raw();
        // d has the same bytes and layout as c's, so memory will be shared.
        let d = interner.intern(&1_i32).erased_raw();

        let groups: [&[RawInterned]; _] = [&[a, b], &[c, d]];
        common::assert_group_addr_eq(&groups);
    }

    fn test_dropless_interner_str() {
        let interner = DroplessInterner::new();

        // "apple"
        let a = interner.intern("apple").erased_raw();
        let b = interner.intern(*Box::new("apple")).erased_raw();
        let c = interner.intern(&*String::from("apple")).erased_raw();

        // "banana"
        let d = interner.intern("banana").erased_raw();

        // ""
        let e = interner.intern("").erased_raw();
        let f = interner.intern(&*String::from("")).erased_raw();

        // "42"
        let g = interner.intern("42").erased_raw();
        let h = interner.intern_formatted_str(&42, 2).unwrap().erased_raw();

        // "43"
        let i = interner
            .intern_formatted_str(&NonZeroU32::new(43).unwrap(), 2)
            .unwrap()
            .erased_raw();
        let j = interner.intern(&*43.to_string()).erased_raw();

        let groups: [&[RawInterned]; _] = [&[a, b, c], &[d], &[e, f], &[g, h], &[i, j]];
        common::assert_group_addr_eq(&groups);
    }

    fn test_dropless_interner_bytes() {
        let interner = DroplessInterner::new();

        let a = interner.intern(&[0, 1]).erased_raw();
        let boxed: Box<[i32]> = Box::new([0, 1]);
        let b = interner.intern(&*boxed).erased_raw();
        let c = interner.intern(&*vec![0, 1]).erased_raw();
        let d = interner.intern(&[2, 3]).erased_raw();

        let groups: [&[RawInterned]; _] = [&[a, b, c], &[d]];
        common::assert_group_addr_eq(&groups);
    }

    fn test_dropless_interner_mixed() {
        let interner = DroplessInterner::new();

        interner.intern("apple");
        interner.intern(&1_u32);
        interner.intern(&[2, 3]);
        interner.intern_formatted_str(&42, 10).unwrap();

        assert_eq!(interner.get("apple").as_deref(), Some("apple"));
        assert!(interner.get("banana").is_none());

        assert_eq!(interner.get(&1_u32).as_deref(), Some(&1_u32));
        assert!(interner.get(&2_u32).is_none());

        assert_eq!(interner.get(&[2, 3]).as_deref(), Some(&[2, 3]));
        assert!(interner.get(&[2]).is_none());

        assert_eq!(interner.get("42").as_deref(), Some("42"));
    }

    fn test_dropless_interner_many() {
        let interner = DroplessInterner::new();

        let mut interned_int = Vec::new();
        let mut interned_str = Vec::new();
        let mut interned_bytes = Vec::new();

        #[cfg(not(miri))]
        const N: usize = 1000;
        #[cfg(miri)]
        const N: usize = 50;

        let strs = (0..N).map(|i| (i * 10_000).to_string()).collect::<Vec<_>>();

        // Interns lots of items.
        for (i, str_) in strs.iter().enumerate().take(N) {
            let int = &(i as u32); // 4 bytes
            let str_ = str_.as_str(); // greater than 4 btyes
            let bytes = &[i as u16]; // 2 bytes
            interned_int.push(interner.intern(int).erased_raw());
            interned_str.push(interner.intern(str_).erased_raw());
            interned_bytes.push(interner.intern(bytes).erased_raw());
        }

        // Verifies every pointer is unique.
        interned_int.sort_unstable();
        interned_int.dedup();
        interned_str.sort_unstable();
        interned_str.dedup();
        interned_bytes.sort_unstable();
        interned_bytes.dedup();
        assert_eq!(interned_int.len(), N);
        assert_eq!(interned_str.len(), N);
        assert_eq!(interned_bytes.len(), N);
        let whole = interned_int
            .iter()
            .chain(&interned_str)
            .chain(&interned_bytes)
            .cloned()
            .collect::<fxhash::FxHashSet<_>>();
        assert_eq!(whole.len(), N * 3);

        // Verifies `get` method for every item.
        for (i, str_) in strs.iter().enumerate().take(N) {
            let int = &(i as u32); // 4 bytes
            let str_ = str_.as_str(); // greater than 4 btyes
            let bytes = &[i as u16]; // 2 bytes
            assert_eq!(interner.get(int).as_deref(), Some(int));
            assert_eq!(interner.get(str_).as_deref(), Some(str_));
            assert_eq!(interner.get(bytes).as_deref(), Some(bytes));
        }
    }

    #[rustfmt::skip]
    fn test_dropless_interner_alignment_handling() {
        #[derive(PartialEq, Eq, Hash, Debug)] #[repr(C, align(4))]  struct T4  (u16, [u8; 2]);
        #[derive(PartialEq, Eq, Hash, Debug)] #[repr(C, align(8))]  struct T8  (u16, [u8; 6]);
        #[derive(PartialEq, Eq, Hash, Debug)] #[repr(C, align(16))] struct T16 (u16, [u8; 14]);

        macro_rules! impl_for_first_2bytes {
            () => {
                unsafe fn hash<S: BuildHasher>(hash_builder: &S, bytes: &[u8]) -> u64 {
                    hash_builder.hash_one(&bytes[0..2])
                }
                unsafe fn eq(a: &[u8], b: &[u8]) -> bool {
                    a[0..2] == b[0..2]
                }
            };
        }

        impl Dropless for T4 { simple_impl_dropless_fn!(from_bytes); impl_for_first_2bytes!(); }
        impl Dropless for T8 { simple_impl_dropless_fn!(from_bytes); impl_for_first_2bytes!(); }
        impl Dropless for T16 { simple_impl_dropless_fn!(from_bytes); impl_for_first_2bytes!(); }

        let interner = DroplessInterner::new();

        let mut interned_4 = Vec::new();
        let mut interned_8 = Vec::new();
        let mut interned_16 = Vec::new();

        #[cfg(not(miri))]
        const N: usize = 1000;
        #[cfg(miri)]
        const N: usize = 50;

        // Items being interned have the same first 2 bytes, so they will look like the smae from
        // perspective of interner. But, they must be distinguished from each other due to their
        // different layouts. Following bytes after the 2 bytes will be used to detect data
        // validity.
        for i in 0..N {
            let t4 = T4(i as u16, [4; _]);
            let t8 = T8(i as u16, [8; _]);
            let t16 = T16(i as u16, [16; _]);
            interned_4.push(interner.intern(&t4).erased_raw());
            interned_8.push(interner.intern(&t8).erased_raw());
            interned_16.push(interner.intern(&t16).erased_raw());
        }

        for i in 0..N {
            let t4 = T4(i as u16, [4; _]);
            let t8 = T8(i as u16, [8; _]);
            let t16 = T16(i as u16, [16; _]);
            unsafe {
                assert_eq!(*<Interned<'_, T4>>::from_erased_raw(interned_4[i]), t4);
                assert_eq!(*<Interned<'_, T8>>::from_erased_raw(interned_8[i]), t8);
                assert_eq!(*<Interned<'_, T16>>::from_erased_raw(interned_16[i]), t16);
            }
        }
    }

    fn test_dropless_interner_complex_display_type() {
        let interner = DroplessInterner::new();

        #[allow(unused)]
        #[derive(Debug)]
        struct A<'a> {
            int: i32,
            float: f32,
            text: &'a str,
            bytes: &'a [u8],
        }

        impl Display for A<'_> {
            fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
                fmt::Debug::fmt(self, f)
            }
        }

        let a = A {
            int: 123,
            float: 456.789,
            text: "this is a text",
            bytes: &[1, 2, 3, 4, 5, 6, 7, 8, 9, 0],
        };

        let interned = interner.intern_formatted_str(&a, 1_000).unwrap();

        let mut s = String::new();
        write!(&mut s, "{a}").unwrap();
        assert_eq!(&*interned, s.as_str());
    }

    // UB if the interner gets a mutable reference to a whole chunk while a reference(i0) is alive.
    fn test_dropless_interner_consecutive_formatted_strs() {
        let dropless = DroplessInterner::new();

        let value = 0;
        let i1 = dropless.intern_formatted_str(&value, 1).unwrap();

        let value = 1;
        let i2 = dropless.intern_formatted_str(&value, 1).unwrap();

        // Prevents i1 and i2 from being optimized away.
        let _c = format!("{i1}{i2}");
    }

    #[test]
    fn test_string_buffer() {
        test_string_buffer_chunk();
        test_string_buffer_discard();
        test_string_buffer_insufficient_upper_size();
        test_string_buffer_long_string();
    }

    fn test_string_buffer_chunk() {
        let mut buf = StringBuffer::default();
        assert_eq!(buf.chunks.len(), 0);

        // Fills the first chunk.
        let s = "a".repeat(INIT_CHUNK_SIZE);
        let mut write_buf = buf.speculative_alloc(s.len());
        write_buf.write_str(&s).unwrap();
        assert_eq!(write_buf.as_bytes(), s.as_bytes());
        write_buf.commit();

        assert_eq!(buf.chunks.len(), 1);
        assert_eq!(buf.last_chunk_start, s.len());

        // Makes another chunk.
        let mut write_buf = buf.speculative_alloc(1);
        write_buf.write_str("a").unwrap();
        assert_eq!(write_buf.as_bytes(), b"a");
        write_buf.commit();

        assert_eq!(buf.chunks.len(), 2);
        assert_eq!(buf.last_chunk_start, 1);

        // Forces to make another chunk
        let mut write_buf = buf.speculative_alloc(GROW_MAX_CHUNK_SIZE);
        write_buf.write_str("aa").unwrap();
        assert_eq!(write_buf.as_bytes(), b"aa");
        write_buf.commit();

        assert_eq!(buf.chunks.len(), 3);
        assert_eq!(buf.last_chunk_start, 2);
    }

    fn test_string_buffer_discard() {
        let mut buf = StringBuffer::default();

        // Write & discard makes no change.
        for _ in 0..10 {
            let s = "a".repeat(INIT_CHUNK_SIZE);
            let mut write_buf = buf.speculative_alloc(s.len());
            write_buf.write_str(&s).unwrap();
            assert_eq!(write_buf.as_bytes(), s.as_bytes());
            assert_eq!(buf.chunks.len(), 1);
            assert_eq!(buf.last_chunk_start, 0);
        }
    }

    fn test_string_buffer_insufficient_upper_size() {
        let mut buf = StringBuffer::default();

        for _ in 0..10 {
            let mut write_buf = buf.speculative_alloc(5);
            let res = write_buf.write_str("this is longer than 5");
            assert!(res.is_err());
            write_buf.commit();

            assert_eq!(buf.chunks.len(), 1);
            assert_eq!(buf.last_chunk_start, 0);
        }
    }

    fn test_string_buffer_long_string() {
        let mut buf = StringBuffer::default();

        let s = "a".repeat(GROW_MAX_CHUNK_SIZE * 10);
        let mut write_buf = buf.speculative_alloc(s.len());
        write_buf.write_str(&s).unwrap();
        assert_eq!(write_buf.as_bytes(), s.as_bytes());
        write_buf.commit();

        assert_eq!(buf.chunks.len(), 1);
        assert_eq!(buf.last_chunk_start, s.len());
    }
}