add role permissions

Author: Arkrissym

CMakeLists.txt

@@ -78,7 +78,7 @@ find_package(OpenSSL REQUIRED)
 if(VCPKG_TOOLCHAIN)
       find_package(unofficial-sodium CONFIG REQUIRED)
       find_package(Opus CONFIG REQUIRED)
-      
+
       target_link_libraries(${PROJECT_NAME}
                         unofficial-sodium::sodium
                         Opus::opus

Discord.C++/Guild.cpp

@@ -5,6 +5,7 @@
 
 #include "ChannelHelper.h"
 #include "Emoji.h"
+#include "Exceptions.h"
 #include "Role.h"
 #include "VoiceChannel.h"
 #include "static.h"
@@ -199,6 +200,32 @@ void DiscordCPP::Guild::unban(const User& user) {
     api_call(url, "DEL");
 }
 
+DiscordCPP::Permissions DiscordCPP::Guild::get_member_permissions(const Member& member) {
+    if (owner_id.has_value() && owner_id.value() == member.get_id()) {
+        return Permissions::All();
+    }
+
+    std::map<std::string, Permissions> permission_map;
+    for (const Role& role : roles) {
+        permission_map.insert({role.get_id(), role.get_permissions()});
+    }
+
+    Permissions permissions;
+    for (std::string role : member.get_roles()) {
+        if (permission_map.find(role) == permission_map.end()) {
+            throw DiscordException("unknown role " + role + " for member " + member.get_id() + " in guild " + get_id());
+        }
+
+        if (permission_map.at(role).has_permission(Permissions::ADMINISTRATOR)) {
+            return Permissions::All();
+        }
+
+        permissions.add(permission_map.at(role));
+    }
+
+    return permissions;
+}
+
 std::optional<DiscordCPP::User> DiscordCPP::Guild::get_owner() {
     if (!owner.has_value() && owner_id.has_value()) {
         owner.emplace(owner_id.value(), get_token());

Discord.C++/Guild.h

@@ -150,6 +150,10 @@ class Guild : public DiscordObject {
     /// unbans the given User from this Guild
     DLL_EXPORT void unban(const User& user);
 
+    /** @return the members permissions in the guild
+        @throws DiscordException if an unknown role is encountered */
+    DLL_EXPORT Permissions get_member_permissions(const Member& member);
+
     /// @return the guild's name
     DLL_EXPORT std::string get_name() { return name; }
     /// @return the guild's icon hash

Discord.C++/Interaction.cpp

@@ -1,6 +1,7 @@
 #include "Interaction.h"
 
 #include "Embed.h"
+#include "Guild.h"
 
 DiscordCPP::Interaction::Interaction(const json& data, const std::string& token)
     : DiscordCPP::DiscordObject(token, data.at("id").get<std::string>()),
@@ -26,6 +27,19 @@ DiscordCPP::Interaction::Interaction(const json& data, const std::string& token)
     guild_locale = get_optional<std::string>(data, "guild_locale");
 }
 
+DiscordCPP::Interaction::Interaction(const Interaction& old)
+    : DiscordCPP::DiscordObject(old) {
+    if (old.guild != nullptr) {
+        guild = new Guild(*old.guild);
+    }
+}
+
+DiscordCPP::Interaction::~Interaction() {
+    if (guild != nullptr) {
+        delete guild;
+    }
+}
+
 void DiscordCPP::Interaction::reply(const std::string& content, const bool tts) {
     std::string url = "/interactions/" + get_id() + "/" + token + "/callback";
 
@@ -37,8 +51,8 @@ void DiscordCPP::Interaction::reply(const std::string& content, const bool tts)
                 //
                 {"content", content},
                 {"tts", tts}  //
-            }                 //
-        }                     //
+            }  //
+        }  //
     };
 
     api_call(url, "POST", data, "application/json");
@@ -74,3 +88,10 @@ void DiscordCPP::Interaction::update_reply(Embed embed) {
 
     api_call(url, "PATCH", data, "application/json");
 }
+
+std::optional<DiscordCPP::Guild> DiscordCPP::Interaction::get_guild() {
+    if (guild_id.has_value() && guild == nullptr) {
+        guild = new Guild(nullptr, guild_id.value(), get_token());
+    }
+    return (guild != nullptr) ? std::optional<Guild>{*guild} : std::nullopt;
+}

Discord.C++/Interaction.h

@@ -9,6 +9,8 @@
 
 namespace DiscordCPP {
 
+class Guild;
+
 class Interaction : public DiscordObject {
    public:
     enum Type {
@@ -31,6 +33,7 @@ class Interaction : public DiscordObject {
     std::optional<InteractionData> data;
     /// The guild the interaction was sent from.
     std::optional<std::string> guild_id;
+    Guild* guild = nullptr;
     /// The channel the interaction was sent from.
     std::optional<std::string> channel_id;
     /// The member that sent the interaction.
@@ -46,6 +49,8 @@ class Interaction : public DiscordObject {
 
    public:
     DLL_EXPORT Interaction(const json& data, const std::string& token);
+    DLL_EXPORT Interaction(const Interaction& old);
+    DLL_EXPORT ~Interaction();
 
     /// Send a Message as a reply to this interaction.
     DLL_EXPORT void reply(const std::string& content, const bool tts = false);
@@ -58,25 +63,27 @@ class Interaction : public DiscordObject {
     DLL_EXPORT void update_reply(Embed embed);
 
     /// @return ID of the application this interaction is for.
-    DLL_EXPORT std::string get_application_id() { return application_id; }
+    DLL_EXPORT std::string get_application_id() const { return application_id; }
     /// @return The type of the invoked command.
-    DLL_EXPORT Type get_type() { return type; }
+    DLL_EXPORT Type get_type() const { return type; }
     /// @return The command data payload.
-    DLL_EXPORT std::optional<InteractionData> get_data() { return data; }
+    DLL_EXPORT std::optional<InteractionData> get_data() const { return data; }
+    /// @return The guild the interaction was sent from.
+    DLL_EXPORT std::optional<std::string> get_guild_id() const { return guild_id; }
     /// @return The guild the interaction was sent from.
-    DLL_EXPORT std::optional<std::string> get_guild_id() { return guild_id; }
+    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() { return channel_id; }
+    DLL_EXPORT std::optional<std::string> get_channel_id() const { return channel_id; }
     /// @return The member that sent the interaction.
-    DLL_EXPORT std::optional<Member> get_member() { return member; }
+    DLL_EXPORT std::optional<Member> get_member() const { return member; }
     /// @return The user that sent the interaction.
-    DLL_EXPORT std::optional<User> get_user() { return user; }
+    DLL_EXPORT std::optional<User> get_user() const { return user; }
     /// @return For components, the message they were attached to.
-    DLL_EXPORT std::optional<Message> get_message() { return message; }
+    DLL_EXPORT std::optional<Message> get_message() const { return message; }
     /// @return The selected locale of the user who sent this interaction.
-    DLL_EXPORT std::optional<std::string> get_locale() { return locale; }
+    DLL_EXPORT std::optional<std::string> get_locale() const { return locale; }
     /// @return The guild's preferred locale.
-    DLL_EXPORT std::optional<std::string> get_guild_locale() { return guild_locale; }
+    DLL_EXPORT std::optional<std::string> get_guild_locale() const { return guild_locale; }
 };
 
 }  // namespace DiscordCPP

Discord.C++/Member.cpp

@@ -1,11 +1,15 @@
 #include "Member.h"
 
 DiscordCPP::Member::Member(const json& data, const std::string& token) : User(data.at("user"), token) {
-    nick = get_or_else<std::string>(data, "nick", "");
     nick = get_or_else<std::string>(data, "nick", "");
     deaf = get_or_else<bool>(data, "deaf", false);
     mute = get_or_else<bool>(data, "mute", false);
-    // roles
+
+    if (has_value(data, "roles")) {
+        for (const json& role : data.at("roles")) {
+            role_ids.push_back(role.get<std::string>());
+        }
+    }
 }
 
 DiscordCPP::Member::operator std::string() {

Discord.C++/Member.h

@@ -1,5 +1,4 @@
 #pragma once
-#include <iostream>
 
 #include "User.h"
 #include "static.h"
@@ -10,7 +9,8 @@ class Member : public User {
    private:
     /// the member's nickname
     std::string nick;
-    // vector<Role> roles;
+    /// the member's roles
+    std::vector<std::string> role_ids;
     /// when the member joined the server
     std::string joined_at;  // ISO8601 timestamp
     /// true, if the member is defeaned
@@ -25,14 +25,16 @@ class Member : public User {
     ///@return Nickname (Username if not present) as std::string
     DLL_EXPORT explicit operator std::string() override;
 
-    /// the member's nickname
-    DLL_EXPORT std::string get_nick() { return nick; }
-    /// when the member joined the server
-    DLL_EXPORT std::string get_joined_at() { return joined_at; }
-    /// true, if the member is defeaned
-    DLL_EXPORT bool is_deaf() { return deaf; }
-    /// true, if the member is muted
-    DLL_EXPORT bool is_mute() { return mute; }
+    /// @return the member's nickname
+    DLL_EXPORT std::string get_nick() const { return nick; }
+    /// @return the member's roles
+    DLL_EXPORT std::vector<std::string> get_roles() const { return role_ids; }
+    /// @return when the member joined the server
+    DLL_EXPORT std::string get_joined_at() const { return joined_at; }
+    /// @return true, if the member is defeaned
+    DLL_EXPORT bool is_deaf() const { return deaf; }
+    /// @return true, if the member is muted
+    DLL_EXPORT bool is_mute() const { return mute; }
 };
 
 }  // namespace DiscordCPP

Discord.C++/Permissions.cpp

@@ -0,0 +1,43 @@
+#include "Permissions.h"
+
+DiscordCPP::Permissions DiscordCPP::Permissions::All() {
+    Permissions permissions;
+    permissions.permissions = UINT64_MAX;
+    return permissions;
+}
+
+DiscordCPP::Permissions::Permissions(const std::string& permissions) {
+    std::istringstream iss(permissions);
+    iss.exceptions(std::ios::failbit | std::ios::badbit);
+    iss >> this->permissions;
+}
+
+void DiscordCPP::Permissions::add(const Permission& permission) {
+    permissions |= (1 << permission);
+}
+
+void DiscordCPP::Permissions::add(const Permissions& permissions) {
+    this->permissions |= permissions.permissions;
+}
+
+bool DiscordCPP::Permissions::has_permission(Permission permission) {
+    return permissions & (1 << permission);
+}
+
+bool DiscordCPP::Permissions::has_all_permissions(std::vector<Permission> permissions) {
+    for (auto permission : permissions) {
+        if (!has_permission(permission)) {
+            return false;
+        }
+    }
+    return true;
+}
+
+bool DiscordCPP::Permissions::has_any_permission(std::vector<Permission> permissions) {
+    for (auto permission : permissions) {
+        if (has_permission(permission)) {
+            return true;
+        }
+    }
+    return false;
+}