Implemented asset resolution

1 files changed, 68 insertions, 0 deletions

diff --git a/src/main/java/shcm/shsupercm/fabric/citresewn/cit/CITType.java b/src/main/java/shcm/shsupercm/fabric/citresewn/cit/CITType.java
index 7ee1e03..3618dfd 100644
--- a/src/main/java/shcm/shsupercm/fabric/citresewn/cit/CITType.java
+++ b/src/main/java/shcm/shsupercm/fabric/citresewn/cit/CITType.java
@@ -1,11 +1,14 @@
 package shcm.shsupercm.fabric.citresewn.cit;
 
 import net.minecraft.resource.ResourceManager;
+import net.minecraft.util.Identifier;
 import shcm.shsupercm.fabric.citresewn.CITResewn;
 import shcm.shsupercm.fabric.citresewn.ex.CITParsingException;
 import shcm.shsupercm.fabric.citresewn.pack.format.PropertyGroup;
 import shcm.shsupercm.fabric.citresewn.pack.format.PropertyValue;
 
+import java.util.Arrays;
+import java.util.LinkedList;
 import java.util.List;
 
 public abstract class CITType {
@@ -14,4 +17,69 @@ public abstract class CITType {
     protected void warn(String message, PropertyValue value, PropertyGroup properties) {
         CITResewn.logWarnLoading("Warning: " + CITParsingException.descriptionOf(message, properties, value.position()));
     }
+
+    /**
+     * Takes a defined path and resolves it to an identifier pointing to the resourcepack's path of the specified extension(returns null if no path can be resolved).<br>
+     * If definedPath is null, will try to resolve a relative file with the same name as the rootIdentifier with the extension, otherwise: <br>
+     * definedPath will be formatted to replace "\\" with "/" the extension will be appended if not there already. <br>
+     * It will first try using definedPath as an absolute path, if it cant resolve(or definedPath starts with ./), definedPath will be considered relative. <br>
+     * Relative paths support going to parent directories using "..".
+     */
+    public static Identifier resolveAsset(Identifier rootIdentifier, String path, String defaultedTypeDirectory, String extension, ResourceManager resourceManager) {
+        if (path == null) {
+            path = rootIdentifier.getPath().substring(0, rootIdentifier.getPath().length() - 11);
+            if (!path.endsWith(extension))
+                path = path + extension;
+            Identifier pathIdentifier = new Identifier(rootIdentifier.getNamespace(), path);
+            return resourceManager.containsResource(pathIdentifier) ? pathIdentifier : null;
+        }
+
+        Identifier pathIdentifier = new Identifier(path);
+
+        path = pathIdentifier.getPath().replace('\\', '/');
+        if (!path.endsWith(extension))
+            path = path + extension;
+
+        if (path.startsWith("./"))
+            path = path.substring(2);
+        else if (!path.contains("..")) {
+            pathIdentifier = new Identifier(pathIdentifier.getNamespace(), path);
+            if (resourceManager.containsResource(pathIdentifier))
+                return pathIdentifier;
+            else if (path.startsWith("assets/")) {
+                path = path.substring(7);
+                int sep = path.indexOf('/');
+                pathIdentifier = new Identifier(path.substring(0, sep), path.substring(sep + 1));
+                if (resourceManager.containsResource(pathIdentifier))
+                    return pathIdentifier;
+            }
+            pathIdentifier = new Identifier(pathIdentifier.getNamespace(), defaultedTypeDirectory + "/" + path);
+            if (resourceManager.containsResource(pathIdentifier))
+                return pathIdentifier;
+        }
+
+        LinkedList<String> pathParts = new LinkedList<>(Arrays.asList(rootIdentifier.getPath().split("/")));
+        pathParts.removeLast();
+
+        if (path.contains("/")) {
+            for (String part : path.split("/")) {
+                if (part.equals("..")) {
+                    if (pathParts.size() == 0)
+                        return null;
+                    pathParts.removeLast();
+                } else
+                    pathParts.addLast(part);
+            }
+        } else
+            pathParts.addLast(path);
+        path = String.join("/", pathParts);
+
+        pathIdentifier = new Identifier(rootIdentifier.getNamespace(), path);
+
+        return resourceManager.containsResource(pathIdentifier) ? pathIdentifier : null;
+    }
+
+    public static Identifier resolveAsset(Identifier rootIdentifier, PropertyValue path, String defaultedTypeDirectory, String extension, ResourceManager resourceManager) {
+        return resolveAsset(rootIdentifier, path == null ? null : path.value(), defaultedTypeDirectory, extension, resourceManager);
+    }
 }