[Poc] Generate IR definitions - superclass approach

build.sbt

@@ -353,6 +353,9 @@ lazy val enso = (project in file("."))
     `runtime-compiler`,
     `runtime-integration-tests`,
     `runtime-parser`,
+    `runtime-parser-dsl`,
+    `runtime-parser-processor`,
+    `runtime-parser-processor-tests`,
     `runtime-language-arrow`,
     `runtime-language-epb`,
     `runtime-instrument-common`,
@@ -3218,14 +3221,73 @@ lazy val `runtime-parser` =
       Compile / moduleDependencies ++= Seq(
         "org.netbeans.api" % "org-openide-util-lookup" % netbeansApiVersion
       ),
+      // Java compiler is not able to correctly find all the annotation processor, because
+      // one of the is on module-path. To overcome this, we explicitly list all of them here.
+      Compile / javacOptions ++= {
+        val processorClasses = Seq(
+          "org.enso.runtime.parser.processor.IRProcessor",
+          "org.enso.persist.impl.PersistableProcessor",
+          "org.netbeans.modules.openide.util.ServiceProviderProcessor",
+          "org.netbeans.modules.openide.util.NamedServiceProcessor"
+        ).mkString(",")
+        Seq(
+          "-processor",
+          processorClasses
+        )
+      },
       Compile / internalModuleDependencies := Seq(
         (`syntax-rust-definition` / Compile / exportedModule).value,
-        (`persistance` / Compile / exportedModule).value
+        (`persistance` / Compile / exportedModule).value,
+        (`runtime-parser-dsl` / Compile / exportedModule).value,
+        (`runtime-parser-processor` / Compile / exportedModule).value
       )
     )
     .dependsOn(`syntax-rust-definition`)
     .dependsOn(`persistance`)
     .dependsOn(`persistance-dsl` % "provided")
+    .dependsOn(`runtime-parser-dsl`)
+    .dependsOn(`runtime-parser-processor`)
+
+lazy val `runtime-parser-dsl` =
+  (project in file("engine/runtime-parser-dsl"))
+    .enablePlugins(JPMSPlugin)
+    .settings(
+      frgaalJavaCompilerSetting
+    )
+
+lazy val `runtime-parser-processor-tests` =
+  (project in file("engine/runtime-parser-processor-tests"))
+    .settings(
+      inConfig(Compile)(truffleRunOptionsSettings),
+      frgaalJavaCompilerSetting,
+      commands += WithDebugCommand.withDebug,
+      annotationProcSetting,
+      Compile / javacOptions ++= Seq(
+        "-processor",
+        "org.enso.runtime.parser.processor.IRProcessor"
+      ),
+      Test / fork := true,
+      libraryDependencies ++= Seq(
+        "junit"                      % "junit"           % junitVersion     % Test,
+        "com.github.sbt"             % "junit-interface" % junitIfVersion   % Test,
+        "org.hamcrest"               % "hamcrest-all"    % hamcrestVersion  % Test,
+        "com.google.testing.compile" % "compile-testing" % "0.21.0"         % Test,
+        "org.scalatest"             %% "scalatest"       % scalatestVersion % Test
+      )
+    )
+    .dependsOn(`runtime-parser-processor`)
+    .dependsOn(`runtime-parser`)
+
+lazy val `runtime-parser-processor` =
+  (project in file("engine/runtime-parser-processor"))
+    .enablePlugins(JPMSPlugin)
+    .settings(
+      frgaalJavaCompilerSetting,
+      Compile / internalModuleDependencies := Seq(
+        (`runtime-parser-dsl` / Compile / exportedModule).value
+      )
+    )
+    .dependsOn(`runtime-parser-dsl`)
 
 lazy val `runtime-compiler` =
   (project in file("engine/runtime-compiler"))

engine/runtime-compiler/src/main/scala/org/enso/compiler/pass/optimise/LambdaConsolidate.scala

@@ -227,7 +227,7 @@ case object LambdaConsolidate extends IRPass {
             }
 
             val shadower: IR =
-              mShadower.getOrElse(Empty(spec.identifiedLocation))
+              mShadower.getOrElse(new Empty(spec.identifiedLocation))
 
             spec.getDiagnostics.add(
               warnings.Shadowed

engine/runtime-compiler/src/main/scala/org/enso/compiler/pass/resolve/SuspendedArguments.scala

@@ -306,7 +306,7 @@ case object SuspendedArguments extends IRPass {
       } else if (args.length > signatureSegments.length) {
         val additionalSegments = signatureSegments ::: List.fill(
           args.length - signatureSegments.length
-        )(Empty(identifiedLocation = null))
+        )(new Empty(null))
 
         args.zip(additionalSegments)
       } else {

engine/runtime-integration-tests/src/test/scala/org/enso/compiler/test/core/ir/DiagnosticStorageTest.scala

@@ -16,7 +16,7 @@ class DiagnosticStorageTest extends CompilerTest {
   def mkDiagnostic(name: String): Diagnostic = {
     warnings.Shadowed.FunctionParam(
       name,
-      Empty(identifiedLocation = null),
+      new Empty(null),
       identifiedLocation = null
     )
   }

engine/runtime-integration-tests/src/test/scala/org/enso/compiler/test/pass/desugar/OperatorToFunctionTest.scala

@@ -84,9 +84,9 @@ class OperatorToFunctionTest extends MiniPassTest {
   // === The Tests ============================================================
   val opName =
     Name.Literal("=:=", isMethod = true, null)
-  val left     = Empty(null)
-  val right    = Empty(null)
-  val rightArg = new CallArgument.Specified(None, Empty(null), false, null)
+  val left     = new Empty(null)
+  val right    = new Empty(null)
+  val rightArg = new CallArgument.Specified(None, new Empty(null), false, null)
 
   val (operator, operatorFn) = genOprAndFn(opName, left, right)
 
@@ -96,11 +96,11 @@ class OperatorToFunctionTest extends MiniPassTest {
   "Operators" should {
     val opName =
       Name.Literal("=:=", isMethod = true, identifiedLocation = null)
-    val left  = Empty(identifiedLocation = null)
-    val right = Empty(identifiedLocation = null)
+    val left  = new Empty(null)
+    val right = new Empty(null)
     val rightArg = new CallArgument.Specified(
       None,
-      Empty(identifiedLocation = null),
+      new Empty(null),
       false,
       identifiedLocation = null
     )

engine/runtime-parser-dsl/src/main/java/module-info.java

@@ -0,0 +1,3 @@
+module org.enso.runtime.parser.dsl {
+  exports org.enso.runtime.parser.dsl;
+}

engine/runtime-parser-dsl/src/main/java/org/enso/runtime/parser/dsl/GenerateFields.java

@@ -0,0 +1,51 @@
+package org.enso.runtime.parser.dsl;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+/**
+ * Parameters of the constructor annotated with this annotation will be scanned by the IR processor
+ * and <b>fields</b>will be generated for them. There can be only a single constructor with this
+ * annotation in a class. The enclosing class must be annotated with {@link GenerateIR}.
+ *
+ * <h2>Fields</h2>
+ *
+ * The generated class will contain 4 <b>meta</b> fields that are required to be present inside
+ * every IR element:
+ *
+ * <ul>
+ *   <li>{@code private DiagnosticStorage diagnostics}
+ *   <li>{@code private MetadataStorage passData}
+ *   <li>{@code private IdentifiedLocation location}
+ *   <li>{@code private UUID id}
+ * </ul>
+ *
+ * Apart from these <b>meta</b> fields, the generated class will also contain <b>user-defined</b>
+ * fields. User-defined fields are inferred from all the parameters of the constructor annotated
+ * with {@link GenerateFields}. The parameter of the constructor can be one of the following:
+ *
+ * <ul>
+ *   <li>Any reference, or primitive type annotated with {@link IRField}
+ *   <li>A subtype of {@code org.enso.compiler.ir.IR} annotated with {@link IRChild}
+ *   <li>One of the <emph>meta</emph> types mentioned above
+ * </ul>
+ *
+ * <p>A user-defined field generated out of constructor parameter annotated with {@link IRChild} is
+ * a child element of this IR element. That means that it will be included in generated
+ * implementation of IR methods that iterate over the IR tree. For example {@code mapExpressions} or
+ * {@code children}.
+ *
+ * <p>A user-defined field generated out of constructor parameter annotated with {@link IRField}
+ * will be a field with generated getters. Such field however, will not be part of the IR tree
+ * traversal methods.
+ *
+ * <p>For a constructor parameter of a meta type, there will be no user-defined field generated, as
+ * the meta fields are always generated.
+ *
+ * <p>Other types of constructor parameters are forbidden.
+ */
+@Retention(RetentionPolicy.SOURCE)
+@Target(ElementType.CONSTRUCTOR)
+public @interface GenerateFields {}

engine/runtime-parser-dsl/src/main/java/org/enso/runtime/parser/dsl/GenerateIR.java

@@ -0,0 +1,31 @@
+package org.enso.runtime.parser.dsl;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+/**
+ * A class annotated with this annotation will be processed by the IR processor. The processor will
+ * generate a super class that is meant to be extended by this class. The generated class will have
+ * the same package as this class, and its name will have the "Gen" suffix. Majority of the methods
+ * in the generated class will be either private or package-private, so that they are not accessible
+ * from the outside.
+ *
+ * <p>The class can be enclosed (nested inside) an interface.
+ *
+ * <p>The class must contain a single constructor annotated with {@link GenerateFields}.
+ */
+@Retention(RetentionPolicy.SOURCE)
+@Target(ElementType.TYPE)
+public @interface GenerateIR {
+
+  /**
+   * Name of the interfaces that the generated superclass must implement. All the interfaces must be
+   * subtypes of the {@code org.enso.compiler.ir.IR} interface. All the abstract parameterless
+   * methods from all the interfaces will be implemented by the generated class. Must not be empty.
+   *
+   * @return
+   */
+  String interfaces() default "org.enso.compiler.core.IR";
+}

engine/runtime-parser-dsl/src/main/java/org/enso/runtime/parser/dsl/IRChild.java

@@ -0,0 +1,19 @@
+package org.enso.runtime.parser.dsl;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+/**
+ * Constructor parameter annotated with this annotation will be represented as a child field in the
+ * generated super class. Children of IR elements for a tree. A child will be part of the methods
+ * traversing the tree, like {@code mapExpression} and {@code children}. The parameter type must be
+ * a subtype of {@code org.enso.compiler.ir.IR}.
+ */
+@Retention(RetentionPolicy.SOURCE)
+@Target(ElementType.PARAMETER)
+public @interface IRChild {
+  /** If true, the child will always be non-null. Otherwise, it can be null. */
+  boolean required() default true;
+}

engine/runtime-parser-dsl/src/main/java/org/enso/runtime/parser/dsl/IRField.java

@@ -0,0 +1,19 @@
+package org.enso.runtime.parser.dsl;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+/**
+ * A constructor parameter annotated with this annotation will have a corresponding user-defined
+ * field generated in the super class (See {@link GenerateFields} for docs about fields).
+ *
+ * <p>There is no restriction on the type of the parameter.
+ */
+@Retention(RetentionPolicy.SOURCE)
+@Target(ElementType.PARAMETER)
+public @interface IRField {
+  /** If true, the field will always be non-null. Otherwise, it can be null. */
+  boolean required() default true;
+}

engine/runtime-parser-processor-tests/src/main/java/org/enso/runtime/parser/processor/test/gen/ir/core/JCallArgument.java

@@ -0,0 +1,42 @@
+package org.enso.runtime.parser.processor.test.gen.ir.core;
+
+import java.util.function.Function;
+import org.enso.compiler.core.IR;
+import org.enso.compiler.core.ir.Expression;
+import org.enso.compiler.core.ir.Name;
+import org.enso.runtime.parser.dsl.GenerateFields;
+import org.enso.runtime.parser.dsl.GenerateIR;
+import org.enso.runtime.parser.dsl.IRChild;
+import org.enso.runtime.parser.dsl.IRField;
+import scala.Option;
+
+/** Call-site arguments in Enso. */
+public interface JCallArgument extends IR {
+  /** The name of the argument, if present. */
+  Option<Name> name();
+
+  /** The expression of the argument, if present. */
+  Expression value();
+
+  /** Flag indicating that the argument was generated by compiler. */
+  boolean isSynthetic();
+
+  @Override
+  JCallArgument mapExpressions(Function<Expression, Expression> fn);
+
+  @Override
+  JCallArgument duplicate(
+      boolean keepLocations,
+      boolean keepMetadata,
+      boolean keepDiagnostics,
+      boolean keepIdentifiers);
+
+  @GenerateIR(interfaces = "JCallArgument")
+  final class JSpecified extends JSpecifiedGen {
+    @GenerateFields
+    public JSpecified(
+        @IRField boolean isSynthetic, @IRChild Option<Name> name, @IRChild Expression value) {
+      super(isSynthetic, name, value);
+    }
+  }
+}

engine/runtime-parser-processor-tests/src/main/java/org/enso/runtime/parser/processor/test/gen/ir/core/JExpression.java

@@ -0,0 +1,42 @@
+package org.enso.runtime.parser.processor.test.gen.ir.core;
+
+import java.util.function.Function;
+import org.enso.compiler.core.IR;
+import org.enso.compiler.core.ir.Expression;
+import org.enso.compiler.core.ir.Name;
+import org.enso.runtime.parser.dsl.GenerateFields;
+import org.enso.runtime.parser.dsl.GenerateIR;
+import org.enso.runtime.parser.dsl.IRChild;
+import org.enso.runtime.parser.dsl.IRField;
+import scala.collection.immutable.List;
+
+public interface JExpression extends IR {
+  @Override
+  JExpression mapExpressions(Function<Expression, Expression> fn);
+
+  @Override
+  JExpression duplicate(
+      boolean keepLocations,
+      boolean keepMetadata,
+      boolean keepDiagnostics,
+      boolean keepIdentifiers);
+
+  @GenerateIR(interfaces = "JExpression")
+  final class JBlock extends JBlockGen {
+    @GenerateFields
+    public JBlock(
+        @IRChild List<JExpression> expressions,
+        @IRChild JExpression returnValue,
+        @IRField boolean suspended) {
+      super(expressions, returnValue, suspended);
+    }
+  }
+
+  @GenerateIR(interfaces = "JExpression")
+  final class JBinding extends JBindingGen {
+    @GenerateFields
+    public JBinding(@IRChild Name name, @IRChild JExpression expression) {
+      super(name, expression);
+    }
+  }
+}

engine/runtime-parser-processor-tests/src/main/java/org/enso/runtime/parser/processor/test/gen/ir/core/package-info.java

@@ -0,0 +1,9 @@
+/**
+ * Contains hierarchy of interfaces that should correspond to the previous {@link
+ * org.enso.compiler.core.IR} element hierarchy. All the classes inside this package have {@code J}
+ * prefix. So for example {@code JCallArgument} correspond to {@code CallArgument}.
+ *
+ * <p>The motivation to put these classes here is to test the generation of {@link
+ * org.enso.runtime.parser.processor.IRProcessor}.
+ */
+package org.enso.runtime.parser.processor.test.gen.ir.core;

engine/runtime-parser-processor-tests/src/main/java/org/enso/runtime/parser/processor/test/gen/ir/package-info.java

@@ -0,0 +1,5 @@
+/**
+ * Contains interfaces with parser-dsl annotations. There will be generated classes for these
+ * interfaces and they are tested. All these interfaces are only for testing.
+ */
+package org.enso.runtime.parser.processor.test.gen.ir;