~singpolyma/biboumi

ref: 88ae2599f6dbf655e8806c9b4619ec089425683b biboumi/src/database/database.cpp -rw-r--r-- 1.5 KiB
88ae2599 — Florent Le Coz Introduce an optional Database module 6 years ago
                                                                                
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
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);
}