// 16 may 2015
#include "uipriv_windows.h"

// TODO
// - write comment here about how tabs and parents work
// - make sure parent Z-orders are always above tab Z-orders

struct uiTab {
	uiWindowsControl c;
	HWND hwnd;
	struct ptrArray *pages;
	HWND parent;
};

static void onDestroy(uiTab *);

uiWindowsDefineControlWithOnDestroy(
	uiTab,								// type name
	uiTabType,							// type function
	onDestroy(this);						// on destroy
)

// utility functions

static LRESULT curpage(uiTab *t)
{
	return SendMessageW(t->hwnd, TCM_GETCURSEL, 0, 0);
}

static void showHidePage(uiTab *t, LRESULT which, int hide)
{
	struct child *page;

	if (which == (LRESULT) (-1))
		return;
	page = ptrArrayIndex(t->pages, struct child *, which);
	if (hide)
		ShowWindow(childTabPage(page), SW_HIDE);
	else {
		ShowWindow(childTabPage(page), SW_SHOW);
		// we only resize the current page, so we have to do this here
		childQueueRelayout(page);
	}
}

// control implementation

static BOOL onWM_NOTIFY(uiControl *c, HWND hwnd, NMHDR *nm, LRESULT *lResult)
{
	uiTab *t = uiTab(c);

	if (nm->code != TCN_SELCHANGING && nm->code != TCN_SELCHANGE)
		return FALSE;
	showHidePage(t, curpage(t), nm->code == TCN_SELCHANGING);
	*lResult = 0;
	if (nm->code == TCN_SELCHANGING)
		*lResult = FALSE;
	return TRUE;
}

static void onDestroy(uiTab *t)
{
	struct child *page;

	while (t->pages->len != 0) {
		page = ptrArrayIndex(t->pages, struct child *, 0);
		ptrArrayDelete(t->pages, 0);
		childDestroy(page);
	}
	ptrArrayDestroy(t->pages);
	uiWindowsUnregisterWM_NOTIFYHandler(t->hwnd);
}

static void tabCommitSetParent(uiWindowsControl *c, HWND parent)
{
	uiTab *t = uiTab(c);
	struct child *page;
	HWND pagehwnd;
	uintmax_t i;

	t->parent = parent;
	uiWindowsEnsureSetParent(t->hwnd, t->parent);
	for (i = 0; i < t->pages->len; i++) {
		page = ptrArrayIndex(t->pages, struct child *, i);
		pagehwnd = childTabPage(page);
		uiWindowsEnsureSetParent(pagehwnd, t->parent);
	}
	uiWindowsRearrangeControlIDsZOrder(uiControl(t));
}

static void minimumSize(uiWindowsControl *c, uiWindowsSizing *d, intmax_t *width, intmax_t *height)
{
	uiTab *t = uiTab(c);
	intmax_t pagewid, pageht;
	struct child *page;
	LRESULT n;
	RECT r;

	// only consider the current page
	pagewid = 0;
	pageht = 0;
	n = curpage(t);
	if (n != (LRESULT) (-1)) {
		page = ptrArrayIndex(t->pages, struct child *, n);
		childMinimumSize(page, d, &pagewid, &pageht);
	}

	r.left = 0;
	r.top = 0;
	r.right = pagewid;
	r.bottom = pageht;
	// this also includes the tabs themselves
	SendMessageW(t->hwnd, TCM_ADJUSTRECT, (WPARAM) TRUE, (LPARAM) (&r));
	*width = r.right - r.left;
	*height = r.bottom - r.top;
}

static void tabRelayout(uiWindowsControl *c, intmax_t x, intmax_t y, intmax_t width, intmax_t height)
{
	uiTab *t = uiTab(c);
	LRESULT n;
	struct child *page;
	RECT r;

	uiWindowsEnsureMoveWindow(t->hwnd, x, y, width, height);
	n = curpage(t);
	if (n == (LRESULT) (-1))
		return;
	page = ptrArrayIndex(t->pages, struct child *, n);

	// now we need to figure out what rect the child goes
	// this rect needs to be in parent window coordinates, but TCM_ADJUSTRECT wants a window rect, which is screen coordinates
	// because we have each page as a sibling of the tab, use the tab's own rect as the input rect
	r.left = x;
	r.top = y;
	r.right = x + width;
	r.bottom = y + height;
	mapWindowRect(t->hwnd, NULL, &r);
	SendMessageW(t->hwnd, TCM_ADJUSTRECT, (WPARAM) FALSE, (LPARAM) (&r));
	mapWindowRect(NULL, t->hwnd, &r);

	childRelayout(page, r.left, r.top, r.right - r.left, r.bottom - r.top);
}

static void tabContainerUpdateState(uiControl *c)
{
	uiTab *t = uiTab(c);
	struct child *page;
	uintmax_t i;

	for (i = 0; i < t->pages->len; i++) {
		page = ptrArrayIndex(t->pages, struct child *, i);
		childUpdateState(page);
	}
}

static void tabArrangeChildrenControlIDsZOrder(uiWindowsControl *c)
{
	uiTab *t = uiTab(c);
	struct child *page;
	uintmax_t i;

	for (i = 0; i < t->pages->len; i++) {
		page = ptrArrayIndex(t->pages, struct child *, i);
		childSetSoleControlID(page);
	}
}

void uiTabAppend(uiTab *t, const char *name, uiControl *child)
{
	uiTabInsertAt(t, name, t->pages->len, child);
}

void uiTabInsertAt(uiTab *t, const char *name, uintmax_t n, uiControl *child)
{
	struct child *page;
	LRESULT hide, show;
	TCITEMW item;
	WCHAR *wname;

	// see below
	hide = curpage(t);

	page = newChildWithTabPage(child, uiControl(t), t->parent);
	childSetSoleControlID(page);
	ptrArrayInsertAt(t->pages, n, page);

	ZeroMemory(&item, sizeof (TCITEMW));
	item.mask = TCIF_TEXT;
	wname = toUTF16(name);
	item.pszText = wname;
	if (SendMessageW(t->hwnd, TCM_INSERTITEM, (WPARAM) n, (LPARAM) (&item)) == (LRESULT) -1)
		logLastError("error adding tab to uiTab in uiTabInsertAt()");
	uiFree(wname);

	// we need to do this because adding the first tab doesn't send a TCN_SELCHANGE; it just shows the page
	show = curpage(t);
	if (show != hide) {
		showHidePage(t, hide, 1);
		showHidePage(t, show, 0);
	}
}

void uiTabDelete(uiTab *t, uintmax_t n)
{
	struct child *page;

	// first delete the tab from the tab control
	// if this is the current tab, no tab will be selected, which is good
	if (SendMessageW(t->hwnd, TCM_DELETEITEM, (WPARAM) n, 0) == FALSE)
		logLastError("error deleting uiTab tab in tabDelete()");

	// now delete the page itself
	page = ptrArrayIndex(t->pages, struct child *, n);
	ptrArrayDelete(t->pages, n);
	childRemove(page);
}

uintmax_t uiTabNumPages(uiTab *t)
{
	return t->pages->len;
}

int uiTabMargined(uiTab *t, uintmax_t n)
{
	struct child *page;

	page = ptrArrayIndex(t->pages, struct child *, n);
	return childMargined(page);
}

void uiTabSetMargined(uiTab *t, uintmax_t n, int margined)
{
	struct child *page;

	page = ptrArrayIndex(t->pages, struct child *, n);
	childSetMargined(page, margined);
}

uiTab *uiNewTab(void)
{
	uiTab *t;

	t = (uiTab *) uiNewControl(uiTabType());

	t->hwnd = uiWindowsEnsureCreateControlHWND(0,
		WC_TABCONTROLW, L"",
		TCS_TOOLTIPS | WS_TABSTOP,
		hInstance, NULL,
		TRUE);

	uiWindowsRegisterWM_NOTIFYHandler(t->hwnd, onWM_NOTIFY, uiControl(t));

	t->pages = newPtrArray();

	t->parent = utilWindow;

	uiWindowsFinishNewControl(t, uiTab);
	uiControl(t)->ContainerUpdateState = tabContainerUpdateState;
	uiWindowsControl(t)->CommitSetParent = tabCommitSetParent;
	uiWindowsControl(t)->Relayout = tabRelayout;
	uiWindowsControl(t)->ArrangeChildrenControlIDsZOrder = tabArrangeChildrenControlIDsZOrder;

	return t;
}