diff options
Diffstat (limited to 'bot')
| -rw-r--r-- | bot/src/commands/command.cpp | 2 | ||||
| -rw-r--r-- | bot/src/localization/line_id.cpp | 10 | ||||
| -rw-r--r-- | bot/src/localization/line_id.hpp | 7 | ||||
| -rw-r--r-- | bot/src/modules/settings.hpp | 119 |
4 files changed, 137 insertions, 1 deletions
diff --git a/bot/src/commands/command.cpp b/bot/src/commands/command.cpp index 3f99d9e..7dd4e47 100644 --- a/bot/src/commands/command.cpp +++ b/bot/src/commands/command.cpp @@ -17,6 +17,7 @@ #include "../modules/massping.hpp" #include "../modules/notify.hpp" #include "../modules/ping.hpp" +#include "../modules/settings.hpp" #include "../modules/spam.hpp" #include "../modules/timer.hpp" #include "../utils/chrono.hpp" @@ -36,6 +37,7 @@ namespace bot { this->add_command(std::make_unique<mod::Help>()); this->add_command(std::make_unique<mod::Chatters>()); this->add_command(std::make_unique<mod::Spam>()); + this->add_command(std::make_unique<mod::Settings>()); } void CommandLoader::add_command(std::unique_ptr<Command> command) { diff --git a/bot/src/localization/line_id.cpp b/bot/src/localization/line_id.cpp index dabd85a..7f17378 100644 --- a/bot/src/localization/line_id.cpp +++ b/bot/src/localization/line_id.cpp @@ -96,6 +96,16 @@ namespace bot { return LineId::ChattersResponse; } + else if (str == "set.locale") { + return LineId::SetLocale; + } else if (str == "set.prefix") { + return LineId::SetPrefix; + } else if (str == "set.feature.disabled") { + return LineId::SetFeatureDisabled; + } else if (str == "set.feature.enabled") { + return LineId::SetFeatureEnabled; + } + else { return std::nullopt; } diff --git a/bot/src/localization/line_id.hpp b/bot/src/localization/line_id.hpp index 6d7729e..e9b90c0 100644 --- a/bot/src/localization/line_id.hpp +++ b/bot/src/localization/line_id.hpp @@ -50,7 +50,12 @@ namespace bot { HelpResponse, - ChattersResponse + ChattersResponse, + + SetLocale, + SetPrefix, + SetFeatureDisabled, + SetFeatureEnabled }; std::optional<LineId> string_to_line_id(const std::string &str); diff --git a/bot/src/modules/settings.hpp b/bot/src/modules/settings.hpp new file mode 100644 index 0000000..75060f3 --- /dev/null +++ b/bot/src/modules/settings.hpp @@ -0,0 +1,119 @@ +#pragma once + +#include <sys/resource.h> +#include <sys/types.h> +#include <unistd.h> + +#include <algorithm> +#include <optional> +#include <string> +#include <vector> + +#include "../bundle.hpp" +#include "../commands/command.hpp" +#include "../commands/response_error.hpp" + +namespace bot { + namespace mod { + class Settings : public command::Command { + std::string get_name() const override { return "set"; } + + schemas::PermissionLevel get_permission_level() const override { + return schemas::PermissionLevel::MODERATOR; + } + + std::vector<std::string> get_subcommand_ids() const override { + return {"locale", "prefix", "feature"}; + } + + command::Response run(const InstanceBundle &bundle, + const command::Request &request) const override { + if (!request.message.has_value()) { + throw ResponseException<ResponseError::NOT_ENOUGH_ARGUMENTS>( + request, bundle.localization, command::CommandArgument::VALUE); + } + + pqxx::work work(request.conn); + + std::vector<std::string> parts = + utils::string::split_text(request.message.value(), ' '); + + std::string &value = parts[0]; + + if (request.subcommand_id == "locale") { + auto locals = bundle.localization.get_loaded_localizations(); + if (!std::any_of(locals.begin(), locals.end(), + [&value](const auto &x) { return x == value; })) { + throw ResponseException<ResponseError::NOT_FOUND>( + request, bundle.localization, value); + } + + work.exec_params( + "UPDATE channel_preferences SET locale = $1 WHERE channel_id = " + "$2", + value, request.channel.get_id()); + + work.commit(); + + return command::Response( + bundle.localization + .get_formatted_line(request, loc::LineId::SetLocale, + {value}) + .value()); + } else if (request.subcommand_id == "prefix") { + work.exec_params( + "UPDATE channel_preferences SET prefix = $1 WHERE channel_id = " + "$2", + value, request.channel.get_id()); + + work.commit(); + + return command::Response( + bundle.localization + .get_formatted_line(request, loc::LineId::SetPrefix, + {value}) + .value()); + } else if (request.subcommand_id == "feature") { + std::optional<schemas::ChannelFeature> feature = + schemas::string_to_channel_feature(value); + if (!feature.has_value()) { + throw ResponseException<ResponseError::NOT_FOUND>( + request, bundle.localization, value); + } + + loc::LineId line_id; + std::string query; + + if (std::any_of(request.channel_preferences.get_features().begin(), + request.channel_preferences.get_features().end(), + [&feature](const auto &x) { + return x == feature.value(); + })) { + line_id = loc::LineId::SetFeatureDisabled; + query = + "UPDATE channel_preferences SET features = " + "array_remove(features, $1) WHERE channel_id = $2"; + } else { + line_id = loc::LineId::SetFeatureEnabled; + query = + "UPDATE channel_preferences SET features = " + "array_append(features, $1) WHERE channel_id = $2"; + } + + work.exec_params(query, (int)feature.value(), + request.channel.get_id()); + work.commit(); + + return command::Response( + bundle.localization + .get_formatted_line(request, line_id, {value}) + .value()); + } + + work.commit(); + throw ResponseException<ResponseError::SOMETHING_WENT_WRONG>( + request, bundle.localization); + } + }; + } +} |