#include <Access/RowPolicyContext.h>
#include <boost/range/adaptor/map.hpp>
#include <boost/range/algorithm/copy.hpp>


namespace DB
{
size_t RowPolicyContext::Hash::operator()(const DatabaseAndTableNameRef & database_and_table_name) const
{
    return std::hash<StringRef>{}(database_and_table_name.first) - std::hash<StringRef>{}(database_and_table_name.second);
}


RowPolicyContext::RowPolicyContext()
    : atomic_map_of_mixed_conditions(std::make_shared<MapOfMixedConditions>())
{
}


RowPolicyContext::~RowPolicyContext() = default;


RowPolicyContext::RowPolicyContext(const String & user_name_)
    : user_name(user_name_)
{}


ASTPtr RowPolicyContext::getCondition(const String & database, const String & table_name, ConditionIndex index) const
{
    /// We don't lock `mutex` here.
    auto map_of_mixed_conditions = std::atomic_load(&atomic_map_of_mixed_conditions);
    auto it = map_of_mixed_conditions->find({database, table_name});
    if (it == map_of_mixed_conditions->end())
        return {};
    return it->second.mixed_conditions[index];
}


std::vector<UUID> RowPolicyContext::getCurrentPolicyIDs() const
{
    /// We don't lock `mutex` here.
    auto map_of_mixed_conditions = std::atomic_load(&atomic_map_of_mixed_conditions);
    std::vector<UUID> policy_ids;
    for (const auto & mixed_conditions : *map_of_mixed_conditions | boost::adaptors::map_values)
        boost::range::copy(mixed_conditions.policy_ids, std::back_inserter(policy_ids));
    return policy_ids;
}


std::vector<UUID> RowPolicyContext::getCurrentPolicyIDs(const String & database, const String & table_name) const
{
    /// We don't lock `mutex` here.
    auto map_of_mixed_conditions = std::atomic_load(&atomic_map_of_mixed_conditions);
    auto it = map_of_mixed_conditions->find({database, table_name});
    if (it == map_of_mixed_conditions->end())
        return {};
    return it->second.policy_ids;
}
}