Store player save & version data on Game Sync data creation (#36)

Author: kuroppoi

src/main/java/entralinked/model/player/PlayerManager.java

@@ -17,6 +17,8 @@
 import com.fasterxml.jackson.databind.ObjectMapper;
 import com.fasterxml.jackson.databind.SerializationFeature;
 
+import entralinked.GameVersion;
+
 /**
  * Manager class for managing {@link Player} information (Global Link users)
  */
@@ -175,7 +177,7 @@ private boolean savePlayer(Player player, File outputFile) {
      * That is, the specified Game Sync ID wasn't already registered and the player data
      * was saved without any errors.
      */
-    public Player registerPlayer(String gameSyncId) {
+    public Player registerPlayer(String gameSyncId, GameVersion version) {
         // Check for duplicate Game Sync ID
         if(playerMap.containsKey(gameSyncId)) {
             logger.warn("Attempted to register duplicate player {}", gameSyncId);
@@ -193,6 +195,7 @@ public Player registerPlayer(String gameSyncId) {
         // Construct player object
         Player player = new Player(gameSyncId);
         player.setStatus(PlayerStatus.AWAKE);
+        player.setGameVersion(version);
         player.setDataDirectory(playerDataDirectory);
 
         // Try to save player data

src/main/java/entralinked/network/http/pgl/PglHandler.java

@@ -1,5 +1,6 @@
 package entralinked.network.http.pgl;
 
+import java.io.ByteArrayInputStream;
 import java.io.File;
 import java.io.FileInputStream;
 import java.io.IOException;
@@ -355,6 +356,12 @@ private void handleMemoryLink(PglRequest request, Context ctx) throws IOExceptio
             return;
         }
 
+        // Version null check because this can happen in specific cases
+        if(player.getGameVersion() == null) {
+            writeStatusCode(outputStream, 5); // No game save data exists for this Game Sync ID
+            return;
+        }
+        
         // Check if the save file belongs to Black or White
         if(player.getGameVersion().isVersion2()) {
             writeStatusCode(outputStream, 10); // Not a Black or White save
@@ -485,13 +492,9 @@ private void handleUploadSaveData(PglRequest request, Context ctx) throws IOExce
     /**
      * POST handler for {@code /dsio/gw?p=account.create.upload}
      */
-    private void handleCreateAccount(PglRequest request, Context ctx) throws IOException {        
-        // It sends the entire save file, but we just skip through it because we don't need anything from it here
-        ServletInputStream inputStream = ctx.req().getInputStream();
-        
-        while(!inputStream.isFinished()) {
-            inputStream.read();
-        }
+    private void handleCreateAccount(PglRequest request, Context ctx) throws IOException {
+        // Have to read all the bytes first for some reason
+        byte[] bytes = ctx.bodyAsBytes();
 
         // Prepare response
         LEOutputStream outputStream = new LEOutputStream(ctx.outputStream());
@@ -509,11 +512,19 @@ private void handleCreateAccount(PglRequest request, Context ctx) throws IOExcep
         }
 
         // Try to register player
-        if(playerManager.registerPlayer(request.gameSyncId()) == null) {
+        Player player = playerManager.registerPlayer(request.gameSyncId(), request.gameVersion());
+        
+        if(player == null) {
             writeStatusCode(outputStream, 3); // Registration error
             return;
         }
 
+        // Try to store save data
+        if(!playerManager.storePlayerGameSaveFile(player, new ByteArrayInputStream(bytes))) {
+            writeStatusCode(outputStream, 4); // Game save data IO error
+            return;
+        }
+        
         // Write status code
         writeStatusCode(outputStream, 0);
     }
@@ -534,7 +545,8 @@ private void handleCreateData(PglRequest request, Context ctx) throws IOExceptio
         }
 
         // Try to register player
-        if(playerManager.registerPlayer(gameSyncId) == null) {
+        // Regrettably, this request does not contain game save & version data.
+        if(playerManager.registerPlayer(gameSyncId, null) == null) {
             writeStatusCode(outputStream, 3); // Registration error
             return;
         }