#pragma once

#include "../pch/pch.h"

#include "CWaimaiOrderItemUI.h"
#include "OrderListUI.h"
#include "CWaimaiOrderInfoUI.h"

class CDialogBuilderCallbackEx : public IDialogBuilderCallback
{
public:
	CControlUI* CreateControl(LPCTSTR pstrClass)
	{
		if (_tcscmp(pstrClass, _T("DeskList")) == 0)
		{
			return new OrderListUI;
		}
		else if (_tcscmp(pstrClass, _T("WaimaiOrderItem")) == 0)
		{
			return new CWaimaiOrderItemUI;
		}
		else if (_tcscmp(pstrClass, _T("WaimaiOrderInfo")) == 0)
		{
			return new CWaimaiOrderInfoUI;
		}

		return NULL;
	}
};