Older/Database/Session.cpp

#include "Session.h"
#include "BoostLog.h"
#include <Wt/Auth/Dbo/UserDatabase.h>
#include <Wt/Dbo/WtJsonSqlTraits.h>
#include <Wt/Dbo/backend/Sqlite3.h>
#include <Wt/Dbo/FixedSqlConnectionPool.h>
#include <Wt/Dbo/SqlConnectionPool.h>

namespace Database {
std::unique_ptr<Wt::Dbo::SqlConnectionPool> sqlConnectionPool;
bool initialize(const std::string &path) {
    try {
        auto connection = std::make_unique<Wt::Dbo::backend::Sqlite3>(path);
        connection->setProperty("show-queries", "true");
        connection->setDateTimeStorage(Wt::Dbo::SqlDateTimeType::DateTime,
                                       Wt::Dbo::backend::DateTimeStorage::PseudoISO8601AsText);
        sqlConnectionPool = std::make_unique<Wt::Dbo::FixedSqlConnectionPool>(std::move(connection), 10);
        session()->createTables();
    } catch (const Wt::Dbo::Exception &e) {
        LOG(error) << e.code() << ": " << e.what();
    }
    return true;
}

std::unique_ptr<Session> session() {
    if (!sqlConnectionPool) return {};
    return std::make_unique<Session>(*sqlConnectionPool);
}
} // namespace Database

Session::Session(Wt::Dbo::SqlConnectionPool &connectionPool) {
    setConnectionPool(connectionPool);

    mapClass<Task>("task");
    mapClass<HomeBox::Item>("homebox_item");
    mapClass<VisitorRecord>("visitor_record");

    mapClass<User>("user");
    mapClass<AuthInfo>("auth_info");
    mapClass<AuthInfo::AuthIdentityType>("auth_identity");
    mapClass<AuthInfo::AuthTokenType>("auth_token");

    m_users = std::make_unique<UserDatabase>(*this);
}

Session::~Session() {
}

Wt::Dbo::ptr<User> Session::user() const {
    if (m_login.loggedIn()) {
        Wt::Dbo::ptr<AuthInfo> authInfo = m_users->find(m_login.user());
        return authInfo->user();
    } else
        return Wt::Dbo::ptr<User>();
}

Wt::Auth::AbstractUserDatabase &Session::users() {
    return *m_users;
}

Wt::Auth::Login &Session::login() {
    return m_login;
}

namespace Wt {
namespace Dbo {
template <>
void JsonSerializer::act(FieldRef<std::chrono::system_clock::time_point> field) {
    using namespace std::chrono;
    writeFieldName(field.name());
    fastJsStringLiteral(std::to_string(duration_cast<seconds>(field.value().time_since_epoch()).count()));
}
} // namespace Dbo
} // namespace Wt

DBO_INSTANTIATE_TEMPLATES(User)
DBO_INSTANTIATE_TEMPLATES(Task)
DBO_INSTANTIATE_TEMPLATES(HomeBox::Item)
DBO_INSTANTIATE_TEMPLATES(VisitorRecord)