Add command visibility check for Velocity and BungeeCord (#437)

3 files changed, 15 insertions, 0 deletions

diff --git a/spark-bungeecord/src/main/java/me/lucko/spark/bungeecord/BungeeCordSparkPlugin.java b/spark-bungeecord/src/main/java/me/lucko/spark/bungeecord/BungeeCordSparkPlugin.java
index 038f635..11f64db 100644
--- a/spark-bungeecord/src/main/java/me/lucko/spark/bungeecord/BungeeCordSparkPlugin.java
+++ b/spark-bungeecord/src/main/java/me/lucko/spark/bungeecord/BungeeCordSparkPlugin.java
@@ -129,5 +129,10 @@ public class BungeeCordSparkPlugin extends Plugin implements SparkPlugin {
         public Iterable<String> onTabComplete(CommandSender sender, String[] args) {
             return this.plugin.platform.tabCompleteCommand(new BungeeCordCommandSender(sender, this.plugin.audienceFactory), args);
         }
+
+        @Override
+        public boolean hasPermission(CommandSender sender) {
+            return this.plugin.platform.hasPermissionForAnyCommand(new BungeeCordCommandSender(sender, this.plugin.audienceFactory));
+        }
     }
 }
diff --git a/spark-velocity/src/main/java/me/lucko/spark/velocity/VelocitySparkPlugin.java b/spark-velocity/src/main/java/me/lucko/spark/velocity/VelocitySparkPlugin.java
index da7bd18..02fbe0b 100644
--- a/spark-velocity/src/main/java/me/lucko/spark/velocity/VelocitySparkPlugin.java
+++ b/spark-velocity/src/main/java/me/lucko/spark/velocity/VelocitySparkPlugin.java
@@ -88,6 +88,11 @@ public class VelocitySparkPlugin implements SparkPlugin, SimpleCommand {
     }
 
     @Override
+    public boolean hasPermission(Invocation inv) {
+        return this.platform.hasPermissionForAnyCommand(new VelocityCommandSender(inv.source()));
+    }
+
+    @Override
     public String getVersion() {
         return VelocitySparkPlugin.class.getAnnotation(Plugin.class).version();
     }
diff --git a/spark-velocity4/src/main/java/me/lucko/spark/velocity/Velocity4SparkPlugin.java b/spark-velocity4/src/main/java/me/lucko/spark/velocity/Velocity4SparkPlugin.java
index 0a0346e..abbade3 100644
--- a/spark-velocity4/src/main/java/me/lucko/spark/velocity/Velocity4SparkPlugin.java
+++ b/spark-velocity4/src/main/java/me/lucko/spark/velocity/Velocity4SparkPlugin.java
@@ -88,6 +88,11 @@ public class Velocity4SparkPlugin implements SparkPlugin, SimpleCommand {
     }
 
     @Override
+    public boolean hasPermission(Invocation inv) {
+        return this.platform.hasPermissionForAnyCommand(new Velocity4CommandSender(inv.source()));
+    }
+
+    @Override
     public String getVersion() {
         return Velocity4SparkPlugin.class.getAnnotation(Plugin.class).version();
     }