1 files changed, 54 insertions, 0 deletions
diff --git a/src/database/database.cpp b/src/database/database.cpp
new file mode 100644
index 0000000..e16465c
--- /dev/null
+++ b/src/database/database.cpp
@@ -0,0 +1,54 @@
+#include <database/database.hpp>
+#include <config/config.hpp>
+#include <utils/xdg.hpp>
+#include <logger/logger.hpp>
+#include <string>
+
+using namespace std::string_literals;
+
+std::unique_ptr<db::BibouDB> Database::db;
+
+db::BibouDB& Database::get_db()
+{
+  if (!Database::db)
+    {
+      const std::string db_filename = Config::get("db_name",
+                                                  xdg_data_path("biboumi.sqlite"));
+      // log_info("Opening database: " << db_filename);
+      std::cout << "Opening database: " << db_filename << std::endl;
+      Database::db = std::make_unique<db::BibouDB>("sqlite3",
+                                                   "database="s + db_filename);
+    }
+
+  if (Database::db->needsUpgrade())
+    Database::db->upgrade();
+
+  return *Database::db.get();
+}
+
+void Database::set_verbose(const bool val)
+{
+  Database::get_db().verbose = val;
+}
+
+db::IrcServerOptions Database::get_irc_server_options(const std::string& owner,
+                                                      const std::string& server)
+{
+  try {
+    auto options = litesql::select<db::IrcServerOptions>(Database::get_db(),
+                             db::IrcServerOptions::Owner == owner &&
+                             db::IrcServerOptions::Server == server).one();
+    return options;
+  } catch (const litesql::NotFound& e) {
+    db::IrcServerOptions options(Database::get_db());
+    options.owner = owner;
+    options.server = server;
+    // options.update();
+    return options;
+  }
+}
+
+void Database::close()
+{
+  Database::db.reset(nullptr);
+}