context.hpp

/* -*- mode: c++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*-
 *
 * Copyright (C) 2025 Marco Craveiro <marco.craveiro@gmail.com>
 *
 * This program is free software; you can redistribute it and/or modify it under
 * the terms of the GNU General Public License as published by the Free Software
 * Foundation; either version 3 of the License, or (at your option) any later
 * version.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
 * details.
 *
 * You should have received a copy of the GNU General Public License along with
 * this program; if not, write to the Free Software Foundation, Inc., 51
 * Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 *
 */
#ifndef ORES_DATABASE_CONTEXT_HPP
#define ORES_DATABASE_CONTEXT_HPP
 
#include <vector>
#include <optional>
#include <sqlgen/postgres.hpp>
#include <boost/uuid/uuid.hpp>
#include "ores.database/domain/tenant_aware_pool.hpp"
#include "ores.utility/uuid/tenant_id.hpp"
 
namespace ores::database {
 
class context {
public:
    using connection_type = sqlgen::postgres::Connection;
    using connection_pool_type = tenant_aware_pool<connection_type>;
 
    explicit context(sqlgen::ConnectionPool<connection_type> connection_pool,
                     sqlgen::postgres::Credentials credentials,
                     utility::uuid::tenant_id tenant_id,
                     std::string actor = "",
                     std::string service_account = "")
    : connection_pool_(std::move(connection_pool), credentials,
                       std::move(tenant_id), std::move(actor), service_account),
      credentials_(std::move(credentials)),
      service_account_(std::move(service_account)) {}
 
    explicit context(sqlgen::ConnectionPool<connection_type> connection_pool,
                     sqlgen::postgres::Credentials credentials,
                     utility::uuid::tenant_id tenant_id,
                     boost::uuids::uuid party_id,
                     std::vector<boost::uuids::uuid> visible_party_ids,
                     std::string actor = "",
                     std::string service_account = "")
    : connection_pool_(std::move(connection_pool), credentials,
                       std::move(tenant_id), party_id,
                       std::move(visible_party_ids), std::move(actor),
                       service_account),
      credentials_(std::move(credentials)),
      service_account_(std::move(service_account)) {}
 
    connection_pool_type& connection_pool() { return connection_pool_; }
 
    const sqlgen::postgres::Credentials& credentials() const { return credentials_; }
 
    const utility::uuid::tenant_id& tenant_id() const {
        return connection_pool_.tenant_id();
    }
 
    std::optional<boost::uuids::uuid> party_id() const {
        return connection_pool_.party_id();
    }
 
    const std::vector<boost::uuids::uuid>& visible_party_ids() const {
        return connection_pool_.visible_party_ids();
    }
 
    const std::string& actor() const { return connection_pool_.actor(); }
 
    const std::string& service_account() const { return service_account_; }
 
    const sqlgen::ConnectionPool<connection_type>& underlying_pool() const {
        return connection_pool_.underlying_pool();
    }
 
    const std::vector<std::string>& roles() const { return roles_; }
 
    [[nodiscard]] context with_roles(std::vector<std::string> roles) const {
        auto copy = *this;
        copy.roles_ = std::move(roles);
        return copy;
    }
 
    [[nodiscard]] context with_tenant(utility::uuid::tenant_id tenant_id,
                                      std::string actor) const {
        return context(connection_pool_.underlying_pool(), credentials_,
                       std::move(tenant_id), std::move(actor), service_account_);
    }
 
    [[nodiscard]] context with_party(
        utility::uuid::tenant_id tenant_id,
        boost::uuids::uuid party_id,
        std::vector<boost::uuids::uuid> visible_party_ids,
        std::string actor) const {
        return context(connection_pool_.underlying_pool(), credentials_,
                       std::move(tenant_id), party_id,
                       std::move(visible_party_ids), std::move(actor),
                       service_account_);
    }
 
private:
    connection_pool_type connection_pool_;
    sqlgen::postgres::Credentials credentials_;
    std::string service_account_;
    std::vector<std::string> roles_;
};
 
}
 
#endif