#include "common/fs_file.h"

#include <filesystem>

namespace Common::FS {

File::File() = default;

File::File(const std::string& path, OpenMode mode) { open(path, mode); }

File::~File() { close(); }

bool File::open(const std::string& path, OpenMode mode) {
    close();
#ifdef _WIN64
    fopen_s(&m_file, path.c_str(), getOpenMode(mode));
#else
    m_file = std::fopen(path.c_str(), getOpenMode(mode));
#endif
    return isOpen();
}

bool File::close() {
    if (!isOpen() || std::fclose(m_file) != 0) [[unlikely]] {
        m_file = nullptr;
        return false;
    }

    m_file = nullptr;
    return true;
}

bool File::write(std::span<const u08> data) { return isOpen() && std::fwrite(data.data(), 1, data.size(), m_file) == data.size(); }

bool File::read(void* data, u64 size) const { return isOpen() && std::fread(data, 1, size, m_file) == size; }

bool File::seek(s64 offset, SeekMode mode) {
#ifdef _WIN64
    if (!isOpen() || _fseeki64(m_file, offset, getSeekMode(mode)) != 0) {
        return false;
    }
#else
    if (!isOpen() || fseeko64(m_file, offset, getSeekMode(mode)) != 0) {
        return false;
    }
#endif
    return true;
}

u64 File::tell() const {
    if (isOpen()) [[likely]] {
#ifdef _WIN64
        return _ftelli64(m_file);
#else
        return ftello64(m_file);
#endif
    }

    return -1;
}

std::vector<DirEntry> File::getDirectoryEntries(const std::string& path) {
    std::string curpath = path;
    if (!curpath.ends_with("/")) {
        curpath = std::string(curpath + "/");
    }
    std::vector<DirEntry> files;

    for (const auto& entry : std::filesystem::directory_iterator(curpath)) {
        DirEntry e = {};
        if (std::filesystem::is_regular_file(entry.path().string())) {
            e.name = entry.path().filename().string();
            e.isFile = true;
        } else {
            DirEntry e = {};
            e.name = entry.path().filename().string() + "/";  // hmmm not sure if it has to be like this...
            e.isFile = false;
        }
        files.push_back(e);
    }

    return files;
}

u64 File::getFileSize() {
#ifdef _WIN64
    const u64 pos = _ftelli64(m_file);
    if (_fseeki64(m_file, 0, SEEK_END) != 0) {
        return 0;
    }

    const u64 size = _ftelli64(m_file);
    if (_fseeki64(m_file, pos, SEEK_SET) != 0) {
        return 0;
    }
#else
    const u64 pos = ftello64(m_file);
    if (fseeko64(m_file, 0, SEEK_END) != 0) {
        return 0;
    }

    const u64 size = ftello64(m_file);
    if (fseeko64(m_file, pos, SEEK_SET) != 0) {
        return 0;
    }
#endif
    return size;
}

}  // namespace Common::FS