Differentiate Floodgate players during login

Author: Smart123s

core/src/main/java/com/github/games647/fastlogin/core/shared/FloodgateManagement.java

@@ -81,6 +81,38 @@ public void run() {
         }
 
         profile = core.getStorage().loadProfile(username);
+
+        if (profile.isSaved()) {
+            if (!profile.isFloodgateMigrated()) {
+                if (isLinked) {
+                    profile.setFloodgate(FloodgateState.LINKED);
+                    core.getPlugin().getLog().info(
+                            "Player {} will be migrated to the v2 database schema as a linked Floodgate user",
+                            username);
+                } else {
+                    profile.setFloodgate(FloodgateState.TRUE);
+                    core.getPlugin().getLog().info(
+                            "Player {} will be migrated to the v2 database schema as a Floodgate user", username);
+                }
+            } else if (profile.getFloodgate() == FloodgateState.TRUE && isLinked) {
+                core.getPlugin().getLog()
+                        .info("Player {} is already stored by FastLogin as a non-linked Bedrock Edition player",
+                                username);
+                return;
+            } else if (profile.getFloodgate() == FloodgateState.FALSE && isLinked) {
+                profile.setFloodgate(FloodgateState.LINKED);
+                core.getPlugin().getLog().info(
+                        "Player {} will be changed from a Java player to a linked Floodgate player",
+                        username);
+            }
+        } else {
+            if (isLinked) {
+                profile.setFloodgate(FloodgateState.LINKED);
+            } else {
+                profile.setFloodgate(FloodgateState.TRUE);
+            }
+        }
+
         AuthPlugin<P> authPlugin = core.getAuthPluginHook();
 
         try {
@@ -126,7 +158,8 @@ public void run() {
 
         //logging in from bedrock for a second time threw an error with UUID
         if (profile == null) {
-            profile = new StoredProfile(getUUID(player), username, true, getAddress(player).toString());
+            profile = new StoredProfile(getUUID(player), username, true, FloodgateState.TRUE,
+                    getAddress(player).toString());
         }
 
         //start Bukkit/Bungee specific tasks

core/src/main/java/com/github/games647/fastlogin/core/shared/JoinManagement.java

@@ -50,19 +50,33 @@ public JoinManagement(FastLoginCore<P, C, ?> core, AuthPlugin<P> authHook, Bedro
 
     public void onLogin(String username, S source) {
         core.getPlugin().getLog().info("Handling player {}", username);
-        StoredProfile profile = core.getStorage().loadProfile(username);
-        if (profile == null) {
-            return;
-        }
 
         //check if the player is connecting through Bedrock Edition
         if (bedrockService != null && bedrockService.isBedrockConnection(username)) {
-            //perform Bedrock specific checks and skip Java checks, if they are not needed
+            //perform Bedrock specific checks
             if (bedrockService.performChecks(username, source)) {
+                //skip Java checks, since they are not needed
                 return;
             }
         }
 
+        StoredProfile profile = core.getStorage().loadProfile(username);
+
+        //can't be a premium Java player, if it's not saved in the database
+        if (profile == null) {
+            return;
+        }
+
+        if (!profile.isFloodgateMigrated()) {
+            profile.setFloodgate(FloodgateState.FALSE);
+            core.getPlugin().getLog().info(
+                    "Player {} will be migrated to the v2 database schema as a JAVA user", username);
+        } else if (profile.getFloodgate() == FloodgateState.TRUE) {
+            core.getPlugin().getLog().info("Player {} is already stored by FastLogin as a Bedrock Edition player",
+                    username);
+            return;
+        }
+
         callFastLoginPreLoginEvent(username, source, profile);
 
         Configuration config = core.getConfig();
@@ -140,6 +154,12 @@ private boolean checkNameChange(S source, String username, Profile profile) {
         if (core.getConfig().get("nameChangeCheck", false)) {
             StoredProfile storedProfile = core.getStorage().loadProfile(profile.getId());
             if (storedProfile != null) {
+                if (storedProfile.getFloodgate() == FloodgateState.TRUE) {
+                    core.getPlugin().getLog()
+                            .info("Player {} is already stored by FastLogin as a Bedrock Edition player.", username);
+                    return false;
+                }
+
                 //uuid exists in the database
                 core.getPlugin().getLog().info("GameProfile {} changed it's username", profile);