From b71ca15a0f9114db38eec23b49d1489a2ff1d0ca Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?louiz=E2=80=99?= <louiz@louiz.org>
Date: Fri, 23 Jun 2017 00:11:07 +0200
Subject: Move a few functions from select_query to query

---
 src/database/query.hpp | 43 ++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 40 insertions(+), 3 deletions(-)

(limited to 'src/database/query.hpp')

diff --git a/src/database/query.hpp b/src/database/query.hpp
index d9638f7..f103fe9 100644
--- a/src/database/query.hpp
+++ b/src/database/query.hpp
@@ -22,16 +22,34 @@ struct Query
 
     Statement prepare(sqlite3* db)
     {
-      sqlite3_stmt* statement;
+      sqlite3_stmt* stmt;
       log_debug(this->body);
       auto res = sqlite3_prepare(db, this->body.data(), static_cast<int>(this->body.size()) + 1,
-                                 &statement, nullptr);
+                                 &stmt, nullptr);
       if (res != SQLITE_OK)
         {
           log_error("Error preparing statement: ", sqlite3_errmsg(db));
           return nullptr;
         }
-      return {statement};
+      Statement statement(stmt);
+      int i = 1;
+      for (const std::string& param: this->params)
+        {
+          if (sqlite3_bind_text(statement.get(), i, param.data(), static_cast<int>(param.size()), SQLITE_TRANSIENT) != SQLITE_OK)
+            log_debug("Failed to bind ", param, " to param ", i);
+          else
+            log_debug("Bound ", param, " to ", i);
+          i++;
+        }
+
+      return statement;
+    }
+
+    void execute(sqlite3* db)
+    {
+      auto statement = this->prepare(db);
+      while (sqlite3_step(statement.get()) != SQLITE_DONE)
+        ;
     }
 };
 
@@ -51,3 +69,22 @@ void actual_add_param(Query& query, const T& val)
 
 void actual_add_param(Query& query, const std::string& val);
 void actual_add_param(Query& query, const OptionalBool& val);
+
+template <typename T>
+typename std::enable_if<!std::is_integral<T>::value, Query&>::type
+operator<<(Query& query, const T&)
+{
+  query.body += T::name;
+  return query;
+}
+
+Query& operator<<(Query& query, const char* str);
+Query& operator<<(Query& query, const std::string& str);
+template <typename Integer>
+typename std::enable_if<std::is_integral<Integer>::value, Query&>::type
+operator<<(Query& query, const Integer& i)
+{
+  query.body += "?";
+  actual_add_param(query, i);
+  return query;
+}
-- 
cgit v1.2.3