#include <fstream>
#include <boost/filesystem/operations.hpp>
#include <boost/lexical_cast.hpp>
#include <boost/date_time/posix_time/posix_time.hpp>
#include <boost/bind.hpp>
#include "../saves/SaveExporter.h"
#include "SaveView.h"
#include "string_cast.h"
#include "win32/Static.h"
#include "win32/FileDialog.h"

#define CLSNAME _T("CSaveView")

using namespace Framework;
using namespace boost;
using namespace std;

CSaveView::CSaveView(HWND hParent)
{
	RECT rc;

	if(!DoesWindowClassExist(CLSNAME))
	{
		WNDCLASSEX wc;
		memset(&wc, 0, sizeof(WNDCLASSEX));
		wc.cbSize			= sizeof(WNDCLASSEX);
		wc.hCursor			= LoadCursor(NULL, IDC_ARROW);
		wc.hbrBackground	= (HBRUSH)GetSysColorBrush(COLOR_BTNFACE); 
		wc.hInstance		= GetModuleHandle(NULL);
		wc.lpszClassName	= CLSNAME;
		wc.lpfnWndProc		= CWindow::WndProc;
		wc.style			= CS_OWNDC | CS_HREDRAW | CS_VREDRAW;
		RegisterClassEx(&wc);
	}

	Create(NULL, CLSNAME, _T(""), WS_VISIBLE | WS_CLIPCHILDREN | WS_CHILD, &rc, hParent, NULL);
	SetClassPtr();

	GetClientRect(&rc);

	m_pNameLine1	= new Win32::CEdit(m_hWnd, &rc, _T(""), ES_READONLY, 0);
	m_pNameLine2	= new Win32::CEdit(m_hWnd, &rc, _T(""), ES_READONLY, 0);
	m_pSize			= new Win32::CEdit(m_hWnd, &rc, _T(""), ES_READONLY, 0);
	m_pId			= new Win32::CEdit(m_hWnd, &rc, _T(""), ES_READONLY, 0);
	m_pLastModified	= new Win32::CEdit(m_hWnd, &rc, _T(""), ES_READONLY, 0);
	m_pOpenFolder	= new Win32::CButton(_T("Open folder..."), m_hWnd, &rc);
	m_pExport		= new Win32::CButton(_T("Export..."), m_hWnd, &rc);
	m_pDelete		= new Win32::CButton(_T("Delete"), m_hWnd, &rc);
	m_pNormalIcon	= new Win32::CButton(_T("Normal Icon"), m_hWnd, &rc, BS_PUSHLIKE | BS_CHECKBOX);
	m_pCopyingIcon	= new Win32::CButton(_T("Copying Icon"), m_hWnd, &rc, BS_PUSHLIKE | BS_CHECKBOX);
	m_pDeletingIcon	= new Win32::CButton(_T("Deleting Icon"), m_hWnd, &rc, BS_PUSHLIKE | BS_CHECKBOX);
	m_pIconViewWnd	= new CIconViewWnd(m_hWnd, &rc);

	m_CommandSink.RegisterCallback(m_pOpenFolder->m_hWnd,	bind(&CSaveView::OpenSaveFolder, this));
	m_CommandSink.RegisterCallback(m_pNormalIcon->m_hWnd,	bind(&CSaveView::SetIconType, this, ICON_NORMAL));
	m_CommandSink.RegisterCallback(m_pCopyingIcon->m_hWnd,	bind(&CSaveView::SetIconType, this, ICON_COPYING));
	m_CommandSink.RegisterCallback(m_pDeletingIcon->m_hWnd,	bind(&CSaveView::SetIconType, this, ICON_DELETING));
	m_CommandSink.RegisterCallback(m_pExport->m_hWnd,		bind(&CSaveView::Export, this));
	m_CommandSink.RegisterCallback(m_pDelete->m_hWnd,		bind(&CSaveView::Delete, this));

    FlatLayoutPtr pSubLayout0 = CHorizontalLayout::Create();
	{
		pSubLayout0->InsertObject(Win32::CLayoutWindow::CreateTextBoxBehavior(300, 23, m_pId));
		pSubLayout0->InsertObject(Win32::CLayoutWindow::CreateButtonBehavior(100, 23, m_pOpenFolder));
	}

    GridLayoutPtr pSubLayout1 = CGridLayout::Create(2, 5);
	{
		pSubLayout1->SetObject(0, 0, Win32::CLayoutWindow::CreateTextBoxBehavior(100, 23, new Win32::CStatic(m_hWnd, _T("Name:"))));
		pSubLayout1->SetObject(0, 2, Win32::CLayoutWindow::CreateTextBoxBehavior(100, 23, new Win32::CStatic(m_hWnd, _T("Size:"))));
		pSubLayout1->SetObject(0, 3, Win32::CLayoutWindow::CreateTextBoxBehavior(100, 23, new Win32::CStatic(m_hWnd, _T("Id:"))));
		pSubLayout1->SetObject(0, 4, Win32::CLayoutWindow::CreateTextBoxBehavior(100, 23, new Win32::CStatic(m_hWnd, _T("Last Modified:")))); 

		pSubLayout1->SetObject(1, 0, Win32::CLayoutWindow::CreateTextBoxBehavior(300, 23, m_pNameLine1));
		pSubLayout1->SetObject(1, 1, Win32::CLayoutWindow::CreateTextBoxBehavior(300, 23, m_pNameLine2));
		pSubLayout1->SetObject(1, 2, Win32::CLayoutWindow::CreateTextBoxBehavior(300, 23, m_pSize));
		pSubLayout1->SetObject(1, 3, pSubLayout0);
		pSubLayout1->SetObject(1, 4, Win32::CLayoutWindow::CreateTextBoxBehavior(300, 23, m_pLastModified));

		pSubLayout1->SetVerticalStretch(0);
	}

    FlatLayoutPtr pSubLayout2 = CHorizontalLayout::Create();
	{
		pSubLayout2->InsertObject(Win32::CLayoutWindow::CreateButtonBehavior(100, 23, m_pExport));
		pSubLayout2->InsertObject(Win32::CLayoutWindow::CreateButtonBehavior(100, 23, m_pDelete));
		pSubLayout2->SetVerticalStretch(0);
	}

    FlatLayoutPtr pSubLayout3 = CHorizontalLayout::Create();
	{
        pSubLayout3->InsertObject(Win32::CLayoutWindow::CreateCustomBehavior(50, 50, 1, 1, m_pIconViewWnd));
		pSubLayout3->SetVerticalStretch(1);
	}

    FlatLayoutPtr pSubLayout4 = CHorizontalLayout::Create();
    {
        pSubLayout4->InsertObject(Win32::CLayoutWindow::CreateCustomBehavior(50, 23, 1, 0, m_pNormalIcon));
        pSubLayout4->InsertObject(Win32::CLayoutWindow::CreateCustomBehavior(50, 23, 1, 0, m_pCopyingIcon));
        pSubLayout4->InsertObject(Win32::CLayoutWindow::CreateCustomBehavior(50, 23, 1, 0, m_pDeletingIcon));
        pSubLayout4->SetVerticalStretch(0);
    }

	SetIconType(ICON_NORMAL);

    m_pLayout = CVerticalLayout::Create();
	m_pLayout->InsertObject(pSubLayout1);
	m_pLayout->InsertObject(pSubLayout2);
	m_pLayout->InsertObject(pSubLayout3);
	m_pLayout->InsertObject(pSubLayout4);

	RefreshLayout();
}

CSaveView::~CSaveView()
{

}

void CSaveView::SetSave(const CSave* pSave)
{
	m_pSave = pSave;

	if(m_pSave != NULL)
	{
		wstring sName(m_pSave->GetName());

		m_pNameLine1->SetText(sName.substr(0, m_pSave->GetSecondLineStartPosition()).c_str());
		m_pNameLine2->SetText(sName.substr(m_pSave->GetSecondLineStartPosition()).c_str());
		m_pSize->SetText((lexical_cast<tstring>(m_pSave->GetSize()) + _T(" bytes")).c_str());
		m_pId->SetText(string_cast<tstring>(m_pSave->GetId()).c_str());
		m_pLastModified->SetText(
			string_cast<tstring>(
			posix_time::to_simple_string(
			posix_time::from_time_t(m_pSave->GetLastModificationTime()))).c_str());
	}
	else
	{
		m_pNameLine1->SetText(_T("--"));
		m_pNameLine2->SetText(_T(""));
		m_pSize->SetText(_T("--"));
		m_pId->SetText(_T("--"));
		m_pLastModified->SetText(_T("--"));
	}

	m_pIconViewWnd->SetSave(pSave);
	m_pOpenFolder->Enable(m_pSave != NULL);
	m_pNormalIcon->Enable(m_pSave != NULL);
	m_pDeletingIcon->Enable(m_pSave != NULL);
	m_pCopyingIcon->Enable(m_pSave != NULL);
	m_pExport->Enable(m_pSave != NULL);
	m_pDelete->Enable(m_pSave != NULL);
}

long CSaveView::OnSize(unsigned int nX, unsigned int nY, unsigned int nType)
{
	RefreshLayout();
	return TRUE;
}

long CSaveView::OnCommand(unsigned short nCmd, unsigned short nId, HWND hWndFrom)
{
	return m_CommandSink.OnCommand(nCmd, nId, hWndFrom);
}

void CSaveView::RefreshLayout()
{
	RECT rc;

	GetClientRect(&rc);

	SetRect(&rc, rc.left + 10, rc.top + 10, rc.right - 10, rc.bottom - 10);

	m_pLayout->SetRect(rc.left, rc.top, rc.right, rc.bottom);
	m_pLayout->RefreshGeometry();

	Redraw();
}

long CSaveView::SetIconType(ICONTYPE nIconType)
{
	m_nIconType = nIconType;

	m_pIconViewWnd->SetIconType(nIconType);

	m_pNormalIcon->SetCheck(m_nIconType == ICON_NORMAL);
	m_pDeletingIcon->SetCheck(m_nIconType == ICON_DELETING);
	m_pCopyingIcon->SetCheck(m_nIconType == ICON_COPYING);

	return FALSE;
}

long CSaveView::OpenSaveFolder()
{
	if(m_pSave == NULL) return FALSE;

	filesystem::path Path(filesystem::complete(m_pSave->GetPath()));

	ShellExecuteA(m_hWnd, "open", Path.string().c_str(), NULL, NULL, SW_SHOW);

	return FALSE;
}

long CSaveView::Export()
{
	if(m_pSave == NULL) return FALSE;

	unsigned int nRet;

	Win32::CFileDialog FileDialog;
	FileDialog.m_OFN.lpstrFilter = _T("EMS Memory Adapter Save Dumps (*.psu)\0*.psu\0");

	EnableWindow(GetParent(), FALSE);
	nRet = FileDialog.Summon(m_hWnd);
	EnableWindow(GetParent(), TRUE);

	if(nRet == 0) return FALSE;

	FILE* pStream;
	pStream = _tfopen(FileDialog.m_sFile, _T("wb"));

	if(pStream == NULL)
	{
		MessageBox(m_hWnd, _T("Couldn't open file for writing."), NULL, 16);
		return FALSE;
	}

	ofstream Output(pStream);

	try
	{
		CSaveExporter::ExportPSU(Output, m_pSave->GetPath().string().c_str());
		Output.close();
	}
	catch(const exception& Exception)
	{
		Output.close();

		string sMessage;
		sMessage  = "Couldn't export save:\r\n\r\n";
		sMessage += Exception.what();
		MessageBoxA(m_hWnd, sMessage.c_str(), NULL, 16);
		return FALSE;
	}

	MessageBox(m_hWnd, _T("Save exported successfully."), NULL, MB_ICONINFORMATION); 

	return FALSE;
}

long CSaveView::Delete()
{
	if(m_pSave == NULL) return FALSE;

	m_OnDeleteClicked(m_pSave);

	return FALSE;
}