1 files changed, 81 insertions, 0 deletions
diff --git a/src/main/java/kr/syeyoung/dungeonsguide/wsresource/StaticResourceCache.java b/src/main/java/kr/syeyoung/dungeonsguide/wsresource/StaticResourceCache.java
new file mode 100644
index 00000000..b134b7c3
--- /dev/null
+++ b/src/main/java/kr/syeyoung/dungeonsguide/wsresource/StaticResourceCache.java
@@ -0,0 +1,81 @@
+/*
+ * Dungeons Guide - The most intelligent Hypixel Skyblock Dungeons Mod
+ * Copyright (C) 2021  cyoung06
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published
+ * by the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <https://www.gnu.org/licenses/>.
+ */
+
+package kr.syeyoung.dungeonsguide.wsresource;
+
+import kr.syeyoung.dungeonsguide.DungeonsGuide;
+import kr.syeyoung.dungeonsguide.events.StompConnectedEvent;
+import kr.syeyoung.dungeonsguide.stomp.StompInterface;
+import kr.syeyoung.dungeonsguide.stomp.StompMessageHandler;
+import kr.syeyoung.dungeonsguide.stomp.StompPayload;
+import kr.syeyoung.dungeonsguide.stomp.StompSubscription;
+import net.minecraftforge.fml.common.eventhandler.SubscribeEvent;
+import org.json.JSONObject;
+
+import java.util.HashMap;
+import java.util.Map;
+import java.util.UUID;
+import java.util.concurrent.CompletableFuture;
+
+public class StaticResourceCache implements StompMessageHandler {
+    public static final StaticResourceCache INSTANCE = new StaticResourceCache();
+
+    private Map<UUID, StaticResource> staticResourceMap = new HashMap<>();
+
+    private Map<UUID, CompletableFuture<StaticResource>> staticResourceRequest = new HashMap<>();
+
+    public void purgeCache() {
+        staticResourceRequest.clear();
+        staticResourceMap.clear();
+    }
+
+    public CompletableFuture<StaticResource> getResource(UUID resourceID) {
+        if (staticResourceMap.containsKey(resourceID)) return CompletableFuture.completedFuture(staticResourceMap.get(resourceID));
+        if (staticResourceRequest.containsKey(resourceID)) return staticResourceRequest.get(resourceID);
+
+        DungeonsGuide.getDungeonsGuide().getStompConnection().send(new StompPayload().header("destination", "/app/staticresource.get").payload(resourceID.toString()));
+        CompletableFuture<StaticResource> comp = new CompletableFuture<>();
+        staticResourceRequest.put(resourceID, comp);
+        return comp;
+    }
+
+    @Override
+    public void handle(StompInterface stompInterface, StompPayload stompPayload) {
+        JSONObject object = new JSONObject(stompPayload.payload());
+        StaticResource staticResource = new StaticResource();
+        staticResource.setResourceID(UUID.fromString(object.getString("resourceID")));
+        staticResource.setExists(object.getBoolean("exists"));
+        staticResource.setValue((!object.has("value") || object.isNull("value")) ? null : object.getString("value"));
+
+        staticResourceMap.put(staticResource.getResourceID(), staticResource);
+        CompletableFuture<StaticResource> completed = staticResourceRequest.remove(staticResource.getResourceID());
+        if (completed != null) completed.complete(staticResource);
+    }
+
+    @SubscribeEvent
+    public void stompConnect(StompConnectedEvent stompConnectedEvent) {
+        stompConnectedEvent.getStompInterface().subscribe(StompSubscription.builder()
+                .stompMessageHandler(this).ackMode(StompSubscription.AckMode.AUTO).destination("/user/queue/staticresource.get").build());
+
+        getResource(BONUS_SCORE);
+        getResource(TRIVIA_ANSWERS);
+    }
+
+    public static final UUID BONUS_SCORE = UUID.fromString("13f10001-66b5-46e5-94f9-1a5161a23429");
+    public static final UUID TRIVIA_ANSWERS = UUID.fromString("5657f2cc-2bb8-4fcd-b55c-ffc0a35b9349");
+}