sql queries

src/Business/PdnsSlave.cpp

@@ -50,5 +50,7 @@ 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;
 }

src/DBO/Actions/Action.cpp

@@ -21,3 +21,8 @@ void Action::setDomain(const std::string &domain)
 {
     _domain = domain;
 }
+
+const std::string Action::getSqlQuery() const
+{
+    return "SET @domain_id=(SELECT get_domain_id(\"" + _domain + "\"));\n" + getSql();
+}

src/DBO/Actions/Action.h

@@ -15,13 +15,15 @@ public:
     Action();
     virtual ~Action();
 
-    virtual const std::string getSql() const = 0;
+    const std::string getSqlQuery() const;
 
     const std::string &getDomain() const;
 
     void setDomain(const std::string &domain);
 
-private:
+protected:
+    virtual const std::string getSql() const = 0;
+
     std::string _domain;
 };
 

src/DBO/Actions/ActionAddDomain.cpp

@@ -10,5 +10,7 @@ ActionAddDomain::ActionAddDomain()
 
 const std::string ActionAddDomain::getSql() const
 {
-    return "";
+    return "INSERT INTO domains (name, type) VALUES (\"" + _domain + "\", \"MASTER\");\n"
+            "INSERT INTO zones (domain_id, owner, zone_templ_id) VALUES ((SELECT get_domain_id(\""
+           + _domain + "\")), (SELECT id FROM users ORDER BY id LIMIT 1), 0);";
 }

src/DBO/Actions/ActionAddDomain.h

@@ -12,6 +12,7 @@ class ActionAddDomain : public Action
 public:
     ActionAddDomain();
 
+protected:
     const std::string getSql() const override;
 };
 

src/DBO/Actions/ActionAddHost.cpp

@@ -1,16 +1,48 @@
 //
 // Created by robin on 8/9/15.
 //
-
+#include <string>
+#include <sstream>
+#include <vector>
 #include "ActionAddHost.h"
 
+std::vector<std::string> &split(const std::string &s, char delim, std::vector<std::string> &elems) {
+    std::stringstream ss(s);
+    std::string item;
+    while (std::getline(ss, item, delim)) {
+        elems.push_back(item);
+    }
+    return elems;
+}
+
+std::vector<std::string> split(const std::string &s, char delim) {
+    std::vector<std::string> elems;
+    split(s, delim, elems);
+    return elems;
+}
+
 ActionAddHost::ActionAddHost()
 {
 }
 
 const std::string ActionAddHost::getSql() const
 {
-    return "";
+    auto query = "INSERT INTO records (domain_id, name, type, content, ttl, prio, change_date)\n"
+            "    VALUES(@domain_id, \"" + _host + "." + _domain + "\", \"" + _recordType + "\","
+            " \"" + _recordValue + "\", 84600, 0, " + std::to_string(time(nullptr)) + ");\n";
+
+    if (_reverseEnabled && _recordType == "A")
+    {
+        auto reversedValue = getReversedValue();
+        auto reverseDomain = _reverseDomain.empty() ? "in-addr.arpa" : _reverseDomain + ".in-addr.arpa";
+
+        query += "INSERT INTO records (domain_id, name, type, content, ttl, prio, change_date)\n"
+                 "    VALUES((SELECT get_domain_id(\"" + reverseDomain + "\")), \""
+                 + reversedValue + "." + reverseDomain + "\", \"PTR\","
+                 " \"" + _host + "." + _domain + "\", 84600, 0, " + std::to_string(time(nullptr)) + ");\n";
+    }
+
+    return query;
 }
 
 const std::string &ActionAddHost::getHost() const
@@ -71,3 +103,22 @@ void ActionAddHost::setReverseEnabled(bool reverseEnabled)
 {
     _reverseEnabled = reverseEnabled;
 }
+
+const std::string ActionAddHost::getReversedValue() const
+{
+    auto splitValue = split(_recordValue, '.');
+    int index = 1;
+
+    for (; index < splitValue.size(); ++index)
+    {
+        std::string reversedValue;
+        for (int i = (int) splitValue.size() - 1 - index; i >= 0; --i)
+            reversedValue += splitValue[i] + (i == 0 ? "" : ".");
+        if (_reverseDomain == reversedValue)
+            break;
+    }
+    std::string reversedValue;
+    for (int i = (int) splitValue.size() - 1; i >= (int) splitValue.size() - index; --i)
+        reversedValue += std::string(splitValue[i]) + (i == (int) splitValue.size() - index ? "" : ".");
+    return reversedValue;
+}

src/DBO/Actions/ActionAddHost.h

@@ -12,8 +12,6 @@ class ActionAddHost : public Action
 public:
     ActionAddHost();
 
-    const std::string getSql() const override;
-
     const std::string &getHost() const;
 
     void setHost(const std::string &host);
@@ -38,7 +36,12 @@ public:
 
     void setReverseEnabled(bool reverseEnabled);
 
+protected:
+    const std::string getSql() const override;
+
 private:
+    const std::string getReversedValue() const;
+
     std::string _host;
 
     std::string _recordValue;

src/DBO/Actions/ActionDelDomain.cpp

@@ -10,5 +10,10 @@ ActionDelDomain::ActionDelDomain()
 
 const std::string ActionDelDomain::getSql() const
 {
-    return "";
+    return "DELETE FROM records WHERE domain_id=@domain_id;\n"
+            "DELETE FROM comments WHERE domain_id=@domain_id;\n"
+            "DELETE FROM zones WHERE domain_id=@domain_id;\n"
+            "DELETE FROM cryptokeys WHERE domain_id=@domain_id;\n"
+            "DELETE FROM domainmetadata WHERE domain_id=@domain_id;\n"
+            "DELETE FROM domains WHERE id=@domain_id;\n";
 }

src/DBO/Actions/ActionDelDomain.h

@@ -12,6 +12,7 @@ class ActionDelDomain : public Action
 public:
     ActionDelDomain();
 
+protected:
     const std::string getSql() const override;
 };
 

src/DBO/Actions/ActionDelHost.cpp

@@ -10,7 +10,8 @@ ActionDelHost::ActionDelHost()
 
 const std::string ActionDelHost::getSql() const
 {
-    return "";
+    return "DELETE FROM records WHERE name=\"" + _host + "." + _domain + "\" AND "
+           "domain_id=@domain_id;";
 }
 
 const std::string &ActionDelHost::getHost() const

src/DBO/Actions/ActionDelHost.h

@@ -12,12 +12,13 @@ class ActionDelHost : public Action
 public:
     ActionDelHost();
 
-    const std::string getSql() const override;
-
     const std::string &getHost() const;
 
     void setHost(const std::string &host);
 
+protected:
+    const std::string getSql() const override;
+
 private:
     std::string _host;
 };