/*

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

Copyright 2000-12 Miranda IM, 2012-13 Miranda NG 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.
*/
#include "..\..\core\commonheaders.h"

struct SoundItem
{
	char*  name;
	TCHAR* ptszSection;
	TCHAR* ptszDescription;
	TCHAR* ptszTempFile;
	int    hLangpack;

	__inline TCHAR* getSection() const { return TranslateTH(hLangpack, ptszSection); }
	__inline TCHAR* getDescr() const { return TranslateTH(hLangpack, ptszDescription); }

	__inline void clear(void)
	{
		mir_free(name);
		mir_free(ptszSection);
		mir_free(ptszDescription);
		mir_free(ptszTempFile);
	}
};

static int CompareSounds(const SoundItem* p1, const SoundItem* p2)
{
	return lstrcmpA(p1->name, p2->name);
}

static OBJLIST<SoundItem> arSounds(10, CompareSounds);

///////////////////////////////////////////////////////////////////////////////

void KillModuleSounds(int hLangpack)
{
	for (int i = arSounds.getCount()-1; i >= 0; i--) {
		SoundItem& p = arSounds[i];
		if (p.hLangpack == hLangpack) {
			p.clear();
			arSounds.remove(i);
		}
	}
}

///////////////////////////////////////////////////////////////////////////////

static BOOL bModuleInitialized = FALSE;
static HANDLE hPlayEvent = NULL;

static INT_PTR ServiceSkinAddNewSound(WPARAM wParam, LPARAM lParam)
{
	SKINSOUNDDESCEX *ssd = (SKINSOUNDDESCEX*)lParam;
	if (ssd->cbSize != sizeof(SKINSOUNDDESCEX) || ssd->pszName == NULL || ssd->pszDescription == NULL)
		return 1;

	SoundItem* item = new SoundItem; // due to OBJLIST
	item->name = mir_strdup(ssd->pszName);
	item->ptszTempFile = NULL;
	item->hLangpack = (int)wParam;
	arSounds.insert(item);
	
	TCHAR* ptszDefaultFile;
	if (ssd->dwFlags & SSDF_UNICODE) {
		item->ptszDescription = mir_tstrdup(ssd->ptszDescription);
		item->ptszSection = mir_tstrdup((ssd->pszSection != NULL) ? ssd->ptszSection : _T("Other"));
		ptszDefaultFile = mir_tstrdup(ssd->ptszDefaultFile);
	}
	else {
		item->ptszDescription = mir_a2t(ssd->pszDescription);
		item->ptszSection = mir_a2t((ssd->pszSection != NULL) ? ssd->pszSection : "Other");
		ptszDefaultFile = mir_a2t(ssd->pszDefaultFile);
	}

	if (ptszDefaultFile) {
		DBVARIANT dbv;
		if (db_get_s(NULL, "SkinSounds", item->name, &dbv))
			db_set_ts(NULL, "SkinSounds", item->name, ptszDefaultFile);
		else
			db_free(&dbv);
		mir_free(ptszDefaultFile);
	}

	return 0;
}

static int SkinPlaySoundDefault(WPARAM wParam, LPARAM lParam)
{
	TCHAR* pszFile = (TCHAR*) lParam;
	if (pszFile && (db_get_b(NULL, "Skin", "UseSound", 0) || (int)wParam == 1))
		PlaySound(pszFile, NULL, SND_ASYNC | SND_FILENAME | SND_NOSTOP);

	return 0;
}

static INT_PTR ServiceSkinPlaySoundFile(WPARAM, LPARAM lParam)
{
	TCHAR *ptszFileName = (TCHAR*)lParam;
	if (ptszFileName == NULL)
		return 1;

	TCHAR tszFull[MAX_PATH];
	PathToAbsoluteT(ptszFileName, tszFull);
	NotifyEventHooks(hPlayEvent, 0, (LPARAM)tszFull);
	return 0;
}

static INT_PTR ServiceSkinPlaySound(WPARAM, LPARAM lParam)
{
	char* pszSoundName = (char*)lParam;
	if (pszSoundName == NULL)
		return 1;

	SoundItem tmp = { pszSoundName };
	int idx = arSounds.getIndex( &tmp );
	if (idx == -1)
		return 1;

	if ( db_get_b(NULL, "SkinSoundsOff", pszSoundName, 0))
		return 1;

	DBVARIANT dbv;
	if ( db_get_ts(NULL, "SkinSounds", pszSoundName, &dbv) == 0) {
		ServiceSkinPlaySoundFile(0, (LPARAM)dbv.ptszVal);
		db_free(&dbv);
		return 0;
	}
	return 1;
}

#define DM_REBUILD_STREE (WM_USER+1)
#define DM_HIDEPANE      (WM_USER+2)
#define DM_SHOWPANE      (WM_USER+3)
#define DM_CHECKENABLED  (WM_USER+4)

INT_PTR CALLBACK DlgProcSoundOpts(HWND hwndDlg, UINT msg, WPARAM wParam, LPARAM lParam)
{
	static HWND hwndTree = NULL;
	switch (msg) {
	case WM_INITDIALOG:
		TranslateDialogDefault(hwndDlg);
		hwndTree = GetDlgItem(hwndDlg, IDC_SOUNDTREE);
		SetWindowLongPtr(hwndTree, GWL_STYLE, GetWindowLongPtr(hwndTree, GWL_STYLE)|TVS_NOHSCROLL|TVS_CHECKBOXES);
		SendMessage(hwndDlg, DM_HIDEPANE, 0, 0);
		SendMessage(hwndDlg, DM_REBUILD_STREE, 0, 0);
		TreeView_SetItemState(hwndTree, 0, TVIS_SELECTED, TVIS_SELECTED);
		CheckDlgButton(hwndDlg, IDC_ENABLESOUNDS, db_get_b(NULL, "Skin", "UseSound", 0));
		SendMessage(hwndDlg, DM_CHECKENABLED, 0, 0);
		return TRUE;

	case DM_REBUILD_STREE:
		TreeView_SelectItem(hwndTree, NULL);
		ShowWindow(hwndTree, SW_HIDE);
		TreeView_DeleteAllItems(hwndTree);
		{
			TVINSERTSTRUCT tvis;
			tvis.hParent = NULL;
			tvis.hInsertAfter = TVI_SORT;
			tvis.item.mask = TVIF_TEXT | TVIF_STATE | TVIF_PARAM;
			tvis.item.state = tvis.item.stateMask = TVIS_EXPANDED;
			for (int i=0; i < arSounds.getCount(); i++) {
				tvis.item.stateMask = TVIS_EXPANDED;
				tvis.item.state = TVIS_EXPANDED;
				tvis.hParent = FindNamedTreeItemAtRoot(hwndTree, arSounds[i].getSection());
				if (tvis.hParent == NULL) {
					tvis.item.lParam = -1;
					tvis.item.pszText = arSounds[i].getSection();
					tvis.hParent = tvis.item.hItem = TreeView_InsertItem(hwndTree, &tvis);
					tvis.item.stateMask = TVIS_STATEIMAGEMASK;
					tvis.item.state = INDEXTOSTATEIMAGEMASK(0);
					TreeView_SetItem(hwndTree, &tvis.item);
				}
				tvis.item.stateMask = TVIS_STATEIMAGEMASK;
				tvis.item.state = INDEXTOSTATEIMAGEMASK( !db_get_b(NULL, "SkinSoundsOff", arSounds[i].name, 0)?2:1);
				tvis.item.lParam = i;
				tvis.item.pszText = arSounds[i].getDescr();
				TreeView_InsertItem(hwndTree, &tvis);
		}	}
		{
			TVITEM tvi;
			tvi.hItem = TreeView_GetRoot(hwndTree);
			while (tvi.hItem != NULL) {
				tvi.mask = TVIF_PARAM | TVIF_HANDLE | TVIF_STATE;
				TreeView_GetItem(hwndTree, &tvi);
				if (tvi.lParam == -1)
					TreeView_SetItemState(hwndTree, tvi.hItem, INDEXTOSTATEIMAGEMASK(0), TVIS_STATEIMAGEMASK);

				tvi.hItem = TreeView_GetNextSibling(hwndTree, tvi.hItem);
		}	}

		ShowWindow(hwndTree, SW_SHOW);
		break;

	case DM_HIDEPANE:
		ShowWindow( GetDlgItem(hwndDlg, IDC_SGROUP), SW_HIDE);
		ShowWindow( GetDlgItem(hwndDlg, IDC_NAME), SW_HIDE);
		ShowWindow( GetDlgItem(hwndDlg, IDC_NAMEVAL), SW_HIDE);
		ShowWindow( GetDlgItem(hwndDlg, IDC_SLOC), SW_HIDE);
		ShowWindow( GetDlgItem(hwndDlg, IDC_LOCATION), SW_HIDE);
		ShowWindow( GetDlgItem(hwndDlg, IDC_CHANGE), SW_HIDE);
		ShowWindow( GetDlgItem(hwndDlg, IDC_PREVIEW), SW_HIDE);
		ShowWindow( GetDlgItem(hwndDlg, IDC_GETMORE), SW_HIDE);
		break;

	case DM_SHOWPANE:
		ShowWindow( GetDlgItem(hwndDlg, IDC_SGROUP), SW_SHOW);
		ShowWindow( GetDlgItem(hwndDlg, IDC_NAME), SW_SHOW);
		ShowWindow( GetDlgItem(hwndDlg, IDC_NAMEVAL), SW_SHOW);
		ShowWindow( GetDlgItem(hwndDlg, IDC_SLOC), SW_SHOW);
		ShowWindow( GetDlgItem(hwndDlg, IDC_LOCATION), SW_SHOW);
		ShowWindow( GetDlgItem(hwndDlg, IDC_CHANGE), SW_SHOW);
		ShowWindow( GetDlgItem(hwndDlg, IDC_PREVIEW), SW_SHOW);
		ShowWindow( GetDlgItem(hwndDlg, IDC_GETMORE), SW_SHOW);
		break;

	case DM_CHECKENABLED:
		EnableWindow( GetDlgItem(hwndDlg, IDC_SOUNDTREE), IsDlgButtonChecked(hwndDlg, IDC_ENABLESOUNDS));
		if ( !IsDlgButtonChecked(hwndDlg, IDC_ENABLESOUNDS))
			SendMessage(hwndDlg, DM_HIDEPANE, 0, 0);
		else if (TreeView_GetSelection(hwndTree) && TreeView_GetParent(hwndTree, TreeView_GetSelection(hwndTree)))
			SendMessage(hwndDlg, DM_SHOWPANE, 0, 0);
		break;

	case WM_COMMAND:
		if (LOWORD(wParam) == IDC_ENABLESOUNDS)
			SendMessage(hwndDlg, DM_CHECKENABLED, 0, 0);

		if (LOWORD(wParam) == IDC_PREVIEW) {
			HTREEITEM hti = TreeView_GetSelection(hwndTree);
			if (hti == NULL)
				break;

			TVITEM tvi = { 0 };
			tvi.mask = TVIF_HANDLE | TVIF_IMAGE | TVIF_SELECTEDIMAGE | TVIF_PARAM | TVIF_TEXT;
			tvi.hItem = hti;
			if (TreeView_GetItem(hwndTree, &tvi) == FALSE)
				break;
			if (tvi.lParam == -1)
				break;

			if (arSounds[tvi.lParam].ptszTempFile)
				NotifyEventHooks(hPlayEvent, 1, (LPARAM)arSounds[tvi.lParam].ptszTempFile);
			else {
				DBVARIANT dbv;
				if ( !db_get_ts(NULL, "SkinSounds", arSounds[tvi.lParam].name, &dbv)) {
					TCHAR szPathFull[MAX_PATH];
					PathToAbsoluteT(dbv.ptszVal, szPathFull);
					NotifyEventHooks(hPlayEvent, 1, (LPARAM)szPathFull);
					db_free(&dbv);
				}
			}
			break;
		}

		if (LOWORD(wParam) == IDC_CHANGE) {
			HTREEITEM hti = TreeView_GetSelection(hwndTree);
			if (hti == NULL)
				break;

			TVITEM tvi = { 0 };
			tvi.mask = TVIF_HANDLE|TVIF_IMAGE|TVIF_SELECTEDIMAGE|TVIF_PARAM|TVIF_TEXT;
			tvi.hItem = hti;
			if (TreeView_GetItem(hwndTree, &tvi) == FALSE)
				break;
			if (tvi.lParam == -1)
				break;

			SoundItem& snd = arSounds[tvi.lParam];

			TCHAR str[MAX_PATH], strFull[MAX_PATH], strdir[MAX_PATH], filter[MAX_PATH];
			if (snd.ptszTempFile)
				_tcsncpy(strFull, snd.ptszTempFile, SIZEOF(strFull));
			else {
				if (db_get_b(NULL, "SkinSoundsOff", snd.name, 0) == 0) {
					DBVARIANT dbv;
					if (db_get_ts(NULL, "SkinSounds", snd.name, &dbv) == 0) {
						PathToAbsoluteT(dbv.ptszVal, strdir);
						db_free(&dbv);
			}	}	}

			mir_sntprintf(strFull, SIZEOF(strFull), _T("%s"), snd.ptszTempFile ? snd.ptszTempFile : _T(""));
			PathToAbsoluteT(strFull, strdir);

			OPENFILENAME ofn;
			ZeroMemory(&ofn, sizeof(ofn));
			if (GetModuleHandle(_T("bass_interface.dll")))
				mir_sntprintf(filter, SIZEOF(filter), _T("%s (*.wav, *.mp3, *.ogg)%c*.wav;*.mp3;*.ogg%c%s (*)%c*%c"), TranslateT("Sound Files"), 0, 0, TranslateT("All Files"), 0, 0);
			else
				mir_sntprintf(filter, SIZEOF(filter), _T("%s (*.wav)%c*.wav%c%s (*)%c*%c"), TranslateT("WAV Files"), 0, 0, TranslateT("All Files"), 0, 0);
			ofn.lStructSize = OPENFILENAME_SIZE_VERSION_400;
			ofn.hwndOwner = GetParent(hwndDlg);
			ofn.hInstance = NULL;
			ofn.lpstrFilter = filter;
			
			TCHAR* slash = _tcsrchr(strdir, '\\');
			if (slash) {
				*slash = 0;
				ofn.lpstrInitialDir = strdir;
			}

			str[0] = 0;
			ofn.lpstrFile = str;
			ofn.Flags = OFN_FILEMUSTEXIST|OFN_HIDEREADONLY|OFN_EXPLORER|OFN_LONGNAMES|OFN_NOCHANGEDIR;
			ofn.nMaxFile = SIZEOF(str);
			ofn.nMaxFileTitle = MAX_PATH;
			ofn.lpstrDefExt = _T("wav");
			if ( !GetOpenFileName(&ofn))
				break;

			PathToRelativeT(str, strFull);
			snd.ptszTempFile = mir_tstrdup(strFull);
			SetDlgItemText(hwndDlg, IDC_LOCATION, strFull);
		}
		if (LOWORD(wParam) == IDC_GETMORE) {
			CallService(MS_UTILS_OPENURL, 1, (LPARAM)"http://miranda-ng.org/");
			break;
		}
		if (LOWORD(wParam) == IDC_LOCATION)
			break;

		SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
		break;

	case WM_NOTIFY:
		switch(((LPNMHDR)lParam)->idFrom) {
		case 0:
			if (((LPNMHDR)lParam)->code == PSN_APPLY) {
				db_set_b(NULL, "Skin", "UseSound", (BYTE) IsDlgButtonChecked(hwndDlg, IDC_ENABLESOUNDS));

				for (int i=0; i < arSounds.getCount(); i++)
					if (arSounds[i].ptszTempFile)
						db_set_ts(NULL, "SkinSounds", arSounds[i].name, arSounds[i].ptszTempFile);

				TVITEM tvi, tvic;
				tvi.hItem = TreeView_GetRoot(hwndTree);
				while (tvi.hItem != NULL) {
					tvi.mask = TVIF_PARAM | TVIF_HANDLE | TVIF_STATE;
					TreeView_GetItem(hwndTree, &tvi);
					if (tvi.lParam == -1) {
						tvic.hItem = TreeView_GetChild(hwndTree, tvi.hItem);
						while (tvic.hItem != NULL) {
							tvic.mask = TVIF_PARAM | TVIF_HANDLE | TVIF_STATE;
							TreeView_GetItem(hwndTree, &tvic);
							if (((tvic.state & TVIS_STATEIMAGEMASK) >> 12 == 2))
								db_unset(NULL, "SkinSoundsOff", arSounds[tvic.lParam].name);
							else
								db_set_b(NULL, "SkinSoundsOff", arSounds[tvic.lParam].name, 1);
							tvic.hItem = TreeView_GetNextSibling(hwndTree, tvic.hItem);
					}	}

					tvi.hItem = TreeView_GetNextSibling(hwndTree, tvi.hItem);
				}
				return TRUE;
			}
			break;

		case IDC_SOUNDTREE:
			switch(((NMHDR*)lParam)->code) {
			case TVN_SELCHANGEDW:
			case TVN_SELCHANGEDA:
				{
					NMTREEVIEW *pnmtv = (NMTREEVIEW*)lParam;
					TVITEM tvi = pnmtv->itemNew;

					if (tvi.lParam == -1)
						SendMessage(hwndDlg, DM_HIDEPANE, 0, 0);
					else {
						TCHAR buf[256];
						mir_sntprintf(buf, SIZEOF(buf), _T("%s: %s"), arSounds[tvi.lParam].getSection(), arSounds[tvi.lParam].getDescr());
						SetDlgItemText(hwndDlg, IDC_NAMEVAL, buf);
						if (arSounds[tvi.lParam].ptszTempFile)
							SetDlgItemText(hwndDlg, IDC_LOCATION, arSounds[tvi.lParam].ptszTempFile);
						else {
							DBVARIANT dbv;
							if ( !db_get_ts(NULL, "SkinSounds", arSounds[tvi.lParam].name, &dbv)) {
								SetDlgItemText(hwndDlg, IDC_LOCATION, dbv.ptszVal);
								db_free(&dbv);
							}
							else SetDlgItemText(hwndDlg, IDC_LOCATION, TranslateT("<not specified>"));
						}
						SendMessage(hwndDlg, DM_SHOWPANE, 0, 0);
					}
				}
				break;
			case TVN_KEYDOWN:
				{
					NMTVKEYDOWN* ptkd = (NMTVKEYDOWN*)lParam;
					if (ptkd && ptkd->wVKey == VK_SPACE && TreeView_GetSelection(ptkd->hdr.hwndFrom))
						SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
				}
				break;
			case NM_CLICK:
				{
					TVHITTESTINFO hti;
					hti.pt.x = (short)LOWORD(GetMessagePos());
					hti.pt.y = (short)HIWORD(GetMessagePos());
					ScreenToClient(((LPNMHDR)lParam)->hwndFrom, &hti.pt);
					if (TreeView_HitTest(((LPNMHDR)lParam)->hwndFrom, &hti))
						if (hti.flags & (TVHT_ONITEM | TVHT_ONITEMSTATEICON))
							if (TreeView_GetParent(hwndTree, hti.hItem) != NULL)
								SendMessage(GetParent(hwndDlg), PSM_CHANGED, 0, 0);
				}
				break;
			}	
		}
		break;

	case WM_DESTROY:
		ImageList_Destroy(TreeView_GetImageList(hwndTree, TVSIL_STATE));
		break;
	}
	return FALSE;
}

static int SkinOptionsInit(WPARAM wParam, LPARAM)
{
	OPTIONSDIALOGPAGE odp = { 0 };
	odp.cbSize = sizeof(odp);
	odp.position = -200000000;
	odp.hInstance = hInst;
	odp.pszTemplate = MAKEINTRESOURCEA(IDD_OPT_SOUND);
	odp.pszTitle = LPGEN("Sounds");
	odp.pfnDlgProc = DlgProcSoundOpts;
	odp.flags = ODPF_BOLDGROUPS;
	Options_AddPage(wParam, &odp);
	return 0;
}

static int SkinSystemModulesLoaded(WPARAM, LPARAM)
{
	HookEvent(ME_OPT_INITIALISE, SkinOptionsInit);
	return 0;
}

int LoadSkinSounds(void)
{
	bModuleInitialized = TRUE;

	CreateServiceFunction("Skin/Sounds/AddNew", ServiceSkinAddNewSound);
	CreateServiceFunction(MS_SKIN_PLAYSOUND, ServiceSkinPlaySound);
	CreateServiceFunction(MS_SKIN_PLAYSOUNDFILE, ServiceSkinPlaySoundFile);
	HookEvent(ME_SYSTEM_MODULESLOADED, SkinSystemModulesLoaded);
	hPlayEvent = CreateHookableEvent(ME_SKIN_PLAYINGSOUND);
	SetHookDefaultForHookableEvent(hPlayEvent, SkinPlaySoundDefault);
	return 0;
}

void UnloadSkinSounds(void)
{
	for (int i=0; i < arSounds.getCount(); i++)
		arSounds[i].clear();

	arSounds.destroy();
}