Initial commit

2022-03-09 22:11:11 +01:00
commit 9f1f62d593
34 changed files with 933 additions and 0 deletions
@@ -0,0 +1,75 @@
+package com.alttd.permissions;
+
+import com.alttd.util.Logger;
+import net.dv8tion.jda.api.entities.*;
+
+import java.util.HashMap;
+import java.util.List;
+import java.util.stream.Collectors;
+
+public class PermissionManager {
+
+    HashMap<Long, List<String>> userPermissions;
+    HashMap<Long, List<String>> groupPermissions;
+    HashMap<Long, List<String>> channelEnabledCommands;
+    List<String> privateEnabledCommands;
+
+    public PermissionManager(HashMap<Long, List<String>> userPermissions,
+                             HashMap<Long, List<String>> groupPermissions,
+                             HashMap<Long, List<String>> channelEnabledCommands,
+                             List<String> privateEnabledCommands) {
+        this.userPermissions = userPermissions;
+        this.groupPermissions = groupPermissions;
+        this.channelEnabledCommands = channelEnabledCommands;
+        this.privateEnabledCommands = privateEnabledCommands;
+    }
+
+    public boolean hasPermission(TextChannel textChannel, User user, String permission) {
+        permission = permission.toLowerCase();
+        if (textChannel instanceof PrivateChannel) {
+            if (isDisabled(privateEnabledCommands, permission))
+                return false;
+            return hasPermission(user.getIdLong(), null, permission);
+        } else {
+            Logger.warning("Using user for Guild channel % ", textChannel.getAsMention());
+            return false;
+        }
+    }
+
+    public boolean hasPermission(TextChannel textChannel, Member member, String permission) {
+        permission = permission.toLowerCase();
+        if (isDisabled(channelEnabledCommands.get(textChannel.getIdLong()), permission.toLowerCase()))
+            return false;
+        return hasPermission(
+                member.getIdLong(),
+                member.getRoles().stream()
+                        .map(Role::getIdLong)
+                        .collect(Collectors.toList()),
+                permission);
+    }
+
+    private boolean isDisabled(List<String> enabledCommandList, String permission) {
+        if (enabledCommandList == null || enabledCommandList.isEmpty())
+            return false;
+        return !enabledCommandList.contains(permission);
+    }
+
+    private boolean hasPermission(long userId, List<Long> groupIds, String permission) {
+        if (hasPermission(userPermissions.get(userId), permission))
+            return true;
+        if (groupIds == null || groupIds.isEmpty())
+            return false;
+        for (long groupId : groupIds) {
+            if (hasPermission(groupPermissions.get(groupId), permission))
+                return true;
+        }
+        return false;
+    }
+
+    private boolean hasPermission(List<String> permissions, String permission) {
+        if (permission == null || permission.isEmpty())
+            return false;
+        return permissions.contains(permission);
+    }
+
+}