#include "StdAfx.h"
#include "UndoRedoMenuHandler.h"
#include "ModelController.h"
#include "UndoRedoMgr.h"

namespace SbjCore
{
	namespace Mvc
	{
		namespace Ribbon
		{
			LRESULT UndoRedoMenuHandler::OnHandleWndMsg(WPARAM wParam, LPARAM lParam, LRESULT* pResult)
			{
				wParam;
				*pResult = 0;
				LRESULT lRslt = 1;

				SbjCore::Mvc::Model::Controller* pCtrlr = SbjCore::Mvc::Model::GetCurController();
				SbjCore::UndoRedo::Manager* pMgr = pCtrlr->GetUndoRedoMgr();

				if (pMgr != NULL)
				{
					CMFCRibbonBaseElement* pElem = (CMFCRibbonBaseElement*) lParam;
					ASSERT_VALID(pElem);

					if (pElem->GetID() == ID_EDIT_UNDO)
					{
						CMFCRibbonUndoButton* pUndo = dynamic_cast<CMFCRibbonUndoButton*>(pElem);
						ASSERT_VALID(pUndo);

						pMgr->SetUndoButton(pUndo);

						pUndo->CleanUpUndoList();

						CStringList& sUndoList = pMgr->GetUndoList();
						for (POSITION pos = sUndoList.GetHeadPosition (); pos != NULL;)
						{
							pUndo->AddUndoAction(sUndoList.GetNext(pos));
						}
					}
					else if (pElem->GetID() == ID_EDIT_REDO)
					{
						CMFCRibbonUndoButton* pUndo = dynamic_cast<CMFCRibbonUndoButton*>(pElem);
						ASSERT_VALID(pUndo);

						pMgr->SetRedoButton(pUndo);

						pUndo->CleanUpUndoList();

						CStringList& sRedoList = pMgr->GetRedoList();
						for (POSITION pos = sRedoList.GetHeadPosition (); pos != NULL;)
						{
							pUndo->AddUndoAction(sRedoList.GetNext(pos));
						}
					}
				}

				return lRslt;
			}
		}
	}
}