From 5d06edd0d18654a5e7b7066431c986fe8c9a7aa8 Mon Sep 17 00:00:00 2001
From: onebeastchris <github@onechris.mozmail.com>
Date: Wed, 18 Oct 2023 12:06:42 +0200
Subject: [PATCH] Fallback system - download the pack to serve the client in
 case cdn fails

---
 .../geyser/pack/path/GeyserPathPackCodec.java |  2 +-
 .../geyser/pack/url/GeyserUrlPackCodec.java   | 63 ++++++-------------
 .../registry/loader/ResourcePackLoader.java   | 16 +----
 3 files changed, 23 insertions(+), 58 deletions(-)

diff --git a/core/src/main/java/org/geysermc/geyser/pack/path/GeyserPathPackCodec.java b/core/src/main/java/org/geysermc/geyser/pack/path/GeyserPathPackCodec.java
index 84067600f..13d07c800 100644
--- a/core/src/main/java/org/geysermc/geyser/pack/path/GeyserPathPackCodec.java
+++ b/core/src/main/java/org/geysermc/geyser/pack/path/GeyserPathPackCodec.java
@@ -84,7 +84,7 @@ public class GeyserPathPackCodec extends PathPackCodec {
     }
 
     @Override
-    protected @NonNull ResourcePack create() {
+    public @NonNull ResourcePack create() {
         return ResourcePackLoader.readPack(this.path);
     }
 
diff --git a/core/src/main/java/org/geysermc/geyser/pack/url/GeyserUrlPackCodec.java b/core/src/main/java/org/geysermc/geyser/pack/url/GeyserUrlPackCodec.java
index fc8665679..391f78c4e 100644
--- a/core/src/main/java/org/geysermc/geyser/pack/url/GeyserUrlPackCodec.java
+++ b/core/src/main/java/org/geysermc/geyser/pack/url/GeyserUrlPackCodec.java
@@ -25,79 +25,54 @@
 
 package org.geysermc.geyser.pack.url;
 
-import lombok.RequiredArgsConstructor;
 import org.checkerframework.checker.nullness.qual.NonNull;
 import org.geysermc.geyser.api.pack.ResourcePack;
 import org.geysermc.geyser.api.pack.UrlPackCodec;
+import org.geysermc.geyser.pack.path.GeyserPathPackCodec;
 import org.geysermc.geyser.registry.loader.ResourcePackLoader;
 
 import java.io.IOException;
-import java.io.InputStream;
-import java.net.HttpURLConnection;
-import java.net.URL;
-import java.net.URLConnection;
-import java.nio.channels.Channels;
 import java.nio.channels.SeekableByteChannel;
-import java.security.MessageDigest;
-import java.security.NoSuchAlgorithmException;
+import java.nio.file.Path;
 
-@RequiredArgsConstructor
 public class GeyserUrlPackCodec extends UrlPackCodec {
-
     private final String url;
     private final String contentKey;
+    private final GeyserPathPackCodec fallback;
 
     public GeyserUrlPackCodec(String url) {
+        this(url, "");
+    }
+
+    public GeyserUrlPackCodec(String url, String contentKey) {
         this.url = url;
-        this.contentKey = "";
+        this.contentKey = contentKey;
+        this.fallback = new GeyserPathPackCodec(getCachePath(url));
+    }
+
+    private static Path getCachePath(String url) {
+        return ResourcePackLoader.downloadPack(url);
     }
 
     @Override
     public byte @NonNull [] sha256() {
-        try {
-            URL resourcePackURL = new URL(this.url);
-            InputStream inputStream = resourcePackURL.openStream();
-            MessageDigest digest = MessageDigest.getInstance("SHA-256");
-            byte[] buffer = new byte[8192];
-            int bytesRead;
-            while ((bytesRead = inputStream.read(buffer)) != -1) {
-                digest.update(buffer, 0, bytesRead);
-            }
-            return digest.digest();
-        } catch (IOException | NoSuchAlgorithmException e) {
-            throw new RuntimeException(e);
-        }
+        return fallback.sha256();
     }
 
     @Override
     public long size() {
-        URLConnection conn = null;
-        try {
-            conn = new URL(this.url).openConnection();
-            if(conn instanceof HttpURLConnection) {
-                ((HttpURLConnection)conn).setRequestMethod("HEAD");
-            }
-            conn.getInputStream();
-            return conn.getContentLength();
-        } catch (IOException e) {
-            throw new RuntimeException(e);
-        } finally {
-            if(conn instanceof HttpURLConnection) {
-                ((HttpURLConnection)conn).disconnect();
-            }
-        }
+        return fallback.size();
     }
 
     @Override
     public @NonNull SeekableByteChannel serialize(@NonNull ResourcePack resourcePack) throws IOException {
-        URL resourcePackURL = new URL(this.url);
-        URLConnection connection = resourcePackURL.openConnection();
-        return (SeekableByteChannel) Channels.newChannel(connection.getInputStream());
+        return fallback.serialize(resourcePack);
     }
 
     @Override
-    protected @NonNull ResourcePack create() {
-        return ResourcePackLoader.downloadPack(this.url, this.contentKey);
+    @NonNull
+    public ResourcePack create() {
+        return fallback.create();
     }
 
     @Override
diff --git a/core/src/main/java/org/geysermc/geyser/registry/loader/ResourcePackLoader.java b/core/src/main/java/org/geysermc/geyser/registry/loader/ResourcePackLoader.java
index a46525632..a30292175 100644
--- a/core/src/main/java/org/geysermc/geyser/registry/loader/ResourcePackLoader.java
+++ b/core/src/main/java/org/geysermc/geyser/registry/loader/ResourcePackLoader.java
@@ -29,7 +29,6 @@ import it.unimi.dsi.fastutil.objects.Object2ObjectOpenHashMap;
 import org.geysermc.geyser.GeyserImpl;
 import org.geysermc.geyser.api.event.lifecycle.GeyserLoadResourcePacksEvent;
 import org.geysermc.geyser.api.pack.ResourcePack;
-import org.geysermc.geyser.api.pack.ResourcePackManifest;
 import org.geysermc.geyser.event.type.GeyserDefineResourcePacksEventImpl;
 import org.geysermc.geyser.pack.GeyserResourcePack;
 import org.geysermc.geyser.pack.GeyserResourcePackManifest;
@@ -190,26 +189,17 @@ public class ResourcePackLoader implements RegistryLoader<Path, Map<String, Reso
 
         for (String url: cdnPacks) {
             GeyserImpl.getInstance().getLogger().info("Loading CDN pack " + url);
-            ResourcePack pack = downloadPack(url, "");
+            ResourcePack pack = new GeyserUrlPackCodec(url).create();
             packMap.put(pack.manifest().header().uuid().toString(), pack);
         }
         return packMap;
     }
 
-    public static ResourcePack downloadPack(String url, String contentKey) throws IllegalArgumentException {
+    public static Path downloadPack(String url) throws IllegalArgumentException {
         int packHash = url.hashCode();
         Path cachedPath = GeyserImpl.getInstance().getBootstrap().getConfigFolder().resolve("cache").resolve("cdn-packs").resolve(packHash + ".zip");
         WebUtils.downloadFile(url, cachedPath.toString());
 
-        ResourcePack temp = readPack(cachedPath);
-        ResourcePackManifest manifest = temp.manifest();
-
-        try {
-            Files.delete(cachedPath);
-        } catch (IOException e) {
-            GeyserImpl.getInstance().getLogger().error("Could not delete cached pack", e);
-        }
-
-        return new GeyserResourcePack(new GeyserUrlPackCodec(url, contentKey), manifest, contentKey);
+        return cachedPath;
     }
 }