add files

This commit is contained in:
Eric Mertens 2023-11-29 09:07:20 -08:00
parent 7ea1d8c322
commit 7eb725fd5b
2 changed files with 78 additions and 0 deletions

47
priv_thread.cpp Normal file
View File

@ -0,0 +1,47 @@
#include "priv_thread.hpp"
#include "connection.hpp"
#include "command_thread.hpp"
PrivThread::PrivThread(Connection& connection) : connection_{connection} {}
auto PrivThread::check_command(CommandEvent& event, std::string priv) -> bool
{
return
(not event.account.empty() &&
(check(account_privs, priv, "") ||
check(account_privs, priv, std::string{event.account}))) ||
(not event.oper.empty() &&
(check(oper_privs, priv, "") ||
check(oper_privs, priv, std::string{event.oper})));
}
auto PrivThread::check(
std::unordered_map<std::string, std::unordered_set<std::string>> const& privs,
std::string const& priv,
std::string const& name
) -> bool
{
auto const cursor = privs.find(name);
return cursor != privs.end() && cursor->second.contains(priv);
}
auto PrivThread::on_command(CommandEvent& event) -> void
{
if (event.command == "setup")
{
oper_privs["glguy"].insert("owner");
}
}
auto PrivThread::start(Connection& connection) -> std::shared_ptr<PrivThread>
{
auto thread = std::make_shared<PrivThread>(connection);
connection.add_listener<CommandEvent>([thread](CommandEvent& event)
{
thread->on_command(event);
});
return thread;
}

31
priv_thread.hpp Normal file
View File

@ -0,0 +1,31 @@
#pragma once
#include <memory>
#include <string>
#include <unordered_map>
#include <unordered_set>
class Connection;
class CommandEvent;
class PrivThread : std::enable_shared_from_this<PrivThread>
{
Connection& connection_;
std::unordered_map<std::string, std::unordered_set<std::string>> oper_privs;
std::unordered_map<std::string, std::unordered_set<std::string>> account_privs;
auto on_command(CommandEvent&) -> void;
auto check(
std::unordered_map<std::string, std::unordered_set<std::string>> const& privs,
std::string const& priv,
std::string const& name) -> bool;
public:
PrivThread(Connection&);
auto check_command(CommandEvent& event, std::string priv) -> bool;
static constexpr char const* owner_priv = "owner";
static auto start(Connection&) -> std::shared_ptr<PrivThread>;
};