Add Dungeon Secrets loading

1 files changed, 90 insertions, 0 deletions

diff --git a/src/main/java/me/xmrvizzy/skyblocker/skyblock/dungeon/secrets/DungeonSecrets.java b/src/main/java/me/xmrvizzy/skyblocker/skyblock/dungeon/secrets/DungeonSecrets.java
new file mode 100644
index 00000000..6e72c1c4
--- /dev/null
+++ b/src/main/java/me/xmrvizzy/skyblocker/skyblock/dungeon/secrets/DungeonSecrets.java
@@ -0,0 +1,90 @@
+package me.xmrvizzy.skyblocker.skyblock.dungeon.secrets;
+
+import com.google.gson.JsonObject;
+import me.xmrvizzy.skyblocker.SkyblockerMod;
+import net.minecraft.client.MinecraftClient;
+import net.minecraft.util.Identifier;
+import org.jetbrains.annotations.Nullable;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.io.BufferedReader;
+import java.io.File;
+import java.io.IOException;
+import java.io.ObjectInputStream;
+import java.util.*;
+import java.util.concurrent.CompletableFuture;
+import java.util.zip.InflaterInputStream;
+
+public class DungeonSecrets {
+    private static final Logger LOGGER = LoggerFactory.getLogger(DungeonSecrets.class);
+    private static final String DUNGEONS_DATA_DIR = "/assets/skyblocker/dungeons";
+    private static final HashMap<String, HashMap<String, HashMap<String, long[]>>> ROOMS = new HashMap<>();
+    private static JsonObject roomsJson;
+    private static JsonObject waypointsJson;
+    @Nullable
+    private static CompletableFuture<Void> roomsLoaded;
+
+    public static boolean isRoomsLoaded() {
+        return roomsLoaded != null && roomsLoaded.isDone();
+    }
+
+    public static void init() {
+        CompletableFuture.runAsync(() -> {
+            List<CompletableFuture<Void>> dungeonFutures = Collections.synchronizedList(new ArrayList<>());
+            try {
+                //noinspection DataFlowIssue
+                File dungeons = new File(SkyblockerMod.class.getResource(DUNGEONS_DATA_DIR).getFile());
+                int resourcePathIndex = dungeons.getPath().indexOf(DUNGEONS_DATA_DIR);
+                //noinspection DataFlowIssue
+                for (File dungeon : dungeons.listFiles()) {
+                    if (!dungeon.isDirectory()) {
+                        continue;
+                    }
+                    File[] roomShapes = dungeon.listFiles();
+                    if (roomShapes == null) {
+                        LOGGER.error("Failed to load dungeon secrets for dungeon {}", dungeon.getName());
+                        continue;
+                    }
+                    ROOMS.put(dungeon.getName(), new HashMap<>());
+                    List<CompletableFuture<Void>> roomShapeFutures = new ArrayList<>();
+                    for (File roomShape : roomShapes) {
+                        roomShapeFutures.add(CompletableFuture.supplyAsync(() -> readRooms(roomShape, resourcePathIndex)).thenAccept(rooms -> ROOMS.get(dungeon.getName()).put(roomShape.getName(), rooms)));
+                    }
+                    dungeonFutures.add(CompletableFuture.allOf(roomShapeFutures.toArray(CompletableFuture[]::new)).thenRun(() -> LOGGER.debug("Loaded dungeon secrets for dungeon {} with {} room shapes and {} rooms total", dungeon.getName(), ROOMS.get(dungeon.getName()).size(), ROOMS.get(dungeon.getName()).values().stream().mapToInt(HashMap::size).sum())));
+                }
+            } catch (NullPointerException e) {
+                LOGGER.error("Failed to load dungeon secrets", e);
+            }
+            try (BufferedReader roomsReader = MinecraftClient.getInstance().getResourceManager().openAsReader(new Identifier(SkyblockerMod.NAMESPACE, "dungeons/dungeonrooms.json")); BufferedReader waypointsReader = MinecraftClient.getInstance().getResourceManager().openAsReader(new Identifier(SkyblockerMod.NAMESPACE, "dungeons/secretlocations.json"))) {
+                roomsJson = SkyblockerMod.GSON.fromJson(roomsReader, JsonObject.class);
+                waypointsJson = SkyblockerMod.GSON.fromJson(waypointsReader, JsonObject.class);
+                LOGGER.debug("Loaded dungeon secrets json");
+            } catch (IOException e) {
+                LOGGER.error("Failed to load dungeon secrets json", e);
+            }
+            roomsLoaded = CompletableFuture.allOf(dungeonFutures.toArray(CompletableFuture[]::new)).thenRun(() -> LOGGER.info("Loaded dungeon secrets for {} dungeon(s), {} room shapes, and {} rooms total", ROOMS.size(), ROOMS.values().stream().mapToInt(HashMap::size).sum(), ROOMS.values().stream().map(HashMap::values).flatMap(Collection::stream).mapToInt(HashMap::size).sum()));
+        });
+    }
+
+    private static HashMap<String, long[]> readRooms(File roomShape, int resourcePathIndex) {
+        HashMap<String, long[]> data = new HashMap<>();
+        File[] rooms = roomShape.listFiles();
+        if (rooms == null) {
+            LOGGER.error("Failed to load dungeon secrets room shape {}", roomShape.getName());
+            return data;
+        }
+        for (File room : rooms) {
+            String name = room.getName();
+            //noinspection DataFlowIssue
+            try (ObjectInputStream in = new ObjectInputStream(new InflaterInputStream(SkyblockerMod.class.getResourceAsStream(room.getPath().substring(resourcePathIndex))))) {
+                data.put(name.substring(0, name.length() - 9), (long[]) in.readObject());
+                LOGGER.debug("Loaded dungeon secrets room {}", name);
+            } catch (NullPointerException | IOException | ClassNotFoundException e) {
+                LOGGER.error("Failed to load dungeon secrets room " + name, e);
+            }
+        }
+        LOGGER.debug("Loaded dungeon secrets room shape {} with {} rooms", roomShape.getName(), data.size());
+        return data;
+    }
+}