WIP work

Author: jaskarth

plugins/unpick/.gitignore

@@ -0,0 +1,4 @@
+/build/
+/out/
+testData/classes/java*/
+bin

plugins/unpick/build.gradle

@@ -0,0 +1,13 @@
+import org.vineflower.build.TestDataRuntimesProvider
+
+archivesBaseName = 'vineflower-unpick'
+
+dependencies {
+  implementation project(":")
+  testImplementation testFixtures(project(":"))
+}
+
+task testDataClasses {
+  group = 'build'
+}
+testClasses.dependsOn(testDataClasses)

plugins/unpick/gradle.properties

@@ -0,0 +1 @@
+does_shadow=false

plugins/unpick/src/main/java/org/vineflower/unpick/UnpickParser.java

@@ -0,0 +1,87 @@
+package org.vineflower.unpick;
+
+import org.vineflower.unpick.def.UnpickTarget;
+
+import java.util.*;
+
+public class UnpickParser {
+  /*
+unpick 1
+
+const unpick_name int pkg/Destination
+  1 = VAL_1
+  2 = VAL_2
+
+const float_pis float
+  1.5707964 = Math.PI / 2
+
+param * float_pis .*
+
+param 1 unpick_name pkg/Use.method(I)V
+return unpick_name pkg/Use.method()I
+  */
+
+  public static void main(String[] args) {
+    System.out.println(parse("unpick 1\n" +
+      "\n" +
+      "const unpick_name int pkg/Destination\n" +
+      "  VAL_1 = 1\n" +
+      "  VAL_2 = 2\n" +
+      "\n" +
+      "param 1 unpick_name pkg/Use.method(I)V\n" +
+      "return unpick_name pkg/Use.method()I"));
+  }
+
+  public static List<UnpickTarget> parse(String string) {
+    System.out.println(string);
+
+    String[] split = string.split("\\n");
+    if (split.length < 1) {
+      return List.of();
+    }
+
+    String first = split[0];
+    if (!first.equals("unpick 1")) {
+      return List.of();
+    }
+
+    Map<String, List<UnpickTarget>> namedTargets = new HashMap<>();
+
+    String current = null;
+
+    for (String s : split) {
+      if (s.indexOf('#') != -1) {
+        s = s.substring(0, s.indexOf('#'));
+      }
+
+      if (s.isBlank()) {
+        continue;
+      }
+
+      boolean leading = !s.stripLeading().equals(s);
+      String[] components = s.stripLeading().split("\\s");
+      if (components[0].equals("const") || components[0].equals("bits")) {
+        parseDefinition(components);
+      }
+    }
+
+    return new ArrayList<>();
+  }
+
+  private static String parseDefinition(String[] line) {
+    String sKind = line[0];
+    if (sKind.equals("const")) {
+
+    } else if (sKind.equals("bits")) {
+
+    }
+
+    String name = line[1];
+
+    String sType = line[2];
+
+    String destination = line[3];
+
+    return null;
+  }
+}

plugins/unpick/src/main/java/org/vineflower/unpick/UnpickPass.java

@@ -0,0 +1,11 @@
+package org.vineflower.unpick;
+
+import org.jetbrains.java.decompiler.api.plugin.pass.Pass;
+import org.jetbrains.java.decompiler.api.plugin.pass.PassContext;
+
+public class UnpickPass implements Pass {
+  @Override
+  public boolean run(PassContext ctx) {
+    return false;
+  }
+}

plugins/unpick/src/main/java/org/vineflower/unpick/UnpickPlugin.java

@@ -0,0 +1,30 @@
+package org.vineflower.unpick;
+
+import org.jetbrains.annotations.Nullable;
+import org.jetbrains.java.decompiler.api.java.JavaPassLocation;
+import org.jetbrains.java.decompiler.api.java.JavaPassRegistrar;
+import org.jetbrains.java.decompiler.api.plugin.Plugin;
+import org.jetbrains.java.decompiler.api.plugin.PluginOptions;
+import org.jetbrains.java.decompiler.api.plugin.pass.NamedPass;
+
+public class UnpickPlugin implements Plugin {
+  @Override
+  public String id() {
+    return "Unpick";
+  }
+
+  @Override
+  public @Nullable PluginOptions getPluginOptions() {
+    return null;
+  }
+
+  @Override
+  public void registerJavaPasses(JavaPassRegistrar registrar) {
+    registrar.register(JavaPassLocation.AT_END, NamedPass.of("Unpick", new UnpickPass()));
+  }
+
+  @Override
+  public String description() {
+    return "Allows constant de-inlining in decompiled code";
+  }
+}

plugins/unpick/src/main/java/org/vineflower/unpick/def/UnpickDefinition.java

@@ -0,0 +1,14 @@
+package org.vineflower.unpick.def;
+
+import org.jetbrains.java.decompiler.util.Either;
+import org.jetbrains.java.decompiler.util.Unit;
+
+import java.util.List;
+
+public class UnpickDefinition {
+  private UnpickKind kind;
+  private String target;
+  private List<UnpickTarget> targets;
+  // <Param, Return>
+  private Either<Integer, Unit> location;
+}

plugins/unpick/src/main/java/org/vineflower/unpick/def/UnpickKind.java

@@ -0,0 +1,6 @@
+package org.vineflower.unpick.def;
+
+public enum UnpickKind {
+  CONST,
+  BITS
+}

plugins/unpick/src/main/java/org/vineflower/unpick/def/UnpickTarget.java

@@ -0,0 +1,21 @@
+package org.vineflower.unpick.def;
+
+import org.jetbrains.java.decompiler.struct.gen.VarType;
+
+public class UnpickTarget {
+  private final String name;
+  private final String value;
+
+  public UnpickTarget(String name, String value, VarType kind, String destination) {
+    this.name = name;
+    this.value = value;
+  }
+
+  public String getName() {
+    return name;
+  }
+
+  public String getValue() {
+    return value;
+  }
+}

plugins/unpick/src/main/resources/services/org.jetbrains.java.decompiler.api.plugin.Plugin

@@ -0,0 +1 @@
+org.vineflower.unpick.UnpickPlugin