OrcCommand: UtilitiesMain: add OutputOption overloads

DFIR-ORC · Nov 9, 2020 · aec3c11 · aec3c11
1 parent 5bf2f6f
commit aec3c11
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/src/OrcCommand/UtilitiesMain.h b/src/OrcCommand/UtilitiesMain.h
@@ -542,11 +542,37 @@ class ORCLIB_API UtilitiesMain
     // Option handling
     //
     bool OutputOption(LPCWSTR szArg, LPCWSTR szOption, OutputSpec::Kind supportedTypes, OutputSpec& anOutput);
+
+    template <typename OptionType>
+    bool
+    OutputOption(LPCWSTR szArg, LPCWSTR szOption, OutputSpec::Kind supportedTypes, std::optional<OptionType>& parameter)
+    {
+        OptionType result;
+        if (OutputOption(szArg, szOption, supportedTypes, result))
+        {
+            parameter.emplace(std::move(result));
+            return true;
+        }
+        return false;
+    }
+
     bool OutputOption(LPCWSTR szArg, LPCWSTR szOption, OutputSpec& anOutput)
     {
         return OutputOption(szArg, szOption, anOutput.supportedTypes, anOutput);
     };
 
+    template <typename OptionType>
+    bool OutputOption(LPCWSTR szArg, LPCWSTR szOption, std::optional<OptionType> parameter)
+    {
+        OptionType result;
+        if (OutputOption(szArg, szOption, result))
+        {
+            parameter.emplace(std::move(result));
+            return true;
+        }
+        return false;
+    }
+
     bool OutputFileOption(LPCWSTR szArg, LPCWSTR szOption, std::wstring& strOutputFile);
     template <typename OptionType>
     bool OutputFileOption(LPCWSTR szArg, LPCWSTR szOption, std::optional<OptionType> parameter)