// MassBuilderSaveTool
// Copyright (C) 2021-2022 Guillaume Jacquemin
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <https://www.gnu.org/licenses/>.

#include <Corrade/Containers/Pair.h>
#include <Corrade/Containers/ScopeGuard.h>
#include <Corrade/Utility/Path.h>
#include <Corrade/Utility/Unicode.h>

#include <shlobj.h>

#include "../FontAwesome/IconsFontAwesome5.h"
#include "../FontAwesome/IconsFontAwesome5Brands.h"

#include "SaveTool.h"

void SaveTool::initEvent(SDL_Event& event) {
    _initThread.join();

    switch(event.user.code) {
        case InitSuccess:
            _uiState = UiState::ProfileManager;
            ImGui::CloseCurrentPopup();
            break;
        case ProfileManagerFailure:
            Utility::Error{} << "Error initialising ProfileManager:" << _profileManager->lastError();
            SDL_ShowSimpleMessageBox(SDL_MESSAGEBOX_ERROR, "Error initialising ProfileManager", _profileManager->lastError().data(), window());
            exit(EXIT_FAILURE);
            break;
        default:
            break;
    }
}

void SaveTool::initialiseConfiguration() {
    Utility::Debug{} << "Reading configuration file...";

    if(_conf.hasValue("cheat_mode"_s)) {
        _cheatMode = _conf.value<bool>("cheat_mode"_s);
    }
    else {
        _conf.setValue("cheat_mode"_s, _cheatMode);
    }

    if(_conf.hasValue("startup_update_check"_s)) {
        _checkUpdatesOnStartup = _conf.value<bool>("startup_update_check"_s);
    }
    else {
        _conf.setValue("startup_update_check"_s, _checkUpdatesOnStartup);
    }

    if(_conf.hasValue("skip_disclaimer"_s)) {
        _skipDisclaimer = _conf.value<bool>("skip_disclaimer"_s);
    }
    else {
        _conf.setValue("skip_disclaimer"_s, _skipDisclaimer);
    }

    if(_conf.hasValue("frame_limit"_s)) {
        std::string frame_limit = _conf.value("frame_limit"_s);
        if(frame_limit == "vsync"_s) {
            _framelimit = Framelimit::Vsync;
        }
        else if(frame_limit == "half_vsync"_s) {
            _framelimit = Framelimit::HalfVsync;
        }
        else {
            _framelimit = Framelimit::FpsCap;
            _fpsCap = std::stoul(frame_limit);
        }
    }
    else {
        _conf.setValue("frame_limit"_s, "vsync"_s);
    }

    _conf.save();
}

void SaveTool::initialiseGui() {
    Utility::Debug{} << "Initialising ImGui...";

    ImGui::CreateContext();

    ImGuiIO& io = ImGui::GetIO();

    auto reg_font = _rs.getRaw("SourceSansPro-Regular.ttf"_s);
    ImFontConfig font_config;
    font_config.FontDataOwnedByAtlas = false;
    std::strcpy(font_config.Name, "Source Sans Pro");
    io.Fonts->AddFontFromMemoryTTF(const_cast<char*>(reg_font.data()), reg_font.size(), 20.0f, &font_config);

    auto icon_font = _rs.getRaw(FONT_ICON_FILE_NAME_FAS);
    static const ImWchar icon_range[] = { ICON_MIN_FA, ICON_MAX_FA, 0 };
    ImFontConfig icon_config;
    icon_config.FontDataOwnedByAtlas = false;
    icon_config.MergeMode = true;
    icon_config.PixelSnapH = true;
    icon_config.OversampleH = icon_config.OversampleV = 1;
    icon_config.GlyphMinAdvanceX = 18.0f;
    io.Fonts->AddFontFromMemoryTTF(const_cast<char*>(icon_font.data()), icon_font.size(), 16.0f, &icon_config, icon_range);

    auto brand_font = _rs.getRaw(FONT_ICON_FILE_NAME_FAB);
    static const ImWchar brand_range[] = { ICON_MIN_FAB, ICON_MAX_FAB, 0 };
    io.Fonts->AddFontFromMemoryTTF(const_cast<char*>(brand_font.data()), brand_font.size(), 16.0f, &icon_config, brand_range);

    auto mono_font = _rs.getRaw("SourceCodePro-Regular.ttf"_s);
    ImVector<ImWchar> range;
    ImFontGlyphRangesBuilder builder;
    builder.AddRanges(io.Fonts->GetGlyphRangesDefault());
    builder.AddChar(u'š'); // This allows displaying Vladimír Vondruš' name in Corrade's and Magnum's licences.
    builder.BuildRanges(&range);
    io.Fonts->AddFontFromMemoryTTF(const_cast<char*>(mono_font.data()), mono_font.size(), 18.0f, &font_config, range.Data);

    _imgui = ImGuiIntegration::Context(*ImGui::GetCurrentContext(), windowSize());

    io.IniFilename = nullptr;

    ImGuiStyle& style = ImGui::GetStyle();

    style.WindowTitleAlign = {0.5f, 0.5f};
    style.FrameRounding = 3.2f;
    style.Colors[ImGuiCol_WindowBg] = ImColor(0xff1f1f1f);
}

void SaveTool::initialiseManager() {
    SDL_Event event;
    SDL_zero(event);
    event.type = _initEventId;

    _profileManager.emplace(_saveDir, _backupsDir);
    if(!_profileManager->ready()) {
        event.user.code = ProfileManagerFailure;
        SDL_PushEvent(&event);
        return;
    }

    event.user.code = InitSuccess;
    SDL_PushEvent(&event);
}

auto SaveTool::initialiseToolDirectories() -> bool {
    Utility::Debug{} << "Initialising Save Tool directories...";

    _backupsDir = Utility::Path::join(Utility::Path::split(*Utility::Path::executableLocation()).first(), "backups");
    _stagingDir = Utility::Path::join(Utility::Path::split(*Utility::Path::executableLocation()).first(), "staging");
    //_armouryDir = Utility::Directory::join(Utility::Directory::path(Utility::Directory::executableLocation()), "armoury");
    //_armoursDir = Utility::Directory::join(_armouryDir, "armours");
    //_weaponsDir = Utility::Directory::join(_armouryDir, "weapons");
    //_stylesDir  = Utility::Directory::join(_armouryDir, "styles");

    if(!Utility::Path::exists(_backupsDir)) {
        Utility::Debug{} << "Backups directory not found, creating...";
        if(!Utility::Path::make(_backupsDir)) {
            Utility::Error{} << (_lastError = "Couldn't create the backups directory.");
            return false;
        }
    }

    if(!Utility::Path::exists(_stagingDir)) {
        Utility::Debug{} << "Staging directory not found, creating...";
        if(!Utility::Path::make(_stagingDir)) {
            Utility::Error{} << (_lastError = "Couldn't create the backups directory.");
            return false;
        }
    }

    //if(!Utility::Directory::exists(_armouryDir)) {
    //    Utility::Debug{} << "Armoury directory not found, creating...";
    //    if(!Utility::Path::make(_armouryDir)) {
    //        Utility::Error{} << (_lastError = "Couldn't create the armoury directory.");
    //        return false;
    //    }
    //}

    //if(!Utility::Directory::exists(_armoursDir)) {
    //    Utility::Debug{} << "Armours directory not found, creating...";
    //    if(!Utility::Path::make(_armoursDir)) {
    //        Utility::Error{} << (_lastError = "Couldn't create the armours directory.");
    //        return false;
    //    }
    //}

    //if(!Utility::Directory::exists(_weaponsDir)) {
    //    Utility::Debug{} << "Weapons directory not found, creating...";
    //    if(!Utility::Path::make(_weaponsDir)) {
    //        Utility::Error{} << (_lastError = "Couldn't create the weapons directory.");
    //        return false;
    //    }
    //}

    //if(!Utility::Directory::exists(_stylesDir)) {
    //    Utility::Debug{} << "Styles directory not found, creating...";
    //    if(!Utility::Path::make(_stylesDir)) {
    //        Utility::Error{} << (_lastError = "Couldn't create the styles directory.");
    //        return false;
    //    }
    //}

    return true;
}

auto SaveTool::findGameDataDirectory() -> bool {
    Utility::Debug{} << "Searching for the game's save directory...";

    wchar_t* localappdata_path = nullptr;
    Containers::ScopeGuard guard{localappdata_path, CoTaskMemFree};
    if(SHGetKnownFolderPath(FOLDERID_LocalAppData, KF_FLAG_NO_APPCONTAINER_REDIRECTION, nullptr, &localappdata_path) != S_OK)
    {
        Utility::Error{} << (_lastError = "SHGetKnownFolderPath() failed in SaveTool::findGameDataDirectory()"_s);
        return false;
    }

    _gameDataDir = Utility::Path::join(Utility::Path::fromNativeSeparators(Utility::Unicode::narrow(localappdata_path)), "MASS_Builder"_s);

    if(!Utility::Path::exists(_gameDataDir)) {
        Utility::Error{} << (_lastError = _gameDataDir + " wasn't found. Make sure to play the game at least once."_s);
        return false;
    }

    _configDir = Utility::Path::join(_gameDataDir, "Saved/Config/WindowsNoEditor"_s);
    _saveDir = Utility::Path::join(_gameDataDir, "Saved/SaveGames"_s);
    _screenshotsDir = Utility::Path::join(_gameDataDir, "Saved/Screenshots/WindowsNoEditor"_s);

    return true;
}

void SaveTool::initialiseMassManager() {
    _massManager.emplace(_saveDir, _currentProfile->account(), _currentProfile->isDemo(), _stagingDir);
}

void SaveTool::initialiseFileWatcher() {
    _fileWatcher.emplace();
    _watchIDs[SaveDir] = _fileWatcher->addWatch(_saveDir, this, false);
    _watchIDs[StagingDir] = _fileWatcher->addWatch(_stagingDir, this, false);
    _fileWatcher->watch();
}