Initial work on script resources

Author: lukebemish

Common/src/main/groovy/org/groovymc/cgl/api/resources/ScriptResourceProvider.groovy

@@ -0,0 +1,33 @@
+package org.groovymc.cgl.api.resources
+
+import com.mojang.datafixers.util.Pair
+import groovy.transform.CompileStatic
+import groovy.transform.TupleConstructor
+import groovy.transform.VisibilityOptions
+import groovy.transform.options.Visibility
+import groovy.transform.stc.ClosureParams
+import groovy.transform.stc.FromAbstractTypeMethods
+import net.minecraft.resources.ResourceLocation
+import org.groovymc.cgl.impl.resources.ScriptResourceListener
+
+@CompileStatic
+@TupleConstructor(visibilityId = 'private')
+@VisibilityOptions(id = 'private', value = Visibility.PRIVATE)
+final class ScriptResourceProvider<T> {
+    final ResourceLocation prefix
+    final Class<? super T> clazz
+
+    Closure getResource(ResourceLocation id) {
+        return ScriptResourceListener.SCRIPTS.getOrDefault(prefix, new LinkedHashMap<ResourceLocation, Closure>()).get(id)
+    }
+
+    void getResources(@ClosureParams(value = FromAbstractTypeMethods, options = ['org.groovymc.cgl.impl.resources.ResourceConsumer']) Closure resourceConsumer) {
+        ScriptResourceListener.SCRIPTS.getOrDefault(prefix, new LinkedHashMap<ResourceLocation, Closure>()).each { id, closure ->
+            if (resourceConsumer.getMaximumNumberOfParameters() == 2) {
+                resourceConsumer.call(id, closure)
+            } else {
+                resourceConsumer.call(new Pair<>(id, closure))
+            }
+        }
+    }
+}

Common/src/main/groovy/org/groovymc/cgl/impl/resources/ResourceCompilerCustomizer.groovy

@@ -0,0 +1,7 @@
+package org.groovymc.cgl.impl.resources
+
+import org.codehaus.groovy.control.CompilerConfiguration
+
+interface ResourceCompilerCustomizer {
+    void customize(CompilerConfiguration compilerConfiguration)
+}

Common/src/main/groovy/org/groovymc/cgl/impl/resources/ResourceConsumer.groovy

@@ -0,0 +1,9 @@
+package org.groovymc.cgl.impl.resources
+
+import com.mojang.datafixers.util.Pair
+import net.minecraft.resources.ResourceLocation
+
+abstract class ResourceConsumer {
+    abstract void accept(ResourceLocation id, Closure resource)
+    abstract void accept(Pair<ResourceLocation, Closure> resource)
+}

Common/src/main/groovy/org/groovymc/cgl/impl/resources/ScriptResourceListener.groovy

@@ -0,0 +1,53 @@
+package org.groovymc.cgl.impl.resources
+
+import groovy.transform.CompileStatic
+import net.minecraft.resources.FileToIdConverter
+import net.minecraft.resources.ResourceLocation
+import net.minecraft.server.packs.resources.ResourceManager
+import net.minecraft.server.packs.resources.SimplePreparableReloadListener
+import net.minecraft.util.profiling.ProfilerFiller
+import org.codehaus.groovy.control.CompilerConfiguration
+
+@CompileStatic
+class ScriptResourceListener extends SimplePreparableReloadListener<Map<ResourceLocation, Map<ResourceLocation, Closure>>> {
+    public static final Map<ResourceLocation, Map<ResourceLocation, Closure>> SCRIPTS = [:]
+    public static final Map<ResourceLocation, Class<?>> PREFIXES = [:]
+
+    public static final ResourceCompilerCustomizer CUSTOMIZER = ServiceLoader.load(ResourceCompilerCustomizer.class, ScriptResourceListener.classLoader)
+            .findFirst().orElseThrow {
+        new NullPointerException("Failed to load compiler customizer service for platform")
+    }
+
+    public static final CompilerConfiguration COMPILER_CONFIGURATION = new CompilerConfiguration().tap {
+        CUSTOMIZER.customize(it)
+    }
+    public static final GroovyShell SHELL = new GroovyShell(ScriptResourceListener.classLoader, COMPILER_CONFIGURATION)
+
+    @Override
+    protected Map<ResourceLocation, Map<ResourceLocation, Closure>> prepare(ResourceManager resourceManager, ProfilerFiller profiler) {
+        SCRIPTS.clear()
+        PREFIXES.each { prefix, contextType ->
+            FileToIdConverter fileToIdConverter = new FileToIdConverter("$prefix.namespace/$prefix.path", ".groovy")
+
+            fileToIdConverter.listMatchingResources(resourceManager).each { fileLocation, resource ->
+                var idLocation = fileToIdConverter.fileToId(fileLocation)
+
+                try (Reader reader = resource.openAsReader()) {
+                    Closure closure = SHELL.evaluate("""
+{ context ->
+${reader.text}
+}
+""") as Closure
+                    SCRIPTS.computeIfAbsent(prefix, { [:] }).put(idLocation, closure)
+                }
+            }
+        }
+
+        return SCRIPTS
+    }
+
+    @Override
+    protected void apply(Map<ResourceLocation, Map<ResourceLocation, Closure>> object, ResourceManager resourceManager, ProfilerFiller profiler) {
+        // We've already provessed scripts
+    }
+}

Common/src/main/groovy/org/groovymc/cgl/impl/resources/package-info.java

@@ -0,0 +1,3 @@
+@ApiStatus.Internal package org.groovymc.cgl.impl.resources;
+
+import org.jetbrains.annotations.ApiStatus;

Forge/src/main/groovy/org/groovymc/cgl/impl/forge/CGLForge.groovy

@@ -7,7 +7,9 @@ package org.groovymc.cgl.impl.forge
 
 import groovy.transform.CompileStatic
 import groovy.transform.stc.POJO
+import net.minecraftforge.event.AddReloadListenerEvent
 import org.groovymc.cgl.impl.CommonGroovyLibrary
+import org.groovymc.cgl.impl.resources.ScriptResourceListener
 import org.groovymc.gml.GMod
 import org.jetbrains.annotations.ApiStatus
 
@@ -16,4 +18,9 @@ import org.jetbrains.annotations.ApiStatus
 @GMod(CommonGroovyLibrary.MOD_ID)
 @ApiStatus.Internal
 class CGLForge {
+    CGLForge() {
+        forgeBus.addListener(AddReloadListenerEvent) { event ->
+            event.addListener(new ScriptResourceListener())
+        }
+    }
 }

Forge/src/main/groovy/org/groovymc/cgl/impl/forge/resources/ResourceCompilerCustomizerImpl.groovy

@@ -0,0 +1,13 @@
+package org.groovymc.cgl.impl.forge.resources
+
+import com.google.auto.service.AutoService
+import org.codehaus.groovy.control.CompilerConfiguration
+import org.groovymc.cgl.impl.resources.ResourceCompilerCustomizer
+
+@AutoService(ResourceCompilerCustomizer.class)
+class ResourceCompilerCustomizerImpl implements ResourceCompilerCustomizer {
+    @Override
+    void customize(CompilerConfiguration compilerConfiguration) {
+        // Nothing needs done on forge
+    }
+}

Forge/src/main/groovy/org/groovymc/cgl/impl/forge/resources/package-info.java

@@ -0,0 +1,4 @@
+@ApiStatus.Internal
+package org.groovymc.cgl.impl.forge.resources;
+
+import org.jetbrains.annotations.ApiStatus;

Quilt/src/main/groovy/org/groovymc/cgl/impl/quilt/CGLQuilt.groovy

@@ -8,3 +8,5 @@ package org.groovymc.cgl.impl.quilt
 import org.groovymc.cgl.impl.CommonGroovyLibrary
 
 CommonGroovyLibrary.LOGGER.info('CGL Quilt initialised')
+
+// TODO: register listener

Quilt/src/main/groovy/org/groovymc/cgl/impl/quilt/package-info.java

@@ -0,0 +1,4 @@
+@ApiStatus.Internal
+package org.groovymc.cgl.impl.quilt;
+
+import org.jetbrains.annotations.ApiStatus;