#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "fs.h"
#include "system.h"
#include "utils.h"

const char *System_GetModel(void)
{
	const char *models[] = 
	{
		"OLD 3DS - CTR",
		"OLD 3DS XL - SPR",
		"NEW 3DS - KTR",
		"OLD 2DS - FTR",
		"NEW 3DS XL - RED",
		"NEW 2DS XL - JAN",
		"Unknown"
	};

	u8 model = 0;
   
	if (R_SUCCEEDED(CFGU_GetSystemModel(&model)))
	{
		if (model < 6)
			return models[model];
	}
	
	return models[6];
}

const char *System_GetRegion(void)
{
	const char *regions[] = 
	{
		"JPN",
		"USA",
		"EUR",
		"AUS",
		"CHN",
		"KOR",
		"TWN",
		"Unknown"
	};

	CFG_Region region = 0;
	
	if (R_SUCCEEDED(CFGU_SecureInfoGetRegion(&region)))
	{
		if (region < 7)
			return regions[region];
	}
	
	return regions[7];
}

const char System_GetFirmRegion(void)
{
	if (strncmp(System_GetRegion(), "JPN", 3) == 0)
		return 'J';
	else if (strncmp(System_GetRegion(), "USA", 3) == 0)
		return 'U';
	else if ((strncmp(System_GetRegion(), "EUR", 3) == 0) || (strncmp(System_GetRegion(), "AUS", 3) == 0))
		return 'E';
	else if (strncmp(System_GetRegion(), "CHN", 3) == 0)
		return 'C';
	else if (strncmp(System_GetRegion(), "KOR", 3) == 0)
		return 'K';
	else if (strncmp(System_GetRegion(), "TWN", 3) == 0)
		return 'T';
	
	return 0;
}

bool System_IsCoppacsSupported()
{
	u8 IsCoppacs = 0;
	
	if (R_SUCCEEDED(CFGU_GetRegionCanadaUSA(&IsCoppacs)))
	{
		if (IsCoppacs)
			return true;
	}
	
	return false;
}

const char *System_GetLang(void)
{
	const char *languages[] = 
	{
		"Japanese",
		"English",
		"French",
		"German",
		"Italian",
		"Spanish",
		"Simplified Chinese",
		"Korean",
		"Dutch",
		"Portugese",
		"Russian",
		"Traditional Chinese",
		"Unknown"
	};

	CFG_Language language;
	
	if (R_SUCCEEDED(CFGU_GetSystemLanguage(&language)))
	{
		if (language < 12)
			return languages[language];
	}
	
	return languages[12];
}

char *System_GetMacAddress(void)
{
	u8 *macByte = (u8 *)WIFI_MACADDR; 
	static char macAddress[0x12];
	
	snprintf(macAddress, 0x12, "%02X:%02X:%02X:%02X:%02X:%02X", *macByte, *(macByte + 0x1), *(macByte + 0x2), *(macByte + 0x3), *(macByte + 0x4), *(macByte + 0x5));

	return macAddress;
}

const char *System_GetRunningHW(void)
{
	const char *runningHW[] = 
	{
		"Retail",
		"Devboard",
		"Debugger",
		"Capture"
	};

	return runningHW[(*(u8 *)RUNNING_HW) - 0x1];
}

char *System_IsDebugUnit(void)
{
	return *(u8 *)UNITINFO ? "" : "(Debug Unit)";
}

char *System_GetScreenType(void)
{	
	static char upperScreen[20];
	static char lowerScreen[20];
	
	static char screenType[32];
	
	if (Utils_IsN3DS())
	{
		u8 screens = 0;
		
		if(R_SUCCEEDED(gspLcdInit()))
		{
			if (R_SUCCEEDED(GSPLCD_GetVendors(&screens)))
				gspLcdExit();
		}	
        
		switch ((screens >> 4) & 0xF)
		{
			case 0x01: // 0x01 = JDI => IPS
				sprintf(upperScreen, "Upper: IPS");
				break;
			case 0x0C: // 0x0C = SHARP => TN
				sprintf(upperScreen, "Upper: TN");
				break;
			default:
				sprintf(upperScreen, "Upper: Unknown");
				break;
		}
		switch (screens & 0xF)
		{
			case 0x01: // 0x01 = JDI => IPS
				sprintf(lowerScreen, " | Lower: IPS");
				break;
			case 0x0C: // 0x0C = SHARP => TN
				sprintf(lowerScreen, " | Lower: TN");
				break;
			default:
				sprintf(lowerScreen, " | Lower: Unknown");
				break;
		}
		
		strcpy(screenType, upperScreen);
		strcat(screenType, lowerScreen);
	}
	else
		sprintf(screenType, "Upper: TN | Lower: TN"); 
	
	return screenType;
}

u64 System_GetLocalFriendCodeSeed(void)
{
	u64 seed = 0;
	
    if (R_SUCCEEDED(CFGI_GetLocalFriendCodeSeed(&seed)))
		return seed;
	
	return 0;
}

char *System_GetNANDLocalFriendCodeSeed(void)
{
	FS_Archive nandArchive;
	Handle handle;
	u32 bytesread = 0;
	char *buf = (char *)malloc(6);
	static char out[0xB];

	FS_OpenArchive(&nandArchive, ARCHIVE_NAND_CTR_FS);

	if (FS_FileExists(nandArchive, "/rw/sys/LocalFriendCodeSeed_B"))
	{
		if (R_FAILED(FSUSER_OpenFile(&handle, nandArchive, fsMakePath(PATH_ASCII, "/rw/sys/LocalFriendCodeSeed_B"), FS_OPEN_READ, 0)))
		{
			FS_CloseArchive(nandArchive);
			return NULL;
		}
	}
	else if (FS_FileExists(nandArchive, "/rw/sys/LocalFriendCodeSeed_A"))
	{
		if (R_FAILED(FSUSER_OpenFile(&handle, nandArchive, fsMakePath(PATH_ASCII, "/rw/sys/LocalFriendCodeSeed_A"), FS_OPEN_READ, 0)))
		{
			FS_CloseArchive(nandArchive);
			return NULL;
		}
	}
	else
	{
		FS_CloseArchive(nandArchive);
		return NULL;
	}
	
	if (R_FAILED(FSFILE_Read(handle, &bytesread, 0x108, (u32 *)buf, 6)))
	{
		FS_CloseArchive(nandArchive);
		return NULL;
	}

	if (R_FAILED(FSFILE_Close(handle)))
	{
		FS_CloseArchive(nandArchive);
		return NULL;
	}

	buf[6] = '\0';
	FS_CloseArchive(nandArchive);
	snprintf(out, 0xB, "%02X%02X%02X%02X%02X", buf[0x4], buf[0x3], buf[0x2], buf[0x1], buf[0x0]);

	return out;
}

u8 *System_GetSerialNumber(void)
{
    static u8 serial[0xF];
	
    if (R_SUCCEEDED(CFGI_SecureInfoGetSerialNumber(serial)))
		return serial;
    
	return NULL;
}

u64 System_GetSoapId(void)
{
    u32 id = 0;
	
    if (R_SUCCEEDED(AM_GetDeviceId(&id)))
		return (id | (((u64) 4) << 32));
	
	return 0;
}