kts gone

2026-07-15 03:13:49 +00:00 · 2026-02-22 08:59:50 -05:00
parent 130073989d
commit 589baafc28
90 changed files with 2798 additions and 3681 deletions
@@ -0,0 +1,162 @@
+import org.gradle.api.DefaultTask;
+import org.gradle.api.GradleException;
+import org.gradle.api.Plugin;
+import org.gradle.api.Project;
+import org.gradle.api.publish.PublishingExtension;
+import org.gradle.api.publish.maven.MavenPublication;
+import org.gradle.api.tasks.InputFile;
+import org.gradle.api.tasks.OutputFile;
+import org.gradle.api.tasks.TaskAction;
+import org.gradle.api.tasks.TaskProvider;
+import org.gradle.jvm.tasks.Jar;
+import org.objectweb.asm.ClassReader;
+import org.objectweb.asm.ClassVisitor;
+import org.objectweb.asm.ClassWriter;
+import org.objectweb.asm.MethodVisitor;
+import org.objectweb.asm.Opcodes;
+
+import java.io.File;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.util.jar.JarEntry;
+import java.util.jar.JarFile;
+import java.util.jar.JarOutputStream;
+
+public class ApiGenerator implements Plugin<Project> {
+    @Override
+    public void apply(Project target) {
+        target.getPlugins().apply("maven-publish");
+        TaskProvider<GenerateApiTask> task = target.getTasks().register("irisApi", GenerateApiTask.class);
+
+        PublishingExtension publishing = target.getExtensions().findByType(PublishingExtension.class);
+        if (publishing == null) {
+            throw new GradleException("Publishing extension not found");
+        }
+
+        publishing.getRepositories().maven(repository -> {
+            repository.setName("deployDir");
+            repository.setUrl(targetDirectory(target).toURI());
+        });
+
+        publishing.getPublications().create("maven", MavenPublication.class, publication -> {
+            publication.setGroupId(target.getName());
+            publication.setVersion(target.getVersion().toString());
+            publication.artifact(task);
+        });
+    }
+
+    public static File targetDirectory(Project project) {
+        String dir = System.getenv("DEPLOY_DIR");
+        if (dir == null) {
+            return project.getLayout().getBuildDirectory().dir("api").get().getAsFile();
+        }
+        return new File(dir);
+    }
+}
+
+abstract class GenerateApiTask extends DefaultTask {
+    private final File inputFile;
+    private final File outputFile;
+
+    public GenerateApiTask() {
+        setGroup("iris");
+        dependsOn("jar");
+        finalizedBy("publishMavenPublicationToDeployDirRepository");
+        doLast(task -> getLogger().lifecycle("The API is located at " + getOutputFile().getAbsolutePath()));
+
+        TaskProvider<Jar> jarTask = getProject().getTasks().named("jar", Jar.class);
+        this.inputFile = jarTask.get().getArchiveFile().get().getAsFile();
+        this.outputFile = ApiGenerator.targetDirectory(getProject()).toPath().resolve(this.inputFile.getName()).toFile();
+    }
+
+    @InputFile
+    public File getInputFile() {
+        return inputFile;
+    }
+
+    @OutputFile
+    public File getOutputFile() {
+        return outputFile;
+    }
+
+    @TaskAction
+    public void generate() throws IOException {
+        File parent = outputFile.getParentFile();
+        if (parent != null) {
+            parent.mkdirs();
+        }
+
+        try (JarFile jar = new JarFile(inputFile);
+             JarOutputStream out = new JarOutputStream(new FileOutputStream(outputFile))) {
+            jar.stream()
+                    .parallel()
+                    .filter(entry -> !entry.isDirectory())
+                    .filter(entry -> entry.getName().endsWith(".class"))
+                    .forEach(entry -> writeStrippedClass(jar, out, entry));
+        }
+    }
+
+    private static void writeStrippedClass(JarFile jar, JarOutputStream out, JarEntry entry) {
+        byte[] bytes;
+        try (InputStream input = jar.getInputStream(entry)) {
+            ClassWriter writer = new ClassWriter(ClassWriter.COMPUTE_MAXS);
+            ClassVisitor visitor = new MethodClearingVisitor(writer);
+            ClassReader reader = new ClassReader(input);
+            reader.accept(visitor, 0);
+            bytes = writer.toByteArray();
+        } catch (IOException e) {
+            throw new RuntimeException(e);
+        }
+
+        synchronized (out) {
+            try {
+                JarEntry outputEntry = new JarEntry(entry.getName());
+                out.putNextEntry(outputEntry);
+                out.write(bytes);
+                out.closeEntry();
+            } catch (IOException e) {
+                throw new RuntimeException(e);
+            }
+        }
+    }
+}
+
+class MethodClearingVisitor extends ClassVisitor {
+    public MethodClearingVisitor(ClassVisitor cv) {
+        super(Opcodes.ASM9, cv);
+    }
+
+    @Override
+    public MethodVisitor visitMethod(int access, String name, String descriptor, String signature, String[] exceptions) {
+        return new ExceptionThrowingMethodVisitor(super.visitMethod(access, name, descriptor, signature, exceptions));
+    }
+}
+
+class ExceptionThrowingMethodVisitor extends MethodVisitor {
+    public ExceptionThrowingMethodVisitor(MethodVisitor mv) {
+        super(Opcodes.ASM9, mv);
+    }
+
+    @Override
+    public void visitCode() {
+        if (mv == null) {
+            return;
+        }
+
+        mv.visitCode();
+        mv.visitTypeInsn(Opcodes.NEW, "java/lang/IllegalStateException");
+        mv.visitInsn(Opcodes.DUP);
+        mv.visitLdcInsn("Only API");
+        mv.visitMethodInsn(
+                Opcodes.INVOKESPECIAL,
+                "java/lang/IllegalStateException",
+                "<init>",
+                "(Ljava/lang/String;)V",
+                false
+        );
+        mv.visitInsn(Opcodes.ATHROW);
+        mv.visitMaxs(0, 0);
+        mv.visitEnd();
+    }
+}