From 053438bfbf8959a0c15d52014734bb54ba089bdf Mon Sep 17 00:00:00 2001
From: fullwall <fullwall25@gmail.com>
Date: Sun, 25 Jun 2023 00:25:49 +0800
Subject: [PATCH] Add /npc bossbar --viewpermission

---
 .../trait/versioned/BossBarTrait.java         | 21 ++++++++++++++++---
 1 file changed, 18 insertions(+), 3 deletions(-)

diff --git a/main/src/main/java/net/citizensnpcs/trait/versioned/BossBarTrait.java b/main/src/main/java/net/citizensnpcs/trait/versioned/BossBarTrait.java
index a573ddafb..d6125e6c9 100644
--- a/main/src/main/java/net/citizensnpcs/trait/versioned/BossBarTrait.java
+++ b/main/src/main/java/net/citizensnpcs/trait/versioned/BossBarTrait.java
@@ -51,6 +51,8 @@ public class BossBarTrait extends Trait {
     @Persist
     private String track;
     @Persist
+    private String viewPermission;
+    @Persist
     private boolean visible = true;
 
     public BossBarTrait() {
@@ -91,6 +93,10 @@ public class BossBarTrait extends Trait {
         return track;
     }
 
+    public String getViewPermission() {
+        return viewPermission;
+    }
+
     private boolean isBoss(Entity entity) {
         boolean isBoss = entity.getType() == EntityType.ENDER_DRAGON || entity.getType() == EntityType.WITHER
                 || entity.getType() == EntityType.GUARDIAN;
@@ -168,6 +174,8 @@ public class BossBarTrait extends Trait {
             barCache.removeAll();
             for (Player player : CitizensAPI.getLocationLookup().getNearbyPlayers(npc.getEntity().getLocation(),
                     range > 0 ? range : Setting.BOSSBAR_RANGE.asInt())) {
+                if (viewPermission != null && !player.hasPermission(viewPermission))
+                    continue;
                 barCache.addPlayer(player);
             }
         }
@@ -201,13 +209,17 @@ public class BossBarTrait extends Trait {
         this.track = variable;
     }
 
+    public void setViewPermission(String viewpermission) {
+        this.viewPermission = viewpermission;
+    }
+
     public void setVisible(boolean visible) {
         this.visible = visible;
     }
 
     @Command(
             aliases = { "npc" },
-            usage = "bossbar --style [style] --color [color] --title [title] --visible [visible] --flags [flags] --track [health | placeholder] --range [range]",
+            usage = "bossbar --style [style] --color [color] --title [title] --visible [visible] --viewpermission [permission] --flags [flags] --track [health | placeholder] --range [range]",
             desc = "Edit bossbar properties",
             modifiers = { "bossbar" },
             min = 1,
@@ -215,8 +227,8 @@ public class BossBarTrait extends Trait {
     @Requirements(selected = true, ownership = true)
     public static void bossbar(CommandContext args, CommandSender sender, NPC npc, @Flag("style") BarStyle style,
             @Flag("track") String track, @Flag("color") BarColor color, @Flag("visible") Boolean visible,
-            @Flag("range") Integer range, @Flag("title") String title, @Flag("flags") String flags)
-            throws CommandException {
+            @Flag("range") Integer range, @Flag("title") String title, @Flag("flags") String flags,
+            @Flag("viewpermission") String viewpermission) throws CommandException {
         BossBarTrait trait = npc.getOrAddTrait(BossBarTrait.class);
         if (style != null) {
             trait.setStyle(style);
@@ -236,6 +248,9 @@ public class BossBarTrait extends Trait {
         if (range != null) {
             trait.setRange(range);
         }
+        if (viewpermission != null) {
+            trait.setViewPermission(viewpermission);
+        }
         if (flags != null) {
             List<BarFlag> parsed = Lists.newArrayList();
             for (String s : Splitter.on(',').omitEmptyStrings().trimResults().split(flags)) {