From 0124b6952c29bf22af5267d7f33b67035a17d932 Mon Sep 17 00:00:00 2001
From: destro174 <40720638+destro174@users.noreply.github.com>
Date: Mon, 23 Aug 2021 12:12:30 +0200
Subject: [PATCH] Add support for primary prefixes

---
 .../java/com/alttd/chat/util/Utility.java     | 23 +++++++++++++++++--
 1 file changed, 21 insertions(+), 2 deletions(-)

diff --git a/api/src/main/java/com/alttd/chat/util/Utility.java b/api/src/main/java/com/alttd/chat/util/Utility.java
index 88245b7..b222e43 100755
--- a/api/src/main/java/com/alttd/chat/util/Utility.java
+++ b/api/src/main/java/com/alttd/chat/util/Utility.java
@@ -78,7 +78,8 @@ public class Utility {
                         }
                     });
         }
-        prefix.append(user.getCachedData().getMetaData().getPrefix());
+        prefix.append(getUserPrefix(user));
+//        prefix.append(user.getCachedData().getMetaData().getPrefix());
 
         return applyColor(prefix.toString());
     }
@@ -91,11 +92,29 @@ public class Utility {
         if(user.getCachedData().getPermissionData().checkPermission("group." + Config.MINIMIUMSTAFFRANK).asBoolean()) {
             Group group = luckPerms.getGroupManager().getGroup(user.getPrimaryGroup());
             if(group != null)
-                prefix.append(group.getCachedData().getMetaData().getPrefix());
+                prefix.append(getGroupPrefix(group));
+//                prefix.append(group.getCachedData().getMetaData().getPrefix());
         }
         return applyColor(prefix.toString());
     }
 
+    public static String getUserPrefix(User user) {
+        LuckPerms luckPerms = ChatAPI.get().getLuckPerms();
+        Group group = luckPerms.getGroupManager().getGroup(user.getPrimaryGroup());
+        if (group == null) {
+            return "";
+        }
+        return ChatAPI.get().getPrefixes().get(group.getName()).replace("<prefix>", user.getCachedData().getMetaData().getPrefix());
+    }
+
+    public static String getGroupPrefix(String groupName) {
+        Group group = ChatAPI.get().getLuckPerms().getGroupManager().getGroup(groupName);
+        if (group == null) {
+            return "";
+        }
+        return getGroupPrefix(group);
+    }
+
     public static String getGroupPrefix(Group group) {
         return ChatAPI.get().getPrefixes().get(group.getName()).replace("<prefix>", group.getCachedData().getMetaData().getPrefix());
     }