#include <cstddef>
#include <cstdint>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <memory>
#include <mutex>
#include <set>
#include <stdexcept>
#include <string>

#include "NpUniversalDataSystem.hpp"
#include "SceTypes.hpp"
#include "prx/libSceNpTrophy2/include/TrophyStore.hpp"
#include "prx/libc/include/General.hpp"

namespace {

Uds::Owned<NpUniversalDataSystemEvent>& events() {
    static Uds::Owned<NpUniversalDataSystemEvent> registry;
    return registry;
}

bool firstProgress(std::int32_t trophyId) {
    static std::mutex mutex;
    static std::set<std::int32_t> logged;
    std::lock_guard lock(mutex);
    return logged.insert(trophyId).second;
}

void ignore(const NpUniversalDataSystemEvent& event, int context, const std::string& reason) {
    std::fprintf(stderr, "[trophy] ignored %s posted to context %d: %s\n", Uds::Describe(event).c_str(), context, reason.c_str());
}

void post(const NpUniversalDataSystemEvent& event, int context) {
    static const bool trace = std::getenv("APS5_TRACE_UDS") != nullptr;
    if (trace) std::fprintf(stderr, "[uds] context %d posted %s\n", context, Uds::Describe(event).c_str());
    const auto trophy = Uds::RecognizeTrophyEvent(event);
    switch (trophy.kind) {
    case Uds::TrophyEventKind::None:
        if (const auto owner = Uds::FindContext(context)) NpTrophy2ApplyEvent_nid_no_patch(owner->userId, owner->serviceLabel, Uds::StatEvent(event));
        return;
    case Uds::TrophyEventKind::Malformed:
        ignore(event, context, trophy.problem);
        return;
    case Uds::TrophyEventKind::Progress:
        if (firstProgress(trophy.trophyId)) ignore(event, context, "trophy progress is not recorded");
        return;
    case Uds::TrophyEventKind::Unlock:
        if (const auto owner = Uds::FindContext(context)) NpTrophy2RecordUnlock_nid_no_patch(owner->userId, owner->serviceLabel, trophy.trophyId);
        else ignore(event, context, "the context does not exist");
        return;
    }
}

}

extern "C" {

int APS5_VABI sceNpUniversalDataSystemCreateEvent(
    const char* event_name,
    const NpUniversalDataSystemEventPropertyObject* prop,
    NpUniversalDataSystemEvent** new_event,
    NpUniversalDataSystemEventPropertyObject** prop_ptr)
{
    if (event_name == nullptr || new_event == nullptr) {
        APS5_INVALID_ARG_EX;
    }
    auto event = std::make_unique<NpUniversalDataSystemEvent>();
    event->name = event_name;
    if (prop != nullptr) event->properties = prop->Clone();
    *new_event = events().Adopt(std::move(event));
    if (prop_ptr != nullptr) {
        *prop_ptr = &(*new_event)->properties;
    }
    return SCE_NP_UNIVERSAL_DATA_SYSTEM_OK;
}

int APS5_VABI sceNpUniversalDataSystemDestroyEvent(NpUniversalDataSystemEvent* event) {
    events().Release(event);
    return SCE_NP_UNIVERSAL_DATA_SYSTEM_OK;
}

int APS5_VABI sceNpUniversalDataSystemPostEvent(int context, int handle, const void* event, uint64_t options) {
    if (event == nullptr) {
        APS5_INVALID_ARG_EX;
    }
    const auto* posted = static_cast<const NpUniversalDataSystemEvent*>(event);
    if (!events().Contains(posted)) {
        throw std::invalid_argument(std::string(__func__) + ": the event was not created by sceNpUniversalDataSystemCreateEvent or was destroyed");
    }
    post(*posted, context);
    return SCE_NP_UNIVERSAL_DATA_SYSTEM_OK;
}

int APS5_VABI sceNpUniversalDataSystemEventEstimateSize(const NpUniversalDataSystemEvent* event, size_t* size) {
    if (event == nullptr || size == nullptr) {
        APS5_INVALID_ARG_EX;
    }
    *size = NP_UNIVERSAL_DATA_SYSTEM_EMPTY_EVENT_SIZE;
    return SCE_NP_UNIVERSAL_DATA_SYSTEM_OK;
}

int APS5_VABI sceNpUniversalDataSystemEventToString(
    const NpUniversalDataSystemEvent* event,
    char* buf,
    size_t buf_size,
    size_t* string_size)
{
    if (event == nullptr) {
        APS5_INVALID_ARG_EX;
    }
    const size_t json_len = std::strlen(NP_UNIVERSAL_DATA_SYSTEM_EMPTY_JSON) + 1;
    if (string_size != nullptr) {
        *string_size = json_len;
    }
    if (buf != nullptr && buf_size > 0) {
        std::snprintf(buf, buf_size, "%s", NP_UNIVERSAL_DATA_SYSTEM_EMPTY_JSON);
    }
    return SCE_NP_UNIVERSAL_DATA_SYSTEM_OK;
}

}