fix registry

Author: Ecdcaeb

src/main/java/mods/Hileb/shotaasm/api/TransformerRegistry.java

@@ -0,0 +1,89 @@
+package mods.Hileb.shotaasm.api;
+
+
+import top.outlands.foundation.TransformerDelegate;
+
+import net.minecraft.launchwrapper.IClassNameTransformer;
+import net.minecraft.launchwrapper.IClassTransformer;
+import top.outlands.foundation.boot.TransformerHolder;
+
+import java.util.*;
+
+import static net.minecraft.launchwrapper.Launch.classLoader;
+import static top.outlands.foundation.boot.Foundation.LOGGER;
+import static top.outlands.foundation.boot.TransformerHolder.*;
+
+import org.objectweb.asm.ClassReader;
+import org.objectweb.asm.ClassWriter;
+import org.objectweb.asm.tree.ClassNode;
+import top.outlands.foundation.IExplicitTransformer;
+
+public class TransformerRegistry {
+    
+    public static void registerRenameTransformer(IClassNameTransformer transformer) {
+        TransformerDelegate.registerRenameTransformer(transformer);
+    }
+
+    public static void registerExplicitTransformer(final int priorty, final IExplicitTransformer transformer, String... targets) {
+        TransformerDelegate.registerExplicitTransformer(new IExplicitTransformer() {
+            
+            @Override
+            public byte[] transform(byte[] basicClass) {
+                return transformer.transform(basicClass);
+            }
+
+            @Override
+            public int getPriority() {
+                return priorty;
+            }
+        }, targets);
+    }
+
+    public static void registerASMExplicitTransformer(final Consumer<ClassNode> transformer, String... targets) {
+        TransformerDelegate.registerExplicitTransformer(new IExplicitTransformer() {
+            
+            @Override
+            public byte[] transform(byte[] basicClass) {
+                ClassReader classReader = new ClassReader(bytes);
+                ClassNode classNode = new ClassNode();
+                classReader.accept(classNode, 0);
+                transformer.apply(classNode);
+                ClassWriter classWriter = new ClassWriter(classNode);
+                classNode.accept(classWriter);
+                return classWriter.toByteArray();
+            }
+
+            @Override
+            public int getPriority() {
+                return priorty;
+            }
+        }, targets);
+    }
+
+    public static void registerASMExplicitTransformer(final int priorty, final Consumer<ClassNode> transformer, String... targets) {
+        TransformerDelegate.registerExplicitTransformer(new IExplicitTransformer() {
+            @Override
+            public byte[] transform(byte[] basicClass) {
+                ClassReader classReader = new ClassReader(bytes);
+                ClassNode classNode = new ClassNode();
+                classReader.accept(classNode, 0);
+                transformer.apply(classNode);
+                ClassWriter classWriter = new ClassWriter(classNode);
+                classNode.accept(classWriter);
+                return classWriter.toByteArray();
+            }
+        }, targets);
+    }
+
+    public static void registerExplicitTransformer(IExplicitTransformer transformer, String... targets) {
+        TransformerDelegate.registerExplicitTransformer(transformer, targets);
+    }
+
+    public static void registerClassTransformer(IClassTransformer transformer) {
+        TransformerDelegate.registerClassTransformer(transformer);
+    }
+
+    public static void unregisterClassTransformer(IClassTransformer transformer) {
+        TransformerDelegate.unRegisterTransformer(transformer);
+    }
+}

src/main/java/mods/Hileb/shotaasm/impl/ShotaCompiler.java

@@ -22,7 +22,7 @@ public Runnable compile(ScriptFile file) {
             builder.append("import ").append(s).append(";\n");
         }
         builder
-                .append("import top.outlands.foundation.TransformerDelegate;")
+                .append("import mods.Hileb.shotaasm.api.TransformerRegistry;")
                 .append("import org.objectweb.asm.*;")
                 .append("import org.objectweb.asm.tree.*;")
                 .append("import org.objectweb.asm.util.*;")
@@ -34,8 +34,16 @@ public Runnable compile(ScriptFile file) {
                 .append(file.text())
                 .append("}}");
         try {
-            return (Runnable) Launch.classLoader.defineClass(name, Compiler.compileSingle(name, builder.toString()))
-                    .getConstructor().newInstance();
+            Compiler compiler = new Compiler();
+            compiler.addSource(name, builder.toString());
+            CompileError error = compiler.compile();
+            if (error != null) throw new RuntimeException(error);
+            else {
+                for (Map.Entry<String, byte[]> entry : compiler.getClasses().entrySet()) {
+                    Launch.classLoader.defineClass(entry.getKey(), entry.getValue());
+                }
+                return (Runnable) Class.forName(name, true, Launch.classLoader).getConstructor().newInstance();
+            }
         } catch (InstantiationException | IllegalAccessException | InvocationTargetException | NoSuchMethodException e) {
             throw new RuntimeException(e);
         }