#include "StdAfx.h" LPCTSTR g_pszXmlValue = L"Value"; LPCTSTR g_pszXmlName = L"Name"; LPCTSTR g_pszXmlSetting = L"Setting"; LPCTSTR g_pszXmlModule = L"Module"; LPCTSTR g_pszXmlContact = L"Contact"; LPCTSTR g_pszXmlContacts = L"Contacts"; LPCTSTR g_pszXmlType = L"type"; LPCTSTR g_pszXmlTypeByte = L"byte"; LPCTSTR g_pszXmlTypeWord = L"word"; LPCTSTR g_pszXmlTypeDword = L"dword"; LPCTSTR g_pszXmlTypeAsciiz = L"asciiz"; LPCTSTR g_pszXmlTypeWchar = L"wchar"; LPCTSTR g_pszXmlTypeUtf8 = L"utf8"; LPCTSTR g_pszXmlTypeBlob = L"blob"; struct CEnumContext { CModuleInfo::TXMLEnginePtr m_pXmlEngine; IXMLNode::TXMLNodePtr m_pNode; MCONTACT m_hContact; LPCSTR m_pszModule; }; struct mir_safety_dbvar { mir_safety_dbvar(DBVARIANT* p) : m_p(p) {} ~mir_safety_dbvar() { db_free(m_p); } DBVARIANT* m_p; }; static int enum_contact_settings(const char* szSetting, LPARAM lp) { CEnumContext* ctx = reinterpret_cast(lp); DBVARIANT dbv; if (0 == db_get(ctx->m_hContact, ctx->m_pszModule, szSetting, &dbv)) { mir_safety_dbvar sdbvar(&dbv); tstring sType; tostringstream sValue; sValue.imbue(GetSystemLocale()); switch (dbv.type) { case DBVT_BYTE: sValue << dbv.bVal; sType = g_pszXmlTypeByte; break; case DBVT_WORD: sValue << dbv.wVal; sType = g_pszXmlTypeWord; break; case DBVT_DWORD: sValue << dbv.dVal; sType = g_pszXmlTypeDword; break; case DBVT_ASCIIZ: sType = g_pszXmlTypeAsciiz; if (dbv.pszVal) sValue << dbv.pszVal; break; case DBVT_WCHAR: sType = g_pszXmlTypeWchar; if (dbv.pwszVal) sValue << dbv.pwszVal; break; case DBVT_UTF8: sType = g_pszXmlTypeUtf8; if (dbv.pszVal) sValue << dbv.pszVal; break; case DBVT_BLOB: sType = g_pszXmlTypeBlob; if (dbv.pbVal) { ptrA buf(mir_base64_encode(dbv.pbVal, dbv.cpbVal)); if (buf) sValue << buf; } break; } IXMLNode::TXMLNodePtr pXmlSet = ctx->m_pXmlEngine->CreateNode(g_pszXmlSetting, tstring()); if (pXmlSet) { IXMLNode::TXMLNodePtr pXmlName = ctx->m_pXmlEngine->CreateNode(g_pszXmlName, quotes_a2t(szSetting)); IXMLNode::TXMLNodePtr pXmlValue = ctx->m_pXmlEngine->CreateNode(g_pszXmlValue, sValue.str()); if (pXmlName && pXmlValue) { pXmlValue->AddAttribute(g_pszXmlType, sType); pXmlSet->AddChild(pXmlName); pXmlSet->AddChild(pXmlValue); ctx->m_pNode->AddChild(pXmlSet); } } } return 0; } int EnumDbModules(const char *szModuleName, DWORD, LPARAM lp) { CEnumContext *ctx = (CEnumContext*)lp; IXMLNode::TXMLNodePtr pXml = ctx->m_pNode; IXMLNode::TXMLNodePtr pModule = ctx->m_pXmlEngine->CreateNode(g_pszXmlModule, quotes_a2t(szModuleName)/*A2CT(szModuleName)*/); if (pModule) { ctx->m_pszModule = szModuleName; ctx->m_pNode = pModule; DBCONTACTENUMSETTINGS dbces; dbces.pfnEnumProc = &enum_contact_settings; dbces.szModule = szModuleName; dbces.lParam = reinterpret_cast(ctx); CallService(MS_DB_CONTACT_ENUMSETTINGS, WPARAM(ctx->m_hContact), reinterpret_cast(&dbces)); if (pModule->GetChildCount() > 0) pXml->AddChild(pModule); ctx->m_pNode = pXml; } return 0; } IXMLNode::TXMLNodePtr export_contact(MCONTACT hContact, const CModuleInfo::TXMLEnginePtr& pXmlEngine) { IXMLNode::TXMLNodePtr pNode = pXmlEngine->CreateNode(g_pszXmlContact, tstring()); if (pNode) { CEnumContext ctx; ctx.m_pXmlEngine = pXmlEngine; ctx.m_pNode = pNode; ctx.m_hContact = hContact; CallService(MS_DB_MODULES_ENUM, (WPARAM)&ctx, (LPARAM)EnumDbModules); } return pNode; } LPCTSTR prepare_filter(LPTSTR pszBuffer, size_t cBuffer) { LPTSTR p = pszBuffer; LPCTSTR pszXml = TranslateT("XML File (*.xml)"); mir_tstrncpy(p, pszXml, (int)cBuffer); size_t nLen = mir_tstrlen(pszXml) + 1; p += nLen; if (nLen < cBuffer) { mir_tstrncpy(p, L"*.xml", (int)(cBuffer - nLen)); p += 6; nLen += 6; } if (nLen < cBuffer) { LPCTSTR pszAll = TranslateT("All files (*.*)"); mir_tstrncpy(p, pszAll, (int)(cBuffer - nLen)); size_t n = mir_tstrlen(pszAll) + 1; nLen += n; p += n; } if (nLen < cBuffer) { mir_tstrncpy(p, L"*.*", (int)(cBuffer - nLen)); p += 4; nLen += 4; } if (nLen < cBuffer) *p = '\0'; return pszBuffer; } bool show_open_file_dialog(bool bOpen, tstring& rsFile) { wchar_t szBuffer[MAX_PATH]; wchar_t szFilter[MAX_PATH]; OPENFILENAME ofn; memset(&ofn, 0, sizeof(ofn)); ofn.lStructSize = sizeof(OPENFILENAME); ofn.hwndOwner = NULL; ofn.lpstrFilter = prepare_filter(szFilter, MAX_PATH); ofn.Flags = OFN_PATHMUSTEXIST | OFN_HIDEREADONLY | OFN_EXPLORER; ofn.lpstrDefExt = L"xml"; if (bOpen) ofn.Flags |= OFN_FILEMUSTEXIST; else ofn.Flags |= OFN_OVERWRITEPROMPT; ofn.nMaxFile = MAX_PATH; ofn.lpstrFile = szBuffer; ofn.lpstrFile[0] = '\0'; if (bOpen) { if (FALSE == GetOpenFileName(&ofn)) return false; } else { if (FALSE == GetSaveFileName(&ofn)) return false; } rsFile = szBuffer; return true; } INT_PTR Quotes_Export(WPARAM wp, LPARAM lp) { tstring sFileName; const char* pszFile = reinterpret_cast(lp); if (NULL == pszFile) { if (false == show_open_file_dialog(false, sFileName)) return -1; } else sFileName = quotes_a2t(pszFile);//A2CT(pszFile); CModuleInfo::TXMLEnginePtr pXmlEngine = CModuleInfo::GetInstance().GetXMLEnginePtr(); CModuleInfo::TQuotesProvidersPtr pProviders = CModuleInfo::GetInstance().GetQuoteProvidersPtr(); IXMLNode::TXMLNodePtr pRoot = pXmlEngine->CreateNode(g_pszXmlContacts, tstring()); MCONTACT hContact = MCONTACT(wp); if (hContact) { CQuotesProviders::TQuotesProviderPtr pProvider = pProviders->GetContactProviderPtr(hContact); if (pProvider) { IXMLNode::TXMLNodePtr pNode = export_contact(hContact, pXmlEngine); if (pNode) pRoot->AddChild(pNode); } } else { for (hContact = db_find_first(QUOTES_MODULE_NAME); hContact; hContact = db_find_next(hContact, QUOTES_MODULE_NAME)) { CQuotesProviders::TQuotesProviderPtr pProvider = pProviders->GetContactProviderPtr(hContact); if (pProvider) { IXMLNode::TXMLNodePtr pNode = export_contact(hContact, pXmlEngine); if (pNode) pRoot->AddChild(pNode); } } } return ((true == pXmlEngine->SaveFile(sFileName, pRoot)) ? 0 : 1); } bool set_contact_settings(MCONTACT hContact, DBCONTACTWRITESETTING& dbs) { assert(DBVT_DELETED != dbs.value.type); return (0 == db_set(hContact, dbs.szModule, dbs.szSetting, &dbs.value)); } bool handle_module(MCONTACT hContact, const IXMLNode::TXMLNodePtr& pXmlModule) { size_t cCreatedRecords = 0; tstring sModuleName = pXmlModule->GetText(); if (false == sModuleName.empty()) { DBCONTACTWRITESETTING dbs; std::string s = quotes_t2a(sModuleName.c_str()); dbs.szModule = s.c_str();//T2CA(sModuleName.c_str()); bool bCListModule = 0 == mir_tstrcmpi(sModuleName.c_str(), L"CList"); size_t cChild = pXmlModule->GetChildCount(); for (size_t i = 0; i < cChild; ++i) { IXMLNode::TXMLNodePtr pSetting = pXmlModule->GetChildNode(i); tstring sSetting = pSetting->GetName(); if (0 == mir_tstrcmpi(g_pszXmlSetting, sSetting.c_str())) { size_t cSetChild = pSetting->GetChildCount(); if (cSetChild >= 2) { tstring sName; tstring sValue; tstring sType; for (size_t j = 0; j < cSetChild; ++j) { IXMLNode::TXMLNodePtr pNode = pSetting->GetChildNode(j); tstring sNode = pNode->GetName(); if (0 == mir_tstrcmpi(g_pszXmlName, sNode.c_str())) { sName = pNode->GetText(); } else if (0 == mir_tstrcmpi(g_pszXmlValue, sNode.c_str())) { sValue = pNode->GetText(); sType = pNode->GetAttributeValue(g_pszXmlType); } } if ((false == sName.empty()) && (false == sType.empty())) { std::string s1 = quotes_t2a(sName.c_str()); dbs.szSetting = s1.c_str(); if (0 == mir_tstrcmpi(g_pszXmlTypeByte, sType.c_str())) { tistringstream in(sValue.c_str()); in.imbue(GetSystemLocale()); dbs.value.cVal = in.get(); if (in.good() && in.eof()) { dbs.value.type = DBVT_BYTE; if (set_contact_settings(hContact, dbs)) ++cCreatedRecords; } } else if (0 == mir_tstrcmpi(g_pszXmlTypeWord, sType.c_str())) { tistringstream in(sValue.c_str()); in.imbue(GetSystemLocale()); in >> dbs.value.wVal; if (in.good() || in.eof()) { dbs.value.type = DBVT_WORD; if (set_contact_settings(hContact, dbs)) ++cCreatedRecords; } } else if (0 == mir_tstrcmpi(g_pszXmlTypeDword, sType.c_str())) { tistringstream in(sValue.c_str()); in.imbue(GetSystemLocale()); in >> dbs.value.dVal; if (in.good() || in.eof()) { dbs.value.type = DBVT_DWORD; if (set_contact_settings(hContact, dbs)) ++cCreatedRecords; } } else if (0 == mir_tstrcmpi(g_pszXmlTypeAsciiz, sType.c_str())) { ptrA v(mir_t2a(sValue.c_str())); dbs.value.pszVal = v; dbs.value.type = DBVT_ASCIIZ; if (set_contact_settings(hContact, dbs)) ++cCreatedRecords; } else if (0 == mir_tstrcmpi(g_pszXmlTypeUtf8, sType.c_str())) { T2Utf szValue(sValue.c_str()); dbs.value.pszVal = szValue; dbs.value.type = DBVT_UTF8; if (set_contact_settings(hContact, dbs)) ++cCreatedRecords; } else if (0 == mir_tstrcmpi(g_pszXmlTypeWchar, sType.c_str())) { ptrW val(mir_t2u(sValue.c_str())); dbs.value.pwszVal = val; dbs.value.type = DBVT_WCHAR; if (set_contact_settings(hContact, dbs)) ++cCreatedRecords; mir_free(dbs.value.pwszVal); } else if (0 == mir_tstrcmpi(g_pszXmlTypeBlob, sType.c_str())) { unsigned bufLen; mir_ptr buf((PBYTE)mir_base64_decode(_T2A(sValue.c_str()), &bufLen)); if (buf) { dbs.value.pbVal = buf; dbs.value.cpbVal = (WORD)bufLen; dbs.value.type = DBVT_BLOB; if (set_contact_settings(hContact, dbs)) ++cCreatedRecords; } } if ((true == bCListModule) && (0 == mir_tstrcmpi(sName.c_str(), L"Group"))) Clist_GroupCreate(NULL, sValue.c_str()); } } } } } return true; } size_t count_contacts(const IXMLNode::TXMLNodePtr& pXmlRoot, bool bInContactsGroup) { size_t cContacts = 0; size_t cChild = pXmlRoot->GetChildCount(); for (size_t i = 0; i < cChild; ++i) { IXMLNode::TXMLNodePtr pNode = pXmlRoot->GetChildNode(i); tstring sName = pNode->GetName(); if (false == bInContactsGroup) { if (0 == mir_tstrcmpi(g_pszXmlContacts, sName.c_str())) cContacts += count_contacts(pNode, true); else cContacts += count_contacts(pNode, false); } else { if (0 == mir_tstrcmpi(g_pszXmlContact, sName.c_str())) ++cContacts; } } return cContacts; } struct CImportContext { CImportContext(size_t cTotalContacts) : m_cTotalContacts(cTotalContacts), m_cHandledContacts(0), m_nFlags(0) {} size_t m_cTotalContacts; size_t m_cHandledContacts; UINT m_nFlags; }; struct CContactState { CContactState() : m_hContact(NULL), m_bNewContact(false) {} MCONTACT m_hContact; CQuotesProviders::TQuotesProviderPtr m_pProvider; bool m_bNewContact; }; IXMLNode::TXMLNodePtr find_quotes_module(const IXMLNode::TXMLNodePtr& pXmlContact) { static const tstring g_sQuotes = quotes_a2t(QUOTES_MODULE_NAME); size_t cChild = pXmlContact->GetChildCount(); for (size_t i = 0; i < cChild; ++i) { IXMLNode::TXMLNodePtr pNode = pXmlContact->GetChildNode(i); tstring sName = pNode->GetName(); if ((0 == mir_tstrcmpi(g_pszXmlModule, sName.c_str())) && (0 == mir_tstrcmpi(g_sQuotes.c_str(), pNode->GetText().c_str()))) { return pNode; } } return IXMLNode::TXMLNodePtr(); } typedef std::pair TNameValue;//first is name,second is value TNameValue parse_setting_node(const IXMLNode::TXMLNodePtr& pXmlSetting) { assert(pXmlSetting); tstring sName, sValue; size_t cSettingChildItems = pXmlSetting->GetChildCount(); for (size_t j = 0; j < cSettingChildItems; ++j) { IXMLNode::TXMLNodePtr pXMLSetChild = pXmlSetting->GetChildNode(j); if (pXMLSetChild) { if (0 == mir_tstrcmpi(g_pszXmlName, pXMLSetChild->GetName().c_str())) { sName = pXMLSetChild->GetText(); } else if (0 == mir_tstrcmpi(g_pszXmlValue, pXMLSetChild->GetName().c_str())) { sValue = pXMLSetChild->GetText(); } } } return std::make_pair(sName, sValue); } CQuotesProviders::TQuotesProviderPtr find_provider(const IXMLNode::TXMLNodePtr& pXmlQuotesModule) { // USES_CONVERSION; static const tstring g_sQuotesProvider = quotes_a2t(DB_STR_QUOTE_PROVIDER);//A2CT(DB_STR_QUOTE_PROVIDER); size_t cChild = pXmlQuotesModule->GetChildCount(); for (size_t i = 0; i < cChild; ++i) { IXMLNode::TXMLNodePtr pXMLSetting = pXmlQuotesModule->GetChildNode(i); if (pXMLSetting && (0 == mir_tstrcmpi(g_pszXmlSetting, pXMLSetting->GetName().c_str()))) { TNameValue Item = parse_setting_node(pXMLSetting); if ((0 == mir_tstrcmpi(g_sQuotesProvider.c_str(), Item.first.c_str())) && (false == Item.second.empty())) { return CModuleInfo::GetInstance().GetQuoteProvidersPtr()->FindProvider(Item.second); } } } return CQuotesProviders::TQuotesProviderPtr(); } bool get_contact_state(const IXMLNode::TXMLNodePtr& pXmlContact, CContactState& cst) { class visitor : public CQuotesProviderVisitor { public: visitor(const IXMLNode::TXMLNodePtr& pXmlQuotes) : m_hContact(NULL), m_pXmlQuotes(pXmlQuotes) { } MCONTACT GetContact()const { return m_hContact; } private: virtual void Visit(const CQuotesProviderDukasCopy& rProvider) { tstring sQuoteID = GetXMLNodeValue(DB_STR_QUOTE_ID); if (false == sQuoteID.empty()) { m_hContact = rProvider.GetContactByQuoteID(sQuoteID); } } virtual void Visit(const CQuotesProviderGoogle& rProvider) { // USES_CONVERSION; static const tstring g_sFromID = quotes_a2t(DB_STR_FROM_ID);//A2CT(DB_STR_FROM_ID); static const tstring g_sToID = quotes_a2t(DB_STR_TO_ID);//A2CT(DB_STR_TO_ID); tstring sFromID; tstring sToID; size_t cChild = m_pXmlQuotes->GetChildCount(); for (size_t i = 0; i < cChild; ++i) { IXMLNode::TXMLNodePtr pNode = m_pXmlQuotes->GetChildNode(i); if (pNode && (0 == mir_tstrcmpi(g_pszXmlSetting, pNode->GetName().c_str()))) { TNameValue Item = parse_setting_node(pNode); if (0 == mir_tstrcmpi(g_sFromID.c_str(), Item.first.c_str())) { sFromID = Item.second; } else if (0 == mir_tstrcmpi(g_sToID.c_str(), Item.first.c_str())) { sToID = Item.second; } } } if ((false == sFromID.empty()) && (false == sToID.empty())) { m_hContact = rProvider.GetContactByID(sFromID, sToID); } } virtual void Visit(const CQuotesProviderFinance& rProvider) { tstring sQuoteID = GetXMLNodeValue(DB_STR_QUOTE_ID); if (false == sQuoteID.empty()) { m_hContact = rProvider.GetContactByQuoteID(sQuoteID); } } tstring GetXMLNodeValue(const char* pszXMLNodeName)const { tstring sXMLNodeName = quotes_a2t(pszXMLNodeName); tstring sValue; size_t cChild = m_pXmlQuotes->GetChildCount(); for (size_t i = 0; i < cChild; ++i) { IXMLNode::TXMLNodePtr pNode = m_pXmlQuotes->GetChildNode(i); if (pNode && (0 == mir_tstrcmpi(g_pszXmlSetting, pNode->GetName().c_str()))) { TNameValue Item = parse_setting_node(pNode); if (0 == mir_tstrcmpi(Item.first.c_str(), sXMLNodeName.c_str())) { sValue = Item.second; break; } } } return sValue; } private: MCONTACT m_hContact; IXMLNode::TXMLNodePtr m_pXmlQuotes; }; IXMLNode::TXMLNodePtr pXmlQuotes = find_quotes_module(pXmlContact); if (pXmlQuotes) { cst.m_pProvider = find_provider(pXmlQuotes); if (cst.m_pProvider) { visitor vs(pXmlQuotes); cst.m_pProvider->Accept(vs); cst.m_hContact = vs.GetContact(); return true; } } return false; } bool import_contact(const IXMLNode::TXMLNodePtr& pXmlContact, CImportContext& impctx) { ++impctx.m_cHandledContacts; CContactState cst; bool bResult = get_contact_state(pXmlContact, cst); if (bResult) { if (NULL == cst.m_hContact) { cst.m_hContact = MCONTACT(CallService(MS_DB_CONTACT_ADD, 0, 0)); cst.m_bNewContact = true; } else if (impctx.m_nFlags & QUOTES_IMPORT_SKIP_EXISTING_CONTACTS) return true; if (cst.m_hContact) { size_t cChild = pXmlContact->GetChildCount(); for (size_t i = 0; i < cChild && bResult; ++i) { IXMLNode::TXMLNodePtr pNode = pXmlContact->GetChildNode(i); tstring sName = pNode->GetName(); if (0 == mir_tstrcmpi(g_pszXmlModule, sName.c_str())) bResult &= handle_module(cst.m_hContact, pNode); } if (cst.m_bNewContact && bResult) { cst.m_pProvider->AddContact(cst.m_hContact); cst.m_pProvider->RefreshContact(cst.m_hContact); } } else bResult = false; } return bResult; } size_t import_contacts(const IXMLNode::TXMLNodePtr& pXmlContacts, CImportContext& impctx) { size_t cContacts = 0; size_t cChild = pXmlContacts->GetChildCount(); for (size_t i = 0; i < cChild; ++i) { IXMLNode::TXMLNodePtr pNode = pXmlContacts->GetChildNode(i); tstring sName = pNode->GetName(); if (0 == mir_tstrcmpi(g_pszXmlContact, sName.c_str())) if (true == import_contact(pNode, impctx)) ++cContacts; } return cContacts; } size_t handle_contacts_node(const IXMLNode::TXMLNodePtr& pXmlRoot, CImportContext& impctx) { size_t cContacts = 0; size_t cChild = pXmlRoot->GetChildCount(); for (size_t i = 0; i < cChild; ++i) { IXMLNode::TXMLNodePtr pNode = pXmlRoot->GetChildNode(i); tstring sName = pNode->GetName(); if (0 == mir_tstrcmpi(g_pszXmlContacts, sName.c_str())) cContacts += import_contacts(pNode, impctx); else cContacts += handle_contacts_node(pNode, impctx); } return cContacts; } bool do_import(const IXMLNode::TXMLNodePtr& pXmlRoot, UINT nFlags) { CImportContext imctx(count_contacts(pXmlRoot, false)); imctx.m_cHandledContacts = 0; imctx.m_nFlags = nFlags; return (handle_contacts_node(pXmlRoot, imctx) > 0); } INT_PTR Quotes_Import(WPARAM wp, LPARAM lp) { // USES_CONVERSION; tstring sFileName; const char* pszFile = reinterpret_cast(lp); if (NULL == pszFile) { if (false == show_open_file_dialog(true, sFileName)) return -1; } else sFileName = quotes_a2t(pszFile);//A2CT(pszFile); CModuleInfo::TXMLEnginePtr pXmlEngine = CModuleInfo::GetInstance().GetXMLEnginePtr(); IXMLNode::TXMLNodePtr pXmlRoot = pXmlEngine->LoadFile(sFileName); if (pXmlRoot) return ((true == do_import(pXmlRoot, wp)) ? 0 : 1); return 1; } INT_PTR QuotesMenu_ImportAll(WPARAM, LPARAM) { return CallService(MS_QUOTES_IMPORT, 0, 0); } INT_PTR QuotesMenu_ExportAll(WPARAM, LPARAM) { return CallService(MS_QUOTES_EXPORT, 0, 0); }