#include <switch_min.h>

#include <dirent.h>
#include <stdio.h>
#include <string.h>
#include <switch_min/kernel/ipc.h>
#include <sys/iosupport.h>
#include <sys/reent.h>

#include "useful/useful.h"

#include "saltysd/saltysd_core.h"
#include "saltysd/saltysd_dynamic.h"
#include "saltysd/saltysd_ipc.h"
#include "saltysd/saltysd_helper.h"

#include "hitbox_visualizer.hpp"
#include "training_mods.hpp"

extern "C" {
extern u32 __start__;

static char g_heap[0x8000];

void __libnx_init(void* ctx, Handle main_thread, void* saved_lr);
void __attribute__((weak)) NORETURN __libnx_exit(int rc);
void __nx_exit(int, void*);
void __libc_fini_array(void);
void __libc_init_array(void);
}

u32 __nx_applet_type = AppletType_None;

Handle orig_main_thread;
void* orig_ctx;
void* orig_saved_lr;

void __libnx_init(void* ctx, Handle main_thread, void* saved_lr) {
    extern char* fake_heap_start;
    extern char* fake_heap_end;

    fake_heap_start = &g_heap[0];
    fake_heap_end = &g_heap[sizeof g_heap];

    orig_ctx = ctx;
    orig_main_thread = main_thread;
    orig_saved_lr = saved_lr;

    // Call constructors.
    // void __libc_init_array(void);
    __libc_init_array();
}

void __attribute__((weak)) NORETURN __libnx_exit(int rc) {
    // Call destructors.
    // void __libc_fini_array(void);
    __libc_fini_array();

    SaltySD_printf("SaltySD Plugin: jumping to %p\n", orig_saved_lr);

    __nx_exit(0, orig_saved_lr);
    while (true)
        ;
}

extern int sprintf(char* s, const char* format, ...) LINKABLE;

int sprintf_intercept(char* s, const char* format, ...) {
    if (strcmp(format, "Ver. %d.%d.%d") == 0 || strcmp(format, "Salty v%d%d%d") == 0) {
        va_list vl;
        va_start(vl, format);
        major = va_arg(vl, int);
        minor = va_arg(vl, int);
        patch = va_arg(vl, int);
        va_end(vl);
        return vsprintf(s, "Training Mods v1.5", vl);
    }
    va_list vl;
    va_start(vl,format);

    int ret = vsprintf(s, format, vl);
    va_end(vl);

    return ret;
}

int main(int argc, char* argv[]) {
    SaltySD_printf("SaltySD Plugin: alive\n");

    // Get anchor for imports
    // do not remove if you plan on using IMPORT
    ANCHOR_ABS = SaltySDCore_getCodeStart();

    /*
            Example of string replacement:
            replaces the title screen's version number with the string
            below.
    */

    const char* ver = "Ver. %d.%d.%d";
    u64 version_string = SaltySDCore_findCode((u8*)ver, strlen(ver));
    if (version_string) {
        SaltySD_Memcpy(version_string, (u64) "Salty v%d%d%d", 13);
    }

    SaltySDCore_ReplaceImport("sprintf", (void*)sprintf_intercept);

    // Add function replacements here
    hitbox_vis_main();
    training_mods_main();
    FILE* f = SaltySDCore_fopen("sdmc:/SaltySD/training_modpack.log", "w");
	if (f) {
		SaltySD_printf("Writing training_modpack.log...\n");
		char buffer[20];
		snprintf(buffer, 20, "%lx", (u64)&menu);
		SaltySDCore_fwrite(buffer, strlen(buffer), 1, f);
		SaltySDCore_fclose(f);
	}

    __libnx_exit(0);
}