/*

Miranda IM: the free IM client for Microsoft* Windows*

Copyright 2000-2005 Miranda ICQ/IM project, 
all portions of this codebase are copyrighted to the people 
listed in contributors.txt.

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 2
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, write to the Free Software
Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.

Created by Pescuma 
Based on work by nullbie

*/
#include <windows.h>
#include <io.h>
#include <stdio.h>
#include <newpluginapi.h>
#include <m_clist.h>
#include <m_skin.h>
#include <m_langpack.h>
#include <m_protocols.h>
#include <m_protosvc.h>
#include <m_utils.h>
#include <m_database.h>

#define MIID_REMOVEPERSONALSETTINGS { 0x5eaec989, 0x8ff, 0x4820, { 0xb8, 0x6c, 0x2b, 0x6e, 0xf0, 0x8e, 0x33, 0x73 } }

#define INI_FILE_NAME  "RemovePersonalSettings.ini"

#define PLUGINDISABLELIST "PluginDisable"

#define METACONTACTS_PROTOCOL_NAME "MetaContacts"


HINSTANCE hInst;
char gIniFile[MAX_PATH];
char gMirandaDir[MAX_PATH];

int hLangpack;


PLUGININFOEX pluginInfo={
	sizeof(PLUGININFOEX),
	"Remove Personal Settings",
	PLUGIN_MAKE_VERSION(0,1,0,4),
	"Remove personal settings to allow to send a profile to other user(s) without sending personal data.",
	"Ricardo Pescuma Domenecci",
	"",
	"© 2007-2009 Ricardo Pescuma Domenecci",
	"http://pescuma.org/miranda/rps",
	0,		//not transient
	{ 0x60e94b84, 0xa799, 0x4021, { 0x94, 0x49, 0x5b, 0x83, 0x8f, 0xc0, 0x6a, 0x7c } } // {60E94B84-A799-4021-9449-5B838FC06A7C}
};


INT_PTR RemoveAllService(WPARAM wParam,LPARAM lParam);
void SetProtocolsOffline();
void RemoveUsers();
void RemoveSettings();
void ExecuteServices();
void RemoveDirectories();
void DisablePlugins();

// Ini access functions
BOOL GetSettingBool(const char *section, const char *key, BOOL defaultValue);
BOOL GetSettings(const char *section, char *buffer, size_t bufferSize);


// Utils
void DeleteFileOrFolder(const char *name);
void DeleteSetting(const char *setting);
void DeleteSettingEx(const char *szModule, const char *szSetting);
BOOL isMetaContact(HANDLE hContact);



BOOL WINAPI DllMain(HINSTANCE hinstDLL,DWORD fdwReason,LPVOID lpvReserved)
{
	hInst=hinstDLL;
	return TRUE;
}


__declspec(dllexport) PLUGININFOEX* MirandaPluginInfoEx(DWORD mirandaVersion)
{
    // Are we running under Unicode Windows version ?
    if ((GetVersion() & 0x80000000) == 0)
		pluginInfo.flags = 1; // UNICODE_AWARE
    
	pluginInfo.cbSize = sizeof(PLUGININFOEX);
	return &pluginInfo;
}


extern "C" __declspec(dllexport) const MUUID MirandaInterfaces[] = { MIID_REMOVEPERSONALSETTINGS, MIID_LAST };

int __declspec(dllexport) Load()
{
	CLISTMENUITEM mi;
	char *strTmp;

	CreateServiceFunction("RemovePersonalSettings/RemoveAll",RemoveAllService);
	ZeroMemory(&mi,sizeof(mi));
	mi.cbSize=sizeof(mi);
	mi.position=-0x7FFFFFFF;
	mi.flags=0;
	mi.hIcon=LoadSkinnedIcon(SKINICON_OTHER_MIRANDA);
	mi.pszName="Remove Personal Settings...";
	mi.pszService="RemovePersonalSettings/RemoveAll";
	Menu_AddMainMenuItem(&mi);

	// Get ini file name
	gMirandaDir[0] = '\0';
	GetModuleFileName(GetModuleHandle(NULL),gMirandaDir,sizeof(gMirandaDir));

	// Remove last name
	strTmp = strrchr(gMirandaDir,'\\');
	if(strTmp != NULL)
		*strTmp = '\0';

	// Set vars
	strcat(gMirandaDir, "\\");
	strcpy(gIniFile, gMirandaDir);

	// Store last pos
	strTmp = &gIniFile[strlen(gIniFile)];

	// Lets try fist name
	strcpy(strTmp, INI_FILE_NAME);

	if (_access(gIniFile, 4) != 0)
	{
		// Not found, lets try the other aproach
		strcpy(strTmp, "plugins\\" INI_FILE_NAME);

		if (_access(gIniFile, 4) != 0)
		{
			// Not found :(
			gIniFile[0] = '\0';
		}
	}

	return 0;
}


int __declspec(dllexport) Unload(void)
{
	return 0;
}


INT_PTR RemoveAllService(WPARAM wParam,LPARAM lParam)
{
	if (gIniFile[0] == '\0')
	{
		MessageBox(NULL, Translate("Configuration file could not be found!"), Translate("Remove Personal Settings"), MB_OK | MB_ICONERROR);
		return -1;
	}

	if (MessageBox(NULL,Translate("All your personal settings will be erased!\n"
					"Make sure you are running this from a copy of your profile (and not over the original one).\n"
					"Running this will erase files/folders under Miranda main folder.\n"
					"\n"
					"Are you sure you want to remove all your personal settings?\n"
					"\n"
					"(You cannot say that I don't told you about the risks :P )"), Translate("Remove Personal Settings"),MB_YESNO)
		== IDYES)
	{
		SetProtocolsOffline();
		RemoveUsers();
		RemoveSettings();
		ExecuteServices();
		RemoveDirectories();
		DisablePlugins();

		MessageBox(NULL,Translate("Settings are deleted now."), Translate("Remove Personal Settings"),MB_OK | MB_ICONINFORMATION);
	}

	return 0;
}

void SetProtocolsOffline()
{
	if ( GetSettingBool("GlobalSettings", "SetProtocolsOffline", TRUE) )
	{
		PROTOACCOUNT **accounts;
		int i,count;

		CallService(MS_PROTO_ENUMACCOUNTS, (WPARAM)&count, (LPARAM)&accounts);

		for (i = 0; i < count; i++)
		{
			/*if (protos[i]->type != PROTOTYPE_PROTOCOL)
				continue;*/
			if (!accounts[i]->bIsEnabled)
				continue;

			if (accounts[i]->szModuleName == NULL || accounts[i]->szModuleName[0] == '\0')
				continue;

			CallProtoService(accounts[i]->szModuleName, PS_SETSTATUS, ID_STATUS_OFFLINE, 0);
		}

		// Give some time to make it really offline
		Sleep(2000);
	}
}

void RemoveUsers()
{
	if ( GetSettingBool("GlobalSettings", "RemoveAllUsers", TRUE) )
	{
		HANDLE hContact;
		HANDLE hDbEvent;
		HANDLE hContactOld;

		// To be faster, remove first all metacontacts (because it syncs histories)
		hContact = db_find_first();
		while(hContact != NULL)
		{
			hContactOld = hContact;
			hContact = db_find_next(hContact);

			if ( isMetaContact(hContactOld) )
				CallService(MS_DB_CONTACT_DELETE, (WPARAM) hContactOld, 0);
		}

		// Now delete all left-overs
		hContact = db_find_first();
		while(hContact != NULL)
		{
			CallService(MS_DB_CONTACT_DELETE, (WPARAM) hContact, 0);

			hContact = db_find_first();
		}

		// Delete events for contacts not in list
		hDbEvent = db_event_first(0);

		while(hDbEvent != NULL)
		{
			int ret = db_event_delete(0, hDbEvent);

			hDbEvent = db_event_first(0);
		}

		// Now delete groups
		DeleteSettingEx("CListGroups", NULL);
	}
}

void RemoveProtocolSettings(const char * protocolName)
{
	char buffer[10000];

	// Remove protocol module settings
	if ( GetSettingBool("GlobalSettings", "RemoveWholeProtocolModule", FALSE) )
	{
		DeleteSettingEx(protocolName, NULL);
	}
	else if ( GetSettings("ProtocolSettings", buffer, sizeof(buffer)) )
	{
		char *name;
		char *value;

		name = buffer;
		while(name[0] != '\0')
		{
			value = strchr(name, '=');
			if (value == NULL)
				value = &name[strlen(name)];

			// Has " ?
			if (*name == '"' && *(value-1) == '"')
			{
				name++;
				*(value-1) = '\0';
			}

			// Disable it
			if (name[0] != '\0')
				DeleteSettingEx(protocolName, name);

			// Get next one
			name = value + strlen(value) + 1;
		}
	}

	// Remove modules by protocol sufixes
	if ( GetSettings("ProtocolModuleSufixes", buffer, sizeof(buffer)) )
	{
		char *name;
		char *value;
		char moduleName[256];

		name = buffer;
		while(name[0] != '\0')
		{
			value = strchr(name, '=');
			if (value == NULL)
				value = &name[strlen(name)];

			// Has " ?
			if (*name == '"' && *(value-1) == '"')
			{
				name++;
				*(value-1) = '\0';
			}

			// Delete it
			if (name[0] != '\0')
			{
				mir_snprintf(moduleName, sizeof(moduleName), "%s%s", protocolName, name);
				DeleteSettingEx(moduleName, NULL);
			}

			// Get next one
			name = value + strlen(value) + 1;
		}
	}
}

void RemoveSettings()
{
	char buffer[10000];

	// Delete protocol settings
	if ( GetSettingBool("GlobalSettings", "RemoveProtocolSettings", TRUE) )
	{
		PROTOACCOUNT **accounts;
		int i,count;

		// TODO MS_PROTO_ENUMACCOUNTS
		CallService(MS_PROTO_ENUMACCOUNTS, (WPARAM)&count, (LPARAM)&accounts);

		for (i = 0; i < count; i++)
		{
			/*if (protos[i]->type != PROTOTYPE_PROTOCOL)
				continue;*/
			if (!accounts[i]->bIsEnabled)
				continue;

			if (accounts[i]->szModuleName == NULL || accounts[i]->szModuleName[0] == '\0')
				continue;

			RemoveProtocolSettings(accounts[i]->szModuleName);
		}

		// Get disabled protocols
		if ( GetSettings("DisabledProtocols", buffer, sizeof(buffer)) )
		{
			char *name;
			char *value;

			name = buffer;
			while(name[0] != '\0')
			{
				value = strchr(name, '=');
				if (value == NULL)
					value = &name[strlen(name)];

				// Has " ?
				if (*name == '"' && *(value-1) == '"')
				{
					name++;
					*(value-1) = '\0';
				}

				// Disable it
				if (name[0] != '\0')
					RemoveProtocolSettings(name);

				// Get next one
				name = value + strlen(value) + 1;
			}
		}
	}


	// Delete other settings
	if ( GetSettings("RemoveSettings", buffer, sizeof(buffer)) )
	{
		char *name;
		char *value;

		name = buffer;
		while(name[0] != '\0')
		{
			value = strchr(name, '=');
			if (value == NULL)
				value = &name[strlen(name)];
 
			// Has " ?
			if (*name == '"' && *(value-1) == '"')
			{
				name++;
				*(value-1) = '\0';
			}

			// Delete it
			if (name[0] != '\0')
				DeleteSetting(name);

			// Get next one
			name = value + strlen(value) + 1;
		}
	}
}

void ExecuteServices()
{
	char buffer[10000];

	if ( GetSettings("ExecuteServices", buffer, sizeof(buffer)) )
	{
		char *name;
		char *value;

		name = buffer;
		while(name[0] != '\0')
		{
			value = strchr(name, '=');
			if (value == NULL)
				value = &name[strlen(name)];
 
			// Has " ?
			if (*name == '"' && *(value-1) == '"')
			{
				name++;
				*(value-1) = '\0';
			}

			// Disable it
			if (name[0] != '\0')
				if (ServiceExists(name))
					CallService(name,0,0);

			// Get next one
			name = value + strlen(value) + 1;
		}
	}
}

void RemoveDirectories()
{
	char buffer[10000];
	char dir[MAX_PATH];

	// Remove protocol folders
	if (GetSettingBool("GlobalSettings", "RemoveProtocolFolders", TRUE))
	{
		PROTOACCOUNT **accounts;
		int i,count;

		CallService(MS_PROTO_ENUMACCOUNTS, (WPARAM)&count, (LPARAM)&accounts);

		for (i = 0; i < count; i++)
		{
			/*if (protos[i]->type != PROTOTYPE_PROTOCOL)
				continue;*/
			if (!accounts[i]->bIsEnabled)
				continue;

			if (accounts[i]->szModuleName == NULL || accounts[i]->szModuleName[0] == '\0')
				continue;

			mir_snprintf(dir, sizeof(dir), "%s%s", gMirandaDir, accounts[i]->szModuleName);
			DeleteFileOrFolder(dir);
		}
	}

	// Remove other folders
	if ( GetSettings("RemoveFilesOrFolders", buffer, sizeof(buffer)) )
	{
		char *name;
		char *value;

		name = buffer;
		while(name[0] != '\0')
		{
			value = strchr(name, '=');
			if (value == NULL)
				value = &name[strlen(name)];
 
			// Has " ?
			if (*name == '"' && *(value-1) == '"')
			{
				name++;
				*(value-1) = '\0';
			}

			// Delete it
			if (name[0] != '\0')
			{
				mir_snprintf(dir, sizeof(dir), "%s%s", gMirandaDir, name);
				DeleteFileOrFolder(dir);
			}

			// Get next one
			name = value + strlen(value) + 1;
		}
	}
}

void DisablePlugins()
{
	char buffer[10000];

	if ( GetSettings("DisablePlugins", buffer, sizeof(buffer)) )
	{
		char *name;
		char *value;

		name = buffer;
		while(name[0] != '\0')
		{
			value = strchr(name, '=');
			if (value == NULL)
				value = &name[strlen(name)];
 
			// Has " ?
			if (*name == '"' && *(value-1) == '"')
			{
				name++;
				*(value-1) = '\0';
			}

			// Disable it
			if (name[0] != '\0')
			{
				CharLower(name);
				if (db_get_b(NULL, PLUGINDISABLELIST, name, 0) != 1)
				{
					db_set_b(NULL, PLUGINDISABLELIST, name, 1);
				}
			}

			// Get next one
			name = value + strlen(value) + 1;
		}
	}
}



// Ini access functions

BOOL GetSettingBool(const char *section, const char *key, BOOL defaultValue)
{
	char tmp[16];

	if ( GetPrivateProfileString(section, key, defaultValue ? "true" : "false", tmp, sizeof(tmp), gIniFile) == 0 )
	{
		return defaultValue;
	}
	else
	{
		return stricmp(tmp, "true") == 0;
	}
}


BOOL GetSettings(const char *section, char *buffer, size_t bufferSize)
{
	buffer[0] = '\0\0';

	return GetPrivateProfileSection(section, buffer, bufferSize, gIniFile) != 0;
}



// Utils

void DeleteFileOrFolder(const char *name)
{
	int attibs = GetFileAttributes(name);

	if (attibs == 0xFFFFFFFF) // Not exists
	{
		// Try to find it
		WIN32_FIND_DATA findData;
		HANDLE hwnd;
		char tmp[MAX_PATH];
		char *strTmp;

		// Delete files
		hwnd = FindFirstFile(name, &findData);
		if (hwnd != INVALID_HANDLE_VALUE)
		{
			strcpy(tmp, name);
			strTmp = strrchr(tmp,'\\');
			if(strTmp != NULL)
			{
				strTmp++;
				*strTmp = '\0';
			}
			else
			{
				strcat(tmp, "\\");
				strTmp = &tmp[strlen(tmp)];
			}

			do 
			{
				if (strcmp(findData.cFileName, ".") && strcmp(findData.cFileName, ".."))
				{
					strcpy(strTmp, findData.cFileName);
					DeleteFileOrFolder(tmp);
				}
			}
			while(FindNextFile(hwnd, &findData) != 0);

			FindClose(hwnd);
		}
	}
	else if (attibs & FILE_ATTRIBUTE_DIRECTORY)	// Is a directory
	{
		// Get all files and delete then
		WIN32_FIND_DATA findData;
		HANDLE hwnd;
		char tmp[MAX_PATH];

		mir_snprintf(tmp, sizeof(tmp), "%s\\*.*", name);

		// Delete files
		hwnd = FindFirstFile(tmp, &findData);
		if (hwnd != INVALID_HANDLE_VALUE)
		{
			do 
			{
				if (strcmp(findData.cFileName, ".") && strcmp(findData.cFileName, ".."))
				{
					mir_snprintf(tmp, sizeof(tmp), "%s\\%s", name, findData.cFileName);
					DeleteFileOrFolder(tmp);
				}
			}
			while(FindNextFile(hwnd, &findData) != 0);

			FindClose(hwnd);
		}

		// Delete directory
		RemoveDirectory(name);
	}
	else // Is a File	
	{
		SetFileAttributes(name, FILE_ATTRIBUTE_ARCHIVE);
		DeleteFile(name);
	}
}


BOOL isMetaContact(HANDLE hContact)
{
	DBVARIANT dbv;
	DBCONTACTGETSETTING dbcgs;
	char name[32];

	dbv.type=DBVT_ASCIIZ;
	dbv.pszVal=name;
	dbv.cchVal=sizeof(name);
	dbcgs.pValue=&dbv;
	dbcgs.szModule="Protocol";
	dbcgs.szSetting="p";

	if(CallService(MS_DB_CONTACT_GETSETTINGSTATIC,(WPARAM)hContact,(LPARAM)&dbcgs)) 
		return FALSE;

	return strcmp(dbcgs.pValue->pszVal, METACONTACTS_PROTOCOL_NAME) == 0;
}



typedef struct {
	char buffer[10000];
	int pos;
	const char *filter;
	size_t lenFilterMinusOne;
} DeleteModuleStruct;


int EnumProc(const char *szName, LPARAM lParam)
{
	DeleteModuleStruct *dms = (DeleteModuleStruct *) lParam;
	size_t len = strlen(szName);

	if (dms->filter != NULL && dms->lenFilterMinusOne > 0)
	{
		if (len >= dms->lenFilterMinusOne)
		{
			if (dms->filter[0] == '*')
			{
				if (strcmp(&dms->filter[1], &szName[len - dms->lenFilterMinusOne]) != 0)
					return 0;
			}
			else // if (dms->filter[dms->lenFilterMinusOne] == '*')
			{
				if (strncmp(dms->filter, szName, dms->lenFilterMinusOne) != 0)
					return 0;
			}
		}
	}

	// Add to the struct
	if (len > 0 && len < sizeof(dms->buffer) - dms->pos - 2)
	{
		strcpy(&dms->buffer[dms->pos], szName);
		dms->pos += len + 1;
	}

	return 0;
}

int ModuleEnumProc(const char *szName, DWORD ofsModuleName, LPARAM lParam)
{
	return EnumProc(szName, lParam);
}

void DeleteSettingEx(const char *szModule, const char *szSetting)
{
	size_t lenModule;

	if (szModule == NULL)
		return;

	lenModule = strlen(szModule);
	if (szModule[0] == '*' || szModule[lenModule-1] == '*')
	{
		DeleteModuleStruct dms;
		ZeroMemory(&dms, sizeof(dms));

		dms.filter = szModule;
		dms.lenFilterMinusOne = lenModule-1;

		CallService(MS_DB_MODULES_ENUM, (WPARAM) &dms, (LPARAM) &ModuleEnumProc);

		// Delete then
		szModule = dms.buffer;
		while(szModule[0] != '\0')
		{
			DeleteSettingEx(szModule, szSetting);

			// Get next one
			szModule += strlen(szModule) + 1;
		}
	}
	else
	{
		size_t lenSetting = szSetting == NULL ? 0 : strlen(szSetting);
		if (szSetting == NULL || szSetting[0] == '*' || szSetting[lenSetting-1] == '*')
		{
			DeleteModuleStruct dms;
			DBCONTACTENUMSETTINGS dbces;

			ZeroMemory(&dms, sizeof(dms));

			dms.filter = szSetting;
			dms.lenFilterMinusOne = lenSetting-1;

			dbces.pfnEnumProc = EnumProc;
			dbces.lParam = (LPARAM) &dms;
			dbces.szModule = szModule;
			dbces.ofsSettings = 0;

			CallService(MS_DB_CONTACT_ENUMSETTINGS, 0, (LPARAM) &dbces);

			// Delete then
			szSetting = dms.buffer;
			while(szSetting[0] != '\0')
			{
				db_unset(NULL, szModule, szSetting);

				// Get next one
				szSetting += strlen(szSetting) + 1;
			}
		}
		else
		{
			db_unset(NULL, szModule, szSetting);
		}
	}
}

void DeleteSetting(const char *setting)
{
	char *szModule;
	char *szSetting;

	if (setting == NULL || setting[0] == '\0')
	{
		return;
	}

	// Split setting
	szModule = strdup(setting);
	szSetting = strrchr(szModule, '/');
	if (szSetting != NULL)
	{
		*szSetting = '\0';
		szSetting ++;
	}

	DeleteSettingEx(szModule, szSetting);

	free(szModule);
}