/*
Copyright (C) 2012 Mataes

This is free software; you can redistribute it and/or
modify it under the terms of the GNU Library General Public
License as published by the Free Software Foundation; either
version 2 of the License, or (at your option) any later version.

This 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
Library General Public License for more details.

You should have received a copy of the GNU Library General Public
License along with this file; see the file license.txt.  If
not, write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330,
Boston, MA 02111-1307, USA.
*/

#include "stdafx.h"

CExportFeed::CExportFeed()
	: CSuper(g_plugin, IDD_FEEDEXPORT),
	m_feedslist(this, IDC_FEEDSLIST), m_feedsexportlist(this, IDC_FEEDSEXPORTLIST),
	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_addfeed.OnClick = Callback(this, &CExportFeed::OnAddFeed);
	m_removefeed.OnClick = Callback(this, &CExportFeed::OnRemoveFeed);
	m_addallfeeds.OnClick = Callback(this, &CExportFeed::OnAddAllFeeds);
	m_removeallfeeds.OnClick = Callback(this, &CExportFeed::OnRemoveAllFeeds);
	m_ok.OnClick = Callback(this, &CExportFeed::OnOk);

	m_feedslist.OnDblClick = Callback(this, &CExportFeed::OnFeedsList);
	m_feedsexportlist.OnDblClick = Callback(this, &CExportFeed::OnFeedsExportList);
}

bool CExportFeed::OnInitDialog()
{
	Utils_RestoreWindowPositionNoSize(m_hwnd, NULL, MODULENAME, "ExportDlg");
	for (auto &hContact : Contacts(MODULENAME)) {
		ptrW message(g_plugin.getWStringA(hContact, "Nick"));
		if (message != nullptr)
			m_feedslist.AddString(message);
	}
	m_removefeed.Disable();
	m_removeallfeeds.Disable();
	m_ok.Disable();
	if (!m_feedslist.GetCount()) {
		m_addfeed.Disable();
		m_addallfeeds.Disable();
	}
	return true;
}

void CExportFeed::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_feedsexportlist.AddString(item);
	m_feedslist.DeleteString(cursel);
	if (!m_feedslist.GetCount()) {
		m_addfeed.Disable();
		m_addallfeeds.Disable();
	}
}

void CExportFeed::OnRemoveFeed(CCtrlBase*)
{
	if (!m_addfeed.Enabled())
		m_addfeed.Enable();
	if (!m_addallfeeds.Enabled())
		m_addallfeeds.Enable();
	int cursel = m_feedsexportlist.GetCurSel();
	wchar_t item[MAX_PATH];
	m_feedsexportlist.GetItemText(cursel, item, _countof(item));
	m_feedslist.AddString(item);
	m_feedsexportlist.DeleteString(cursel);
	if (!m_feedsexportlist.GetCount()) {
		m_removefeed.Disable();
		m_removeallfeeds.Disable();
		m_ok.Disable();
	}
}

void CExportFeed::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_feedsexportlist.AddString(item);
	}
	for (int i = count - 1; i > -1; i--)
		m_feedslist.DeleteString(i);
	m_addfeed.Disable();
	m_addallfeeds.Disable();
}

void CExportFeed::OnRemoveAllFeeds(CCtrlBase*)
{
	if (!m_addfeed.Enabled())
		m_addfeed.Enable();
	if (!m_addallfeeds.Enabled())
		m_addallfeeds.Enable();
	int count = m_feedsexportlist.GetCount();
	for (int i = 0; i < count; i++) {
		wchar_t item[MAX_PATH];
		m_feedsexportlist.GetItemText(i, item, _countof(item));
		m_feedslist.AddString(item);
	}
	for (int i = count - 1; i > -1; i--)
		m_feedsexportlist.DeleteString(i);
	m_removefeed.Disable();
	m_removeallfeeds.Disable();
	m_ok.Disable();
}

void CExportFeed::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_feedsexportlist.AddString(item);
	m_feedslist.DeleteString(cursel);
	if (!m_feedslist.GetCount()) {
		m_addfeed.Disable();
		m_addallfeeds.Disable();
	}
}

void CExportFeed::OnFeedsExportList(CCtrlBase*)
{
	if (!m_addfeed.Enabled())
		m_addfeed.Enable();
	if (!m_addallfeeds.Enabled())
		m_addallfeeds.Enable();
	int cursel = m_feedsexportlist.GetCurSel();
	wchar_t item[MAX_PATH];
	m_feedsexportlist.GetItemText(cursel, item, _countof(item));
	m_feedslist.AddString(item);
	m_feedsexportlist.DeleteString(cursel);
	if (!m_feedsexportlist.GetCount()) {
		m_removefeed.Disable();
		m_removeallfeeds.Disable();
		m_ok.Disable();
	}
}

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;
}

void CExportFeed::OnOk(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)%c*.opml%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_SHAREAWARE | OFN_PATHMUSTEXIST | OFN_OVERWRITEPROMPT;
	ofn.lpstrInitialDir = tszMirDir;
	*FileName = '\0';
	ofn.lpstrDefExt = L"";
	if (!GetSaveFileName(&ofn))
		return;

	TiXmlDocument doc;
	auto *hXml = doc.NewElement("opml"); doc.InsertEndChild(hXml);
	hXml->SetAttribute("version", "1.0"); 

	auto *xmlHeader = doc.NewElement("head"); hXml->InsertEndChild(xmlHeader);
	auto *xmlTitle = doc.NewElement("title"); xmlTitle->SetText("Miranda NG NewsAggregator plugin export"); xmlHeader->InsertEndChild(xmlTitle);
		
	auto *xmlBody = doc.NewElement("body"); hXml->InsertEndChild(xmlBody);

	int count = m_feedsexportlist.GetCount();
	for (int i = 0; i < count; i++) {
		wchar_t item[MAX_PATH];
		m_feedsexportlist.GetItemText(i, item, _countof(item));
		MCONTACT hContact = GetContactByNick(item);
		wchar_t
			*title = g_plugin.getWStringA(hContact, "Nick"),
			*url = g_plugin.getWStringA(hContact, "URL"),
			*siteurl = g_plugin.getWStringA(hContact, "Homepage"),
			*group = Clist_GetGroup(hContact);

		TiXmlElement *elem = xmlBody;
		if (group) {
			wchar_t *section = wcstok(group, L"\\");
			while (section != nullptr) {
				TiXmlElement *existgroup = 0;
				for (auto *it : TiXmlFilter(elem, "outline")) {
					if (it->Attribute("title", T2Utf(section))) {
						existgroup = (TiXmlElement*)it;
						break;
					}
				}

				if (!existgroup) {
					auto *pNew = doc.NewElement("outline");
					pNew->SetAttribute("title", section); pNew->SetAttribute("text", section);
					elem->InsertEndChild(pNew);
					elem = pNew;
				}
				else elem = existgroup;

				section = wcstok(nullptr, L"\\");
			}
		}

		auto *pNew = doc.NewElement("outline"); elem->InsertEndChild(pNew);
		pNew->SetAttribute("text", title);
		pNew->SetAttribute("title", title);
		pNew->SetAttribute("type", "rss");
		pNew->SetAttribute("xmlUrl", url);
		pNew->SetAttribute("htmlUrl", siteurl);

		mir_free(title);
		mir_free(url);
		mir_free(siteurl);
		mir_free(group);
	}

	FILE *out = _wfopen(FileName, L"wb");
	if (out) {
		tinyxml2::XMLPrinter printer(out);
		doc.Print(&printer);
		fclose(out);
	}
}

bool CExportFeed::OnClose()
{
	Utils_SaveWindowPosition(m_hwnd, NULL, MODULENAME, "ExportDlg");
	if (pExportDialog)
		pExportDialog = nullptr;
	return true;
}

/////////////////////////////////////////////////////////////////////////////////////////

CImportFeed::CImportFeed(CCtrlListView *m_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 = m_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_ok.OnClick = Callback(this, &CImportFeed::OnOk);

	m_feedslist.OnDblClick = Callback(this, &CImportFeed::OnFeedsList);
	m_feedsimportlist.OnDblClick = Callback(this, &CImportFeed::OnFeedsImportList);
}

bool CImportFeed::OnInitDialog()
{
	Utils_RestoreWindowPositionNoSize(m_hwnd, NULL, MODULENAME, "ImportDlg");
	m_removefeed.Disable();
	m_removeallfeeds.Disable();
	m_ok.Disable();
	m_addfeed.Disable();
	m_addallfeeds.Disable();
	return true;
}

void CImportFeed::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 CImportFeed::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 CImportFeed::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 CImportFeed::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 CImportFeed::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 CImportFeed::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 CImportFeed::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();
	}
}

void CImportFeed::OnOk(CCtrlBase*)
{
	wchar_t FileName[MAX_PATH];
	m_importfile.GetText(FileName, _countof(FileName));

	FILE *in = _wfopen(FileName, L"rb");
	if (in == nullptr)
		return;

	TiXmlDocument doc;
	int res = doc.LoadFile(in);
	fclose(in);
	if (res != 0)
		return;

	auto *node = TiXmlConst(&doc)["opml"]["body"]["outline"].ToElement();
	if (!node)
		node = TiXmlConst(&doc)["body"]["outline"].ToElement();
	if (node == nullptr)
		return;

	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 && siteurl) {
				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);
				g_plugin.setWString(hContact, "Nick", text);
				g_plugin.setWString(hContact, "URL", url);
				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);
}

bool CImportFeed::OnClose()
{
	Utils_SaveWindowPosition(m_hwnd, NULL, MODULENAME, "ImportDlg");
	if (pImportDialog)
		pImportDialog = nullptr;
	return true;
}

/////////////////////////////////////////////////////////////////////////////////////////

CFeedEditor::CFeedEditor(int iItem, CCtrlListView *m_feeds, MCONTACT Contact)
	: CSuper(g_plugin, IDD_ADDFEED),
	m_feedtitle(this, IDC_FEEDTITLE), m_feedurl(this, IDC_FEEDURL),
	m_checktime(this, IDC_CHECKTIME), m_checktimespin(this, IDC_TIMEOUT_VALUE_SPIN, 999),
	m_checkfeed(this, IDC_DISCOVERY), m_useauth(this, IDC_USEAUTH),
	m_login(this, IDC_LOGIN), m_password(this, IDC_PASSWORD),
	m_tagedit(this, IDC_TAGSEDIT), m_reset(this, IDC_RESET),
	m_help(this, IDC_TAGHELP), m_ok(this, IDOK), m_iItem(iItem)
{
	m_list = m_feeds;
	m_hContact = Contact;
	m_checkfeed.OnClick = Callback(this, &CFeedEditor::OnCheckFeed);
	m_useauth.OnChange = Callback(this, &CFeedEditor::OnUseAuth);
	m_reset.OnClick = Callback(this, &CFeedEditor::OnReset);
	m_help.OnClick = Callback(this, &CFeedEditor::OnHelp);
	m_ok.OnClick = Callback(this, &CFeedEditor::OnOk);
}

bool CFeedEditor::OnInitDialog()
{
	if (m_iItem == -1 && m_hContact == NULL)
		SetWindowText(m_hwnd, TranslateT("Add Feed"));
	else
		SetWindowText(m_hwnd, TranslateT("Change Feed"));
	m_checktime.SetMaxLength(3);

	if (m_iItem > -1 && m_hContact == 0) {
		wchar_t SelNick[MAX_PATH], SelUrl[MAX_PACKAGE_NAME];
		m_list->GetItemText(m_iItem, 0, SelNick, _countof(SelNick));
		m_list->GetItemText(m_iItem, 1, SelUrl, _countof(SelNick));

		for (auto &hContact : Contacts(MODULENAME)) {
			ptrW dbNick(g_plugin.getWStringA(hContact, "Nick"));
			if ((dbNick == NULL) || (mir_wstrcmp(dbNick, SelNick) != 0))
				continue;

			ptrW dbURL(g_plugin.getWStringA(hContact, "URL"));
			if ((dbURL == NULL) || (mir_wstrcmp(dbURL, SelUrl) != 0))
				continue;

			m_hContact = hContact;
			m_feedtitle.SetText(SelNick);
			m_feedurl.SetText(SelUrl);
			m_checktime.SetInt(g_plugin.getDword(hContact, "UpdateTime", DEFAULT_UPDATE_TIME));

			ptrW szMsgFormat(g_plugin.getWStringA(hContact, "MsgFormat"));
			if (szMsgFormat)
				m_tagedit.SetText(szMsgFormat);

			if (g_plugin.getByte(hContact, "UseAuth", 0)) {
				m_useauth.SetState(1);
				m_login.Enable();
				m_password.Enable();

				ptrW szLogin(g_plugin.getWStringA(hContact, "Login"));
				if (szLogin)
					m_login.SetText(szLogin);

				pass_ptrA pwd(g_plugin.getStringA(hContact, "Password"));
				m_password.SetTextA(pwd);
			}
			g_arFeeds.insert(this);
			Utils_RestoreWindowPositionNoSize(m_hwnd, hContact, MODULENAME, "ChangeDlg");
			break;
		}
	}
	else if (m_iItem == -1 && m_hContact == NULL) {
		m_feedurl.SetText(L"http://");
		m_tagedit.SetText(TAGSDEFAULT);
		m_checktime.SetInt(DEFAULT_UPDATE_TIME);
		Utils_RestoreWindowPositionNoSize(m_hwnd, NULL, MODULENAME, "AddDlg");
	}
	else if (m_hContact != NULL) {
		ptrW dbNick(g_plugin.getWStringA(m_hContact, "Nick"));
		ptrW dbURL(g_plugin.getWStringA(m_hContact, "URL"));

		m_feedtitle.SetText(dbNick);
		m_feedurl.SetText(dbURL);
		m_checktime.SetInt(g_plugin.getDword(m_hContact, "UpdateTime", DEFAULT_UPDATE_TIME));

		ptrW szMsgFormat(g_plugin.getWStringA(m_hContact, "MsgFormat"));
		if (szMsgFormat)
			m_tagedit.SetText(szMsgFormat);

		if (g_plugin.getByte(m_hContact, "UseAuth")) {
			m_useauth.SetState(1);
			m_login.Enable();
			m_password.Enable();

			ptrW szLogin(g_plugin.getWStringA(m_hContact, "Login"));
			if (szLogin)
				m_login.SetText(szLogin);

			pass_ptrA pwd(g_plugin.getStringA(m_hContact, "Password"));
			m_password.SetTextA(pwd);
		}
		g_arFeeds.insert(this);
		Utils_RestoreWindowPositionNoSize(m_hwnd, m_hContact, MODULENAME, "ChangeDlg");
	}
	return true;
}

void CFeedEditor::OnCheckFeed(CCtrlBase*)
{
	m_checkfeed.Disable();
	m_checkfeed.SetText(TranslateT("Wait..."));
	wchar_t *tszTitle = nullptr;
	ptrW strfeedurl(m_feedurl.GetText());
	if (strfeedurl || mir_wstrcmp(strfeedurl, L"http://") != 0 || mir_wstrcmp(strfeedurl, L"") != 0)
		tszTitle = (wchar_t*)CheckFeed(strfeedurl, this);
	else
		MessageBox(m_hwnd, TranslateT("Enter Feed URL"), TranslateT("Error"), MB_OK);
	m_feedtitle.SetText(tszTitle);
	mir_free(tszTitle);
	m_checkfeed.Enable();
	m_checkfeed.SetText(TranslateT("Check Feed"));
}

void CFeedEditor::OnReset(CCtrlBase*)
{
	if (MessageBox(m_hwnd, TranslateT("Are you sure?"), TranslateT("Tags Mask Reset"), MB_YESNO | MB_ICONWARNING) == IDYES)
		m_tagedit.SetText(TAGSDEFAULT);
}

void CFeedEditor::OnHelp(CCtrlBase*)
{
	CMStringW wszTagHelp;
	wszTagHelp.Format(L"%s - %s\n%s - %s\n%s - %s\n%s - %s\n%s - %s\n%s - %s\n%s - %s",
		L"#<title>#", TranslateT("The title of the item."),
		L"#<description>#", TranslateT("The item synopsis."),
		L"#<link>#", TranslateT("The URL of the item."),
		L"#<author>#", TranslateT("Email address of the author of the item."),
		L"#<comments>#", TranslateT("URL of a page for comments relating to the item."),
		L"#<guid>#", TranslateT("A string that uniquely identifies the item."),
		L"#<category>#", TranslateT("Specify one or more categories that the item belongs to."));
	MessageBox(m_hwnd, wszTagHelp, TranslateT("Feed Tag Help"), MB_OK);
}

void CFeedEditor::OnOk(CCtrlBase*)
{
	ptrW strfeedtitle(m_feedtitle.GetText());
	if (!strfeedtitle || mir_wstrcmp(strfeedtitle, L"") == 0) {
		MessageBox(m_hwnd, TranslateT("Enter Feed name"), TranslateT("Error"), MB_OK);
		return;
	}

	ptrW strfeedurl(m_feedurl.GetText());
	if (!strfeedurl || mir_wstrcmp(strfeedurl, L"http://") == 0 || mir_wstrcmp(strfeedurl, L"") == 0) {
		MessageBox(m_hwnd, TranslateT("Enter Feed URL"), TranslateT("Error"), MB_OK);
		return;
	}

	ptrW strtagedit(m_tagedit.GetText());
	if (!strtagedit || mir_wstrcmp(strtagedit, L"") == 0) {
		MessageBox(m_hwnd, TranslateT("Enter message format"), TranslateT("Error"), MB_OK);
		return;
	}

	MCONTACT hContact;
	if (m_iItem == -1 && m_hContact == NULL) {
		hContact = db_add_contact();
		Proto_AddToContact(hContact, MODULENAME);
		g_plugin.setByte(hContact, "CheckState", 1);
	}
	else hContact = m_hContact;

	g_plugin.setWString(hContact, "Nick", strfeedtitle);
	g_plugin.setWString(hContact, "URL", strfeedurl);
	g_plugin.setDword(hContact, "UpdateTime", m_checktime.GetInt());
	g_plugin.setWString(hContact, "MsgFormat", strtagedit);
	g_plugin.setWord(hContact, "Status", Proto_GetStatus(MODULENAME));
	if (m_useauth.IsChecked()) {
		g_plugin.setByte(hContact, "UseAuth", 1);
		g_plugin.setWString(hContact, "Login", m_login.GetText());
		g_plugin.setString(hContact, "Password", m_password.GetTextA());
	}
	else {
		g_plugin.delSetting(hContact, "UseAuth");
		g_plugin.delSetting(hContact, "Login");
		g_plugin.delSetting(hContact, "Password");
	}

	if (m_iItem == -1 && m_list != nullptr && m_hContact == NULL) {
		int iItem = m_list->AddItem(strfeedtitle, -1);
		m_list->SetItem(iItem, 1, strfeedurl);
		m_list->SetCheckState(iItem, 1);
	}
	else if (m_iItem > -1) {
		m_list->SetItem(m_iItem, 0, strfeedtitle);
		m_list->SetItem(m_iItem, 1, strfeedurl);
	}
}

bool CFeedEditor::OnClose()
{
	g_arFeeds.remove(this);
	Utils_SaveWindowPosition(m_hwnd, NULL, MODULENAME, m_iItem == -1 ? "AddDlg" : "ChangeDlg");
	if (pAddFeedDialog == this)
		pAddFeedDialog = nullptr;
	return true;
}

void CFeedEditor::OnUseAuth(CCtrlBase*)
{
	m_login.Enable(m_useauth.GetState());
	m_password.Enable(m_useauth.GetState());
}

void COptionsMain::UpdateList()
{
	for (auto &hContact : Contacts(MODULENAME)) {
		UpdateListFlag = TRUE;
		ptrW ptszNick(g_plugin.getWStringA(hContact, "Nick"));
		if (ptszNick) {
			int iItem = m_feeds.AddItem(ptszNick, -1);

			ptrW ptszURL(g_plugin.getWStringA(hContact, "URL"));
			if (ptszURL) {
				m_feeds.SetItem(iItem, 1, ptszURL);
				m_feeds.SetCheckState(iItem, g_plugin.getByte(hContact, "CheckState", 1));
			}
		}
	}
	UpdateListFlag = FALSE;
}

COptionsMain::COptionsMain() :
	CDlgBase(g_plugin, IDD_OPTIONS),
	m_feeds(this, IDC_FEEDLIST),
	m_add(this, IDC_ADD),
	m_change(this, IDC_CHANGE),
	m_delete(this, IDC_REMOVE),
	m_import(this, IDC_IMPORT),
	m_export(this, IDC_EXPORT),
	m_checkonstartup(this, IDC_STARTUPRETRIEVE)
{
	CreateLink(m_checkonstartup, "StartupRetrieve", DBVT_BYTE, 1);

	m_add.OnClick = Callback(this, &COptionsMain::OnAddButtonClick);
	m_change.OnClick = Callback(this, &COptionsMain::OnChangeButtonClick);
	m_delete.OnClick = Callback(this, &COptionsMain::OnDeleteButtonClick);
	m_import.OnClick = Callback(this, &COptionsMain::OnImportButtonClick);
	m_export.OnClick = Callback(this, &COptionsMain::OnExportButtonClick);

	m_feeds.OnItemChanged = Callback(this, &COptionsMain::OnFeedListItemChanged);
	m_feeds.OnDoubleClick = Callback(this, &COptionsMain::OnFeedListDoubleClick);

}

bool COptionsMain::OnInitDialog()
{
	CDlgBase::OnInitDialog();
	m_change.Disable();
	m_delete.Disable();
	m_feeds.SetExtendedListViewStyle(LVS_EX_FULLROWSELECT | LVS_EX_CHECKBOXES);
	m_feeds.AddColumn(0, TranslateT("Feed"), 160);
	m_feeds.AddColumn(1, TranslateT("URL"), 276);
	UpdateList();
	return true;
}

bool COptionsMain::OnApply()
{
	for (auto &hContact : Contacts(MODULENAME)) {
		ptrW dbNick(g_plugin.getWStringA(hContact, "Nick"));
		for (int i = 0; i < m_feeds.GetItemCount(); i++) {
			wchar_t nick[MAX_PATH];
			m_feeds.GetItemText(i, 0, nick, _countof(nick));
			if (mir_wstrcmp(dbNick, nick) == 0) {
				g_plugin.setByte(hContact, "CheckState", m_feeds.GetCheckState(i));
				if (!m_feeds.GetCheckState(i))
					db_set_b(hContact, "CList", "Hidden", 1);
				else
					db_unset(hContact, "CList", "Hidden");
			}
		}
	}
	return true;
}

void COptionsMain::OnAddButtonClick(CCtrlBase*)
{
	if (pAddFeedDialog == nullptr) {
		pAddFeedDialog = new CFeedEditor(-1, &m_feeds, NULL);
		pAddFeedDialog->SetParent(m_hwnd);
		pAddFeedDialog->Show();
	}
	else {
		SetForegroundWindow(pAddFeedDialog->GetHwnd());
		SetFocus(pAddFeedDialog->GetHwnd());
	}
}

void COptionsMain::OnChangeButtonClick(CCtrlBase*)
{
	int isel = m_feeds.GetSelectionMark();
	CFeedEditor *pDlg = nullptr;
	for (auto &it : g_arFeeds) {
		wchar_t nick[MAX_PATH], url[MAX_PATH];
		m_feeds.GetItemText(isel, 0, nick, _countof(nick));
		m_feeds.GetItemText(isel, 1, url, _countof(url));

		ptrW dbNick(g_plugin.getWStringA(it->getContact(), "Nick"));
		if ((dbNick == NULL) || (mir_wstrcmp(dbNick, nick) != 0))
			continue;

		ptrW dbURL(g_plugin.getWStringA(it->getContact(), "URL"));
		if ((dbURL == NULL) || (mir_wstrcmp(dbURL, url) != 0))
			continue;

		pDlg = it;
	}

	if (pDlg == nullptr) {
		pDlg = new CFeedEditor(isel, &m_feeds, NULL);
		pDlg->SetParent(m_hwnd);
		pDlg->Show();
	}
	else {
		SetForegroundWindow(pDlg->GetHwnd());
		SetFocus(pDlg->GetHwnd());
	}
}

void COptionsMain::OnDeleteButtonClick(CCtrlBase*)
{
	if (MessageBox(m_hwnd, TranslateT("Are you sure?"), TranslateT("Contact deleting"), MB_YESNO | MB_ICONWARNING) == IDYES) {
		wchar_t nick[MAX_PATH], url[MAX_PATH];
		int isel = m_feeds.GetSelectionMark();
		m_feeds.GetItemText(isel, 0, nick, _countof(nick));
		m_feeds.GetItemText(isel, 1, url, _countof(url));

		for (auto &hContact : Contacts(MODULENAME)) {
			ptrW dbNick(g_plugin.getWStringA(hContact, "Nick"));
			if (dbNick == NULL)
				break;
			if (mir_wstrcmp(dbNick, nick))
				continue;

			ptrW dbURL(g_plugin.getWStringA(hContact, "URL"));
			if (dbURL == NULL)
				break;
			if (mir_wstrcmp(dbURL, url))
				continue;

			db_delete_contact(hContact);
			m_feeds.DeleteItem(isel);
			break;
		}
	}
}

void COptionsMain::OnImportButtonClick(CCtrlBase*)
{
	if (pImportDialog == nullptr) {
		pImportDialog = new CImportFeed(&m_feeds);
		pImportDialog->Show();
		pImportDialog->SetParent(m_hwnd);
	}
}

void COptionsMain::OnExportButtonClick(CCtrlBase*)
{
	if (pExportDialog == nullptr) {
		pExportDialog = new CExportFeed();
		pExportDialog->Show();
		pExportDialog->SetParent(m_hwnd);
	}
}

void COptionsMain::OnFeedListItemChanged(CCtrlListView::TEventInfo *evt)
{
	int isel = m_feeds.GetSelectionMark();
	if (isel == -1) {
		m_change.Disable();
		m_delete.Disable();
	}
	else {
		m_change.Enable();
		m_delete.Enable();
	}
	if (((evt->nmlv->uNewState ^ evt->nmlv->uOldState) & LVIS_STATEIMAGEMASK) && !UpdateListFlag)
		NotifyChange();
}

void COptionsMain::OnFeedListDoubleClick(CCtrlBase*)
{
	int isel = m_feeds.GetHotItem();
	if (isel != -1) {
		CFeedEditor *pDlg = new CFeedEditor(isel, &m_feeds, 0);
		pDlg->SetParent(m_hwnd);
		pDlg->Show();
	}
}

int OptInit(WPARAM wParam, LPARAM)
{
	OPTIONSDIALOGPAGE odp = {};
	odp.flags = ODPF_BOLDGROUPS | ODPF_UNICODE;
	odp.szGroup.w = LPGENW("Network");
	odp.szTitle.w = LPGENW("News Aggregator");
	odp.pDialog = new COptionsMain();
	g_plugin.addOptions(wParam, &odp);
	return 0;
}