feat: add abstract arg class

Author: nader-zouaoui

parser/arg-parser-arg.cpp

@@ -0,0 +1,95 @@
+#include "arg-parser-arg.h"
+
+arg_parser_arg::arg_parser_arg(
+    const std::wstring name,
+    const std::vector<std::wstring> alias,
+    const std::wstring description,
+    const int arg_count
+) : m_name(name), m_aliases(alias), m_description(description), m_arg_count(arg_count) {};
+
+inline bool arg_parser_arg::operator==(const std::wstring& other_arg) const
+{
+    return is_match(other_arg);
+}
+
+bool arg_parser_arg::is_match(const std::wstring& other_arg) const
+{
+    return !other_arg.empty() && (other_arg == m_name || std::find(m_aliases.begin(), m_aliases.end(), other_arg) != m_aliases.end());
+}
+
+std::wstring arg_parser_arg::get_help() const
+{
+    return m_name + L" or " + get_alias_string() + L"/n         " + m_description;
+}
+
+std::wstring arg_parser_arg::get_usage_text() const
+{
+    return m_description;
+}
+
+arg_parser_arg arg_parser_arg::add_alias(std::wstring new_alias) 
+{
+    m_aliases.push_back(new_alias);
+    return *this;
+}
+
+int arg_parser_arg::get_arg_count() const
+{
+    return m_arg_count;
+}
+
+std::wstring arg_parser_arg::get_name() const
+{
+    return m_name;
+}
+
+std::wstring arg_parser_arg::get_alias_string() const
+{
+    // convert alias vector to wstring
+    std::wstring alias_string;
+    for (auto& m_alias : m_aliases) {
+        alias_string.append(m_alias);
+    }
+    return alias_string;
+}
+
+arg_parser_arg arg_parser_arg::add_check_func(std::function<bool(const std::wstring&)> check_func)
+{
+    m_check_funcs.push_back(check_func);
+    return *this;
+}
+
+void arg_parser_arg::set_is_parsed()
+{
+    m_is_parsed = true;
+}
+
+bool arg_parser_arg::parse(std::vector<std::wstring> arg_vect)
+{
+    if (m_arg_count == -1 && arg_vect.size() > 0) m_arg_count = arg_vect.size() - 1;
+
+    if (arg_vect.size() < m_arg_count + 1)
+        return false;
+
+    if (!is_match(arg_vect[0]))
+        return false;
+    if (m_arg_count == 0)
+    {
+        set_is_parsed();
+        return true;
+    }
+    
+    for (int i = 1; i < m_arg_count + 1; ++i)
+    {
+        for (auto& check_func : m_check_funcs)
+        {
+            if (!check_func(arg_vect[i]))
+                return false;
+        }
+        m_values.push_back(arg_vect[i]);
+       
+    }
+    set_is_parsed();
+    return true;
+}
+

parser/arg-parser-arg.h

@@ -0,0 +1,66 @@
+#pragma once
+#include <string>
+#include <vector>
+#include <set>
+#include <functional>
+
+struct arg_parser_arg {
+    const std::wstring m_name;
+    std::vector<std::wstring> m_aliases{};
+    const std::wstring m_description;
+
+    int m_arg_count; // -1 for variable number of arguments
+    bool m_is_parsed = false;
+    std::vector<std::wstring> m_values{};
+
+    std::vector <std::function<bool(const std::wstring&)>> m_check_funcs = {};
+
+    inline bool operator==(const std::wstring& other) const;
+
+public:
+    arg_parser_arg(
+        const std::wstring name,
+        const std::vector<std::wstring> alias,
+        const std::wstring description,
+        const int arg_count
+    );
+    bool is_match(const std::wstring& arg) const;
+    virtual std::wstring get_help() const;
+    virtual std::wstring get_usage_text() const;
+    arg_parser_arg add_alias(std::wstring new_alias);
+    int get_arg_count() const;
+    std::wstring get_name() const;
+    std::wstring get_alias_string() const;
+    arg_parser_arg add_check_func(std::function<bool(const std::wstring&)> check_func);
+    void set_is_parsed();
+    bool parse(std::vector<std::wstring> arg_vect);
+};
+
+class arg_parser_arg_opt : arg_parser_arg {
+    arg_parser_arg_opt(
+        const std::wstring name,
+        const std::vector<std::wstring> alias,
+        const std::wstring description,
+        const int arg_count
+    ) : arg_parser_arg(name, alias, description, 0) {};
+};
+
+class arg_parser_arg_pos : arg_parser_arg {
+    arg_parser_arg_pos(
+        const std::wstring name,
+        const std::vector<std::wstring> alias,
+        const std::wstring description,
+        const int arg_count
+    ) : arg_parser_arg(name, alias, description, arg_count) {};
+};
+
+class arg_parser_arg_command : arg_parser_arg {
+    arg_parser_arg_command(
+        const std::wstring name,
+        const std::vector<std::wstring> alias,
+        const std::wstring description,
+        const int arg_count,
+        const std::wstring examples
+    ) : arg_parser_arg(name, alias, description, 1) {};
+    const std::wstring m_examples;
+};

parser/arg-parser.h

@@ -37,6 +37,7 @@
 #include <string>
 #include <set>
 #include <unordered_map>
+#include "arg.h"
 
 using namespace std;
 

parser/parser.vcxproj

@@ -131,10 +131,12 @@
   </ItemDefinitionGroup>
   <ItemGroup>
     <ClCompile Include="arg-parser.cpp" />
+    <ClCompile Include="arg-parser-arg.cpp" />
     <ClCompile Include="main.cpp" />
   </ItemGroup>
   <ItemGroup>
     <ClInclude Include="arg-parser.h" />
+    <ClInclude Include="arg-parser-arg.h" />
   </ItemGroup>
   <Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" />
   <ImportGroup Label="ExtensionTargets">

parser/parser.vcxproj.filters

@@ -21,10 +21,16 @@
     <ClCompile Include="arg-parser.cpp">
       <Filter>Source Files</Filter>
     </ClCompile>
+    <ClCompile Include="arg-parser-arg.cpp">
+      <Filter>Source Files</Filter>
+    </ClCompile>
   </ItemGroup>
   <ItemGroup>
     <ClInclude Include="arg-parser.h">
       <Filter>Header Files</Filter>
     </ClInclude>
+    <ClInclude Include="arg-parser-arg.h">
+      <Filter>Header Files</Filter>
+    </ClInclude>
   </ItemGroup>
 </Project>