Fix remapping of records (#1)

Use latest JLS for parser and add missing overrides to RemapperVisitor
architectury · May 26, 2022 · b980f7c · b980f7c
1 parent e869f1a
commit b980f7c
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/src/main/java/org/cadixdev/mercury/Mercury.java b/src/main/java/org/cadixdev/mercury/Mercury.java
@@ -190,7 +190,7 @@ private void cleanup() {
     }
 
     private void run() throws Exception {
-        ASTParser parser = ASTParser.newParser(AST.JLS10);
+        ASTParser parser = ASTParser.newParser(AST.getJLSLatest());
 
         // Set Java version
         Map<String, String> options = JavaCore.getOptions();

diff --git a/src/main/java/org/cadixdev/mercury/remapper/RemapperVisitor.java b/src/main/java/org/cadixdev/mercury/remapper/RemapperVisitor.java
@@ -39,6 +39,7 @@
 import org.eclipse.jdt.core.dom.PackageDeclaration;
 import org.eclipse.jdt.core.dom.QualifiedName;
 import org.eclipse.jdt.core.dom.QualifiedType;
+import org.eclipse.jdt.core.dom.RecordDeclaration;
 import org.eclipse.jdt.core.dom.SimpleName;
 import org.eclipse.jdt.core.dom.SimpleType;
 import org.eclipse.jdt.core.dom.TagElement;
@@ -428,6 +429,12 @@ public boolean visit(EnumDeclaration node) {
         return true;
     }
 
+    @Override
+    public boolean visit(RecordDeclaration node) {
+        pushImportContext(node.resolveBinding());
+        return true;
+    }
+
     @Override
     public boolean visit(TypeDeclaration node) {
         pushImportContext(node.resolveBinding());
@@ -449,6 +456,11 @@ public void endVisit(EnumDeclaration node) {
         this.importStack.pop();
     }
 
+    @Override
+    public void endVisit(RecordDeclaration node) {
+        this.importStack.pop();
+    }
+
     @Override
     public void endVisit(TypeDeclaration node) {
         this.importStack.pop();