database.cpp 2.88 KB
Newer Older
1 2 3
#include "biboumi.h"
#ifdef USE_DATABASE

4 5 6 7 8 9 10 11
#include <database/database.hpp>
#include <logger/logger.hpp>
#include <string>

using namespace std::string_literals;

std::unique_ptr<db::BibouDB> Database::db;

12
void Database::open(const std::string& filename, const std::string& db_type)
13
{
14
  try
15
    {
16 17 18 19 20 21 22 23
      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;
24 25 26 27 28
    }
}

void Database::set_verbose(const bool val)
{
29
  Database::db->verbose = val;
30 31 32 33 34 35
}

db::IrcServerOptions Database::get_irc_server_options(const std::string& owner,
                                                      const std::string& server)
{
  try {
36
    auto options = litesql::select<db::IrcServerOptions>(*Database::db,
37 38 39 40
                             db::IrcServerOptions::Owner == owner &&
                             db::IrcServerOptions::Server == server).one();
    return options;
  } catch (const litesql::NotFound& e) {
41
    db::IrcServerOptions options(*Database::db);
42 43 44 45 46 47 48
    options.owner = owner;
    options.server = server;
    // options.update();
    return options;
  }
}

49 50 51 52 53
db::IrcChannelOptions Database::get_irc_channel_options(const std::string& owner,
                                                        const std::string& server,
                                                        const std::string& channel)
{
  try {
54
    auto options = litesql::select<db::IrcChannelOptions>(*Database::db,
55 56 57 58 59
                                                         db::IrcChannelOptions::Owner == owner &&
                                                         db::IrcChannelOptions::Server == server &&
                                                         db::IrcChannelOptions::Channel == channel).one();
    return options;
  } catch (const litesql::NotFound& e) {
60
    db::IrcChannelOptions options(*Database::db);
61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81
    options.owner = owner;
    options.server = server;
    options.channel = channel;
    return options;
  }
}

db::IrcChannelOptions Database::get_irc_channel_options_with_server_default(const std::string& owner,
                                                                            const std::string& server,
                                                                            const std::string& channel)
{
  auto coptions = Database::get_irc_channel_options(owner, server, channel);
  auto soptions = Database::get_irc_server_options(owner, server);
  if (coptions.encodingIn.value().empty())
    coptions.encodingIn = soptions.encodingIn;
  if (coptions.encodingOut.value().empty())
    coptions.encodingOut = soptions.encodingOut;

  return coptions;
}

82 83 84 85
void Database::close()
{
  Database::db.reset(nullptr);
}
86 87

#endif