CWinMTRCommandLineParser.ixx

/*
WinMTR
Copyright (C)  2010-2019 Appnor MSP S.A. - http://www.appnor.com
Copyright (C) 2019-2022 Leetsoftwerx

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, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
*/

module;
#pragma warning (disable : 4005)
#include "targetver.h"
#define WIN32_LEAN_AND_MEAN
#define VC_EXTRALEAN

#include <afxwin.h>
#include <afxext.h>

export module WinMTR.CommandLineParser;

import WinMTR.Dialog;

export namespace utils {

	class CWinMTRCommandLineParser final :
		public CCommandLineInfo
	{
	public:
		CWinMTRCommandLineParser(WinMTRDialog& dlg) noexcept
			:dlg(dlg) {}
	private:

		void ParseParam(const WCHAR* pszParam, BOOL bFlag, [[maybe_unused]] BOOL bLast) noexcept override final;
#ifdef _UNICODE
		void ParseParam([[maybe_unused]] const char* pszParam, [[maybe_unused]] BOOL bFlag, [[maybe_unused]] BOOL bLast) noexcept override final
		{
		}
#endif
	private:

		WinMTRDialog& dlg;

		enum class expect_next {
			none,
			interval,
			ping_size,
			lru
		};
		expect_next next = expect_next::none;
		bool m_help = false;

	public:
		bool isAskingForHelp() const noexcept {
			return m_help;
		}
		
	};
}


module : private;

import <string_view>;
import WinMTRUtils;


void utils::CWinMTRCommandLineParser::ParseParam(const WCHAR* pszParam, BOOL bFlag, [[maybe_unused]] BOOL bLast) noexcept
{
	using namespace std::literals;
	if (bFlag) {
		if (L"h"sv == pszParam || L"-help"sv == pszParam) {
			this->m_help = true;
		}
		else if (L"n"sv == pszParam || L"-numeric"sv == pszParam) {
			this->dlg.SetUseDNS(false, WinMTRDialog::options_source::cmd_line);
		}
		else if (L"i"sv == pszParam || L"-interval"sv == pszParam) {
			this->next = expect_next::interval;
		}
		else if (L"m"sv == pszParam || L"-maxLRU"sv == pszParam) {
			this->next = expect_next::lru;
		}
		else if (L"s"sv == pszParam || L"-size"sv == pszParam) {
			this->next = expect_next::ping_size;
		}
		return;
	}
	wchar_t* end = nullptr;
	switch (this->next) {
	case expect_next::lru:
	{
		auto parsed = std::wcstol(pszParam, &end, 10);
		if (parsed < WinMTRUtils::MIN_MAX_LRU || parsed > WinMTRUtils::MAX_MAX_LRU) {
			parsed = WinMTRUtils::DEFAULT_MAX_LRU;
		}
		this->dlg.SetMaxLRU(parsed, WinMTRDialog::options_source::cmd_line);
	}
	break;
	case expect_next::interval:
	{
		auto parsed = std::wcstof(pszParam, &end);
		if (parsed > WinMTRUtils::MAX_INTERVAL || parsed < WinMTRUtils::MIN_INTERVAL) {
			parsed = WinMTRUtils::DEFAULT_INTERVAL;
		}
		this->dlg.SetInterval(parsed, WinMTRDialog::options_source::cmd_line);
	}
	break;
	case expect_next::ping_size:
	{
		auto parsed = std::wcstol(pszParam, &end, 10);
		if (parsed > WinMTRUtils::MAX_PING_SIZE || parsed < WinMTRUtils::MIN_PING_SIZE) {
			parsed = WinMTRUtils::DEFAULT_PING_SIZE;
		}
		this->dlg.SetPingSize(parsed, WinMTRDialog::options_source::cmd_line);
	}
	break;
	default:
		break;
	}
	this->next = expect_next::none;
}