//============================================================================
// Name        : plugin.cpp
// Copyright   : 2012 FIX94
//============================================================================

#include <gccore.h>
#include <ogcsys.h>
#include <stdio.h>
#include <string.h>
#include "plugin.hpp"
#include "memory/mem2.hpp"
#include "gui/text.hpp"
#include "gecko/gecko.h"
#include "devicemounter/PartitionHandle.h"
#include "devicemounter/DeviceHandler.hpp"
#include "defines.h"

static const string emptyString;
static const string emptyString2("/");
static char* emptyChar = (char*)" ";

void Plugin::init(string m_pluginsDir)
{
	pluginsDir = m_pluginsDir;
	//Ready to add plugins
	adding = true;
}

void Plugin::EndAdd()
{
	adding = false;
}

void Plugin::Cleanup()
{
	for(u8 pos = 0; pos < Plugins.size(); pos++)
	{
		if(Plugins[pos].BannerSound != NULL)
			MEM2_free(Plugins[pos].BannerSound);
	}
}

bool Plugin::AddPlugin(Config &plugin)
{
	if(!adding)
		return false;

	PluginOptions NewPlugin;
	NewPlugin.DolName = plugin.getString("PLUGIN","dolFile","");
	NewPlugin.coverFolder = plugin.getString("PLUGIN","coverFolder","");
	sscanf(plugin.getString("PLUGIN","magic","").c_str(), "%08x", &NewPlugin.magicWord);
	sscanf(plugin.getString("PLUGIN","coverColor","").c_str(), "%08x", &NewPlugin.caseColor);
	NewPlugin.ReturnLoader = plugin.getBool("PLUGIN","ReturnLoader");

	string PluginName = plugin.getString("PLUGIN","displayname","");
	if(PluginName == emptyString || PluginName == emptyString2)
	{
		PluginName = NewPlugin.DolName;
		PluginName.erase(PluginName.end() - 4, PluginName.end());
	}
	NewPlugin.DisplayName.fromUTF8(PluginName.c_str());

	string bannerfilepath = sfmt("%s/%s", pluginsDir.c_str(), plugin.getString("PLUGIN","bannerSound","").c_str());
	ifstream infile;
	infile.open(bannerfilepath.c_str(), ios::binary);
	if(infile.is_open())
	{
		int size;
		infile.seekg(0, ios::end);
		size = infile.tellg();
		infile.seekg(0, ios::beg);
		//Don't free that, otherwise you would delete the sound
		char* FileReadBuffer = (char*)MEM2_alloc(size);
		infile.read(FileReadBuffer, size);
		NewPlugin.BannerSound = (u8*)FileReadBuffer;
		NewPlugin.BannerSoundSize = size;
		Plugins.push_back(NewPlugin);
		return true;
	}
	else
	{
		NewPlugin.BannerSound = 0;
		NewPlugin.BannerSoundSize = 0;
		Plugins.push_back(NewPlugin);
	}
	return false;
}

s8 Plugin::GetPluginPosition(u32 magic)
{
	for(u8 pos = 0; pos < Plugins.size(); pos++)
	{
		if(magic == Plugins[pos].magicWord)
			return pos;
	}
	return -1;
}

bool Plugin::UseReturnLoader(u32 magic)
{
	if((Plugin_Pos = GetPluginPosition(magic)) >= 0)
		return Plugins[Plugin_Pos].ReturnLoader;
	return false;
}

u8* Plugin::GetBannerSound(u32 magic)
{
	if((Plugin_Pos = GetPluginPosition(magic)) >= 0)
		return Plugins[Plugin_Pos].BannerSound;
	return NULL;
}

u32 Plugin::GetBannerSoundSize()
{
	//We call that directly after GetBannerSound, so no need to search for the magic again
	if(Plugin_Pos >= 0)
		return Plugins[Plugin_Pos].BannerSoundSize;
	return 0;
}

char* Plugin::GetDolName(u32 magic)
{
	if((Plugin_Pos = GetPluginPosition(magic)) >= 0)
		return (char*)Plugins[Plugin_Pos].DolName.c_str();
	return emptyChar;
}

char* Plugin::GetCoverFolderName(u32 magic)
{
	if((Plugin_Pos = GetPluginPosition(magic)) >= 0)
		return (char*)Plugins[Plugin_Pos].coverFolder.c_str();
	return emptyChar;
}

bool Plugin::PluginExist(u8 pos)
{
	if(pos < Plugins.size())
		return true;
	return false;
}

wstringEx Plugin::GetPluginName(u8 pos)
{
	return Plugins[pos].DisplayName;
}

void Plugin::SetEnablePlugin(Config &cfg, u8 pos, u8 ForceMode)
{
	if(pos < Plugins.size())
	{
		char PluginMagicWord[8];
		snprintf(PluginMagicWord, sizeof(PluginMagicWord), "%08x", Plugins[pos].magicWord);
		if(ForceMode == 1)
			cfg.setBool("PLUGIN", PluginMagicWord, false);
		else if(ForceMode == 2)
			cfg.setBool("PLUGIN", PluginMagicWord, true);
		else
			cfg.setBool("PLUGIN", PluginMagicWord, cfg.getBool("PLUGIN", PluginMagicWord) ? false : true);
	}
}

vector<bool> Plugin::GetEnabledPlugins(Config &cfg)
{
	vector<bool> enabledPlugins;
	char PluginMagicWord[8];
	u8 enabledPluginsNumber = 0;
	for(u8 i = 0; i < Plugins.size(); i++)
	{
		snprintf(PluginMagicWord, sizeof(PluginMagicWord), "%08x", Plugins[i].magicWord);
		if(cfg.getBool("PLUGIN", PluginMagicWord, true))
		{
			enabledPluginsNumber++;
			enabledPlugins.push_back(true);
		}
		else
			enabledPlugins.push_back(false);
	}
	if(enabledPluginsNumber == Plugins.size())
		enabledPlugins.clear();
	return enabledPlugins;
}

u32 Plugin::getPluginMagic(u8 pos)
{
	return Plugins[pos].magicWord;
}

vector<dir_discHdr> Plugin::ParseScummvmINI(Config &ini, string Device)
{
	gprintf("Parsing scummvm.ini\n");
	vector<dir_discHdr> gameHeader;
	if(!ini.loaded())
		return gameHeader;
	string game(ini.firstDomain());
	string GameName;
	dir_discHdr tmp;
	while(1)
	{
		if(game == emptyString || game == emptyString2)
			break;
		GameName = ini.getString(game, "description");
		if(GameName == emptyString || GameName == emptyString2 ||
		strncasecmp(ini.getWString(game, "path").toUTF8().c_str(), Device.c_str(), 2) != 0)
		{
			game = ini.nextDomain();
			continue;
		}
		memset(&tmp, 0, sizeof(dir_discHdr));
		tmp.hdr.casecolor = Plugins.back().caseColor;
		wstringEx tmpString;
		tmpString.fromUTF8(GameName.c_str());
		wcsncpy(tmp.title, tmpString.c_str(), 64);
		strncpy(tmp.path, game.c_str(), sizeof(tmp.path));
		gprintf("Found: %ls\n", tmp.title);
		tmp.hdr.magic = Plugins.back().magicWord;
		tmp.hdr.gc_magic = EMU_MAGIC;
		gameHeader.push_back(tmp);
		game = ini.nextDomain();
	}
	return gameHeader;
}

/* Thanks to dimok for this */
vector<string> Plugin::CreateMplayerCEArguments(const char *filepath)
{
	vector<string> args;
	char dst[1024];

	int i = 0;
	char device[10];

	while(filepath[i] != ':')
	{
		device[i] = filepath[i];
		device[i+1] = 0;
		i++;
	}

	char * ptr = (char *) &filepath[i];

	while(ptr[0] != '/' || ptr[1] == '/')
		ptr++;

	if(strncmp(DeviceHandler::PathToFSName(filepath), "NTF", 3) == 0)
	{
		sprintf(dst, "ntfs_usb:%s", ptr);
	}
	else if(strncmp(device, "usb", 3) == 0)
	{
		sprintf(dst, "usb:%s", ptr);
	}
	else
	{
		sprintf(dst, "%s:%s", device, ptr);
	}

	args.push_back(dst);
	args.push_back(string("-quiet"));
	return args;
}

bool Plugin::isMplayerCE(u32 magic)
{
	if((Plugin_Pos = GetPluginPosition(magic)) >= 0)
		return (Plugins[Plugin_Pos].magicWord == 0x4D504345);
	return false;
}