#include "StdAfx.h"

typedef boost::shared_ptr<CAdvProviderSettings> TAdvSettingsPtr;
typedef std::map<const IQuotesProvider*, TAdvSettingsPtr> TAdvSettings;

TAdvSettings g_aAdvSettings;

CAdvProviderSettings* get_adv_settings(const IQuotesProvider* pProvider, bool bCreateIfNonExist)
{
	TAdvSettings::iterator i = g_aAdvSettings.find(pProvider);
	if (i != g_aAdvSettings.end())
		return i->second.get();

	if (true == bCreateIfNonExist) {
		TAdvSettingsPtr pAdvSet(new CAdvProviderSettings(pProvider));
		g_aAdvSettings.insert(std::make_pair(pProvider, pAdvSet));
		return pAdvSet.get();
	}

	return nullptr;
}

void remove_adv_settings(const IQuotesProvider *pProvider)
{
	TAdvSettings::iterator i = g_aAdvSettings.find(pProvider);
	if (i != g_aAdvSettings.end())
		g_aAdvSettings.erase(i);
}

void CommonOptionDlgProc(HWND hWnd, UINT msg, WPARAM wp, LPARAM lp, CCommonDlgProcData& rData)
{
	switch (msg) {
	case WM_INITDIALOG:
		{
			assert(rData.m_pQuotesProvider);

			CQuotesProviderVisitorDbSettings visitor;
			rData.m_pQuotesProvider->Accept(visitor);
			assert(visitor.m_pszDbRefreshRateType);
			assert(visitor.m_pszDbRefreshRateValue);
			assert(visitor.m_pszDbDisplayNameFormat);
			assert(visitor.m_pszDbStatusMsgFormat);
			assert(visitor.m_pszDbTendencyFormat);

			// set contact list display format
			tstring sDspNameFrmt = Quotes_DBGetStringT(NULL, QUOTES_MODULE_NAME, visitor.m_pszDbDisplayNameFormat, visitor.m_pszDefDisplayFormat);
			::SetDlgItemText(hWnd, IDC_EDIT_CONTACT_LIST_FORMAT, sDspNameFrmt.c_str());

			// set status message display format
			tstring sStatusMsgFrmt = Quotes_DBGetStringT(NULL, QUOTES_MODULE_NAME, visitor.m_pszDbStatusMsgFormat, visitor.m_pszDefStatusMsgFormat);
			::SetDlgItemText(hWnd, IDC_EDIT_STATUS_MESSAGE_FORMAT, sStatusMsgFrmt.c_str());

			// set tendency format
			tstring sTendencyFrmt = Quotes_DBGetStringT(NULL, QUOTES_MODULE_NAME, visitor.m_pszDbTendencyFormat, visitor.m_pszDefTendencyFormat);
			::SetDlgItemText(hWnd, IDC_EDIT_TENDENCY_FORMAT, sTendencyFrmt.c_str());

			// refresh rate
			HWND hwndCombo = ::GetDlgItem(hWnd, IDC_COMBO_REFRESH_RATE);
			LPCTSTR pszRefreshRateTypes[] = { TranslateT("Seconds"), TranslateT("Minutes"), TranslateT("Hours") };
			for (int i = 0; i < _countof(pszRefreshRateTypes); ++i)
				::SendMessage(hwndCombo, CB_ADDSTRING, 0, reinterpret_cast<LPARAM>(pszRefreshRateTypes[i]));

			int nRefreshRateType = db_get_w(NULL, QUOTES_MODULE_NAME, visitor.m_pszDbRefreshRateType, RRT_MINUTES);
			if (nRefreshRateType < RRT_SECONDS || nRefreshRateType > RRT_HOURS)
				nRefreshRateType = RRT_MINUTES;

			UINT nRate = db_get_w(NULL, QUOTES_MODULE_NAME, visitor.m_pszDbRefreshRateValue, 1);
			switch (nRefreshRateType) {
			default:
			case RRT_SECONDS:
			case RRT_MINUTES:
				if (nRate < 1 || nRate > 60)
					nRate = 1;

				spin_set_range(::GetDlgItem(hWnd, IDC_SPIN_REFRESH_RATE), 1, 60);
				break;
			case RRT_HOURS:
				if (nRate < 1 || nRate > 24)
					nRate = 1;

				spin_set_range(::GetDlgItem(hWnd, IDC_SPIN_REFRESH_RATE), 1, 24);
				break;
			}

			::SendMessage(hwndCombo, CB_SETCURSEL, nRefreshRateType, 0);
			::SetDlgItemInt(hWnd, IDC_EDIT_REFRESH_RATE, nRate, FALSE);

			PropSheet_UnChanged(::GetParent(hWnd), hWnd);
		}
		break;

	case WM_COMMAND:
		switch (HIWORD(wp)) {
		case CBN_SELCHANGE:
			if (IDC_COMBO_REFRESH_RATE == LOWORD(wp)) {
				ERefreshRateType nType = static_cast<ERefreshRateType>(::SendMessage(reinterpret_cast<HWND>(lp), CB_GETCURSEL, 0, 0));
				switch (nType) {
				default:
				case RRT_SECONDS:
				case RRT_MINUTES:
					spin_set_range(::GetDlgItem(hWnd, IDC_SPIN_REFRESH_RATE), 1, 60);
					break;
				case RRT_HOURS:
					spin_set_range(::GetDlgItem(hWnd, IDC_SPIN_REFRESH_RATE), 1, 24);
					BOOL bOk = FALSE;
					UINT nRefreshRate = ::GetDlgItemInt(hWnd, IDC_EDIT_REFRESH_RATE, &bOk, FALSE);
					if (TRUE == bOk && nRefreshRate > 24)
						::SetDlgItemInt(hWnd, IDC_EDIT_REFRESH_RATE, 24, FALSE);
					break;
				}

				PropSheet_Changed(::GetParent(hWnd), hWnd);
			}
			break;

		case EN_CHANGE:
			switch (LOWORD(wp)) {
			case IDC_EDIT_REFRESH_RATE:
			case IDC_EDIT_CONTACT_LIST_FORMAT:
			case IDC_EDIT_STATUS_MESSAGE_FORMAT:
			case IDC_EDIT_TENDENCY_FORMAT:
				if (reinterpret_cast<HWND>(lp) == ::GetFocus())
					PropSheet_Changed(::GetParent(hWnd), hWnd);
				break;
			}
			break;

		case BN_CLICKED:
			switch (LOWORD(wp)) {
			case IDC_BUTTON_DESCRIPTION:
				show_variable_list(hWnd, rData.m_pQuotesProvider);
				break;
			case IDC_BUTTON_ADVANCED_SETTINGS:
				CAdvProviderSettings* pAdvSet = get_adv_settings(rData.m_pQuotesProvider, true);
				assert(pAdvSet);
				if (true == ShowSettingsDlg(hWnd, pAdvSet))
					PropSheet_Changed(::GetParent(hWnd), hWnd);
				break;
			}
			break;
		}
		break;

	case WM_NOTIFY:
		{
			LPNMHDR pNMHDR = reinterpret_cast<LPNMHDR>(lp);
			switch (pNMHDR->code) {
			case PSN_KILLACTIVE:
				{
					BOOL bOk = FALSE;
					UINT nRefreshRate = ::GetDlgItemInt(hWnd, IDC_EDIT_REFRESH_RATE, &bOk, FALSE);
					ERefreshRateType nType = static_cast<ERefreshRateType>(::SendDlgItemMessage(hWnd, IDC_COMBO_REFRESH_RATE, CB_GETCURSEL, 0, 0));
					switch (nType) {
					default:
					case RRT_MINUTES:
					case RRT_SECONDS:
						if (FALSE == bOk || nRefreshRate < 1 || nRefreshRate > 60) {
							prepare_edit_ctrl_for_error(::GetDlgItem(hWnd, IDC_EDIT_REFRESH_RATE));
							Quotes_MessageBox(hWnd, TranslateT("Enter integer value between 1 and 60."), MB_OK | MB_ICONERROR);
							bOk = FALSE;
						}
						break;
					case RRT_HOURS:
						if (FALSE == bOk || nRefreshRate < 1 || nRefreshRate > 24) {
							prepare_edit_ctrl_for_error(::GetDlgItem(hWnd, IDC_EDIT_REFRESH_RATE));
							Quotes_MessageBox(hWnd, TranslateT("Enter integer value between 1 and 24."), MB_OK | MB_ICONERROR);
							bOk = FALSE;
						}
						break;
					}

					if (TRUE == bOk) {
						HWND hEdit = ::GetDlgItem(hWnd, IDC_EDIT_CONTACT_LIST_FORMAT);
						assert(IsWindow(hEdit));

						tstring s = get_window_text(hEdit);
						if (true == s.empty()) {
							prepare_edit_ctrl_for_error(hEdit);
							Quotes_MessageBox(hWnd, TranslateT("Enter text to display in contact list."), MB_OK | MB_ICONERROR);
							bOk = FALSE;
						}
					}

					::SetWindowLongPtr(hWnd, DWLP_MSGRESULT, (TRUE == bOk) ? FALSE : TRUE);
				}
				break;

			case PSN_APPLY:
				BOOL bOk = FALSE;
				UINT nRefreshRate = ::GetDlgItemInt(hWnd, IDC_EDIT_REFRESH_RATE, &bOk, FALSE);
				assert(TRUE == bOk);
				ERefreshRateType nType = static_cast<ERefreshRateType>(::SendDlgItemMessage(hWnd, IDC_COMBO_REFRESH_RATE, CB_GETCURSEL, 0, 0));

				assert(rData.m_pQuotesProvider);

				CQuotesProviderVisitorDbSettings visitor;
				rData.m_pQuotesProvider->Accept(visitor);
				assert(visitor.m_pszDbRefreshRateType);
				assert(visitor.m_pszDbRefreshRateValue);
				assert(visitor.m_pszDbDisplayNameFormat);
				assert(visitor.m_pszDbStatusMsgFormat);

				rData.m_bFireSetingsChangedEvent = true;
				db_set_w(NULL, QUOTES_MODULE_NAME, visitor.m_pszDbRefreshRateType, nType);
				db_set_w(NULL, QUOTES_MODULE_NAME, visitor.m_pszDbRefreshRateValue, nRefreshRate);

				tstring s = get_window_text(::GetDlgItem(hWnd, IDC_EDIT_CONTACT_LIST_FORMAT));
				db_set_ws(NULL, QUOTES_MODULE_NAME, visitor.m_pszDbDisplayNameFormat, s.c_str());

				s = get_window_text(::GetDlgItem(hWnd, IDC_EDIT_STATUS_MESSAGE_FORMAT));
				db_set_ws(NULL, QUOTES_MODULE_NAME, visitor.m_pszDbStatusMsgFormat, s.c_str());

				s = get_window_text(::GetDlgItem(hWnd, IDC_EDIT_TENDENCY_FORMAT));
				db_set_ws(NULL, QUOTES_MODULE_NAME, visitor.m_pszDbTendencyFormat, s.c_str());

				CAdvProviderSettings* pAdvSet = get_adv_settings(rData.m_pQuotesProvider, false);
				if (pAdvSet)
					pAdvSet->SaveToDb();
				break;
			}
		}
		break;

	case WM_DESTROY:
		remove_adv_settings(rData.m_pQuotesProvider);
		break;
	}
}