#include "prx/libSceSharePlay/libSceSharePlay.hpp"
#include "SceTypes.hpp"
#include "prx/libc/include/general/VabiMacros.hpp"
#include <cstddef>
#include <cstdlib>
#include <cstring>
#include <exception>
#include <span>

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

template <typename TCall>
static bool Throws(TCall call) {
    try {
        call();
    } catch (const std::exception&) {
        return true;
    }
    return false;
}

int main() {
    alignas(16) static unsigned char heap[0x1800];
    constexpr int notInitialized = static_cast<int>(0x810E0004);
    constexpr int alreadyInitialized = static_cast<int>(0x810E0003);
    constexpr int invalidArgument = static_cast<int>(0x810E0001);
    Require(sceSharePlayTerminate() == notInitialized);
    Require(sceSharePlayGetCurrentConnectionInfoA(nullptr) == notInitialized);
    Require(sceSharePlayGetCurrentConnectionInfo(nullptr) == notInitialized);
    Require(sceSharePlayProhibitController(2) == notInitialized);
    Require(sceSharePlayInitialize(nullptr, sizeof(heap)) == 0);
    Require(sceSharePlayInitialize(heap, 0) == alreadyInitialized);
    Require(sceSharePlayTerminate() == 0);
    Require(sceSharePlayInitialize(heap, sizeof(heap)) == 0);
    Require(sceSharePlayTerminate() == 0);
    for (const std::size_t size : {std::size_t{0}, std::size_t{1}, sizeof(heap) - 1}) Require(sceSharePlayInitialize(heap, size) == invalidArgument);
    Require(sceSharePlayInitialize(nullptr, 0) == 0);
    SharePlayConnectionInfoA info;
    std::memset(&info, 0x5a, sizeof(info));
    Require(sceSharePlayGetCurrentConnectionInfoA(&info) == 0);
    Require(info.status == 0 && info.mode == 0 && info.hostAccountId == 0 && info.visitorAccountId == 0);
    Require(info.hostOnlineId.data[0] == 0 && info.visitorOnlineId.data[0] == 0 && info.hostOnlineId.pad[2] == 0);
    Require(info.hostUserId == -1 && info.visitorUserId == -1);
    Require(sceSharePlayGetCurrentConnectionInfoA(nullptr) == static_cast<int>(0x810E0001));
    unsigned char plain[sizeof(SharePlayConnectionInfo) + 8];
    std::memset(plain, 0x5a, sizeof(plain));
    auto* plainInfo = reinterpret_cast<SharePlayConnectionInfo*>(plain);
    Require(sceSharePlayGetCurrentConnectionInfo(plainInfo) == 0);
    Require(plainInfo->status == 0 && plainInfo->mode == 0 && plainInfo->hostOnlineId.data[0] == 0 && plainInfo->visitorOnlineId.pad[2] == 0);
    Require(plainInfo->hostUserId == -1 && plainInfo->visitorUserId == -1);
    for (std::size_t i = sizeof(SharePlayConnectionInfo); i < sizeof(plain); ++i) Require(plain[i] == 0x5a);
    Require(sceSharePlayGetCurrentConnectionInfo(nullptr) == invalidArgument);
    Require(sceSharePlayProhibitController(1) == 0);
    Require(sceSharePlayProhibitController(1) == 0);
    Require(sceSharePlayProhibitController(0) == 0);
    Require(sceSharePlayProhibitController(2) == invalidArgument);
    Require(sceSharePlayProhibitController(255) == invalidArgument);
    Require(sceSharePlayTerminate() == 0);
    std::memset(&info, 0x5a, sizeof(info));
    Require(sceSharePlayGetCurrentConnectionInfoA(&info) == notInitialized);
    for (const auto value : std::as_bytes(std::span(&info, 1))) Require(value == std::byte{0x5a});
    Require(sceSharePlayTerminate() == notInitialized);
}