add permission overwrites for channels

Author: Arkrissym

Discord.C++/Channel.cpp

@@ -1,5 +1,6 @@
 #include "Channel.h"
 
+#include "Permissions.h"
 #include "static.h"
 
 /**	@param[in]	data	JSON data
@@ -9,9 +10,14 @@ DiscordCPP::Channel::Channel(const json& data, const std::string& token)
     : DiscordCPP::DiscordObject(token, data["id"].get<std::string>()),
       type(static_cast<Type>(data["type"].get<int>())) {
     position = get_or_else<int>(data, "position", 0);
-    // permission_overwrites
     name = get_or_else<std::string>(data, "name", "");
     icon = get_or_else<std::string>(data, "icon", "");
+
+    if (has_value(data, "permission_overwrites")) {
+        for (auto d : data.at("permission_overwrites")) {
+            permission_overwrites.push_back(PermissionOverwrites(d));
+        }
+    }
 }
 
 /**	@param[in]	id		the channel's id
@@ -35,3 +41,21 @@ void DiscordCPP::Channel::delete_channel() {
 
     api_call(url, "DELETE");
 }
+
+DiscordCPP::Permissions DiscordCPP::Channel::merge_permission_overwrites(const Permissions& permissions, const std::string& id) {
+    if (permissions.has_permission(Permissions::ADMINISTRATOR)) {
+        return Permissions::All();
+    }
+
+    Permissions result;
+    result.add(permissions);
+
+    for (auto overwrite : permission_overwrites) {
+        if (overwrite.get_id() == id) {
+            result.add(overwrite.get_allowed_permissions());
+            result.remove(overwrite.get_denied_permissions());
+        }
+    }
+
+    return result;
+}

Discord.C++/Channel.h

@@ -1,6 +1,9 @@
 #pragma once
 
+#include <vector>
+
 #include "DiscordObject.h"
+#include "Permissions.h"
 #include "static.h"
 
 namespace DiscordCPP {
@@ -30,7 +33,8 @@ class Channel : public DiscordObject {
     Type type;
     /// the sorting position
     int32_t position;
-    // vector<Overwrite *> permission_overwrites;
+    /// explicit permission overwrites for members and roles
+    std::vector<PermissionOverwrites> permission_overwrites;
     /// the channel's name
     std::string name;
     /// the channel's icon hash
@@ -48,10 +52,17 @@ class Channel : public DiscordObject {
     /// @return Channelname as std::string
     DLL_EXPORT explicit operator std::string() { return name; };
 
+    /** Merge permission overwrites into given permission set
+        @param permissions permission set
+        @param id member or role id */
+    DLL_EXPORT Permissions merge_permission_overwrites(const Permissions& permissions, const std::string& id);
+
     /// @return the ChannelType of the channel
     DLL_EXPORT Type get_type() { return type; }
     /// @return the sorting position
     DLL_EXPORT int32_t get_position() { return position; }
+    /// @return explicit permission overwrites for members and roles
+    DLL_EXPORT std::vector<PermissionOverwrites> get_permission_overwrites() const { return permission_overwrites; }
     /// @return the channel's name
     DLL_EXPORT std::string get_name() { return name; }
     /// @return channel's icon hash

Discord.C++/Interaction.cpp

@@ -95,3 +95,10 @@ std::optional<DiscordCPP::Guild> DiscordCPP::Interaction::get_guild() {
     }
     return (guild != nullptr) ? std::optional<Guild>{*guild} : std::nullopt;
 }
+
+std::optional<DiscordCPP::Channel> DiscordCPP::Interaction::get_channel() {
+    if (channel_id.has_value() && channel == nullptr) {
+        channel = new Channel(channel_id.value(), get_token());
+    }
+    return (channel != nullptr) ? std::optional<Channel>{*channel} : std::nullopt;
+}

Discord.C++/Interaction.h

@@ -2,6 +2,7 @@
 
 #include <optional>
 
+#include "Channel.h"
 #include "InteractionData.h"
 #include "Member.h"
 #include "Message.h"
@@ -36,6 +37,8 @@ class Interaction : public DiscordObject {
     Guild* guild = nullptr;
     /// The channel the interaction was sent from.
     std::optional<std::string> channel_id;
+    Channel* channel = nullptr;
+
     /// The member that sent the interaction.
     std::optional<Member> member;
     /// The user that sent the interaction.
@@ -74,6 +77,8 @@ class Interaction : public DiscordObject {
     DLL_EXPORT std::optional<DiscordCPP::Guild> get_guild();
     /// @return The channel the interaction was sent from.
     DLL_EXPORT std::optional<std::string> get_channel_id() const { return channel_id; }
+    /// @return The channel the interaction was sent from.
+    DLL_EXPORT std::optional<Channel> get_channel();
     /// @return The member that sent the interaction.
     DLL_EXPORT std::optional<Member> get_member() const { return member; }
     /// @return The user that sent the interaction.

Discord.C++/Permissions.cpp

@@ -20,11 +20,19 @@ void DiscordCPP::Permissions::add(const Permissions& permissions) {
     this->permissions |= permissions.permissions;
 }
 
-bool DiscordCPP::Permissions::has_permission(Permission permission) {
+void DiscordCPP::Permissions::remove(const Permission& permission) {
+    permissions &= ~(1 << permission);
+}
+
+void DiscordCPP::Permissions::remove(const Permissions& permissions) {
+    this->permissions &= ~permissions.permissions;
+}
+
+bool DiscordCPP::Permissions::has_permission(Permission permission) const {
     return permissions & (1 << permission);
 }
 
-bool DiscordCPP::Permissions::has_all_permissions(std::vector<Permission> permissions) {
+bool DiscordCPP::Permissions::has_all_permissions(std::vector<Permission> permissions) const {
     for (auto permission : permissions) {
         if (!has_permission(permission)) {
             return false;
@@ -33,11 +41,22 @@ bool DiscordCPP::Permissions::has_all_permissions(std::vector<Permission> permis
     return true;
 }
 
-bool DiscordCPP::Permissions::has_any_permission(std::vector<Permission> permissions) {
+bool DiscordCPP::Permissions::has_any_permission(std::vector<Permission> permissions) const {
     for (auto permission : permissions) {
         if (has_permission(permission)) {
             return true;
         }
     }
     return false;
 }
+
+DiscordCPP::PermissionOverwrites::PermissionOverwrites(const json& data) {
+    id = data.at("id").get<std::string>();
+    type = static_cast<Type>(data["type"].get<int>());
+    if (get_or_else<std::string>(data, "allow", "").length() > 0) {
+        allow = Permissions(data.at("allow").get<std::string>());
+    }
+    if (get_or_else<std::string>(data, "deny", "").length() > 0) {
+        deny = Permissions(data.at("deny").get<std::string>());
+    }
+}

Discord.C++/Permissions.h

@@ -118,11 +118,50 @@ class Permissions {
     DLL_EXPORT Permissions() : permissions(0) {};
     DLL_EXPORT Permissions(const std::string& permissions);
 
+    /// add the given permission
     DLL_EXPORT void add(const Permission& permission);
+    /// merge the given permission set
     DLL_EXPORT void add(const Permissions& permissions);
+    /// remove the given permission
+    DLL_EXPORT void remove(const Permission& permission);
+    /// remove the given permission set
+    DLL_EXPORT void remove(const Permissions& permissions);
 
-    DLL_EXPORT bool has_permission(Permission permission);
-    DLL_EXPORT bool has_all_permissions(std::vector<Permission> permissions);
-    DLL_EXPORT bool has_any_permission(std::vector<Permission> permissions);
+    /// check if the given permission is set
+    DLL_EXPORT bool has_permission(Permission permission) const;
+    /// check if all given permissions is set
+    DLL_EXPORT bool has_all_permissions(std::vector<Permission> permissions) const;
+    /// check if at least one of the given permissions is set
+    DLL_EXPORT bool has_any_permission(std::vector<Permission> permissions) const;
+};
+
+class PermissionOverwrites {
+   public:
+    enum Type {
+        ROLE = 0,
+        MEMBER = 1
+    };
+
+   private:
+    /// role or user id
+    std::string id;
+    /// overwite type
+    Type type;
+    /// allowed permissions
+    Permissions allow;
+    /// denied permissions
+    Permissions deny;
+
+   public:
+    DLL_EXPORT PermissionOverwrites(const json& data);
+
+    /// @return role or user id
+    DLL_EXPORT std::string get_id() const { return id; }
+    /// @return overwite type
+    DLL_EXPORT Type get_type() const { return type; }
+    /// @return allowed permissions
+    DLL_EXPORT Permissions get_allowed_permissions() const { return allow; }
+    /// @return denied permissions
+    DLL_EXPORT Permissions get_denied_permissions() const { return deny; }
 };
 }  // namespace DiscordCPP

test_bot/main.cpp

@@ -2,8 +2,6 @@
 #include <thread>
 #include <vector>
 
-#include "ApplicationCommand.h"
-
 #ifndef _WIN32
 #include <cstdlib>
 #endif
@@ -107,13 +105,13 @@ class Client : public Discord {
         react.set_type(ApplicationCommand::Type::MESSAGE);
         create_application_command(react);
 
-        ApplicationCommand admin = ApplicationCommand();
-        admin.set_name("admin");
-        admin.set_description("check if you are an admin");
-        admin.set_type(DiscordCPP::ApplicationCommand::CHAT_INPUT);
-        admin.add_contexts(DiscordCPP::ApplicationCommand::GUILD);
-        admin.add_integration_types(DiscordCPP::ApplicationCommand::GUILD_INSTALL);
-        create_application_command(admin);
+        ApplicationCommand permissions = ApplicationCommand();
+        permissions.set_name("permissions");
+        permissions.set_description("check some permissions");
+        permissions.set_type(DiscordCPP::ApplicationCommand::CHAT_INPUT);
+        permissions.add_contexts(DiscordCPP::ApplicationCommand::GUILD);
+        permissions.add_integration_types(DiscordCPP::ApplicationCommand::GUILD_INSTALL);
+        create_application_command(permissions);
 
         update_presence(DiscordStatus::Online, Activity("test", Activity::Type::Game));
     }
@@ -467,11 +465,25 @@ class Client : public Discord {
                 this_thread::sleep_for(chrono::seconds(1));
                 message.remove_all_reactions();
             }
-        } else if (name == "admin") {
-            if (interaction.get_member().has_value() && interaction.get_guild().has_value()) {
-                Permissions permissions = interaction.get_guild().value().get_member_permissions(interaction.get_member().value());
+        } else if (name == "permissions") {
+            if (interaction.get_member().has_value() && interaction.get_guild().has_value() && interaction.get_channel().has_value()) {
+                Member member = interaction.get_member().value();
+                Channel channel = interaction.get_channel().value();
+                Guild guild = interaction.get_guild().value();
+
+                Permissions permissions = guild.get_member_permissions(member);
+                permissions = channel.merge_permission_overwrites(permissions, member.get_id());
+                for (auto role : member.get_roles()) {
+                    permissions = channel.merge_permission_overwrites(permissions, role);
+                }
+
                 bool is_admin = permissions.has_permission(DiscordCPP::Permissions::ADMINISTRATOR);
-                interaction.reply("You are " + std::string(is_admin ? "an" : "no") + " admin");
+                bool can_delete_messages = permissions.has_permission(DiscordCPP::Permissions::MANAGE_MESSAGES);
+                bool can_kick_members = permissions.has_permission(DiscordCPP::Permissions::KICK_MEMBERS);
+
+                interaction.reply("You are " + std::string(is_admin ? "an" : "no") + " admin"                                     //
+                                  + "\nYou are " + std::string(can_delete_messages ? "" : "not ") + "allowed to delete messages"  //
+                                  + "\nYou can " + std::string(can_kick_members ? "" : "not ") + "kick members");
             }
         }
     }