#include "StdAfx.h"

CQuotesProviderGoogle::CQuotesProviderGoogle()
{
}

CQuotesProviderGoogle::~CQuotesProviderGoogle()
{
}

inline tstring make_contact_name(const tstring& rsSymbolFrom, const tstring& rsSymbolTo)
{
	tostringstream o;
	o << rsSymbolFrom << L"/" << rsSymbolTo;
	return o.str();
}

inline bool is_rate_watched(MCONTACT hContact,
	const CQuotesProviderBase::CQuote& from,
	const CQuotesProviderBase::CQuote& to)
{
	tstring sFrom = Quotes_DBGetStringT(hContact, QUOTES_PROTOCOL_NAME, DB_STR_FROM_ID);
	tstring sTo = Quotes_DBGetStringT(hContact, QUOTES_PROTOCOL_NAME, DB_STR_TO_ID);
	return ((0 == mir_tstrcmpi(from.GetID().c_str(), sFrom.c_str()))
		&& (0 == mir_tstrcmpi(to.GetID().c_str(), sTo.c_str())));
}

bool CQuotesProviderGoogle::WatchForRate(const CRateInfo& ri,
	bool bWatch)
{
	TContracts::const_iterator i = std::find_if(m_aContacts.begin(), m_aContacts.end(),
		boost::bind(is_rate_watched, _1, ri.m_from, ri.m_to));
	if ((true == bWatch) && (i == m_aContacts.end())) {
		tstring sName = make_contact_name(ri.m_from.GetSymbol(), ri.m_to.GetSymbol());
		MCONTACT hContact = CreateNewContact(sName);
		if (hContact) {
			db_set_ts(hContact, QUOTES_PROTOCOL_NAME, DB_STR_FROM_ID, ri.m_from.GetID().c_str());
			db_set_ts(hContact, QUOTES_PROTOCOL_NAME, DB_STR_TO_ID, ri.m_to.GetID().c_str());
			if (false == ri.m_from.GetName().empty()) {
				db_set_ts(hContact, QUOTES_PROTOCOL_NAME, DB_STR_FROM_DESCRIPTION, ri.m_from.GetName().c_str());
			}
			if (false == ri.m_to.GetName().empty()) {
				db_set_ts(hContact, QUOTES_PROTOCOL_NAME, DB_STR_TO_DESCRIPTION, ri.m_to.GetName().c_str());
			}

			return true;
		}
	}
	else if ((false == bWatch) && (i != m_aContacts.end())) {
		MCONTACT hContact = *i;
		{// for CCritSection
			mir_cslock lck(m_cs);
			m_aContacts.erase(i);
		}

		CallService(MS_DB_CONTACT_DELETE, WPARAM(hContact), 0);
		return true;
	}

	return false;
}

size_t CQuotesProviderGoogle::GetWatchedRateCount()const
{
	return m_aContacts.size();
}

bool CQuotesProviderGoogle::GetWatchedRateInfo(size_t nIndex, CRateInfo& rRateInfo)
{
	if (nIndex < m_aContacts.size()) {
		MCONTACT hContact = m_aContacts[nIndex];
		tstring sSymbolFrom = Quotes_DBGetStringT(hContact, QUOTES_PROTOCOL_NAME, DB_STR_FROM_ID);
		tstring sSymbolTo = Quotes_DBGetStringT(hContact, QUOTES_PROTOCOL_NAME, DB_STR_TO_ID);
		tstring sDescFrom = Quotes_DBGetStringT(hContact, QUOTES_PROTOCOL_NAME, DB_STR_FROM_DESCRIPTION);
		tstring sDescTo = Quotes_DBGetStringT(hContact, QUOTES_PROTOCOL_NAME, DB_STR_TO_DESCRIPTION);

		rRateInfo.m_from = CQuote(sSymbolFrom, sSymbolFrom, sDescFrom);
		rRateInfo.m_to = CQuote(sSymbolTo, sSymbolTo, sDescTo);
		return true;
	}
	else {
		return false;
	}
}

static tstring build_url(const tstring& rsURL, const tstring& from, const tstring& to, double dAmount)
{
	tostringstream o;
	o << rsURL << L"?a=" << std::fixed << dAmount << L"&from=" << from << L"&to=" << to;
	return o.str();
}

static tstring build_url(MCONTACT hContact, const tstring& rsURL, double dAmount = 1.0)
{
	tstring sFrom = Quotes_DBGetStringT(hContact, QUOTES_PROTOCOL_NAME, DB_STR_FROM_ID);
	tstring sTo = Quotes_DBGetStringT(hContact, QUOTES_PROTOCOL_NAME, DB_STR_TO_ID);
	return build_url(rsURL, sFrom, sTo, dAmount);
}

typedef IHTMLNode::THTMLNodePtr THTMLNodePtr;

bool parse_html_node(const THTMLNodePtr& pNode, double& rdRate)
{
	tstring sID = pNode->GetAttribute(L"id");
	if ((false == sID.empty()) && (0 == mir_tstrcmpi(sID.c_str(), L"currency_converter_result"))) {
		size_t cChild = pNode->GetChildCount();
		// 			assert(1 == cChild);
		if (cChild > 0) {
			THTMLNodePtr pChild = pNode->GetChildPtr(0);
			tstring sRate = pChild->GetText();

			tistringstream input(sRate);
			input >> rdRate;

			return ((false == input.bad()) && (false == input.fail()));
		}
	}
	else {
		size_t cChild = pNode->GetChildCount();
		for (size_t i = 0; i < cChild; ++i) {
			THTMLNodePtr pChild = pNode->GetChildPtr(i);
			if (pChild && (true == parse_html_node(pChild, rdRate))) {
				return true;
			}
		}
	}

	return false;
}

bool parse_responce(const tstring& rsHTML, double& rdRate)
{
	IHTMLEngine::THTMLParserPtr pHTMLParser = CModuleInfo::GetHTMLEngine()->GetParserPtr();
	THTMLNodePtr pRoot = pHTMLParser->ParseString(rsHTML);
	if (pRoot) {
		return parse_html_node(pRoot, rdRate);
	}
	else {
		return false;
	}
}

void CQuotesProviderGoogle::RefreshQuotes(TContracts& anContacts)
{
	CHTTPSession http;
	tstring sURL = GetURL();

	bool bUseExtendedStatus = CModuleInfo::GetInstance().GetExtendedStatusFlag();

	for (TContracts::const_iterator i = anContacts.begin(); i != anContacts.end() && IsOnline(); ++i) {
		MCONTACT hContact = *i;

		if (bUseExtendedStatus) {
			SetContactStatus(hContact, ID_STATUS_OCCUPIED);
		}

		tstring sFullURL = build_url(hContact, sURL);
		// 		LogIt(Info,sFullURL);
		if ((true == http.OpenURL(sFullURL)) && (true == IsOnline())) {
			tstring sHTML;
			if ((true == http.ReadResponce(sHTML)) && (true == IsOnline())) {
				double dRate = 0.0;
				if ((true == parse_responce(sHTML, dRate)) && (true == IsOnline())) {
					WriteContactRate(hContact, dRate);
					continue;
				}
			}
		}

		SetContactStatus(hContact, ID_STATUS_NA);
	}
}

inline tstring make_quote_name(const CQuotesProviderGoogle::CQuote& rQuote)
{
	const tstring& rsDesc = rQuote.GetName();
	return((false == rsDesc.empty()) ? rsDesc : rQuote.GetSymbol());
}

static CQuotesProviderGoogle* get_google_provider()
{
	CModuleInfo::TQuotesProvidersPtr pProviders = CModuleInfo::GetQuoteProvidersPtr();
	const CQuotesProviders::TQuotesProviders& rapQuotesProviders = pProviders->GetProviders();
	for (CQuotesProviders::TQuotesProviders::const_iterator i = rapQuotesProviders.begin(); i != rapQuotesProviders.end(); ++i) {
		const CQuotesProviders::TQuotesProviderPtr& pProvider = *i;
		CQuotesProviderGoogle* pGoogle = dynamic_cast<CQuotesProviderGoogle*>(pProvider.get());
		if (pGoogle)
			return pGoogle;
	}

	assert(!"We should never get here!");
	return NULL;
}

CQuotesProviderGoogle::CQuoteSection get_quotes()
{
	const CQuotesProviderGoogle* pProvider = get_google_provider();
	if (pProvider) {
		const CQuotesProviderGoogle::CQuoteSection& rQuotes = pProvider->GetQuotes();
		if (rQuotes.GetSectionCount() > 0)
			return rQuotes.GetSection(0);
	}

	return CQuotesProviderGoogle::CQuoteSection();
}

tstring make_rate_name(const CQuotesProviderGoogle::CQuote& rFrom,
	const CQuotesProviderGoogle::CQuote& rTo)
{
	if ((false == rFrom.GetName().empty()) && (false == rTo.GetName().empty()))
		return make_contact_name(rFrom.GetName(), rTo.GetName());

	return make_contact_name(rFrom.GetSymbol(), rTo.GetSymbol());
}

typedef std::vector<CQuotesProviderGoogle::CRateInfo> TWatchedRates;
TWatchedRates g_aWatchedRates;

bool is_equal_rate(const CQuotesProviderGoogle::CRateInfo& riL, const CQuotesProviderGoogle::CRateInfo& riR)
{
	return ((0 == mir_tstrcmpi(riL.m_from.GetID().c_str(), riR.m_from.GetID().c_str()))
		&& ((0 == mir_tstrcmpi(riL.m_to.GetID().c_str(), riR.m_to.GetID().c_str()))));
}

INT_PTR CALLBACK GoogleOptDlgProc(HWND hdlg, UINT message, WPARAM wParam, LPARAM lParam)
{
	CQuotesProviderGoogle* pProvider = get_google_provider();

	CCommonDlgProcData d(pProvider);
	CommonOptionDlgProc(hdlg, message, wParam, lParam, d);

	switch (message) {
	case WM_NOTIFY:
		{
			LPNMHDR pNMHDR = reinterpret_cast<LPNMHDR>(lParam);
			switch (pNMHDR->code) {
			case PSN_APPLY:
				{
					if (pProvider) {
						TWatchedRates aTemp(g_aWatchedRates);
						TWatchedRates aRemove;
						size_t cWatchedRates = pProvider->GetWatchedRateCount();
						for (size_t i = 0; i < cWatchedRates; ++i) {
							CQuotesProviderGoogle::CRateInfo ri;
							if (true == pProvider->GetWatchedRateInfo(i, ri)) {
								TWatchedRates::iterator it =
									std::find_if(aTemp.begin(), aTemp.end(),
										boost::bind(is_equal_rate, _1, boost::cref(ri)));
								if (it == aTemp.end()) {
									aRemove.push_back(ri);
								}
								else {
									aTemp.erase(it);
								}
							}
						}

						std::for_each(aRemove.begin(), aRemove.end(), boost::bind(&CQuotesProviderGoogle::WatchForRate, pProvider, _1, false));
						std::for_each(aTemp.begin(), aTemp.end(), boost::bind(&CQuotesProviderGoogle::WatchForRate, pProvider, _1, true));

						pProvider->RefreshSettings();
					}
				}
				break;
			}
		}
		break;

	case WM_INITDIALOG:
		TranslateDialogDefault(hdlg);
		{
			g_aWatchedRates.clear();

			HWND hcbxFrom = ::GetDlgItem(hdlg, IDC_COMBO_CONVERT_FROM);
			HWND hcbxTo = ::GetDlgItem(hdlg, IDC_COMBO_CONVERT_INTO);

			const CQuotesProviderGoogle::CQuoteSection& rSection = get_quotes();
			size_t cQuotes = rSection.GetQuoteCount();
			for (size_t i = 0; i < cQuotes; ++i) {
				const CQuotesProviderGoogle::CQuote& rQuote = rSection.GetQuote(i);
				tstring sName = make_quote_name(rQuote);
				LPCTSTR pszName = sName.c_str();
				::SendMessage(hcbxFrom, CB_ADDSTRING, 0, reinterpret_cast<LPARAM>(pszName));
				::SendMessage(hcbxTo, CB_ADDSTRING, 0, reinterpret_cast<LPARAM>(pszName));
			}

			pProvider = get_google_provider();
			if (pProvider) {
				size_t cWatchedRates = pProvider->GetWatchedRateCount();
				for (size_t i = 0; i < cWatchedRates; ++i) {
					CQuotesProviderGoogle::CRateInfo ri;
					if (true == pProvider->GetWatchedRateInfo(i, ri)) {
						g_aWatchedRates.push_back(ri);
						tstring sRate = make_rate_name(ri.m_from, ri.m_to);
						LPCTSTR pszRateName = sRate.c_str();
						::SendDlgItemMessage(hdlg, IDC_LIST_RATES, LB_ADDSTRING, 0, reinterpret_cast<LPARAM>(pszRateName));
					}
				}
			}

			::EnableWindow(::GetDlgItem(hdlg, IDC_BUTTON_ADD), FALSE);
			::EnableWindow(::GetDlgItem(hdlg, IDC_BUTTON_REMOVE), FALSE);
		}
		return TRUE;

	case WM_COMMAND:
		switch (HIWORD(wParam)) {
		case CBN_SELCHANGE:
			switch (LOWORD(wParam)) {
			case IDC_COMBO_REFRESH_RATE:
				break;
			case IDC_COMBO_CONVERT_FROM:
			case IDC_COMBO_CONVERT_INTO:
				{
					int nFrom = static_cast<int>(::SendDlgItemMessage(hdlg, IDC_COMBO_CONVERT_FROM, CB_GETCURSEL, 0, 0));
					int nTo = static_cast<int>(::SendDlgItemMessage(hdlg, IDC_COMBO_CONVERT_INTO, CB_GETCURSEL, 0, 0));
					bool bEnableAddButton = ((CB_ERR != nFrom) && (CB_ERR != nTo) && (nFrom != nTo));
					EnableWindow(GetDlgItem(hdlg, IDC_BUTTON_ADD), bEnableAddButton);
				}
				break;
			case IDC_LIST_RATES:
				{
					int nSel = ::SendDlgItemMessage(hdlg, IDC_LIST_RATES, LB_GETCURSEL, 0, 0);
					::EnableWindow(::GetDlgItem(hdlg, IDC_BUTTON_REMOVE), (LB_ERR != nSel));
				}
				break;
			}
			break;

		case BN_CLICKED:
			switch (LOWORD(wParam)) {
			case IDC_BUTTON_ADD:
				{
					size_t nFrom = static_cast<size_t>(::SendDlgItemMessage(hdlg, IDC_COMBO_CONVERT_FROM, CB_GETCURSEL, 0, 0));
					size_t nTo = static_cast<size_t>(::SendDlgItemMessage(hdlg, IDC_COMBO_CONVERT_INTO, CB_GETCURSEL, 0, 0));
					if ((CB_ERR != nFrom) && (CB_ERR != nTo) && (nFrom != nTo)) {
						const CQuotesProviderGoogle::CQuoteSection& rSection = get_quotes();
						size_t cQuotes = rSection.GetQuoteCount();
						if ((nFrom < cQuotes) && (nTo < cQuotes)) {
							CQuotesProviderGoogle::CRateInfo ri;
							ri.m_from = rSection.GetQuote(nFrom);
							ri.m_to = rSection.GetQuote(nTo);

							g_aWatchedRates.push_back(ri);

							tstring sRate = make_rate_name(ri.m_from, ri.m_to);
							LPCTSTR pszRateName = sRate.c_str();
							::SendDlgItemMessage(hdlg, IDC_LIST_RATES, LB_ADDSTRING, 0, reinterpret_cast<LPARAM>(pszRateName));
							PropSheet_Changed(::GetParent(hdlg), hdlg);
						}
					}
				}
				break;

			case IDC_BUTTON_REMOVE:
				HWND hWnd = ::GetDlgItem(hdlg, IDC_LIST_RATES);
				int nSel = ::SendMessage(hWnd, LB_GETCURSEL, 0, 0);
				if (LB_ERR != nSel) {
					if ((LB_ERR != ::SendMessage(hWnd, LB_DELETESTRING, nSel, 0))
						&& (nSel < static_cast<int>(g_aWatchedRates.size()))) {

						TWatchedRates::iterator i = g_aWatchedRates.begin();
						std::advance(i, nSel);
						g_aWatchedRates.erase(i);
						PropSheet_Changed(::GetParent(hdlg), hdlg);
					}
				}

				nSel = ::SendMessage(hWnd, LB_GETCURSEL, 0, 0);
				::EnableWindow(::GetDlgItem(hdlg, IDC_BUTTON_REMOVE), (LB_ERR != nSel));
				break;
			}
			break;
		}
		break;
	}

	return FALSE;
}

void CQuotesProviderGoogle::ShowPropertyPage(WPARAM wp, OPTIONSDIALOGPAGE &odp)
{
	odp.pszTemplate = MAKEINTRESOURCEA(IDD_DIALOG_OPT_GOOGLE);
	odp.pfnDlgProc = GoogleOptDlgProc;
	odp.ptszTab = const_cast<LPTSTR>(GetInfo().m_sName.c_str());
	Options_AddPage(wp, &odp);
}

void CQuotesProviderGoogle::Accept(CQuotesProviderVisitor& visitor)const
{
	CQuotesProviderBase::Accept(visitor);
	visitor.Visit(*this);
}

double CQuotesProviderGoogle::Convert(double dAmount, const CQuote& from, const CQuote& to)const
{
	tstring sFullURL = build_url(GetURL(), from.GetID(), to.GetID(), dAmount);

	CHTTPSession http;
	if ((true == http.OpenURL(sFullURL))) {
		tstring sHTML;
		if ((true == http.ReadResponce(sHTML))) {
			double dResult = 0.0;
			if ((true == parse_responce(sHTML, dResult)))
				return dResult;

			throw std::runtime_error(Translate("Error occurred during html parsing."));
		}
		else throw std::runtime_error(Translate("Error occurred during site access."));
	}
	else throw std::runtime_error(Translate("Error occurred during site access."));

	return 0.0;
}

namespace
{
	bool is_equal_ids(MCONTACT hContact, const tstring& rsFromID, const tstring& rsToID)
	{
		tstring sFrom = Quotes_DBGetStringT(hContact, QUOTES_PROTOCOL_NAME, DB_STR_FROM_ID);
		tstring sTo = Quotes_DBGetStringT(hContact, QUOTES_PROTOCOL_NAME, DB_STR_TO_ID);
		return ((0 == mir_tstrcmpi(rsFromID.c_str(), sFrom.c_str()))
			&& (0 == mir_tstrcmpi(rsToID.c_str(), sTo.c_str())));
	}
}

MCONTACT CQuotesProviderGoogle::GetContactByID(const tstring& rsFromID, const tstring& rsToID)const
{
	mir_cslock lck(m_cs);

	TContracts::const_iterator i = std::find_if(m_aContacts.begin(), m_aContacts.end(),
		boost::bind(is_equal_ids, _1, boost::cref(rsFromID), boost::cref(rsToID)));
	if (i != m_aContacts.end())
		return *i;

	return NULL;
}