#include "prx/libScePlayerSelectionDialog/include/Api.hpp"
#include "prx/libc/include/general/VabiMacros.hpp"
#include <algorithm>
#include <cstdint>
#include <cstdlib>
#include <cstring>
#include "prx/libSceCommonDialog/CommonDialog.hpp"
#include "prx/libSceUserService/UserService.hpp"

namespace {

void Require(bool value) { if (!value) std::abort(); }

constexpr int kStatusFinished = 3;
constexpr int kErrNotInitialized = static_cast<int>(0x80B80003);
constexpr int kErrAlreadyInitialized = static_cast<int>(0x80B80004);
constexpr int kErrNotFinished = static_cast<int>(0x80B80005);
constexpr int kErrArgNull = static_cast<int>(0x80B8000D);
constexpr int kResultUserCanceled = 1;

}

int main() {
    Require(sceUserServiceInitialize(nullptr) == 0);
    struct GuardedParam {
        PlayerSelectionDialogParam param;
        std::uint8_t guard[8];
    };
    GuardedParam guarded;
    std::memset(&guarded, 0xA5, sizeof(guarded));
    auto& param = guarded.param;
    PlayerSelectionDialogResult result{-1, 99, nullptr, 77, 55};

    Require(scePlayerSelectionDialogTerminate() == kErrNotInitialized);
    Require(scePlayerSelectionDialogOpen(nullptr) == kErrArgNull);
    Require(scePlayerSelectionDialogParamInitialize(nullptr) == kErrArgNull);
    Require(scePlayerSelectionDialogParamInitialize(&param) == 0);
    Require(param.base.size == sizeof(param.base));
    Require(param.base.magic == static_cast<std::uint32_t>(reinterpret_cast<std::uintptr_t>(&param)) - 1060003575u);
    Require(param.size == sizeof(param));
    Require(std::all_of(std::begin(param.base.reserved), std::end(param.base.reserved), [](std::uint8_t value) { return value == 0; }));
    Require(param.userId == 0);
    Require(std::all_of(std::begin(param.title), std::end(param.title), [](char value) { return value == 0; }));
    Require(param.maxPlayers == 0);
    Require(param.mode == 0);
    Require(param.padding0 == 0);
    Require(param.players == nullptr);
    Require(param.playerCount == 0);
    Require(std::all_of(std::begin(param.reserved), std::end(param.reserved), [](std::uint8_t value) { return value == 0; }));
    Require(param.padding1 == 0);
    Require(std::all_of(std::begin(guarded.guard), std::end(guarded.guard), [](std::uint8_t value) { return value == 0xA5; }));
    param.userId = USER_SERVICE_INITIAL_USER_ID;
    Require(scePlayerSelectionDialogOpen(&param) == kErrNotInitialized);
    Require(scePlayerSelectionDialogClose() == kErrNotInitialized);
    Require(scePlayerSelectionDialogGetResult(nullptr) == kErrArgNull);
    Require(scePlayerSelectionDialogGetResult(&result) == kErrNotInitialized);
    Require(scePlayerSelectionDialogGetStatus() == 0);
    Require(scePlayerSelectionDialogUpdateStatus() == 0);

    Require(scePlayerSelectionDialogInitialize() == 0);
    Require(scePlayerSelectionDialogInitialize() == kErrAlreadyInitialized);
    Require(scePlayerSelectionDialogGetStatus() == 1);
    Require(scePlayerSelectionDialogUpdateStatus() == 1);

    Require(scePlayerSelectionDialogOpen(nullptr) == kErrArgNull);
    Require(scePlayerSelectionDialogGetResult(&result) == kErrNotFinished);
    Require(scePlayerSelectionDialogOpen(&param) == 0);
    Require(scePlayerSelectionDialogGetStatus() == kStatusFinished);

    param.size = 0;
    Require(scePlayerSelectionDialogOpen(&param) == static_cast<int>(0x80B8000Au));
    param.size = sizeof(param);
    param.maxPlayers = 257;
    Require(scePlayerSelectionDialogOpen(&param) == static_cast<int>(0x80B8000Au));
    param.maxPlayers = 1;
    param.mode = 4;
    Require(scePlayerSelectionDialogOpen(&param) == static_cast<int>(0x80B8000Au));
    param.mode = 0;
    param.playerCount = 1;
    Require(scePlayerSelectionDialogOpen(&param) == static_cast<int>(0x80B8000Au));
    PlayerSelectionDialogPlayer player{};
    param.players = &player;
    Require(scePlayerSelectionDialogOpen(&param) == static_cast<int>(0x80B8000Au));
    player.accountId = 1;
    player.option0 = 1;
    Require(scePlayerSelectionDialogOpen(&param) == static_cast<int>(0x80B8000Au));
    param.maxPlayers = 2;
    player.option1 = 2;
    Require(scePlayerSelectionDialogOpen(&param) == static_cast<int>(0x80B8000Au));
    player.option1 = 1;
    player.mode = 3;
    Require(scePlayerSelectionDialogOpen(&param) == static_cast<int>(0x80B8000Au));
    player.mode = 2;
    player.reserved[63] = 1;
    Require(scePlayerSelectionDialogOpen(&param) == static_cast<int>(0x80B8000Au));
    player.reserved[63] = 0;
    param.reserved[63] = 1;
    Require(scePlayerSelectionDialogOpen(&param) == static_cast<int>(0x80B8000Au));
    param.reserved[63] = 0;
    Require(scePlayerSelectionDialogOpen(&param) == 0);
    Require(scePlayerSelectionDialogGetStatus() == kStatusFinished);
    Require(scePlayerSelectionDialogUpdateStatus() == kStatusFinished);

    Require(scePlayerSelectionDialogGetResult(nullptr) == kErrArgNull);
    Require(scePlayerSelectionDialogGetResult(&result) == kResultUserCanceled);
    Require(result.result == kResultUserCanceled);
    Require(result.option == 99 && result.accountIds == nullptr && result.count == 77 && result.padding == 55);

    Require(scePlayerSelectionDialogClose() == static_cast<int>(0x80B8000Bu));
    Require(scePlayerSelectionDialogTerminate() == 0);
    Require(scePlayerSelectionDialogTerminate() == kErrNotInitialized);
    Require(scePlayerSelectionDialogOpen(nullptr) == kErrArgNull);
    Require(scePlayerSelectionDialogOpen(&param) == kErrNotInitialized);
    Require(scePlayerSelectionDialogClose() == kErrNotInitialized);
    Require(scePlayerSelectionDialogGetResult(nullptr) == kErrArgNull);
    Require(scePlayerSelectionDialogGetResult(&result) == kErrNotInitialized);
    Require(scePlayerSelectionDialogGetStatus() == 0);
    Require(scePlayerSelectionDialogUpdateStatus() == 0);
    Require(sceUserServiceTerminate() == 0);
}