#include "stdafx.h"
#include "dlgoption.h"

/*
 * DlgOption::SubOutput
 */

DlgOption::SubOutput::SubOutput() :
	m_hRemoveEmptyContacts(nullptr),
	m_hRemoveInChatsZero(nullptr),
	m_hRemoveInBytesZero(nullptr),
	m_hRemoveOutChatsZero(nullptr),
	m_hRemoveOutBytesZero(nullptr),
	m_hOmitContacts(nullptr),
	m_hOmitByValue(nullptr),
	m_hOmitByValueData(nullptr),
	m_hOmitByValueLimit(nullptr),
	m_hOmitByTime(nullptr),
	m_hOmitByTimeDays(nullptr),
	m_hOmitByRank(nullptr),
	m_hOmitNumOnTop(nullptr),
	m_hOmittedInTotals(nullptr),
	m_hOmittedInExtraRow(nullptr),
	m_hCalcTotals(nullptr),
	m_hTableHeader(nullptr),
	m_hTableHeaderRepeat(nullptr),
	m_hTableHeaderVerbose(nullptr),
	m_hHeaderTooltips(nullptr),
	m_hHeaderTooltipsIfCustom(nullptr),
	m_hSort(nullptr),
	m_hNick(nullptr),
	m_hOutputVariables(nullptr),
	m_hOutputFile(nullptr),
	m_hOutputExtraToFolder(nullptr),
	m_hOutputExtraFolder(nullptr),
	m_hOverwriteAlways(nullptr),
	m_hAutoOpenOptions(nullptr),
	m_hAutoOpenStartup(nullptr),
	m_hAutoOpenMenu(nullptr)
{
	array_each_(i, m_hSortBy)
	{
		m_hSortBy[i] = nullptr;
	}

	array_each_(i, m_hSortDir)
	{
		m_hSortDir[i] = nullptr;
	}
}

DlgOption::SubOutput::~SubOutput()
{
}

BOOL DlgOption::SubOutput::handleMsg(UINT msg, WPARAM, LPARAM lParam)
{
	switch (msg) {
	case WM_WINDOWPOSCHANGED:
		{
			RECT rClient, rWnd;

			GetClientRect(getHWnd(), &rClient);

			// options tree
			rWnd = utils::getWindowRect(getHWnd(), m_Options);
			rWnd.right = rClient.right;
			rWnd.bottom = rClient.bottom;
			utils::moveWindow(m_Options, rWnd);
		}
		break;

	case WM_NOTIFY:
		NMHDR* p = reinterpret_cast<NMHDR*>(lParam);

		if (p->idFrom == IDC_OPTIONS) {
			if (p->code == OptionsCtrl::OCN_MODIFIED) {
				OptionsCtrl::NMOPTIONSCTRL* pNM = reinterpret_cast<OptionsCtrl::NMOPTIONSCTRL*>(lParam);

				onChanged(pNM->hItem);
				return TRUE;
			}
		}
		break;
	}

	return FALSE;
}

void DlgOption::SubOutput::onWMInitDialog()
{
	TranslateDialogDefault(getHWnd());

	// init option tree(s)
	m_Options << GetDlgItem(getHWnd(), IDC_OPTIONS);

	// settings
	OptionsCtrl::Item hTemp;
	OptionsCtrl::Item hTemp2;

	hTemp                              = m_Options.insertGroup(nullptr, TranslateT("Contact filtering and totals"), OptionsCtrl::OCF_ROOTGROUP | OptionsCtrl::OCF_NODISABLECHILDS);
		m_hRemoveEmptyContacts          = m_Options.insertCheck(hTemp, TranslateT("Remove contacts with empty history"));
		m_hRemoveOutChatsZero           = m_Options.insertCheck(hTemp, TranslateT("Remove contacts with only incoming chats"), OptionsCtrl::OCF_DISABLECHILDSONUNCHECK);
			m_hRemoveOutBytesZero        = m_Options.insertCheck(m_hRemoveOutChatsZero, TranslateT("Remove only if you never answered"));
		m_hRemoveInChatsZero            = m_Options.insertCheck(hTemp, TranslateT("Remove contacts with only outgoing chats"), OptionsCtrl::OCF_DISABLECHILDSONUNCHECK);
			m_hRemoveInBytesZero         = m_Options.insertCheck(m_hRemoveInChatsZero, TranslateT("Remove only if contact never answered"));
		m_hOmitContacts                 = m_Options.insertCheck(hTemp, TranslateT("Limit number of contacts in statistics"), OptionsCtrl::OCF_DISABLECHILDSONUNCHECK);
			hTemp2                       = m_Options.insertGroup(m_hOmitContacts, TranslateT("Criteria"));
				m_hOmitByValue            = m_Options.insertCheck(hTemp2, TranslateT("Omit contacts that didn't produce a certain amount of data"), OptionsCtrl::OCF_DISABLECHILDSONUNCHECK);
					m_hOmitByValueData     = m_Options.insertCombo(m_hOmitByValue, TranslateT("Omit if"));
					m_hOmitByValueLimit    = m_Options.insertEdit(m_hOmitByValue, TranslateT("...is less than"), L"", OptionsCtrl::OCF_NUMBER);
				m_hOmitByTime             = m_Options.insertCheck(hTemp2, TranslateT("Omit contacts that were inactive for some time"), OptionsCtrl::OCF_DISABLECHILDSONUNCHECK);
					m_hOmitByTimeDays      = m_Options.insertEdit(m_hOmitByTime, TranslateT("Maximum inactivity time (days)"), L"", OptionsCtrl::OCF_NUMBER);
				m_hOmitByRank             = m_Options.insertCheck(hTemp2, TranslateT("Omit all contacts not in \"Top n\""), OptionsCtrl::OCF_DISABLECHILDSONUNCHECK);
					m_hOmitNumOnTop        = m_Options.insertEdit(m_hOmitByRank, TranslateT("Number of contacts in \"Top n\""), L"10", OptionsCtrl::OCF_NUMBER);
			m_hOmittedInTotals           = m_Options.insertCheck(m_hOmitContacts, TranslateT("Include omitted contacts in totals"));
			m_hOmittedInExtraRow         = m_Options.insertCheck(m_hOmitContacts, TranslateT("Include totals of omitted contacts in additional row"));
		m_hCalcTotals                   = m_Options.insertCheck(hTemp, TranslateT("Include totals in statistics"));
	hTemp                              = m_Options.insertGroup(nullptr, TranslateT("Table header"), OptionsCtrl::OCF_ROOTGROUP | OptionsCtrl::OCF_NODISABLECHILDS);
		m_hTableHeader                  = m_Options.insertCheck(hTemp, TranslateT("Output header"), OptionsCtrl::OCF_DISABLECHILDSONUNCHECK | OptionsCtrl::OCF_DISABLED | OptionsCtrl::OCF_NODISABLECHILDS);
			m_hTableHeaderRepeat         = m_Options.insertEdit(m_hTableHeader, TranslateT("Repeat header every n contacts (0=don't repeat)"), L"0", OptionsCtrl::OCF_NUMBER);
			m_hTableHeaderVerbose        = m_Options.insertCheck(m_hTableHeader, TranslateT("Make column titles more verbose"));
			m_hHeaderTooltips            = m_Options.insertCheck(m_hTableHeader, TranslateT("Show tooltips with detailed information in column titles"), OptionsCtrl::OCF_DISABLECHILDSONUNCHECK);
				m_hHeaderTooltipsIfCustom = m_Options.insertCheck(m_hHeaderTooltips, TranslateT("Only show if a custom title was entered or if titles are not verbose"));
	m_hSort                            = m_Options.insertGroup(nullptr, TranslateT("Sorting"), OptionsCtrl::OCF_ROOTGROUP);
	hTemp                              = m_Options.insertGroup(nullptr, TranslateT("HTML file generation"), OptionsCtrl::OCF_ROOTGROUP);
		m_hNick                         = m_Options.insertEdit(hTemp, TranslateT("Own nick for statistics"));
	    hTemp2                         = m_Options.insertGroup(hTemp, TranslateT("Output files and folders"));
			m_hOutputVariables           = m_Options.insertCheck(hTemp2, TranslateT("Substitute variables in output file name and subfolder for additional files"));
			m_hOutputFile                = m_Options.insertEdit(hTemp2, TranslateT("Output file"));
			m_hOutputExtraToFolder       = m_Options.insertCheck(hTemp2, TranslateT("Output additional files to subfolder"), OptionsCtrl::OCF_DISABLECHILDSONUNCHECK);
				m_hOutputExtraFolder      = m_Options.insertEdit(m_hOutputExtraToFolder, TranslateT("Subfolder for additional files"));
			m_hOverwriteAlways           = m_Options.insertCheck(hTemp2, TranslateT("Always overwrite already existing files (dangerous!)"));
		hTemp2                          = m_Options.insertGroup(hTemp, TranslateT("Auto open statistics after being generated..."));
			m_hAutoOpenOptions           = m_Options.insertCheck(hTemp2, TranslateT("...via button \"Create statistics\" in options"));
			m_hAutoOpenStartup           = m_Options.insertCheck(hTemp2, TranslateT("...on Miranda NG startup"));
			m_hAutoOpenMenu              = m_Options.insertCheck(hTemp2, TranslateT("...via Miranda NG main menu"));

	// our sort levels
	static const wchar_t* sortLevels[Settings::cNumSortLevels] = {
		LPGENW("First sort by"),
		LPGENW("Then sort by"),
		LPGENW("Finally sort by")
	};

	// our sort keys
	static const struct {
		int key;
		wchar_t* desc;
	} sortKeys[] = {
		{ Settings::skNick              , LPGENW("Nick")                                  },
		{ Settings::skProtocol          , LPGENW("Protocol")                              },
		{ Settings::skGroup             , LPGENW("Group")                                 },
		{ Settings::skBytesIn           , LPGENW("Characters (incoming, absolute)")       },
		{ Settings::skBytesOut          , LPGENW("Characters (outgoing, absolute)")       },
		{ Settings::skBytesTotal        , LPGENW("Characters (all, absolute)")            },
		{ Settings::skBytesInAvg        , LPGENW("Characters (incoming, average)")        },
		{ Settings::skBytesOutAvg       , LPGENW("Characters (outgoing, average)")        },
		{ Settings::skBytesTotalAvg     , LPGENW("Characters (all, average)")             },
		{ Settings::skMessagesIn        , LPGENW("Messages (incoming, absolute)")         },
		{ Settings::skMessagesOut       , LPGENW("Messages (outgoing, absolute)")         },
		{ Settings::skMessagesTotal     , LPGENW("Messages (all, absolute)")              },
		{ Settings::skMessagesInAvg     , LPGENW("Messages (incoming, average)")          },
		{ Settings::skMessagesOutAvg    , LPGENW("Messages (outgoing, average)")          },
		{ Settings::skMessagesTotalAvg  , LPGENW("Messages (all, average)")               },
		{ Settings::skChatsIn           , LPGENW("Chats (incoming, absolute)")            },
		{ Settings::skChatsOut          , LPGENW("Chats (outgoing, absolute)")            },
		{ Settings::skChatsTotal        , LPGENW("Chats (all, absolute)")                 },
		{ Settings::skChatsInAvg        , LPGENW("Chats (incoming, average)")             },
		{ Settings::skChatsOutAvg       , LPGENW("Chats (outgoing, average)")             },
		{ Settings::skChatsTotalAvg     , LPGENW("Chats (all, average)")                  },
		{ Settings::skChatDurationTotal , LPGENW("Chat duration (total)")                 },
		{ Settings::skChatDurationMin   , LPGENW("Chat duration (minimum)")               },
		{ Settings::skChatDurationAvg   , LPGENW("Chat duration (average)")               },
		{ Settings::skChatDurationMax   , LPGENW("Chat duration (maximum)")               },
		{ Settings::skTimeOfFirstMessage, LPGENW("Time of first message to/from contact") },
		{ Settings::skTimeOfLastMessage , LPGENW("Time of last message to/from contact")  },
	};

	m_SortKeyToIndex.resize(Settings::skLAST - Settings::skFIRST + 1);
	m_IndexToSortKey.resize(_countof(sortKeys));

	array_each_(i, sortKeys)
	{
		m_IndexToSortKey[i] = sortKeys[i].key;
		m_SortKeyToIndex[sortKeys[i].key] = i;
	}

	array_each_(i, sortLevels)
	{
		m_hSortBy[i] = m_Options.insertCombo(m_hSort, TranslateW(sortLevels[i]), (i == 0) ? 0 : OptionsCtrl::OCF_DISABLECHILDSONINDEX0);

		if (i != 0)
		{
			m_Options.addComboItem(m_hSortBy[i], TranslateT("(nothing)"));
		}

		array_each_(j, sortKeys)
		{
			m_Options.addComboItem(m_hSortBy[i], TranslateW(sortKeys[j].desc));
		}

		m_hSortDir[i] = m_Options.insertRadio(m_hSortBy[i], nullptr, TranslateT("Ascending"));
		                m_Options.insertRadio(m_hSortBy[i], m_hSortDir[i], TranslateT("Descending"));
	}

	// our "omit by value" data
	static const wchar_t* omitData[] = {
		LPGENW("Characters (incoming, absolute)"),
		LPGENW("Characters (outgoing, absolute)"),
		LPGENW("Characters (all, absolute)"),
		LPGENW("Characters (incoming, average per week)"),
		LPGENW("Characters (outgoing, average per week)"),
		LPGENW("Characters (all, average per week)"),
		LPGENW("Messages (incoming, absolute)"),
		LPGENW("Messages (outgoing, absolute)"),
		LPGENW("Messages (all, absolute)"),
		LPGENW("Messages (incoming, average per week)"),
		LPGENW("Messages (outgoing, average per week)"),
		LPGENW("Messages (all, average per week)"),
		LPGENW("Chats (incoming, absolute)"),
		LPGENW("Chats (outgoing, absolute)"),
		LPGENW("Chats (all, absolute)"),
		LPGENW("Chats (incoming, average per week)"),
		LPGENW("Chats (outgoing, average per week)"),
		LPGENW("Chats (all, average per week)"),
		LPGENW("Chat duration (total, hours)"),
	};

	array_each_(i, omitData)
	{
		m_Options.addComboItem(m_hOmitByValueData, TranslateW(omitData[i]));
	}

	m_Options.ensureVisible(nullptr);
}

void DlgOption::SubOutput::loadSettings()
{
	// read settings from local settings store
	Settings& localS = getParent()->getLocalSettings();

	m_Options.checkItem       (m_hRemoveEmptyContacts   , localS.m_RemoveEmptyContacts      );
	m_Options.checkItem       (m_hRemoveOutChatsZero    , localS.m_RemoveOutChatsZero       );
	m_Options.checkItem       (m_hRemoveOutBytesZero    , localS.m_RemoveOutBytesZero       );
	m_Options.checkItem       (m_hRemoveInChatsZero     , localS.m_RemoveInChatsZero        );
	m_Options.checkItem       (m_hRemoveInBytesZero     , localS.m_RemoveInBytesZero        );
	m_Options.checkItem       (m_hOmitContacts          , localS.m_OmitContacts             );
	m_Options.checkItem       (m_hOmitByValue           , localS.m_OmitByValue              );
	m_Options.setComboSelected(m_hOmitByValueData       , localS.m_OmitByValueData          );
	m_Options.setEditNumber   (m_hOmitByValueLimit      , localS.m_OmitByValueLimit         );
	m_Options.checkItem       (m_hOmitByTime            , localS.m_OmitByTime               );
	m_Options.setEditNumber   (m_hOmitByTimeDays        , localS.m_OmitByTimeDays           );
	m_Options.checkItem       (m_hOmitByRank            , localS.m_OmitByRank               );
	m_Options.setEditNumber   (m_hOmitNumOnTop          , localS.m_OmitNumOnTop             );
	m_Options.checkItem       (m_hOmittedInTotals       , localS.m_OmittedInTotals          );
	m_Options.checkItem       (m_hOmittedInExtraRow     , localS.m_OmittedInExtraRow        );
	m_Options.checkItem       (m_hCalcTotals            , localS.m_CalcTotals               );
	m_Options.checkItem       (m_hTableHeader           , localS.m_TableHeader              );
	m_Options.setEditNumber   (m_hTableHeaderRepeat     , localS.m_TableHeaderRepeat        );
	m_Options.checkItem       (m_hTableHeaderVerbose    , localS.m_TableHeaderVerbose       );
	m_Options.checkItem       (m_hHeaderTooltips        , localS.m_HeaderTooltips           );
	m_Options.checkItem       (m_hHeaderTooltipsIfCustom, localS.m_HeaderTooltipsIfCustom   );
	m_Options.setEditString   (m_hNick                  , localS.m_OwnNick.c_str()          );
	m_Options.checkItem       (m_hOutputVariables       , localS.m_OutputVariables          );
	m_Options.setEditString   (m_hOutputFile            , localS.m_OutputFile.c_str()       );
	m_Options.checkItem       (m_hOutputExtraToFolder   , localS.m_OutputExtraToFolder      );
	m_Options.setEditString   (m_hOutputExtraFolder     , localS.m_OutputExtraFolder.c_str());
	m_Options.checkItem       (m_hOverwriteAlways       , localS.m_OverwriteAlways          );
	m_Options.checkItem       (m_hAutoOpenOptions       , localS.m_AutoOpenOptions          );
	m_Options.checkItem       (m_hAutoOpenStartup       , localS.m_AutoOpenStartup          );
	m_Options.checkItem       (m_hAutoOpenMenu          , localS.m_AutoOpenMenu             );

	// our sort keys
	upto_each_(i, Settings::cNumSortLevels)
	{
		int by = (localS.m_Sort[i].by >= 0 && localS.m_Sort[i].by < m_SortKeyToIndex.size()) ? m_SortKeyToIndex[localS.m_Sort[i].by] : -1;

		m_Options.setComboSelected(m_hSortBy[i], by + ((i != 0) ? 1 : 0));
		m_Options.setRadioChecked(m_hSortDir[i], localS.m_Sort[i].asc ? 0 : 1);
	}

	// update UI
	onChanged(m_hSortBy[1]);
	onChanged(m_hCalcTotals);
}

void DlgOption::SubOutput::saveSettings()
{
	Settings& localS = getParent()->getLocalSettings();

	localS.m_RemoveEmptyContacts    = m_Options.isItemChecked   (m_hRemoveEmptyContacts   );
	localS.m_RemoveOutChatsZero     = m_Options.isItemChecked   (m_hRemoveOutChatsZero    );
	localS.m_RemoveOutBytesZero     = m_Options.isItemChecked   (m_hRemoveOutBytesZero    );
	localS.m_RemoveInChatsZero      = m_Options.isItemChecked   (m_hRemoveInChatsZero     );
	localS.m_RemoveInBytesZero      = m_Options.isItemChecked   (m_hRemoveInBytesZero     );
	localS.m_OmitContacts           = m_Options.isItemChecked   (m_hOmitContacts          );
	localS.m_OmitByValue            = m_Options.isItemChecked   (m_hOmitByValue           );
	localS.m_OmitByValueData        = m_Options.getComboSelected(m_hOmitByValueData       );
	localS.m_OmitByValueLimit       = m_Options.getEditNumber   (m_hOmitByValueLimit      );
	localS.m_OmitByTime             = m_Options.isItemChecked   (m_hOmitByTime            );
	localS.m_OmitByTimeDays         = m_Options.getEditNumber   (m_hOmitByTimeDays        );
	localS.m_OmitByRank             = m_Options.isItemChecked   (m_hOmitByRank            );
	localS.m_OmitNumOnTop           = m_Options.getEditNumber   (m_hOmitNumOnTop          );
	localS.m_OmittedInTotals        = m_Options.isItemChecked   (m_hOmittedInTotals       );
	localS.m_OmittedInExtraRow      = m_Options.isItemChecked   (m_hOmittedInExtraRow     );
	localS.m_CalcTotals             = m_Options.isItemChecked   (m_hCalcTotals            );
	localS.m_TableHeader            = m_Options.isItemChecked   (m_hTableHeader           );
	localS.m_TableHeaderRepeat      = m_Options.getEditNumber   (m_hTableHeaderRepeat     );
	localS.m_TableHeaderVerbose     = m_Options.isItemChecked   (m_hTableHeaderVerbose    );
	localS.m_HeaderTooltips         = m_Options.isItemChecked   (m_hHeaderTooltips        );
	localS.m_HeaderTooltipsIfCustom = m_Options.isItemChecked   (m_hHeaderTooltipsIfCustom);

	upto_each_(i, Settings::cNumSortLevels)
	{
		int index = m_Options.getComboSelected(m_hSortBy[i]) - ((i != 0) ? 1 : 0);

		localS.m_Sort[i].by = (index >= 0 && index < m_IndexToSortKey.size()) ? m_IndexToSortKey[index] : -1;
		localS.m_Sort[i].asc = (m_Options.getRadioChecked(m_hSortDir[i]) == 0);
	}

	localS.m_OwnNick             = m_Options.getEditString(m_hNick               );
	localS.m_OutputVariables     = m_Options.isItemChecked(m_hOutputVariables    );
	localS.m_OutputFile          = m_Options.getEditString(m_hOutputFile         );
	localS.m_OutputExtraToFolder = m_Options.isItemChecked(m_hOutputExtraToFolder);
	localS.m_OutputExtraFolder   = m_Options.getEditString(m_hOutputExtraFolder  );
	localS.m_OverwriteAlways     = m_Options.isItemChecked(m_hOverwriteAlways    );
	localS.m_AutoOpenOptions     = m_Options.isItemChecked(m_hAutoOpenOptions    );
	localS.m_AutoOpenStartup     = m_Options.isItemChecked(m_hAutoOpenStartup    );
	localS.m_AutoOpenMenu        = m_Options.isItemChecked(m_hAutoOpenMenu       );
}

void DlgOption::SubOutput::onChanged(HANDLE hItem)
{
	if (hItem == m_hSortBy[1])
		m_Options.enableItem(m_hSortBy[2], m_Options.getComboSelected(m_hSortBy[1]) != 0);
	else if (hItem == m_hCalcTotals || hItem == m_hOmitContacts)
		m_Options.enableItem(m_hOmittedInTotals, m_Options.isItemChecked(m_hCalcTotals) && m_Options.isItemChecked(m_hOmitContacts));

	getParent()->settingsChanged();
}