ArgsMan: Add Get*Arg functions returning optional

This allows the caller to not provide a default at all and just check
inside the optional to see if the arg was set or not.
This commit is contained in:
Carl Dong 2021-10-24 02:26:19 -04:00
parent 57a491bee1
commit fc02f77ca6
2 changed files with 54 additions and 6 deletions

View File

@ -612,36 +612,76 @@ bool ArgsManager::IsArgNegated(const std::string& strArg) const
} }
std::string ArgsManager::GetArg(const std::string& strArg, const std::string& strDefault) const std::string ArgsManager::GetArg(const std::string& strArg, const std::string& strDefault) const
{
return GetArg(strArg).value_or(strDefault);
}
std::optional<std::string> ArgsManager::GetArg(const std::string& strArg) const
{ {
const util::SettingsValue value = GetSetting(strArg); const util::SettingsValue value = GetSetting(strArg);
return SettingToString(value, strDefault); return SettingToString(value);
}
std::optional<std::string> SettingToString(const util::SettingsValue& value)
{
if (value.isNull()) return std::nullopt;
if (value.isFalse()) return "0";
if (value.isTrue()) return "1";
if (value.isNum()) return value.getValStr();
return value.get_str();
} }
std::string SettingToString(const util::SettingsValue& value, const std::string& strDefault) std::string SettingToString(const util::SettingsValue& value, const std::string& strDefault)
{ {
return value.isNull() ? strDefault : value.isFalse() ? "0" : value.isTrue() ? "1" : value.isNum() ? value.getValStr() : value.get_str(); return SettingToString(value).value_or(strDefault);
} }
int64_t ArgsManager::GetIntArg(const std::string& strArg, int64_t nDefault) const int64_t ArgsManager::GetIntArg(const std::string& strArg, int64_t nDefault) const
{
return GetIntArg(strArg).value_or(nDefault);
}
std::optional<int64_t> ArgsManager::GetIntArg(const std::string& strArg) const
{ {
const util::SettingsValue value = GetSetting(strArg); const util::SettingsValue value = GetSetting(strArg);
return SettingToInt(value, nDefault); return SettingToInt(value);
}
std::optional<int64_t> SettingToInt(const util::SettingsValue& value)
{
if (value.isNull()) return std::nullopt;
if (value.isFalse()) return 0;
if (value.isTrue()) return 1;
if (value.isNum()) return value.getInt<int64_t>();
return LocaleIndependentAtoi<int64_t>(value.get_str());
} }
int64_t SettingToInt(const util::SettingsValue& value, int64_t nDefault) int64_t SettingToInt(const util::SettingsValue& value, int64_t nDefault)
{ {
return value.isNull() ? nDefault : value.isFalse() ? 0 : value.isTrue() ? 1 : value.isNum() ? value.getInt<int64_t>() : LocaleIndependentAtoi<int64_t>(value.get_str()); return SettingToInt(value).value_or(nDefault);
} }
bool ArgsManager::GetBoolArg(const std::string& strArg, bool fDefault) const bool ArgsManager::GetBoolArg(const std::string& strArg, bool fDefault) const
{
return GetBoolArg(strArg).value_or(fDefault);
}
std::optional<bool> ArgsManager::GetBoolArg(const std::string& strArg) const
{ {
const util::SettingsValue value = GetSetting(strArg); const util::SettingsValue value = GetSetting(strArg);
return SettingToBool(value, fDefault); return SettingToBool(value);
}
std::optional<bool> SettingToBool(const util::SettingsValue& value)
{
if (value.isNull()) return std::nullopt;
if (value.isBool()) return value.get_bool();
return InterpretBool(value.get_str());
} }
bool SettingToBool(const util::SettingsValue& value, bool fDefault) bool SettingToBool(const util::SettingsValue& value, bool fDefault)
{ {
return value.isNull() ? fDefault : value.isBool() ? value.get_bool() : InterpretBool(value.get_str()); return SettingToBool(value).value_or(fDefault);
} }
bool ArgsManager::SoftSetArg(const std::string& strArg, const std::string& strValue) bool ArgsManager::SoftSetArg(const std::string& strArg, const std::string& strValue)

View File

@ -161,8 +161,13 @@ struct SectionInfo
}; };
std::string SettingToString(const util::SettingsValue&, const std::string&); std::string SettingToString(const util::SettingsValue&, const std::string&);
std::optional<std::string> SettingToString(const util::SettingsValue&);
int64_t SettingToInt(const util::SettingsValue&, int64_t); int64_t SettingToInt(const util::SettingsValue&, int64_t);
std::optional<int64_t> SettingToInt(const util::SettingsValue&);
bool SettingToBool(const util::SettingsValue&, bool); bool SettingToBool(const util::SettingsValue&, bool);
std::optional<bool> SettingToBool(const util::SettingsValue&);
class ArgsManager class ArgsManager
{ {
@ -335,6 +340,7 @@ protected:
* @return command-line argument or default value * @return command-line argument or default value
*/ */
std::string GetArg(const std::string& strArg, const std::string& strDefault) const; std::string GetArg(const std::string& strArg, const std::string& strDefault) const;
std::optional<std::string> GetArg(const std::string& strArg) const;
/** /**
* Return path argument or default value * Return path argument or default value
@ -356,6 +362,7 @@ protected:
* @return command-line argument (0 if invalid number) or default value * @return command-line argument (0 if invalid number) or default value
*/ */
int64_t GetIntArg(const std::string& strArg, int64_t nDefault) const; int64_t GetIntArg(const std::string& strArg, int64_t nDefault) const;
std::optional<int64_t> GetIntArg(const std::string& strArg) const;
/** /**
* Return boolean argument or default value * Return boolean argument or default value
@ -365,6 +372,7 @@ protected:
* @return command-line argument or default value * @return command-line argument or default value
*/ */
bool GetBoolArg(const std::string& strArg, bool fDefault) const; bool GetBoolArg(const std::string& strArg, bool fDefault) const;
std::optional<bool> GetBoolArg(const std::string& strArg) const;
/** /**
* Set an argument if it doesn't already have a value * Set an argument if it doesn't already have a value