mm/2s2h/Enhancements/DifficultyOptions/DeleteFileOnDeath.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55

#include <libultraship/bridge/consolevariablebridge.h>
#include <ship/window/gui/ConsoleWindow.h>
#include <ship/Context.h>
#include <ship/window/Window.h>
#include "2s2h/GameInteractor/GameInteractor.h"
#include "2s2h/ShipInit.hpp"

extern "C" {
#include "variables.h"
}

#define CVAR_NAME "gEnhancements.DifficultyOptions.DeleteFileOnDeath"
#define CVAR CVarGetInteger(CVAR_NAME, 0)

void SaveManager_DeleteSaveFile(const std::filesystem::path& fileName);
std::string SaveManager_GetFileName(int fileNum, bool isBackup);

void RegisterDeleteFileOnDeath() {
    COND_HOOK(OnGameStateUpdate, CVAR, []() {
        if (!gPlayState) {
            return;
        }

        static bool fileDeleted = false;

        if (gPlayState->gameOverCtx.state == GAMEOVER_DEATH_WAIT_GROUND && !fileDeleted) {
            fileDeleted = true;
            if (gSaveContext.fileNum >= 0 && gSaveContext.fileNum <= 2) {
                std::string fileName = SaveManager_GetFileName(gSaveContext.fileNum + 1, false);
                std::string backupFileName = SaveManager_GetFileName(gSaveContext.fileNum + 1, true);
                SaveManager_DeleteSaveFile(fileName);
                SaveManager_DeleteSaveFile(backupFileName);
            }
        }

        static int resetTimer = 0;

        if (gPlayState->gameOverCtx.state == 4) { // After GAMEOVER_DEATH_FADE_OUT, no enum
            resetTimer++;

            if (resetTimer >= 20) { // Completely black
                std::reinterpret_pointer_cast<Ship::ConsoleWindow>(
                    Ship::Context::GetRawInstance()->GetWindow()->GetGui()->GetGuiWindow("Console"))
                    ->Dispatch("reset");
            }
        }

        if (gPlayState->gameOverCtx.state == GAMEOVER_INACTIVE) {
            fileDeleted = false;
            resetTimer = 0;
        }
    });
}

static RegisterShipInitFunc initFunc(RegisterDeleteFileOnDeath, { CVAR_NAME });