1 files changed, 397 insertions, 0 deletions
diff --git a/protocols/NewsAggregator/Src/ImportFeed.cpp b/protocols/NewsAggregator/Src/ImportFeed.cpp
new file mode 100644
index 0000000000..a906e27959
--- /dev/null
+++ b/protocols/NewsAggregator/Src/ImportFeed.cpp
@@ -0,0 +1,397 @@
+/*
+Copyright (C) 2012-23 Miranda NG team (https://miranda-ng.org)
+
+This program is free software; you can redistribute it and/or
+modify it under the terms of the GNU General Public License
+as published by the Free Software Foundation version 2
+of the License.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program. If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include "stdafx.h"
+
+static CDlgBase *pImportDialog = nullptr;
+
+static const TiXmlElement* AdviceNode(const TiXmlElement *node)
+{
+	auto *tmpnode = node;
+
+	// try to rest on the same level first
+	node = node->NextSiblingElement();
+	if (node)
+		return node;
+
+	do {
+		// go up one level
+		node = tmpnode->Parent()->ToElement();
+		tmpnode = node;
+		node = node->NextSiblingElement();
+		if (node)
+			return node;
+	} while (mir_strcmpi(tmpnode->Name(), "body"));
+
+	// nothing found or we reached body
+	return nullptr;
+}
+
+/////////////////////////////////////////////////////////////////////////////////////////
+
+class CImportFeed : public CDlgBase
+{
+	typedef CDlgBase CSuper;
+
+	CCtrlListView *m_list;
+
+	CCtrlEdit m_importfile;
+	CCtrlButton m_browsefile;
+	CCtrlListBox m_feedslist;
+	CCtrlListBox m_feedsimportlist;
+	CCtrlButton m_addfeed;
+	CCtrlButton m_removefeed;
+	CCtrlButton m_addallfeeds;
+	CCtrlButton m_removeallfeeds;
+	CCtrlButton m_ok;
+
+public:
+	CImportFeed(CCtrlListView *feeds) :
+		CSuper(g_plugin, IDD_FEEDIMPORT),
+		m_importfile(this, IDC_IMPORTFILEPATH), m_browsefile(this, IDC_BROWSEIMPORTFILE),
+		m_feedslist(this, IDC_FEEDSLIST), m_feedsimportlist(this, IDC_FEEDSIMPORTLIST),
+		m_addfeed(this, IDC_ADDFEED), m_removefeed(this, IDC_REMOVEFEED),
+		m_addallfeeds(this, IDC_ADDALLFEEDS), m_removeallfeeds(this, IDC_REMOVEALLFEEDS),
+		m_ok(this, IDOK)
+	{
+		m_list = feeds;
+		m_browsefile.OnClick = Callback(this, &CImportFeed::OnBrowseFile);
+		m_addfeed.OnClick = Callback(this, &CImportFeed::OnAddFeed);
+		m_removefeed.OnClick = Callback(this, &CImportFeed::OnRemoveFeed);
+		m_addallfeeds.OnClick = Callback(this, &CImportFeed::OnAddAllFeeds);
+		m_removeallfeeds.OnClick = Callback(this, &CImportFeed::OnRemoveAllFeeds);
+
+		m_feedslist.OnDblClick = Callback(this, &CImportFeed::OnFeedsList);
+		m_feedsimportlist.OnDblClick = Callback(this, &CImportFeed::OnFeedsImportList);
+	}
+
+	bool OnInitDialog() override
+	{
+		Utils_RestoreWindowPositionNoSize(m_hwnd, NULL, MODULENAME, "ImportDlg");
+		m_removefeed.Disable();
+		m_removeallfeeds.Disable();
+		m_ok.Disable();
+		m_addfeed.Disable();
+		m_addallfeeds.Disable();
+		return true;
+	}
+
+	bool OnApply() override
+	{
+		wchar_t FileName[MAX_PATH];
+		m_importfile.GetText(FileName, _countof(FileName));
+
+		FILE *in = _wfopen(FileName, L"rb");
+		if (in == nullptr)
+			return false;
+
+		TiXmlDocument doc;
+		int res = doc.LoadFile(in);
+		fclose(in);
+		if (res != 0)
+			return false;
+
+		auto *node = TiXmlConst(&doc)["opml"]["body"]["outline"].ToElement();
+		if (!node)
+			node = TiXmlConst(&doc)["body"]["outline"].ToElement();
+		if (node == nullptr)
+			return false;
+
+		int count = m_feedsimportlist.GetCount();
+		int DUPES = 0;
+
+		while (node) {
+			auto *pszUrl = node->Attribute("xmlUrl");
+			if (!pszUrl && node->NoChildren())
+				node = AdviceNode(node);
+			else if (!pszUrl && !node->NoChildren())
+				node = node->FirstChildElement();
+			else if (pszUrl) {
+				wchar_t *text = nullptr, *url = nullptr, *siteurl = nullptr;
+				bool bNeedToImport = false;
+
+				if (auto *pszText = node->Attribute("text")) {
+					text = mir_utf8decodeW(pszText);
+
+					for (int j = 0; j < count; j++) {
+						wchar_t item[MAX_PATH];
+						m_feedsimportlist.GetItemText(j, item, _countof(item));
+						if (!mir_wstrcmpi(item, text)) {
+							bNeedToImport = true;
+							break;
+						}
+					}
+				}
+
+				if (auto *pszText = node->Attribute("xmlUrl")) {
+					url = mir_utf8decodeW(pszText);
+					if (GetContactByURL(url) && bNeedToImport) {
+						bNeedToImport = false;
+						DUPES++;
+					}
+				}
+
+				if (auto *pszText = node->Attribute("htmlUrl"))
+					siteurl = mir_utf8decodeW(pszText);
+
+				if (bNeedToImport && text && url) {
+					CMStringW wszGroup;
+					auto *parent = node->Parent()->ToElement();
+					while (mir_strcmpi(parent->Name(), "body")) {
+						if (auto *pszText = parent->Attribute("text")) {
+							if (!wszGroup.IsEmpty())
+								wszGroup.Insert(0, L"\\");
+							wszGroup.Insert(0, Utf2T(pszText));
+						}
+						parent = parent->Parent()->ToElement();
+					}
+
+					MCONTACT hContact = db_add_contact();
+					Proto_AddToContact(hContact, MODULENAME);
+					Contact::Readonly(hContact);
+					g_plugin.setWString(hContact, "Nick", text);
+					g_plugin.setWString(hContact, "URL", url);
+					if (siteurl)
+						g_plugin.setWString(hContact, "Homepage", siteurl);
+					g_plugin.setByte(hContact, "CheckState", 1);
+					g_plugin.setDword(hContact, "UpdateTime", DEFAULT_UPDATE_TIME);
+					g_plugin.setWString(hContact, "MsgFormat", TAGSDEFAULT);
+					g_plugin.setWord(hContact, "Status", Proto_GetStatus(MODULENAME));
+
+					if (m_list != nullptr) {
+						int iItem = m_list->AddItem(text, -1);
+						m_list->SetItem(iItem, 1, url);
+						m_list->SetCheckState(iItem, 1);
+					}
+
+					if (!wszGroup.IsEmpty()) {
+						Clist_GroupCreate(0, wszGroup);
+						Clist_SetGroup(hContact, wszGroup);
+					}
+				}
+				mir_free(text);
+				mir_free(url);
+				mir_free(siteurl);
+
+				node = AdviceNode(node);
+			}
+		}
+
+		wchar_t mes[MAX_PATH];
+		if (DUPES)
+			mir_snwprintf(mes, TranslateT("Imported %d feed(s)\r\nNot imported %d duplicate(s)."), count - DUPES, DUPES);
+		else
+			mir_snwprintf(mes, TranslateT("Imported %d feed(s)."), count);
+		MessageBox(m_hwnd, mes, TranslateT("News Aggregator"), MB_OK | MB_ICONINFORMATION);
+		return true;
+	}
+
+	void OnDestroy() override
+	{
+		Utils_SaveWindowPosition(m_hwnd, NULL, MODULENAME, "ImportDlg");
+		pImportDialog = nullptr;
+	}
+
+	void OnBrowseFile(CCtrlBase *)
+	{
+		wchar_t FileName[MAX_PATH];
+		VARSW tszMirDir(L"%miranda_path%");
+
+		OPENFILENAME ofn = { 0 };
+		ofn.lStructSize = sizeof(ofn);
+		wchar_t tmp[MAX_PATH];
+		mir_snwprintf(tmp, L"%s (*.opml, *.xml)%c*.opml;*.xml%c%c", TranslateT("OPML files"), 0, 0, 0);
+		ofn.lpstrFilter = tmp;
+		ofn.hwndOwner = nullptr;
+		ofn.lpstrFile = FileName;
+		ofn.nMaxFile = MAX_PATH;
+		ofn.nMaxFileTitle = MAX_PATH;
+		ofn.Flags = OFN_HIDEREADONLY;
+		ofn.lpstrInitialDir = tszMirDir;
+		*FileName = '\0';
+		ofn.lpstrDefExt = L"";
+		if (!GetOpenFileName(&ofn))
+			return;
+
+		FILE *in = _wfopen(FileName, L"rb");
+		if (in == nullptr)
+			return;
+
+		TiXmlDocument doc;
+		int res = doc.LoadFile(in);
+		fclose(in);
+		if (res != 0) {
+			MessageBox(m_hwnd, TranslateT("Not valid import file."), TranslateT("Error"), MB_OK | MB_ICONERROR);
+			return;
+		}
+
+		m_importfile.SetText(FileName);
+
+		auto *node = TiXmlConst(&doc)["opml"]["body"]["outline"].ToElement();
+		if (!node)
+			node = TiXmlConst(&doc)["body"]["outline"].ToElement();
+		if (node == nullptr) {
+			MessageBox(m_hwnd, TranslateT("Not valid import file."), TranslateT("Error"), MB_OK | MB_ICONERROR);
+			return;
+		}
+
+		while (node) {
+			auto *pszUrl = node->Attribute("xmlUrl");
+			if (!pszUrl && node->NoChildren())
+				node = AdviceNode(node);
+			else if (!pszUrl && !node->NoChildren())
+				node = node->FirstChildElement();
+			else if (pszUrl) {
+				if (auto *pszText = node->Attribute("text")) {
+					Utf2T text(pszText);
+					m_feedslist.AddString(text);
+					m_addfeed.Enable();
+					m_addallfeeds.Enable();
+				}
+
+				node = AdviceNode(node);
+			}
+		}
+	}
+
+	void OnAddFeed(CCtrlBase *)
+	{
+		if (!m_removefeed.Enabled())
+			m_removefeed.Enable();
+		if (!m_removeallfeeds.Enabled())
+			m_removeallfeeds.Enable();
+		if (!m_ok.Enabled())
+			m_ok.Enable();
+		int cursel = m_feedslist.GetCurSel();
+		wchar_t item[MAX_PATH];
+		m_feedslist.GetItemText(cursel, item, _countof(item));
+		m_feedsimportlist.AddString(item);
+		m_feedslist.DeleteString(cursel);
+		if (!m_feedslist.GetCount()) {
+			m_addfeed.Disable();
+			m_addallfeeds.Disable();
+		}
+	}
+
+	void OnRemoveFeed(CCtrlBase *)
+	{
+		if (!m_addfeed.Enabled())
+			m_addfeed.Enable();
+		if (!m_addallfeeds.Enabled())
+			m_addallfeeds.Enable();
+		int cursel = m_feedsimportlist.GetCurSel();
+		wchar_t item[MAX_PATH];
+		m_feedsimportlist.GetItemText(cursel, item, _countof(item));
+		m_feedslist.AddString(item);
+		m_feedsimportlist.DeleteString(cursel);
+		if (!m_feedsimportlist.GetCount()) {
+			m_removefeed.Disable();
+			m_removeallfeeds.Disable();
+			m_ok.Disable();
+		}
+	}
+
+	void OnAddAllFeeds(CCtrlBase *)
+	{
+		if (!m_removefeed.Enabled())
+			m_removefeed.Enable();
+		if (!m_removeallfeeds.Enabled())
+			m_removeallfeeds.Enable();
+		if (!m_ok.Enabled())
+			m_ok.Enable();
+		int count = m_feedslist.GetCount();
+		for (int i = 0; i < count; i++) {
+			wchar_t item[MAX_PATH];
+			m_feedslist.GetItemText(i, item, _countof(item));
+			m_feedsimportlist.AddString(item);
+		}
+		for (int i = count - 1; i > -1; i--)
+			m_feedslist.DeleteString(i);
+		m_addfeed.Disable();
+		m_addallfeeds.Disable();
+	}
+
+	void OnRemoveAllFeeds(CCtrlBase *)
+	{
+		if (!m_addfeed.Enabled())
+			m_addfeed.Enable();
+		if (!m_addallfeeds.Enabled())
+			m_addallfeeds.Enable();
+		int count = m_feedsimportlist.GetCount();
+		for (int i = 0; i < count; i++) {
+			wchar_t item[MAX_PATH];
+			m_feedsimportlist.GetItemText(i, item, _countof(item));
+			m_feedslist.AddString(item);
+		}
+		for (int i = count - 1; i > -1; i--)
+			m_feedsimportlist.DeleteString(i);
+		m_removefeed.Disable();
+		m_removeallfeeds.Disable();
+		m_ok.Disable();
+	}
+
+	void OnFeedsList(CCtrlBase *)
+	{
+		if (!m_removefeed.Enabled())
+			m_removefeed.Enable();
+		if (!m_removeallfeeds.Enabled())
+			m_removeallfeeds.Enable();
+		if (!m_ok.Enabled())
+			m_ok.Enable();
+		int cursel = m_feedslist.GetCurSel();
+		wchar_t item[MAX_PATH];
+		m_feedslist.GetItemText(cursel, item, _countof(item));
+		m_feedsimportlist.AddString(item);
+		m_feedslist.DeleteString(cursel);
+		if (!m_feedslist.GetCount()) {
+			m_addfeed.Disable();
+			m_addallfeeds.Disable();
+		}
+	}
+
+	void OnFeedsImportList(CCtrlBase *)
+	{
+		if (!m_addfeed.Enabled())
+			m_addfeed.Enable();
+		if (!m_addallfeeds.Enabled())
+			m_addallfeeds.Enable();
+		int cursel = m_feedsimportlist.GetCurSel();
+		wchar_t item[MAX_PATH];
+		m_feedsimportlist.GetItemText(cursel, item, _countof(item));
+		m_feedslist.AddString(item);
+		m_feedsimportlist.DeleteString(cursel);
+		if (!m_feedsimportlist.GetCount()) {
+			m_removefeed.Disable();
+			m_removeallfeeds.Disable();
+			m_ok.Disable();
+		}
+	}
+};
+
+/////////////////////////////////////////////////////////////////////////////////////////
+// Module entry point
+
+INT_PTR ImportFeeds(WPARAM wParam, LPARAM)
+{
+	if (pImportDialog == nullptr) {
+		pImportDialog = new CImportFeed(nullptr);
+		if (wParam)
+			pImportDialog->SetParent(MWindow(wParam));
+	}
+	pImportDialog->Show();
+	return 0;
+}