summaryrefslogtreecommitdiff
path: root/components/ratelimitindicator.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'components/ratelimitindicator.cpp')
-rw-r--r--components/ratelimitindicator.cpp137
1 files changed, 0 insertions, 137 deletions
diff --git a/components/ratelimitindicator.cpp b/components/ratelimitindicator.cpp
deleted file mode 100644
index ac4ef4b..0000000
--- a/components/ratelimitindicator.cpp
+++ /dev/null
@@ -1,137 +0,0 @@
-#include "ratelimitindicator.hpp"
-#include "abaddon.hpp"
-#include <filesystem>
-
-RateLimitIndicator::RateLimitIndicator()
- : Gtk::Box(Gtk::ORIENTATION_HORIZONTAL) {
- m_label.set_text("");
- m_label.set_ellipsize(Pango::ELLIPSIZE_START);
- m_label.set_valign(Gtk::ALIGN_END);
- get_style_context()->add_class("ratelimit-indicator");
-
- m_img.set_margin_start(7);
-
- add(m_label);
- add(m_img);
- m_label.show();
-
- const static auto clock_path = Abaddon::GetResPath("/clock.png");
- if (std::filesystem::exists(clock_path)) {
- try {
- const auto pixbuf = Gdk::Pixbuf::create_from_file(clock_path);
- int w, h;
- GetImageDimensions(pixbuf->get_width(), pixbuf->get_height(), w, h, 20, 10);
- m_img.property_pixbuf() = pixbuf->scale_simple(w, h, Gdk::INTERP_BILINEAR);
- } catch (...) {}
- }
-
- Abaddon::Get().GetDiscordClient().signal_message_create().connect(sigc::mem_fun(*this, &RateLimitIndicator::OnMessageCreate));
- Abaddon::Get().GetDiscordClient().signal_message_send_fail().connect(sigc::mem_fun(*this, &RateLimitIndicator::OnMessageSendFail));
- Abaddon::Get().GetDiscordClient().signal_channel_update().connect(sigc::mem_fun(*this, &RateLimitIndicator::OnChannelUpdate));
-}
-
-void RateLimitIndicator::SetActiveChannel(Snowflake id) {
- m_active_channel = id;
- const auto channel = Abaddon::Get().GetDiscordClient().GetChannel(m_active_channel);
- if (channel.has_value() && channel->RateLimitPerUser.has_value())
- m_rate_limit = *channel->RateLimitPerUser;
- else
- m_rate_limit = 0;
-
- UpdateIndicator();
-}
-
-bool RateLimitIndicator::CanSpeak() const {
- const auto rate_limit = GetRateLimit();
- if (rate_limit == 0) return true;
-
- const auto it = m_times.find(m_active_channel);
- if (it == m_times.end())
- return true;
-
- const auto now = std::chrono::steady_clock::now();
- const auto sec_diff = std::chrono::duration_cast<std::chrono::seconds>(it->second - now).count();
- return sec_diff <= 0;
-}
-
-int RateLimitIndicator::GetTimeLeft() const {
- if (CanSpeak()) return 0;
-
- auto it = m_times.find(m_active_channel);
- if (it == m_times.end()) return 0;
-
- const auto now = std::chrono::steady_clock::now();
- const auto sec_diff = std::chrono::duration_cast<std::chrono::seconds>(it->second - now).count();
-
- if (sec_diff <= 0)
- return 0;
- else
- return sec_diff;
-}
-
-int RateLimitIndicator::GetRateLimit() const {
- return m_rate_limit;
-}
-
-bool RateLimitIndicator::UpdateIndicator() {
- if (const auto rate_limit = GetRateLimit(); rate_limit != 0) {
- m_img.show();
-
- auto &discord = Abaddon::Get().GetDiscordClient();
- if (discord.HasAnyChannelPermission(discord.GetUserData().ID, m_active_channel, Permission::MANAGE_MESSAGES | Permission::MANAGE_CHANNELS)) {
- m_label.set_text("You may bypass slowmode.");
- set_has_tooltip(false);
- } else {
- const auto time_left = GetTimeLeft();
- if (time_left > 0)
- m_label.set_text(std::to_string(time_left) + "s");
- else
- m_label.set_text("");
- set_tooltip_text("Slowmode is enabled. Members can send one message every " + std::to_string(rate_limit) + " seconds.");
- }
- } else {
- m_img.hide();
-
- m_label.set_text("");
- set_has_tooltip(false);
- }
-
- if (m_connection)
- m_connection.disconnect();
- m_connection = Glib::signal_timeout().connect_seconds(sigc::mem_fun(*this, &RateLimitIndicator::UpdateIndicator), 1);
-
- return false;
-}
-
-void RateLimitIndicator::OnMessageCreate(const Message &message) {
- auto &discord = Abaddon::Get().GetDiscordClient();
- if (message.Author.ID != discord.GetUserData().ID) return;
- if (!message.GuildID.has_value()) return;
- const bool can_bypass = discord.HasAnyChannelPermission(discord.GetUserData().ID, m_active_channel, Permission::MANAGE_MESSAGES | Permission::MANAGE_CHANNELS);
- const auto rate_limit = GetRateLimit();
- if (rate_limit > 0 && !can_bypass) {
- m_times[message.ChannelID] = std::chrono::steady_clock::now() + std::chrono::duration<int>(rate_limit + 1);
- UpdateIndicator();
- }
-}
-
-void RateLimitIndicator::OnMessageSendFail(const std::string &nonce, float retry_after) {
- if (retry_after != 0) { // failed to rate limit
- const auto msg = Abaddon::Get().GetDiscordClient().GetMessage(nonce);
- const auto channel_id = msg->ChannelID;
- m_times[channel_id] = std::chrono::steady_clock::now() + std::chrono::duration<int>(std::lroundf(retry_after + 0.5f) + 1); // + 0.5 will ceil it
- UpdateIndicator();
- }
-}
-
-void RateLimitIndicator::OnChannelUpdate(Snowflake channel_id) {
- if (channel_id != m_active_channel) return;
- const auto chan = Abaddon::Get().GetDiscordClient().GetChannel(m_active_channel);
- if (!chan.has_value()) return;
- const auto r = chan->RateLimitPerUser;
- if (r.has_value())
- m_rate_limit = *r;
- else
- m_rate_limit = 0;
- UpdateIndicator();
-}