API: Primary group support

Author: Innectic

API/src/main/java/me/innectic/permissify/api/database/DatabaseHandler.java

@@ -148,9 +148,9 @@ protected void displayError(ConnectionError error, Exception e) {
     /**
      * Remove a player from a group.
      *
-     * @param uuid the player to remove
+     * @param uuid  the player to remove
      * @param group the group to remove from
-     * @return if the player was removed
+     * @return      if the player was removed
      */
     public abstract boolean removePlayerFromGroup(UUID uuid, PermissionGroup group);
 
@@ -165,10 +165,26 @@ protected void displayError(ConnectionError error, Exception e) {
      * Get all permission groups a player is in.
      *
      * @param uuid the uuid of the player
-     * @return the groups the player is in
+     * @return     the groups the player is in
      */
     public abstract List<PermissionGroup> getGroups(UUID uuid);
 
+    /**
+     * Set the primary group of the player
+     *
+     * @param group the group to set as the primary
+     * @param uuid  the uuid of the player to set the primary of
+     */
+    public abstract boolean setPrimaryGroup(PermissionGroup group, UUID uuid);
+
+    /**
+     * Get the primary group of a player.
+     *
+     * @param uuid the uuid of the player to get
+     * @return     the primary group of the player
+     */
+    public abstract Optional<PermissionGroup> getPrimaryGroup(UUID uuid);
+
     /**
      * Update a player's cache.
      *

API/src/main/java/me/innectic/permissify/api/database/handlers/MySQLHandler.java

@@ -335,6 +335,35 @@ public List<PermissionGroup> getGroups(UUID uuid) {
         return cachedGroups.stream().filter(group -> group.hasPlayer(uuid)).collect(Collectors.toList());
     }
 
+    @Override
+    public boolean setPrimaryGroup(PermissionGroup group, UUID uuid) {
+        // @Return: Should primary group put the player in the group if they're not already in it?
+        if (!group.hasPlayer(uuid)) return false;
+        if (group.isPrimaryGroup(uuid)) return false;
+        group.setPrimaryGroup(uuid, true);
+        Optional<Connection> connection = getConnection();
+        if (!connection.isPresent()) {
+            displayError(ConnectionError.REJECTED);
+            return false;
+        }
+
+        try {
+            PreparedStatement statement = connection.get().prepareStatement("UPDATE groupMembers SET `primary`=? WHERE uuid=? AND `group`=?");
+            statement.setBoolean(1, true);
+            statement.setString(2, uuid.toString());
+            statement.setString(3, group.getName());
+            statement.execute();
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+        return true;
+    }
+
+    @Override
+    public Optional<PermissionGroup> getPrimaryGroup(UUID uuid) {
+        return getGroups(uuid).stream().filter(group -> group.isPrimaryGroup(uuid)).findFirst();
+    }
+
     @Override
     public void updateCache(UUID uuid) {
         // TODO: Make this method not exist. It shouldn't be needed.

API/src/main/java/me/innectic/permissify/api/permission/PermissionGroup.java

@@ -26,11 +26,9 @@
 
 import lombok.Getter;
 import lombok.RequiredArgsConstructor;
+import lombok.Setter;
 
-import java.util.ArrayList;
-import java.util.List;
-import java.util.Optional;
-import java.util.UUID;
+import java.util.*;
 
 /**
  * @author Innectic
@@ -43,7 +41,7 @@ public class PermissionGroup {
     @Getter private final String prefix;
     @Getter private final String suffix;
     @Getter private List<Permission> permissions = new ArrayList<>();
-    @Getter private List<UUID> players = new ArrayList<>();
+    @Getter private Map<UUID, Boolean> players = new HashMap<>();
 
     /**
      * Remove a permission from the group
@@ -74,14 +72,22 @@ public boolean hasPermission(String permission) {
     }
 
     public void addPlayer(UUID uuid) {
-        if (!players.contains(uuid)) players.add(uuid);
+        if (!hasPlayer(uuid)) players.put(uuid, false);
     }
 
     public void removePlayer(UUID uuid) {
-        players.removeIf(entry -> entry.equals(uuid));
+        players.remove(uuid);
     }
 
     public boolean hasPlayer(UUID uuid) {
-        return players.contains(uuid);
+        return players.containsKey(uuid);
+    }
+
+    public boolean isPrimaryGroup(UUID uuid) {
+        return players.containsKey(uuid) && players.get(uuid).equals(true);
+    }
+
+    public void setPrimaryGroup(UUID uuid, boolean isPrimary) {
+        players.put(uuid, isPrimary);
     }
 }