From 020c10292bc1dc211443c1fb0927c262a2826d91 Mon Sep 17 00:00:00 2001
From: Luck <git@lucko.me>
Date: Fri, 5 Mar 2021 21:19:07 +0000
Subject: Add permissions for each individual subcommand

---
 .../java/me/lucko/spark/common/SparkPlatform.java  | 32 ++++++++++++++++------
 .../me/lucko/spark/common/command/Command.java     |  4 +++
 2 files changed, 27 insertions(+), 9 deletions(-)

(limited to 'spark-common/src/main/java/me/lucko/spark')

diff --git a/spark-common/src/main/java/me/lucko/spark/common/SparkPlatform.java b/spark-common/src/main/java/me/lucko/spark/common/SparkPlatform.java
index f03bab2..cf3b0ee 100644
--- a/spark-common/src/main/java/me/lucko/spark/common/SparkPlatform.java
+++ b/spark-common/src/main/java/me/lucko/spark/common/SparkPlatform.java
@@ -163,10 +163,20 @@ public class SparkPlatform {
         return this.serverNormalOperationStartTime;
     }
 
+    private List<Command> getAvailableCommands(CommandSender sender) {
+        if (sender.hasPermission("spark")) {
+            return this.commands;
+        }
+        return this.commands.stream()
+                .filter(c -> sender.hasPermission("spark." + c.primaryAlias()))
+                .collect(Collectors.toList());
+    }
+
     public void executeCommand(CommandSender sender, String[] args) {
         CommandResponseHandler resp = new CommandResponseHandler(this, sender);
+        List<Command> commands = getAvailableCommands(sender);
 
-        if (!sender.hasPermission("spark")) {
+        if (commands.isEmpty()) {
             resp.replyPrefixed(text("You do not have permission to use this command.", RED));
             return;
         }
@@ -197,7 +207,7 @@ public class SparkPlatform {
         ArrayList<String> rawArgs = new ArrayList<>(Arrays.asList(args));
         String alias = rawArgs.remove(0).toLowerCase();
 
-        for (Command command : this.commands) {
+        for (Command command : commands) {
             if (command.aliases().contains(alias)) {
                 try {
                     command.executor().execute(this, sender, resp, new Arguments(rawArgs));
@@ -208,25 +218,29 @@ public class SparkPlatform {
             }
         }
 
-        sendUsage(resp);
+        sendUsage(commands, resp);
     }
 
     public List<String> tabCompleteCommand(CommandSender sender, String[] args) {
-        if (!sender.hasPermission("spark")) {
+        List<Command> commands = getAvailableCommands(sender);
+        if (commands.isEmpty()) {
             return Collections.emptyList();
         }
 
         List<String> arguments = new ArrayList<>(Arrays.asList(args));
 
         if (args.length <= 1) {
-            List<String> mainCommands = this.commands.stream().map(c -> c.aliases().get(0)).collect(Collectors.toList());
+            List<String> mainCommands = commands.stream()
+                    .map(Command::primaryAlias)
+                    .collect(Collectors.toList());
+
             return TabCompleter.create()
                     .at(0, CompletionSupplier.startsWith(mainCommands))
                     .complete(arguments);
         }
 
         String alias = arguments.remove(0);
-        for (Command command : this.commands) {
+        for (Command command : commands) {
             if (command.aliases().contains(alias)) {
                 return command.tabCompleter().completions(this, sender, arguments);
             }
@@ -235,15 +249,15 @@ public class SparkPlatform {
         return Collections.emptyList();
     }
 
-    private void sendUsage(CommandResponseHandler sender) {
+    private void sendUsage(List<Command> commands, CommandResponseHandler sender) {
         sender.replyPrefixed(text()
                 .append(text("spark", WHITE))
                 .append(space())
                 .append(text("v" + getPlugin().getVersion(), GRAY))
                 .build()
         );
-        for (Command command : this.commands) {
-            String usage = "/" + getPlugin().getCommandName() + " " + command.aliases().get(0);
+        for (Command command : commands) {
+            String usage = "/" + getPlugin().getCommandName() + " " + command.primaryAlias();
             ClickEvent clickEvent = ClickEvent.suggestCommand(usage);
             sender.reply(text()
                     .append(text(">", GOLD, BOLD))
diff --git a/spark-common/src/main/java/me/lucko/spark/common/command/Command.java b/spark-common/src/main/java/me/lucko/spark/common/command/Command.java
index db23454..e1a5146 100644
--- a/spark-common/src/main/java/me/lucko/spark/common/command/Command.java
+++ b/spark-common/src/main/java/me/lucko/spark/common/command/Command.java
@@ -62,6 +62,10 @@ public class Command {
         return this.tabCompleter;
     }
 
+    public String primaryAlias() {
+        return this.aliases.get(0);
+    }
+
     public static final class Builder {
         private final ImmutableList.Builder<String> aliases = ImmutableList.builder();
         private final ImmutableList.Builder<ArgumentInfo> arguments = ImmutableList.builder();
-- 
cgit