/*==LICENSE==*

CyanWorlds.com Engine - MMOG client, server and tools
Copyright (C) 2011  Cyan Worlds, Inc.

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 3 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, see <http://www.gnu.org/licenses/>.

You can contact Cyan Worlds, Inc. by email legal@cyan.com
 or by snail mail at:
      Cyan Worlds, Inc.
      14617 N Newport Hwy
      Mead, WA   99021

*==LICENSE==*/
#include "HeadSpin.h"
#include "plGetLocationDlg.h"
#include "plMaxNode.h"
#include "../MaxComponent/plMiscComponents.h"
#include "resource.h"
#include "../MaxExport/plErrorMsg.h"
#include "../MaxComponent/plComponent.h"
#include "hsUtils.h"

plGetLocationDlg::plGetLocationDlg() : fNode(nil), fErrMsg(nil), fDefaultLocation(nil)
{
}

plGetLocationDlg& plGetLocationDlg::Instance()
{
	static plGetLocationDlg theInstance;
	return theInstance;
}

bool plGetLocationDlg::GetLocation(plMaxNode *node, plErrorMsg *errMsg)
{
	fNode = node;
	fErrMsg = errMsg;

	// If an XRef doesn't have a location, tell the user and stop the export
	if (node->IsXRef())
	{
		char buf[256];
		sprintf(buf, "XRef object \"%s\" does not have a location", node->GetName());
		fErrMsg->Set(true, "Convert Error", buf).Show();
		return false;
	}

	// If we have a default location, use it and exit
	if (fDefaultLocation)
	{
		ISetLocation(fDefaultLocation);
		return true;
	}

	// If we're not showing prompts, just fail if there isn't a location
	if (hsMessageBox_SuppressPrompts)
	{
		fErrMsg->Set(true, "Convert Error", "Object %s doesn't have a location component", node->GetName());
		fErrMsg->Show();
		return false;
	}

	int ret = DialogBox(hInstance,
						MAKEINTRESOURCE(IDD_GET_LOCATION),
						GetCOREInterface()->GetMAXHWnd(),
						ForwardDlgProc);

	return (ret != 0);
}

void plGetLocationDlg::ResetDefaultLocation()
{
	fDefaultLocation = nil;
}

void plGetLocationDlg::IListRooms(plMaxNode *node, HWND hList)
{
	// If node is a room component, add it's name to the list
	plComponentBase *comp = node->ConvertToComponent();
	if(comp && (comp->ClassID() == ROOM_CID || comp->ClassID() == PAGEINFO_CID))
	{
		int idx = SendMessage(hList, LB_ADDSTRING, 0, (LPARAM)node->GetName());
		SendMessage(hList, LB_SETITEMDATA, idx, (LPARAM)node);
	}

	// Recursively add all the nodes children
	for (int i = 0; i < node->NumberOfChildren(); i++)
		IListRooms((plMaxNode*)node->GetChildNode(i), hList);
}

void plGetLocationDlg::IAddSelection(HWND hList, bool setDefault)
{
	int sel = SendMessage(hList, LB_GETCURSEL, 0, 0);
	if (sel != LB_ERR)
	{
		// Get the node and component for the selected room component
		plMaxNode *node = (plMaxNode*)SendMessage(hList, LB_GETITEMDATA, sel, 0);
		ISetLocation(node);

		if (setDefault)
			fDefaultLocation = node;
	}
}

void plGetLocationDlg::ISetLocation(plMaxNode *locNode)
{
	plComponent *comp = (plComponent*)locNode->ConvertToComponent();

	// Add the roomless node to the target list and run the convert pass that gives it a room
	comp->AddTarget(fNode);			// Might want to fix this...
	comp->SetupProperties(fNode, nil);
}

INT_PTR plGetLocationDlg::ForwardDlgProc(HWND hDlg, UINT msg, WPARAM wParam, LPARAM lParam)
{
	return Instance().DlgProc(hDlg, msg, wParam, lParam);
}

INT_PTR plGetLocationDlg::DlgProc(HWND hDlg, UINT msg, WPARAM wParam, LPARAM lParam)
{
	switch (msg)
	{
	case WM_INITDIALOG:
		{
			HWND hList = GetDlgItem(hDlg, IDC_LIST_LOC);
			SendMessage(hList, LB_RESETCONTENT, 0, 0);
			IListRooms((plMaxNode*)GetCOREInterface()->GetRootNode(), hList);

			// Set the prompt text
			char buf[256];
			sprintf(buf, "The object \"%s\" does not have a location. Either pick a location and press OK, or press Cancel to stop the convert.", fNode->GetName());
			SetDlgItemText(hDlg, IDC_PROMPT, buf);

			// No room components found.  Tell user to create one and cancel convert.
			if (SendMessage(hList, LB_GETCOUNT, 0, 0) == 0)
			{
				fErrMsg->Set(true, "Convert Error", "No location component found.  Create one and convert again.");
				fErrMsg->Show();
				EndDialog(hDlg, 0);
			}
			// Else set the first room as the current selection
			else
				SendMessage(hList, LB_SETCURSEL, 0, 0);
		}
		return TRUE;

	case WM_COMMAND:
		switch (LOWORD(wParam))
		{
		case IDOK:
			{
			bool setDefault = (IsDlgButtonChecked(hDlg, IDC_CHECK_DEFAULT) == BST_CHECKED);
			IAddSelection(GetDlgItem(hDlg, IDC_LIST_LOC), setDefault);
			EndDialog(hDlg, 1);
			}
			return TRUE;

		case IDCANCEL:
			// Stop the convert.  No message since they know they canceled.
			fErrMsg->Set(true);
			EndDialog(hDlg, 0);
			return TRUE;
		}
		break;
	}

	return FALSE;
}