#include "SceTypes.hpp"
#include "prx/libc/include/general/VabiMacros.hpp"
#include <cstdint>
#include <cstdlib>
#include <initializer_list>
#include <stdexcept>

extern "C" {
int APS5_VABI sceConvertKeycodeGetVirtualKeycode(std::uint16_t, std::uint32_t, std::uint16_t*);
int APS5_VABI sceConvertKeycodeGetImeKeyboardType(std::int32_t, std::uint32_t*);
int APS5_VABI sceConvertKeycodeGetCharacter(std::uint16_t, std::uint32_t, std::uint32_t, char16_t*);
}

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

int main() {
    constexpr int invalidAddress = static_cast<int>(0x80BC0031u);
    constexpr int invalidUserId = static_cast<int>(0x80BC0010u);

    Require(sceConvertKeycodeGetImeKeyboardType(0, nullptr) == invalidAddress);
    Require(sceConvertKeycodeGetImeKeyboardType(-1, nullptr) == invalidAddress);

    std::uint16_t virtualKey = 0xABCD;
    Require(sceConvertKeycodeGetVirtualKeycode(4, 0, nullptr) == invalidAddress);
    Require(sceConvertKeycodeGetVirtualKeycode(4, 0, &virtualKey) == static_cast<int>(0x80BC001Du));
    Require(virtualKey == 0xABCD);
    Require(sceConvertKeycodeGetVirtualKeycode(4, 40, &virtualKey) == static_cast<int>(0x80BC001Du));
    Require(sceConvertKeycodeGetVirtualKeycode(4, 4, &virtualKey) == 0 && virtualKey == 65);
    Require(sceConvertKeycodeGetVirtualKeycode(4, 9, &virtualKey) == 0 && virtualKey == 81);
    Require(sceConvertKeycodeGetVirtualKeycode(100, 4, &virtualKey) == 0 && virtualKey == 255);
    Require(sceConvertKeycodeGetVirtualKeycode(136, 24, &virtualKey) == 0 && virtualKey == 21);
    Require(sceConvertKeycodeGetVirtualKeycode(65535, 4, &virtualKey) == 0 && virtualKey == 255);
    for (std::uint32_t layout : {2u, 3u, 12u, 31u, 39u}) {
        virtualKey = 0xABCD;
        bool rejected = false;
        try {
            sceConvertKeycodeGetVirtualKeycode(28, layout, &virtualKey);
        } catch (const std::runtime_error&) {
            rejected = true;
        }
        Require(rejected && virtualKey == 0xABCD);
    }

    std::uint32_t type = 0xFFFFFFFFu;
    Require(sceConvertKeycodeGetImeKeyboardType(-1, &type) == invalidUserId);
    Require(type == 0xFFFFFFFFu);

    Require(sceConvertKeycodeGetImeKeyboardType(0, &type) == 0);
    Require(type == 0);

    type = 0xFFFFFFFFu;
    Require(sceConvertKeycodeGetImeKeyboardType(1, &type) == 0);
    Require(type == 0);

    Require(sceConvertKeycodeGetCharacter(0x04, 0, 0, nullptr) == invalidAddress);
    for (std::uint32_t state : {0u, 0x4000000u}) {
        char16_t character[3] = {u'?', 0, 0};
        Require(sceConvertKeycodeGetCharacter(0x04, state, 0, character) == 0);
        Require(character[0] == u'a');
        Require(sceConvertKeycodeGetCharacter(0x1E, state, 0, character) == 0);
        Require(character[0] == u'1');
        Require(sceConvertKeycodeGetCharacter(0x3A, state, 0, character) == 0);
        Require(character[0] == 0);
        Require(sceConvertKeycodeGetCharacter(0x64, state, 0, character) == 0);
        Require(character[0] == u'\\');
        Require(sceConvertKeycodeGetCharacter(0x32, state, 0, character) == 0);
        Require(character[0] == 0);
        Require(sceConvertKeycodeGetCharacter(0x27, state, 0, character) == 0);
        Require(character[0] == u'0');
        Require(sceConvertKeycodeGetCharacter(0x2C, state, 0, character) == 0);
        Require(character[0] == u' ');
        Require(sceConvertKeycodeGetCharacter(0x2D, state, 0, character) == 0);
        Require(character[0] == u'-');
        Require(sceConvertKeycodeGetCharacter(0x54, state, 0, character) == 0);
        Require(character[0] == u'/');
        Require(sceConvertKeycodeGetCharacter(0x59, state, 0, character) == 0);
        Require(character[0] == u'1');
        Require(sceConvertKeycodeGetCharacter(0x63, state, 0, character) == 0);
        Require(character[0] == u'.');
    }
    return 0;
}