#include "shared_string.hh"
#include "debug.hh"

namespace Kakoune
{

SharedString StringRegistry::intern(StringView str)
{
    auto it = m_strings.find({str, SharedString::NoCopy{}});
    if (it == m_strings.end())
    {
        SharedString shared_str = str;
        it = m_strings.emplace(shared_str).first;
    }
    return *it;
}

void StringRegistry::purge_unused()
{
    for (auto it = m_strings.begin(); it != m_strings.end(); )
    {
        if (it->m_storage->refcount == 1)
            it = m_strings.erase(it);
        else
            ++it;
    }
}

void StringRegistry::debug_stats() const
{
    write_debug("Shared Strings stats:");
    size_t total_refcount = 0;
    size_t total_size = 0;
    size_t count = m_strings.size();
    for (auto& st : m_strings)
    {
        total_refcount += st.m_storage->refcount - 1;
        total_size += (int)st.m_storage->length;
    }
    write_debug(format("  data size: {}, mean: {}", total_size, (float)total_size/count));
    write_debug(format("  refcounts: {}, mean: {}", total_refcount, (float)total_refcount/count));
}

}