dhcp override

src/Business/PdnsSlave.cpp

@@ -7,6 +7,14 @@
 #include "DataAccess/HostsConfig.h"
 #include "PdnsSlave.h"
 
+bool replace(std::string& str, const std::string& from, const std::string& to) {
+    size_t start_pos = str.find(from);
+    if(start_pos == std::string::npos)
+        return false;
+    str.replace(start_pos, from.length(), to);
+    return true;
+}
+
 PdnsSlave::PdnsSlave(const std::string &filePath)
     : _filePath(filePath)
 {
@@ -50,7 +58,26 @@ Result<Actions> PdnsSlave::readHosts()
     if (!res)
         return res;
     _actions = res.getData();
-    for (auto action : _actions)
-        std::cout << action->getSqlQuery() << std::endl << std::endl;
     return res;
 }
+
+BResult PdnsSlave::overridePdns()
+{
+    return true;
+}
+
+BResult PdnsSlave::overrideDhcp()
+{
+    std::string hosts;
+    for (auto a : _actions)
+        hosts += a->getDhcpConf();
+
+    auto dhcp = _dhcpdTemplateContent;
+    replace(dhcp, "%%HOSTS%%", hosts);
+    std::ofstream file(_dhcpdFilePath);
+    if (!file)
+        return BResult().error("Could not open file");
+    file << dhcp;
+    file.close();
+    return true;
+}

src/Business/PdnsSlave.h

@@ -21,6 +21,10 @@ public:
 
     Result<Actions> readHosts();
 
+    BResult overridePdns();
+
+    BResult overrideDhcp();
+
 private:
     std::string _filePath;
 

src/DBO/Actions/Action.cpp

@@ -26,3 +26,13 @@ const std::string Action::getSqlQuery() const
 {
     return "SET @domain_id=(SELECT get_domain_id(\"" + _domain + "\"));\n" + getSql();
 }
+
+const std::string Action::getDhcpConf() const
+{
+    return getDhcp();
+}
+
+const std::string Action::getDhcp() const
+{
+    return "";
+}

src/DBO/Actions/Action.h

@@ -15,6 +15,8 @@ public:
     Action();
     virtual ~Action();
 
+    const std::string getDhcpConf() const;
+
     const std::string getSqlQuery() const;
 
     const std::string &getDomain() const;
@@ -24,6 +26,8 @@ public:
 protected:
     virtual const std::string getSql() const = 0;
 
+    virtual const std::string getDhcp() const;
+
     std::string _domain;
 };
 

src/DBO/Actions/ActionAddHost.cpp

@@ -122,3 +122,16 @@ const std::string ActionAddHost::getReversedValue() const
         reversedValue += std::string(splitValue[i]) + (i == (int) splitValue.size() - index ? "" : ".");
     return reversedValue;
 }
+
+const std::string ActionAddHost::getDhcp() const
+{
+    if (!_dhcpMac.empty() && _recordType == "A")
+    {
+        return "\thost " + _host + " {\n"
+                "\t\thardware ethernet " + _dhcpMac + ";\n"
+                "\t\tfixed-address " + _host + "." + _domain + ";\n"
+                "\t}\n";
+    }
+    else
+        return "";
+}

src/DBO/Actions/ActionAddHost.h

@@ -39,6 +39,8 @@ public:
 protected:
     const std::string getSql() const override;
 
+    const std::string getDhcp() const override;
+
 private:
     const std::string getReversedValue() const;
 

src/Interface/MainClass.cpp

@@ -19,10 +19,12 @@ int MainClass::main()
 {
     CommandLineOption configFile("config", 'c', "Path to the configuration file",
                                  "FILE", "/etc/pdns-slave/pdns-slave.json");
+    CommandLineOption check("check", 'C', "Only check configurations", "", "");
     CommandLineOption help("help", 'h', "Displays this help", "", "");
     CommandLineParser parser(_argc, _argv);
     parser.addOption(&help);
     parser.addOption(&configFile);
+    parser.addOption(&check);
     if (!parser.parse())
         return parser.showHelp(EX_USAGE);
     if (help.isSet())
@@ -62,5 +64,26 @@ int MainClass::main()
     }
     res2.print();
 
+    if (!check.isSet())
+    {
+        std::cout << "Overriding dns" << std::endl;
+        if (!(res = pdnsSlave.overridePdns()))
+        {
+            std::cerr << "Failed to override dns" << std::endl;
+            res.print();
+            return 4;
+        }
+        res.print();
+
+        std::cout << "Overriding dhcp" << std::endl;
+        if (!(res = pdnsSlave.overrideDhcp()))
+        {
+            std::cerr << "Failed to override dhcp" << std::endl;
+            res.print();
+            return 5;
+        }
+        res.print();
+    }
+
     return 0;
 }