From 88c61caa3f602847a175168667694688d942f159 Mon Sep 17 00:00:00 2001
From: Stefan Pranger <stefan.pranger@student.tugraz.at>
Date: Tue, 29 Jun 2021 12:55:06 +0200
Subject: [PATCH] introduced statement builder

---
 db/db.h                                  |  3 +
 db/statements/builder/Statement.cpp      | 22 +++++++
 db/statements/builder/Statement.h        | 28 +++++++++
 db/statements/builder/StatementBuilder.h | 74 ++++++++++++++++++++++++
 4 files changed, 127 insertions(+)
 create mode 100644 db/statements/builder/Statement.cpp
 create mode 100644 db/statements/builder/Statement.h
 create mode 100644 db/statements/builder/StatementBuilder.h

diff --git a/db/db.h b/db/db.h
index 4196cde..40b4fb0 100644
--- a/db/db.h
+++ b/db/db.h
@@ -11,5 +11,8 @@ namespace sql = SQLite;
 #include <db/Schema.h>
 #include <db/migrations/builder/Table.h>
 #include <db/migrations/builder/TableBuilder.h>
+#include <db/statements/builder/Statement.h>
+#include <db/statements/builder/StatementBuilder.h>
 
 namespace db_builder = db::migrations::builder;
+namespace stmt = db::statements::builder;
diff --git a/db/statements/builder/Statement.cpp b/db/statements/builder/Statement.cpp
new file mode 100644
index 0000000..cf21df4
--- /dev/null
+++ b/db/statements/builder/Statement.cpp
@@ -0,0 +1,22 @@
+#include "Statement.h"
+#include "StatementBuilder.h"
+
+namespace db {
+  namespace statements {
+    namespace builder {
+      class InsertStatementBuilder;
+
+      InsertStatementBuilder Statement::insert() {
+        return InsertStatementBuilder{};
+      }
+
+      std::string Statement::str() const {
+        return body + "\n";
+      }
+
+      std::ostream &operator<<(std::ostream &os, const Statement &s) {
+        return os << s.body << std::endl;
+      }
+    }
+  }
+}
diff --git a/db/statements/builder/Statement.h b/db/statements/builder/Statement.h
new file mode 100644
index 0000000..5fc0ccb
--- /dev/null
+++ b/db/statements/builder/Statement.h
@@ -0,0 +1,28 @@
+#pragma once
+
+#include <iostream>
+#include <string>
+#include <sstream>
+
+namespace db {
+  namespace statements {
+    namespace builder {
+      class StatementBuilder;
+      class InsertStatementBuilder;
+
+      class Statement {
+      public:
+        Statement() = default;
+
+        friend class InsertStatementBuilder;
+        friend std::ostream &operator<<(std::ostream &os, const Statement &s);
+
+        static InsertStatementBuilder insert();
+
+        std::string str() const;
+      private:
+        std::string body;
+      };
+    }
+  }
+}
diff --git a/db/statements/builder/StatementBuilder.h b/db/statements/builder/StatementBuilder.h
new file mode 100644
index 0000000..b8c78fd
--- /dev/null
+++ b/db/statements/builder/StatementBuilder.h
@@ -0,0 +1,74 @@
+#pragma once
+
+#include "Statement.h"
+#include <string>
+#include <vector>
+
+namespace db {
+  namespace statements {
+    namespace builder {
+      class InsertStatementBuilder;
+
+      class AbstractStatementBuilder {
+      protected:
+        Statement &m_statement;
+        explicit AbstractStatementBuilder(Statement &statement) : m_statement(statement) {}
+      public:
+        operator Statement() const {
+          return std::move(m_statement);
+        }
+      };
+
+      class StatementBuilder : public AbstractStatementBuilder {
+      public:
+        Statement m_statement;
+        StatementBuilder() : AbstractStatementBuilder{m_statement} {
+        }
+      };
+
+      class InsertStatementBuilder : public StatementBuilder {
+      public:
+        explicit InsertStatementBuilder() : StatementBuilder() {
+        }
+
+        InsertStatementBuilder &table(const std::string &table) {
+          m_statement.body += "INSERT INTO " + table + " ";
+          return *this;
+        }
+
+        InsertStatementBuilder &columns(const std::string &cols) {
+          m_statement.body += "(" + cols + ")\n";
+          return *this;
+        }
+
+        InsertStatementBuilder &values() {
+          m_statement.body += " VALUES ";
+          return *this;
+        }
+
+        InsertStatementBuilder &row(const std::string &values) {
+          m_statement.body += " (" + values + "),\n";
+          return *this;
+        }
+
+        InsertStatementBuilder &row(const std::vector<std::string> &values) {
+          m_statement.body += " (";
+          bool first = true;
+          for(auto const& v : values) {
+            if(first) { first = false; } else { m_statement.body += ","; }
+            m_statement.body += v;
+          }
+          m_statement.body += "),\n";
+          return *this;
+        }
+
+        InsertStatementBuilder &close() {
+          m_statement.body.pop_back();
+          m_statement.body.pop_back();
+          m_statement.body += ";\n";
+          return *this;
+        }
+      };
+    }
+  }
+}