command line parser

src/CMakeLists.txt

@@ -2,5 +2,11 @@ include_directories(.)
 set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11")
 set(SOURCE_FILES
   main.cpp
+  Interface/MainClass.cpp
+  Interface/MainClass.h
+  Interface/CommandLineParser.cpp
+  Interface/CommandLineParser.h
+  Interface/CommandLineOption.cpp
+  Interface/CommandLineOption.h
   )
 add_executable(pdns-slave ${SOURCE_FILES})

src/Interface/CommandLineOption.cpp

@@ -0,0 +1,98 @@
+//
+// Created by robin on 8/8/15.
+//
+
+#include "CommandLineOption.h"
+
+CommandLineOption::CommandLineOption(const std::string &longName, char shortName, const std::string &description,
+                                     const std::string &valueName, const std::string &defaultValue)
+    : _longName(longName)
+    , _shortName(shortName)
+    , _description(description)
+    , _valueName(valueName)
+    , _defaultValue(defaultValue)
+    , _isSet(false)
+{
+}
+
+const std::string &CommandLineOption::getLongName() const
+{
+    return _longName;
+}
+
+void CommandLineOption::setLongName(const std::string &longName)
+{
+    _longName = longName;
+}
+
+char CommandLineOption::getShortName() const
+{
+    return _shortName;
+}
+
+void CommandLineOption::setShortName(char shortName)
+{
+    _shortName = shortName;
+}
+
+const std::string &CommandLineOption::getDescription() const
+{
+    return _description;
+}
+
+void CommandLineOption::setDescription(const std::string &description)
+{
+    _description = description;
+}
+
+const std::string &CommandLineOption::getValueName() const
+{
+    return _valueName;
+}
+
+void CommandLineOption::setValueName(const std::string &valueName)
+{
+    _valueName = valueName;
+}
+
+const std::string &CommandLineOption::getDefaultValue() const
+{
+    return _defaultValue;
+}
+
+void CommandLineOption::setDefaultValue(const std::string &defaultValue)
+{
+    _defaultValue = defaultValue;
+}
+
+const std::vector<std::string> &CommandLineOption::getValues() const
+{
+    return _values;
+}
+
+void CommandLineOption::addValue(const std::string &value)
+{
+    _values.push_back(value);
+}
+
+bool CommandLineOption::isSet() const
+{
+    return _isSet;
+}
+
+void CommandLineOption::setIsSet(bool isSet)
+{
+    _isSet = isSet;
+}
+
+bool CommandLineOption::hasValue() const
+{
+    return !_valueName.empty();
+}
+
+const std::string &CommandLineOption::getValue() const
+{
+    if (_values.empty())
+        return _defaultValue;
+    return _values[_values.size() - 1];
+}

src/Interface/CommandLineOption.h

@@ -0,0 +1,66 @@
+//
+// Created by robin on 8/8/15.
+//
+
+#ifndef PDNS_SLAVE_COMMANDLINEOPTION_H
+#define PDNS_SLAVE_COMMANDLINEOPTION_H
+
+# include <string>
+# include <vector>
+
+class CommandLineOption
+{
+public:
+    CommandLineOption(const std::string& longName, char shortName, const std::string& description,
+                      const std::string& valueName = "", const std::string& defaultValue = "");
+
+    const std::string &getLongName() const;
+
+    void setLongName(const std::string &longName);
+
+    char getShortName() const;
+
+    void setShortName(char shortName);
+
+    const std::string &getDescription() const;
+
+    void setDescription(const std::string &description);
+
+    const std::string &getValueName() const;
+
+    void setValueName(const std::string &valueName);
+
+    const std::string &getDefaultValue() const;
+
+    void setDefaultValue(const std::string &defaultValue);
+
+    const std::vector<std::string> &getValues() const;
+
+    const std::string &getValue() const;
+
+    void addValue(const std::string &value);
+
+    bool isSet() const;
+
+    void setIsSet(bool isSet);
+
+    bool hasValue() const;
+
+private:
+    std::string _longName;
+
+    char _shortName;
+
+    std::string _description;
+
+    std::string _valueName;
+
+    std::string _defaultValue;
+
+    std::vector<std::string> _values;
+
+    bool _isSet;
+};
+
+
+#endif //PDNS_SLAVE_COMMANDLINEOPTION_H

src/Interface/CommandLineParser.cpp

@@ -0,0 +1,74 @@
+//
+// Created by robin on 8/8/15.
+//
+
+#include <getopt.h>
+#include <iostream>
+#include "CommandLineParser.h"
+
+CommandLineParser::CommandLineParser(int argc, char **argv)
+    : _argc(argc)
+    , _argv(argv)
+{
+}
+
+bool CommandLineParser::parse()
+{
+    std::string shortOpts;
+    option opts[_options.size() + 1];
+    for (unsigned i = 0; i < _options.size(); ++i)
+    {
+        auto opt = _options[i];
+        shortOpts += opt->getShortName();
+        if (opt->hasValue())
+            shortOpts += ":";
+        opts[i].name = opt->getLongName().c_str();
+        opts[i].has_arg = opt->hasValue();
+        opts[i].flag = 0;
+        opts[i].val = opt->getShortName();
+    }
+    opts[_options.size()] = {0, 0, 0, 0};
+    int option;
+    extern int optind;
+    extern char* optarg;
+    bool valid = false;
+    while ((option = getopt_long(_argc, _argv, shortOpts.c_str(), opts, 0)) != -1)
+    {
+        for (unsigned i = 0; i < _options.size(); ++i)
+        {
+            bool optValid = false;
+            auto opt = _options[i];
+            if (opt->getShortName() == option)
+            {
+                optValid = true;
+                opt->setIsSet(true);
+                if (opt->hasValue())
+                    opt->addValue(optarg);
+            }
+            if (!optValid)
+                valid = false;
+        }
+    }
+    return optind == _argc && valid;
+}
+
+void CommandLineParser::addOption(CommandLineOption* opt)
+{
+    _options.push_back(opt);
+}
+
+int CommandLineParser::showHelp(int status, bool stdErr)
+{
+    auto& out = stdErr ? std::cerr : std::cout;
+    out << "Options:" << std::endl;
+    for (auto opt : _options)
+    {
+        out << "  -" << opt->getShortName() << ", --" << opt->getLongName();
+        if (opt->hasValue())
+        {
+            out << " <" << opt->getValueName() << "=" << opt->getDefaultValue() << ">";
+        }
+        out << " " << opt->getDescription() << std::endl;
+    }
+    return status;
+}

src/Interface/CommandLineParser.h

@@ -0,0 +1,30 @@
+//
+// Created by robin on 8/8/15.
+//
+
+#ifndef PDNS_SLAVE_COMMANDLINEPARSER_H
+#define PDNS_SLAVE_COMMANDLINEPARSER_H
+
+# include "CommandLineOption.h"
+
+class CommandLineParser
+{
+public:
+    CommandLineParser(int argc, char** argv);
+
+    bool parse();
+
+    int showHelp(int status = 0, bool stdErr = true);
+
+    void addOption(CommandLineOption* opt);
+
+private:
+    int _argc;
+
+    char** _argv;
+
+    std::vector<CommandLineOption*> _options;
+};
+
+
+#endif //PDNS_SLAVE_COMMANDLINEPARSER_H

src/Interface/MainClass.cpp

@@ -0,0 +1,36 @@
+//
+// Created by robin on 8/8/15.
+//
+
+#include <iostream>
+#include <sysexits.h>
+#include "CommandLineParser.h"
+#include "MainClass.h"
+
+MainClass::MainClass(int argc, char *argv[])
+    : _argc(argc)
+    , _argv(argv)
+{
+}
+
+int MainClass::main()
+{
+    CommandLineOption configFile("config", 'c', "Path to the configuration directory", "PATH", "/etc/pdns-slave/");
+    CommandLineOption help("help", 'h', "Displays this help", "", "");
+    CommandLineParser parser(_argc, _argv);
+    parser.addOption(&help);
+    parser.addOption(&configFile);
+    if (!parser.parse())
+        return parser.showHelp(EX_USAGE);
+    if (help.isSet())
+        return parser.showHelp(0, false);
+
+    auto path = configFile.getValue();
+    std::cout << "Using configuration directory " << path << std::endl;
+
+    auto sqlPath = path + "/sql.json";
+    auto hostsPath = path + "/hosts.json";
+    auto dhcpPath = path + "/dhcp.conf";
+
+    return 0;
+}

src/Interface/MainClass.h

@@ -0,0 +1,23 @@
+//
+// Created by robin on 8/8/15.
+//
+
+#ifndef PDNS_SLAVE_MAINCLASS_H
+#define PDNS_SLAVE_MAINCLASS_H
+
+
+class MainClass {
+public:
+    MainClass(int argc, char* argv[]);
+
+    int main();
+
+private:
+    int _argc;
+
+    char** _argv;
+
+};
+
+
+#endif //PDNS_SLAVE_MAINCLASS_H

src/main.cpp

@@ -1,4 +1,7 @@
+#include "Interface/MainClass.h"
+
 int main(int argc, char* argv[])
 {
-  return 0;
+    MainClass m(argc, argv);
+    return m.main();
 }