#include "specter/Translator.hpp"
#include "logvisor/logvisor.hpp"

namespace specter
{
static logvisor::Module Log("specter::Translator");

Locale::Locale(std::string_view name, std::string_view fullName,
               const unsigned char* yamlSource, size_t yamlLength)
: m_name(name), m_fullName(fullName)
{
    athena::io::YAMLDocReader reader;
    yaml_parser_set_input_string(reader.getParser(), yamlSource, yamlLength);
    reader.parse(nullptr);
    m_rootNode = reader.releaseRootNode();
    if (m_rootNode)
    {
        m_langNode = m_rootNode->findMapChild(name.data());
        if (!m_langNode)
            Log.report(logvisor::Fatal, "no root node '%s' found in locale", name.data());
    }
    else
        Log.report(logvisor::Warning, "locale empty");
}

void Translator::setLocale(const Locale* targetLocale)
{
    if (!targetLocale)
        Log.report(logvisor::Fatal, "null locale");
    m_targetLocale = targetLocale;
}

static std::string_view RecursiveLookup(const athena::io::YAMLNode* node,
                                          std::string_view::const_iterator start,
                                          std::string_view::const_iterator end)
{
    for (std::string_view::const_iterator it = start ; it != end ; ++it)
    {
        if (*it == '/')
        {
            const athena::io::YAMLNode* ch = node->findMapChild(std::string(start, it));
            if (!ch)
                return nullptr;
            return RecursiveLookup(ch, it+1, end);
        }
    }
    const athena::io::YAMLNode* ch = node->findMapChild(std::string(start, end));
    if (!ch)
        return {};
    return ch->m_scalarString;
}

std::string_view Translator::translate(std::string_view key) const
{
    if (!m_targetLocale->rootNode())
        return nullptr;

    return RecursiveLookup(m_targetLocale->rootNode(), key.cbegin(), key.cend());
}

std::string_view Translator::translateOr(std::string_view key, std::string_view vor) const
{
    std::string_view find = translate(key);
    if (!find.empty())
        return find;
    return vor;
}

}