Properly catch and handle database errors

Do not use a singleton for the database.

fix #3203

3 files changed, 27 insertions, 31 deletions

diff --git a/src/database/database.cpp b/src/database/database.cpp
index 0c7f425..61e1b47 100644
--- a/src/database/database.cpp
+++ b/src/database/database.cpp
@@ -2,8 +2,6 @@
 #ifdef USE_DATABASE
 
 #include <database/database.hpp>
-#include <config/config.hpp>
-#include <utils/xdg.hpp>
 #include <logger/logger.hpp>
 #include <string>
 
@@ -11,37 +9,36 @@ using namespace std::string_literals;
 
 std::unique_ptr<db::BibouDB> Database::db;
 
-db::BibouDB& Database::get_db()
+void Database::open(const std::string& filename, const std::string& db_type)
 {
-  if (!Database::db)
+  try
     {
-      const std::string db_filename = Config::get("db_name",
-                                                  xdg_data_path("biboumi.sqlite"));
-      Database::db = std::make_unique<db::BibouDB>("sqlite3",
-                                                   "database="s + db_filename);
+      auto new_db = std::make_unique<db::BibouDB>(db_type,
+                                             "database="s + filename);
+      if (new_db->needsUpgrade())
+        new_db->upgrade();
+      Database::db.reset(new_db.release());
+    } catch (const litesql::DatabaseError& e) {
+      log_error("Failed to open database ", filename, ". ", e.what());
+      throw;
     }
-
-  if (Database::db->needsUpgrade())
-    Database::db->upgrade();
-
-  return *Database::db.get();
 }
 
 void Database::set_verbose(const bool val)
 {
-  Database::get_db().verbose = val;
+  Database::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(),
+    auto options = litesql::select<db::IrcServerOptions>(*Database::db,
                              db::IrcServerOptions::Owner == owner &&
                              db::IrcServerOptions::Server == server).one();
     return options;
   } catch (const litesql::NotFound& e) {
-    db::IrcServerOptions options(Database::get_db());
+    db::IrcServerOptions options(*Database::db);
     options.owner = owner;
     options.server = server;
     // options.update();
@@ -54,13 +51,13 @@ db::IrcChannelOptions Database::get_irc_channel_options(const std::string& owner
                                                         const std::string& channel)
 {
   try {
-    auto options = litesql::select<db::IrcChannelOptions>(Database::get_db(),
+    auto options = litesql::select<db::IrcChannelOptions>(*Database::db,
                                                          db::IrcChannelOptions::Owner == owner &&
                                                          db::IrcChannelOptions::Server == server &&
                                                          db::IrcChannelOptions::Channel == channel).one();
     return options;
   } catch (const litesql::NotFound& e) {
-    db::IrcChannelOptions options(Database::get_db());
+    db::IrcChannelOptions options(*Database::db);
     options.owner = owner;
     options.server = server;
     options.channel = channel;
diff --git a/src/database/database.hpp b/src/database/database.hpp
index 0131669..7173bcd 100644
--- a/src/database/database.hpp
+++ b/src/database/database.hpp
@@ -26,7 +26,7 @@ public:
   template<typename PersistentType>
   static size_t count()
   {
-    return litesql::select<PersistentType>(Database::get_db()).count();
+    return litesql::select<PersistentType>(*Database::db).count();
   }
   /**
    * Return the object from the db. Create it beforehand (with all default
@@ -42,11 +42,11 @@ public:
                                                                            const std::string& channel);
 
   static void close();
+  static void open(const std::string& filename, const std::string& db_type="sqlite3");
+
 
 private:
   static std::unique_ptr<db::BibouDB> db;
-
-  static db::BibouDB& get_db();
 };
 #endif /* USE_DATABASE */
 
diff --git a/src/main.cpp b/src/main.cpp
index ed05d36..422219a 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -3,22 +3,15 @@
 #include <network/poller.hpp>
 #include <config/config.hpp>
 #include <logger/logger.hpp>
-#include <utils/reload.hpp>
 #include <utils/xdg.hpp>
-
-#include <iostream>
-#include <memory>
-#include <atomic>
-
-#include <signal.h>
+#include <utils/reload.hpp>
 
 #ifdef CARES_FOUND
 # include <network/dns_handler.hpp>
 #endif
 
-#ifdef SYSTEMD_FOUND
-# include <systemd/sd-daemon.h>
-#endif
+#include <atomic>
+#include <signal.h>
 
 // A flag set by the SIGINT signal handler.
 static volatile std::atomic<bool> stop(false);
@@ -71,6 +64,12 @@ int main(int ac, char** av)
   if (hostname.empty())
     return config_help("hostname");
 
+  try {
+      open_database();
+    } catch (...) {
+      return 1;
+    }
+
   // Block the signals we want to manage. They will be unblocked only during
   // the epoll_pwait or ppoll calls. This avoids some race conditions,
   // explained in man 2 pselect on linux