diff options
| -rw-r--r-- | bot/src/schemas/channel.hpp | 24 |
1 files changed, 22 insertions, 2 deletions
diff --git a/bot/src/schemas/channel.hpp b/bot/src/schemas/channel.hpp index 2560331..e346b9f 100644 --- a/bot/src/schemas/channel.hpp +++ b/bot/src/schemas/channel.hpp @@ -4,9 +4,11 @@ #include <optional> #include <pqxx/pqxx> #include <string> +#include <vector> #include "../constants.hpp" #include "../utils/chrono.hpp" +#include "../utils/string.hpp" namespace bot::schemas { class Channel { @@ -45,22 +47,36 @@ namespace bot::schemas { std::optional<std::chrono::system_clock::time_point> opted_out_at; }; + enum ChannelFeature { MARKOV_RESPONSES }; + class ChannelPreferences { public: ChannelPreferences(const pqxx::row &row) { this->channel_id = row[0].as<int>(); - if (!row[2].is_null()) { + if (!row[1].is_null()) { this->prefix = row[1].as<std::string>(); } else { this->prefix = DEFAULT_PREFIX; } - if (!row[3].is_null()) { + if (!row[2].is_null()) { this->locale = row[2].as<std::string>(); } else { this->locale = DEFAULT_LOCALE_ID; } + + if (!row[3].is_null()) { + std::string x = row[3].as<std::string>(); + x = x.substr(1, x.length() - 2); + std::vector<std::string> split_text = + utils::string::split_text(x, ','); + + for (const std::string &part : split_text) { + ChannelFeature feature = (ChannelFeature)std::stoi(part); + this->features.push_back(feature); + } + } } ~ChannelPreferences() = default; @@ -68,9 +84,13 @@ namespace bot::schemas { const int &get_channel_id() const { return this->channel_id; } const std::string &get_prefix() const { return this->prefix; } const std::string &get_locale() const { return this->locale; } + const std::vector<ChannelFeature> &get_features() const { + return this->features; + } private: int channel_id; std::string prefix, locale; + std::vector<ChannelFeature> features; }; } |