jit: use javac API

Author: ice1000

Diff for: jit-compiler/src/test/java/CompileTest.java

@@ -1,7 +1,6 @@
 // Copyright (c) 2020-2024 Tesla (Yinsen) Zhang.
 // Use of this source code is governed by the MIT license that can be found in the LICENSE.md file.
 
-import com.javax0.sourcebuddy.Compiler;
 import kala.collection.immutable.ImmutableSeq;
 import org.aya.compiler.FileSerializer;
 import org.aya.compiler.ModuleSerializer;
@@ -28,6 +27,7 @@
 import org.jetbrains.annotations.NotNull;
 import org.junit.jupiter.api.Test;
 
+import java.io.IOException;
 import java.nio.file.Path;
 
 public class CompileTest {
@@ -37,19 +37,18 @@ public class CompileTest {
       open data Vec (n : Nat) Type
       | O, A   => vnil
       | S n, A => vcons A (Vec n A)
-
+      
       def plus (a b : Nat) : Nat elim a
       | O => b
       | S n => S (plus n b)
       """); // .filter(x -> x instanceof FnDef || x instanceof DataDef);
 
     var code = serializeFrom(result);
 
-    System.out.println(code);
-
     try {
-      var clazz = Compiler.java().from("AYA.baka", code).compile().load().get();
-      var loader = clazz.getClassLoader();
+      var tester = new CompileTester(code);
+      tester.compile();
+      var loader = tester.cl;
 
       var fieldO = loader.loadClass("AYA.baka$Nat$O").getField("INSTANCE");
       var fieldS = loader.loadClass("AYA.baka$Nat$S").getField("INSTANCE");
@@ -67,7 +66,7 @@ def plus (a b : Nat) : Nat elim a
 
       var mResult = plus.invoke(zero, ImmutableSeq.of(two, three));
       System.out.println(mResult.debuggerOnlyToString());
-    } catch (ClassNotFoundException | IllegalAccessException | NoSuchFieldException | Compiler.CompileException e) {
+    } catch (ClassNotFoundException | IllegalAccessException | NoSuchFieldException | IOException e) {
       throw new RuntimeException(e);
     }
 

Diff for: jit-compiler/src/test/java/CompileTester.java

@@ -1,44 +1,54 @@
 // Copyright (c) 2020-2024 Tesla (Yinsen) Zhang.
 // Use of this source code is governed by the MIT license that can be found in the LICENSE.md file.
 
-import com.javax0.sourcebuddy.Compiler;
-import com.javax0.sourcebuddy.Fluent;
 import kala.collection.immutable.ImmutableSeq;
 import org.aya.compiler.AyaSerializer;
-import org.aya.compiler.ModuleSerializer;
-import org.aya.generic.NameGenerator;
-import org.aya.primitive.ShapeFactory;
 import org.aya.resolve.module.DumbModuleLoader;
 import org.aya.syntax.compile.JitDef;
-import org.aya.syntax.core.def.DataDef;
-import org.aya.syntax.core.def.FnDef;
-import org.aya.syntax.core.def.TyckDef;
-import org.intellij.lang.annotations.Language;
 import org.jetbrains.annotations.NotNull;
 
+import javax.tools.ToolProvider;
+import java.io.IOException;
 import java.lang.reflect.Field;
+import java.net.URL;
+import java.net.URLClassLoader;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
 
 public class CompileTester {
-  public final @Language("Java") @NotNull String code;
-  private final Fluent.AddSource compiler = Compiler.java();
-  private Class<?> output = null;
+  public final @NotNull String code;
+  private final Path baka;
+  public final ClassLoader cl;
 
-  public CompileTester(@NotNull String code) { this.code = code; }
+  public CompileTester(@NotNull String code) throws IOException {
+    this.code = code;
+
+    var genDir = Paths.get("src/test/gen");
+    Files.createDirectories(genDir);
+    Files.writeString(baka = genDir.resolve("baka.java"), code);
+    cl = new URLClassLoader(new URL[]{baka.toUri().toURL()});
+  }
 
   public void compile() {
     try {
-      output = compiler.from(STR."\{AyaSerializer.PACKAGE_BASE}.\{DumbModuleLoader.DUMB_MODULE_NAME}", code)
-        .compile().load().get();
-    } catch (ClassNotFoundException | Compiler.CompileException e) {
+      var compiler = ToolProvider.getSystemJavaCompiler();
+      var fileManager = compiler.getStandardFileManager(null, null, null);
+      var compilationUnits = fileManager.getJavaFileObjects(baka);
+      var task = compiler.getTask(null, fileManager, null, null, null, compilationUnits);
+      task.call();
+      var fqName = STR."\{AyaSerializer.PACKAGE_BASE}.\{DumbModuleLoader.DUMB_MODULE_NAME}";
+      cl.loadClass(fqName);
+    } catch (ClassNotFoundException e) {
       throw new RuntimeException(e);
     }
   }
 
   @SuppressWarnings("unchecked")
   public <T> @NotNull Class<T> load(String... qualified) {
     try {
-      return (Class<T>) output.getClassLoader()
-        .loadClass(STR."\{AyaSerializer.PACKAGE_BASE}.\{ImmutableSeq.from(qualified).joinToString("$")}");
+      return (Class<T>) cl.loadClass(
+        STR."\{AyaSerializer.PACKAGE_BASE}.\{ImmutableSeq.from(qualified).joinToString("$")}");
     } catch (ClassNotFoundException e) {
       throw new RuntimeException(e);
     }

Diff for: jit-compiler/src/test/java/RedBlackTreeTest.java

@@ -15,8 +15,6 @@
 import org.junit.jupiter.api.Test;
 
 import java.io.IOException;
-import java.nio.file.Files;
-import java.nio.file.Paths;
 import java.util.Random;
 import java.util.function.Function;
 import java.util.function.IntFunction;
@@ -33,9 +31,6 @@ public class RedBlackTreeTest {
     var result = CompileTest.tyck(code);
 
     var tester = new CompileTester(CompileTest.serializeFrom(result));
-    var genDir = Paths.get("src/test/gen");
-    Files.createDirectories(genDir);
-    Files.writeString(genDir.resolve("baka.java"), tester.code);
     tester.compile();
 
     JitData List = tester.loadInstance("baka", "List");