#include "prx/libkernel/include/Api.hpp"
#include "prx/libc/include/Api.hpp"
#include "prx/libSceLibcInternal/include/Api.hpp"
#include "prx/libc/include/GuestDirectory.hpp"
#include <chrono>
#include <cstdlib>
#include <filesystem>
#include <fstream>
#include <map>
#include <string>

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

static std::map<std::string, int> Entries(void* directory) {
    std::map<std::string, int> entries;
    while (auto* entry = readdir_nid_postfix(directory)) entries[entry->name] = entry->type;
    Require(closedir_nid_postfix(directory) == 0);
    return entries;
}

int main() {
    const auto root = std::filesystem::path("anyps5-opendir2-test-" +
        std::to_string(std::chrono::steady_clock::now().time_since_epoch().count()));
    Require(std::filesystem::create_directory(root));
    Require(std::filesystem::create_directory(root / "subdirectory"));
    { std::ofstream file(root / "sample.txt"); file << "test"; }
    const auto path = root.string();
    void* plain = opendir_nid_postfix(path.c_str());
    Require(plain != nullptr);
    const auto expected = Entries(plain);
    Require(expected.at("sample.txt") == 8 && expected.at("subdirectory") == 4);
    for (const int flags : {0x7, 0x6, 0x0, 0x1, 0x20, 0x27, 0x100, 0x40000000, static_cast<int>(0xffffffe7u)}) {
        void* directory = __opendir2_nid_postfix(path.c_str(), flags);
        Require(directory != nullptr);
        Require(Entries(directory) == expected);
    }
    for (const int flags : {0x8, 0x10, 0x18, 0xf, 0x1f, 0x7fffffff, -1}) {
        *__error_nid_postfix() = 1234;
        Require(__opendir2_nid_postfix(path.c_str(), flags) == nullptr && *__error_nid_postfix() == 1234);
        Require(__opendir2_nid_postfix(nullptr, flags) == nullptr && *__error_nid_postfix() == 1234);
    }
    const auto missing = (root / "missing").string();
    Require(opendir_nid_postfix(missing.c_str()) == nullptr);
    const int missingError = *__error_nid_postfix();
    for (const int flags : {0x7, 0x20}) {
        *__error_nid_postfix() = 0;
        Require(__opendir2_nid_postfix(missing.c_str(), flags) == nullptr && *__error_nid_postfix() == missingError);
    }
    std::filesystem::remove(root / "sample.txt");
    std::filesystem::remove(root / "subdirectory");
    std::filesystem::remove(root);
}