/*****************************************************************************
 * 
 *  Copyright (c) 2020 by SonicMastr <sonicmastr@gmail.com>
 * 
 *  This file is part of Pigs In A Blanket
 * 
 *  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 <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <psp2/types.h>
#include <psp2/kernel/modulemgr.h>
#include <psp2/shacccg.h>
#include <psp2/io/fcntl.h>
#include <psp2/kernel/clib.h>
#include "../include/hooks.h"
#include "../include/debug.h"
#include "../include/sha1.h"
#include <taihen.h>

static SceUID modID[4];
static SceBool pibIsInit = SCE_FALSE;

static PibError loadModules(PibOptions options)
{
    if (!(options & PIB_NOSTDLIB)) {
        if (modID[3] = sceKernelLoadStartModule("vs0:sys/external/libfios2.suprx", 0, NULL, 0, NULL, 0), modID[3] < 0 && modID[3] != 0x8002D014 && modID[3] != 0x8002D013)
            return PIB_ERROR_FIOS2_FAILED;
        if (modID[2] = sceKernelLoadStartModule("vs0:sys/external/libc.suprx", 0, NULL, 0, NULL, 0), modID[2] < 0 && modID[2] != 0x8002D014 && modID[2] != 0x8002D013)
            return PIB_ERROR_LIBC_FAILED;
    }
    if (modID[1] = sceKernelLoadStartModule("ur0:data/external/libScePiglet.suprx", 0, NULL, 0, NULL, 0), modID[1] < 0)
        return PIB_ERROR_PIGLET_FAILED;
    if (options & PIB_SHACCCG) {
        if (modID[0] = sceKernelLoadStartModule("ur0:data/external/libshacccg.suprx", 0, NULL, 0, NULL, 0), modID[0] < 0)
            return PIB_ERROR_SHACCCG_FAILED;
        sceShaccCgSetDefaultAllocator(malloc, free);
    }
    return PIB_SUCCESS;
}

static void unloadModules(void)
{
    for (int i = 0; i < 4; i++) {
        if (modID[i])
            sceKernelStopUnloadModule(modID[i], 0, NULL, 0, NULL, 0);
    }
}

static void getResolutionConfig()
{
    SceUID fd;
	fd = sceIoOpen("ur0:data/external/resolution.bin", SCE_O_RDONLY, 0666);
	if(fd < 0){
        LOG("Error: File Open Error, 0x%08X.\nNo custom resolution set\n", fd);
        customResolutionMode = 0;
		return;
	}
	sceIoRead(fd, &customResolutionMode, sizeof(int));
	sceIoClose(fd);
}

PibError pibInit(PibOptions options)
{
    if(pibIsInit) {
        LOG("PIB is already Initialized!\n");
        return PIB_ERROR_ALREADY_INIT; // PIB already Initialized
    }
    PibError ret = loadModules(options);
    if (ret) return ret;

    if (!(options & PIB_SYSTEM_MODE))
        getResolutionConfig();
    else
        customResolutionMode = 7; // Custom System Mode Resolution Configuration 
   
    loadHooks(options);
    pibIsInit = SCE_TRUE;
    return PIB_SUCCESS;
}

PibError pibTerm(void)
{
    if(!pibIsInit) {
        LOG("PIB is not Initialized!\n");
        return PIB_ERROR_NOT_INIT; // PIB isn't Initialized
    }
    unloadModules();
    releaseHooks();
    pibIsInit = SCE_FALSE;
    return PIB_SUCCESS;	
}