#include "provider_registry.h"
#include <stdexcept>

ProviderRegistry::ProviderRegistry(const ShuangpinProfile &shuangpin_profile, metasequoia::RuntimePaths paths)
    : pinyin_provider_(shuangpin_profile, paths),
      wubi_provider_(metasequoia::path_to_utf8(paths.dictionary(metasequoia::assets::main_dictionary)), paths),
      japanese_provider_(metasequoia::path_to_utf8(paths.dictionary(metasequoia::assets::main_dictionary)),
                         metasequoia::path_to_utf8(paths.resource(metasequoia::assets::japanese_model)))
{
}

ICandidateProvider &ProviderRegistry::resolve(SchemeType scheme_type)
{
    switch (scheme_type)
    {
    case SchemeType::Quanpin:
    case SchemeType::Shuangpin:
        return pinyin_provider_;
    case SchemeType::Wubi:
        return wubi_provider_;
    case SchemeType::JapaneseRomaji:
        return japanese_provider_;
    default:
        throw std::runtime_error("Unknown scheme type.");
    }
}

void ProviderRegistry::reset_cache(SchemeType scheme_type)
{
    switch (scheme_type)
    {
    case SchemeType::Quanpin:
    case SchemeType::Shuangpin:
        pinyin_provider_.reset_cache();
        return;
    case SchemeType::Wubi:
        wubi_provider_.reset_cache();
        return;
    case SchemeType::JapaneseRomaji:
        japanese_provider_.reset_cache();
        return;
    default:
        throw std::runtime_error("Unknown scheme type.");
    }
}

int ProviderRegistry::create_word(SchemeType scheme_type, std::string pinyin, std::string word)
{
    return resolve(scheme_type).create_word(scheme_type, std::move(pinyin), std::move(word));
}

int ProviderRegistry::update_weight_by_pinyin_and_word(SchemeType scheme_type, std::string pinyin, std::string word)
{
    return resolve(scheme_type).update_weight_by_pinyin_and_word(scheme_type, std::move(pinyin), std::move(word));
}

int ProviderRegistry::delete_by_pinyin_and_word(SchemeType scheme_type, std::string pinyin, std::string word)
{
    return resolve(scheme_type).delete_by_pinyin_and_word(scheme_type, std::move(pinyin), std::move(word));
}

int ProviderRegistry::cache_dynamic_candidate(SchemeType scheme_type, const std::string &pinyin,
                                              const std::string &word, CandidateSource source)
{
    return resolve(scheme_type).cache_dynamic_candidate(scheme_type, pinyin, word, source);
}

std::optional<WordItem> ProviderRegistry::find_candidate(SchemeType scheme_type, const std::string &key,
                                                         const std::string &value)
{
    return resolve(scheme_type).find_candidate(scheme_type, key, value);
}

bool ProviderRegistry::can_expand_initial_candidates(const QueryRequest &request,
                                                     const std::vector<WordItem> &candidates) const
{
    return pinyin_provider_.can_expand_initial_candidates(request, candidates);
}

bool ProviderRegistry::expand_initial_candidates(const QueryRequest &request, std::vector<WordItem> &candidates)
{
    return pinyin_provider_.expand_initial_candidates(request, candidates);
}

int ProviderRegistry::cache_dynamic_candidate_for_request(const QueryRequest &request, const std::string &word,
                                                          CandidateSource source)
{
    return resolve(request.scheme).cache_dynamic_candidate_for_request(request, word, source);
}