#pragma once

#include "common/common.h"
#include "utils/pugixml.hpp"
#include <coreinit/memorymap.h>
#include <optional>

class InstallerService {
public:
    enum eResults {
        SUCCESS                                 = 0,
        NO_COMPATIBLE_APP_INSTALLED             = -1,
        FAILED_TO_COPY_FILES                    = -2,
        FAILED_TO_CHECK_HASH_COPIED_FILES       = -3,
        SYSTEM_XML_INFORMATION_NOT_FOUND        = -4,
        SYSTEM_XML_PARSING_FAILED               = -5,
        SYSTEM_XML_HASH_MISMATCH_RESTORE_FAILED = -6,
        SYSTEM_XML_HASH_MISMATCH                = -7,
        RPX_HASH_MISMATCH                       = -8,
        RPX_HASH_MISMATCH_RESTORE_FAILED        = -9,
        COS_XML_PARSING_FAILED                  = -10,
        COS_XML_HASH_MISMATCH                   = -11,
        COS_XML_HASH_MISMATCH_RESTORE_FAILED    = -12,
        MALLOC_FAILED                           = -13,
        FST_HASH_MISMATCH                       = -14,
        FST_HASH_MISMATCH_RESTORE_FAILED        = -15,
        FST_HEADER_MISMATCH                     = -16,
        FST_NO_USABLE_SECTION_FOUND             = -17,
        FAILED_TO_LOAD_FILE                     = -18,
    };

    static bool isColdBootAllowed() {
        /* if (OSIsAddressValid(0x00FFFFF8)) { */
        uint64_t bootedFrom = *((uint64_t *) 0x00FFFFF8);
        if (
                bootedFrom == 0x000500101004E000L || // H&S JPN
                bootedFrom == 0x000500101004E100L || // H&S USA
                bootedFrom == 0x000500101004E200L    // H&S EUR
        ) {
            return true;
        }
        // }
        return false;
    }

    static bool isBackupAvailable(const std::string &path);

    static eResults restoreAppFiles(const std::string &path);

    static eResults backupAppFiles(const std::string &path);

    static eResults patchCOS(const std::string &path, char *hash);

    static eResults checkCOS(const std::string &path, char *hash);

    static eResults checkSystemXML(const std::string &path, uint64_t titleId);

    static eResults checkFST(const std::string &path, const char *fstHash);

    static std::optional<appInformation> getInstalledAppInformation();

    static std::optional<uint64_t> getSystemMenuTitleId();

    static std::string ErrorMessage(eResults error);

    static std::string ErrorDescription(eResults error);

    static eResults patchFST(const std::string &path, const char *hash);

    static eResults copyRPX(const std::string &path, const uint8_t *rpx_data, size_t size, const std::string &rpx_hash);

    static eResults patchSystemXML(const std::string &path, uint64_t id);

    static uint64_t getColdbootTitleId(const std::string &path);

    static eResults checkFSTAlreadyValid(const std::string &path, const std::string &hash);

    static eResults checkTMDValid(const std::string &path, const std::string &hash, const std::string &tmdWithCertHash);

    static eResults checkCOSAlreadyValid(const std::string &path, const std::string &hash);

    static eResults checkRPXAlreadyValid(const std::string &path, const std::string &hash);

    static eResults setBootTitle(uint64_t titleId);

private:
    static eResults patchFSTData(uint8_t *fstData, uint32_t size);

    static bool patchCOSXMLData(pugi::xml_document *doc);

    static eResults checkFileHash(const std::string &filePath, const std::string &hash);
};