#include <windows.h>
#include <commctrl.h>
#include <TCHAR.H>
#include <SWI-Prolog.h>

#include "resource.h"
#include "defs.h"

HWND HElv;
int ISort;
LVITEM LviElvFocus;	/* Focused episode. */
static int CALLBACK ElvSort(LPARAM, LPARAM, LPARAM);

HWND
ElvCreate()
{
	LVCOLUMN lvc;
	term_t t;

	HElv = LvCreate((HMENU)IDC_EPISODELISTVIEW, 0);

	lvc.mask = LVCF_WIDTH|LVCF_TEXT|LVCF_SUBITEM;
	lvc.iSubItem = 0;
	lvc.pszText = TEXT("#");
	lvc.cx = Dpi(42);
	ListView_InsertColumn(HElv, 0, &lvc);

	lvc.iSubItem = 1;
	lvc.pszText = TEXT("/");
	lvc.cx = Dpi(20);
	ListView_InsertColumn(HElv, 1, &lvc);

	lvc.iSubItem = 2;
	lvc.pszText = TEXT("Title");
	lvc.cx = 500;
	ListView_InsertColumn(HElv, 2, &lvc);

	t = T(1);
	P("cfg","get_sort",1,t);
	GI(t,&ISort) ISort = 1;

	return HElv;
}

LRESULT
ElvHandleNotify(LPARAM lParam)
{
	LPNMLISTVIEW lpNmLv;
	lpNmLv = (LPNMLISTVIEW)lParam;

	switch (lpNmLv->hdr.code) {
	case LVN_ITEMCHANGED:	/* Select/focus episode. */
		if ((lpNmLv->uChanged & LVIF_STATE) &&
		    (lpNmLv->uNewState & LVIS_FOCUSED)) {
			LviElvFocus.iItem = lpNmLv->iItem;
			LviElvFocus.lParam = lpNmLv->lParam;
			ElvUpdateItem(&LviElvFocus);
			DlvShowEpisode(lpNmLv->lParam);
		}
		break;
	case LVN_COLUMNCLICK:	/* Sort by column. */
	    {
		int iColumn, iEpFocus;
		term_t t;
		t = T(1);
		iColumn = lpNmLv->iSubItem+1;
		ISort = abs(ISort) == iColumn? -ISort: iColumn;
		PI(t,ISort) goto s;
		P("cfg","set_sort",1,t);
	    s:	ListView_SortItemsEx(HElv, ElvSort, ISort);
		iEpFocus = ListView_GetNextItem(HElv, -1, LVNI_FOCUSED);
		if (iEpFocus == -1) break;
		ListView_EnsureVisible(HElv, iEpFocus, TRUE);
		break;
	    }
	case NM_CUSTOMDRAW:	/* Make unwatched episodes bold. */
	    {
		LPNMLVCUSTOMDRAW lpLvCd;
		lpLvCd = (LPNMLVCUSTOMDRAW)lParam;
		switch (lpLvCd->nmcd.dwDrawStage) {
		case CDDS_PREPAINT:
			return CDRF_NOTIFYITEMDRAW;
			break;
		case CDDS_ITEMPREPAINT:
		    {
			term_t t;
			extern HFONT HfBold;
			t = T(1);
			PI(t,lpLvCd->nmcd.lItemlParam) break;
			P("track_episodes","watched",1,t) {
				SelectObject(lpLvCd->nmcd.hdc, HfBold);
				return CDRF_NEWFONT;
			}
			break;
		    }
		}
		break;
	    }
	case NM_DBLCLK:		/* Open clicked episode. */
	    {
		term_t t;
		t = T(1);
		PI(t,LviElvFocus.lParam) break;
		P("local_episodes","open_episode",1,t);
		break;
	    }
	case NM_RETURN:		/* Open all selected episodes. */
	    {
		LVITEM lvi;
		term_t t;
		extern HWND HElv;
		lvi.mask = LVIF_PARAM;
		lvi.iItem = -1;
		t = T(1);
		while ((lvi.iItem = ListView_GetNextItem(
			    HElv, lvi.iItem, LVNI_SELECTED)) != -1) {
			if (!ListView_GetItem(HElv, &lvi)) goto b;
			PI(t,lvi.lParam) goto b;
			P("local_episodes","open_episode",1,t);
		}
	    b:	break;
	    }
	case NM_RCLICK:
	    {
		DWORD dwPos;
		extern HWND HWnd;
		extern HMENU HPopupMenu;
		dwPos = GetMessagePos();
		TrackPopupMenu(HPopupMenu, TPM_RIGHTBUTTON,
		    LOWORD(dwPos), HIWORD(dwPos), 0,
		    HWnd, NULL);
		break;
	    }
	}

	return 0;
}
void
ElvRedraw()
{
	RedrawWindow(HElv, NULL, NULL,
	    RDW_ERASE|RDW_FRAME|RDW_INVALIDATE|RDW_ALLCHILDREN);
}

void
ElvSetTop(int iItem)
{
	int iLast;
	iLast = ListView_GetItemCount(HElv)-1;
	ListView_EnsureVisible(HElv, iLast, TRUE);
	ListView_EnsureVisible(HElv, iItem, TRUE);
}

/* Select most recent episode. */
void
ElvSelectRecent()
{
	int i, iEpisode, iItem;
	LVFINDINFO lvfi;
	term_t t;

	t = T(1);
	iItem = 0;
	P("track_episodes","most_recently_watched",1,t) goto s;
	GI(t,&iEpisode) return;

	i = 0;
	lvfi.flags = LVFI_PARAM;
	lvfi.lParam = iEpisode;
	do
		lvfi.lParam = iEpisode+i;
	while ((iItem = ListView_FindItem(HElv, -1, &lvfi)) == -1 && i++ < 100);
	if (iItem == -1) return;

s:	ListView_SetItemState(HElv, -1, LVIF_STATE, LVIS_SELECTED);
	ElvSetTop(iItem > 5? iItem-5: 0);
	ListView_SetItemState(HElv, iItem,
	    LVIS_SELECTED|LVIS_FOCUSED, LVIS_SELECTED|LVIS_FOCUSED);
	LviElvFocus.iItem = iItem;
	LviElvFocus.lParam = iEpisode;
	ElvUpdateItem(&LviElvFocus);
	DlvShowEpisode(iEpisode);
}

/* Sort list view items, iSort being the 1-based index of the column
 * to sort by. If iSort is negative, the order is descending. */
int CALLBACK
ElvSort(LPARAM iItem1, LPARAM iItem2, LPARAM iSort)
{
	int iOrder;
	LVITEM lvi1, lvi2;
	lvi1.mask = lvi2.mask = LVIF_PARAM;
	lvi1.iItem = iItem1; lvi2.iItem = iItem2;
	if (!ListView_GetItem(HElv, &lvi1)) return 0;
	if (!ListView_GetItem(HElv, &lvi2)) return 0;
	switch (abs(iSort)) {
	case 1:	/* Sort by episode number. */
		iOrder = lvi1.lParam > lvi2.lParam? 1: -1;
		break;
	case 2:	/* Sort by rating. */
	    {
		int iRating1, iRating2;
		term_t t, t2;
		iRating1 = iSort > 0? 99: -1;
		iRating2 = iSort > 0? 99: -1;
		t = T(2);
		PI(t,lvi1.lParam) goto e;
		P("episode_data","episode_rating",2,t) goto e;
		GI(t+1,&iRating1);
	    e:	t2 = T(2);
		PI(t2,lvi2.lParam) goto f;
		P("episode_data","episode_rating",2,t2) goto f;
		GI(t2+1,&iRating2);
	    f:	iOrder = iRating1 > iRating2? 1: -1;
		break;
	    }
	case 3:	/* Sort by title. */
	    {
		char *sz1, *sz2;
		int cch, cch1, cch2;
		term_t t, t2;
		t = T(2);
		PI(t,lvi1.lParam) return 0;
		P("episode_data","episode_title",2,t) return 0;
		GAC(t+1,&sz1) return 0;
		t2 = T(2);
		PI(t2,lvi2.lParam) return 0;
		P("episode_data","episode_title",2,t2) return 0;
		GAC(t2+1,&sz2) return 0;
		cch1 = strlen(sz1);
		cch2 = strlen(sz2);
		cch = cch1 > cch2? cch2: cch1;
		iOrder = _strnicmp(sz1, sz2, cch);
		break;
	    }
	default:
		return 0;
	}
	return iSort > 0? iOrder: iOrder*-1;
}

/* Update episode list. */
void
ElvUpdate()
{
	int cEp, i, iEp, iEpFocus, iItem, iItemTopNew;
	LVITEM lvi, lviEpisode, lviTop;
	LVFINDINFO lvfi;
	term_t t;
	extern HWND HWndStatus;
	static TCHAR tszDisp[16], tszEpisode[16], tszTotal[16];
	static int aEpSel[2048];

	lviEpisode.mask = LVIF_TEXT|LVIF_PARAM;

	/* Save scrolling position. */
	lviTop.iItem = ListView_GetTopIndex(HElv);
	lviTop.mask = LVIF_PARAM;
	ListView_GetItem(HElv, &lviTop);

	/* Save selected episodes. */
	i = 0;
	lvi.mask = LVIF_PARAM;
	lvi.iItem = -1;
	while ((lvi.iItem = ListView_GetNextItem(
		    HElv, lvi.iItem, LVNI_SELECTED)) != -1
	    && i < 2048) {
		if (!ListView_GetItem(HElv, &lvi)) continue;
		aEpSel[i++] = lvi.lParam;
	}
	aEpSel[i] = 0;

	/* Save focus. */
	iEpFocus = 0;
	while ((lvi.iItem = ListView_GetNextItem(
		    HElv, lvi.iItem, LVNI_FOCUSED)) != -1) {
		if (!ListView_GetItem(HElv, &lvi)) continue;
		iEpFocus = lvi.lParam;
	}

	SendMessage(HElv, WM_SETREDRAW, FALSE, 0);
	ListView_DeleteAllItems(HElv);

	t = T(1);
	P("episode_data","ensure_episode_data",0,t) return;
	P("episode_data","episode_count",1,t) return;
	GI(t,&cEp) return;

	for (iEp = 1, iItem = 0; iEp <= cEp; iEp++) {
		extern int BViewTVOriginal, BViewWatched;

		if (!BViewWatched) {
			term_t t;
			t = T(1);
			PI(t,iEp) goto ep;
			P("track_episodes","watched",1,t) goto ep;
			continue;
		}

		if (!BViewTVOriginal) {
			term_t t;
			t = T(1);
			PI(t,iEp) goto ep;
			P("episode_data","tv_original",1,t) goto ep;
			continue;
		}

		/* Format episode number string. */
	ep:    	_stprintf_s(tszEpisode, sizeof(tszEpisode), TEXT("%d"), iEp);

		/* Insert item. */
		lviEpisode.iItem = iItem++;
		lviEpisode.iSubItem = 0;
		lviEpisode.pszText = tszEpisode;
		lviEpisode.lParam = iEp;
		ListView_InsertItem(HElv, &lviEpisode);
		ElvUpdateItem(&lviEpisode);
	}

	lvfi.flags = LVFI_PARAM;

	/* Reset scrolling position (roughly). */
	i = 0;
	do
		lvfi.lParam = lviTop.lParam+i;
	while ((iItemTopNew = ListView_FindItem(HElv, -1, &lvfi)) == -1
	    && i++ < 100);
	if (iItemTopNew != -1)
		ElvSetTop(iItemTopNew);

	/* Reset selection. */
	for (i = 0; aEpSel[i]; i++) {
		int iSel;
		lvfi.lParam = aEpSel[i];
		if ((iSel = ListView_FindItem(HElv, -1, &lvfi)) != -1)
			ListView_SetItemState(HElv, iSel,
			    LVIS_SELECTED, LVIS_SELECTED);
	}

	/* Reset focus. */
	if (iEpFocus) {
		int iFocus;
		i = 0;
		do
			lvfi.lParam = iEpFocus+i;
		while ((iFocus = ListView_FindItem(HElv, -1, &lvfi)) == -1
		    && i++ < 100);
		if (iFocus != -1)
			ListView_SetItemState(HElv, iFocus,
			    LVIS_FOCUSED, LVIS_FOCUSED);
	}

	_stprintf_s(tszDisp, sizeof(tszDisp), TEXT("%d"), iItem);
	SendMessage(HWndStatus, SB_SETTEXT, MAKEWPARAM(1,0), (LPARAM)tszDisp);
	_stprintf_s(tszTotal, sizeof(tszTotal), TEXT("%d"), cEp);
	SendMessage(HWndStatus, SB_SETTEXT, MAKEWPARAM(2,0), (LPARAM)tszTotal);
	SendMessage(HElv, WM_SETREDRAW, TRUE, 0);
	if (ISort != 1)
		ListView_SortItemsEx(HElv, ElvSort, ISort);
	ElvRedraw();
}

/* Update episode name and rating. */
void
ElvUpdateItem(LPLVITEM lpLvi)
{
	char *szName;
	int iRating;
	TCHAR *tszName, tszRating[3];
	term_t t, t2;

	t = T(2);
	tszName = NULL;
	PI(t,lpLvi->lParam) goto r;
	P("episode_data","episode_title",2,t) {
		P("episode_data","fetch_episode_data",0,t) goto r;
		P("episode_data","episode_title",2,t) goto r;
	}
	GAC(t+1,&szName) goto r;
	tszName = TszFromSz(szName, CP_UTF8);
	if (tszName)
		ListView_SetItemText(HElv, lpLvi->iItem, 2, tszName);

r:	t2 = T(2);
	PI(t2,lpLvi->lParam) goto f;
	P("episode_data","episode_rating",2,t2) {
		ListView_SetItemText(HElv, lpLvi->iItem, 1, TEXT(""));
		goto f;
	}
	GI(t2+1,&iRating) goto f;
	_itot(iRating % 11, tszRating, sizeof(tszRating));
	ListView_SetItemText(HElv, lpLvi->iItem, 1, tszRating);

f:	if (tszName) free(tszName);
}