4 files changed, 10 insertions, 9 deletions

diff --git a/enigma/src/main/java/cuchaz/enigma/source/cfr/EnigmaDumper.java b/enigma/src/main/java/cuchaz/enigma/source/cfr/EnigmaDumper.java
index 4e8940a..14fd168 100644
--- a/enigma/src/main/java/cuchaz/enigma/source/cfr/EnigmaDumper.java
+++ b/enigma/src/main/java/cuchaz/enigma/source/cfr/EnigmaDumper.java
@@ -150,7 +150,7 @@ public class EnigmaDumper extends StringStreamDumper {
 
                     String javaDoc = mapping.javadoc();
                     if (javaDoc != null) {
-                        recordComponentDocs.add(String.format("@param %s %s", field.getFieldName(), javaDoc));
+                        recordComponentDocs.add(String.format("@param %s %s", mapping.targetName(), javaDoc));
                     }
                 }
             }
diff --git a/enigma/src/main/java/cuchaz/enigma/translation/mapping/EntryRemapper.java b/enigma/src/main/java/cuchaz/enigma/translation/mapping/EntryRemapper.java
index 0977b74..0268834 100644
--- a/enigma/src/main/java/cuchaz/enigma/translation/mapping/EntryRemapper.java
+++ b/enigma/src/main/java/cuchaz/enigma/translation/mapping/EntryRemapper.java
@@ -88,9 +88,9 @@ public class EntryRemapper {
                 }
         }
 
-        // A little bit of a hack to also map the getter method for record fields/components.
+        // A little bit of a hack to also map the getter method for record fields.
         private void mapRecordComponentGetter(ValidationContext vc, ClassEntry classEntry, FieldEntry fieldEntry, EntryMapping fieldMapping) {
-                if (!jarIndex.getEntryIndex().getClassAccess(classEntry).isRecord() || jarIndex.getEntryIndex().getFieldAccess(fieldEntry).isStatic()) {
+                if (!jarIndex.getEntryIndex().getDefinition(classEntry).isRecord() || jarIndex.getEntryIndex().getFieldAccess(fieldEntry).isStatic()) {
                         return;
                 }
 
@@ -102,7 +102,7 @@ public class EntryRemapper {
                 MethodEntry methodEntry = null;
 
                 for (MethodEntry method : classMethods) {
-                        // Find the matching record component getter via matching the names. My understanding is this is safe, failing this it may need to be a bit more intelligent
+                        // Find the matching record component getter via matching the names. TODO: Support when the record field and method names do not match
                         if (method.getName().equals(fieldEntry.getName()) && method.getDesc().toString().equals("()" + fieldEntry.getDesc())) {
                                 methodEntry = method;
                                 break;
@@ -114,7 +114,8 @@ public class EntryRemapper {
                         return;
                 }
 
-                putMapping(vc, methodEntry, fieldMapping != null ? new EntryMapping(fieldMapping.targetName()) : null);
+                // Also remap the associated method, without the javadoc.
+                doPutMapping(vc, methodEntry, new EntryMapping(fieldMapping.targetName()), false);
         }
 
         @Nonnull
diff --git a/enigma/src/main/java/cuchaz/enigma/translation/representation/AccessFlags.java b/enigma/src/main/java/cuchaz/enigma/translation/representation/AccessFlags.java
index 21e6ef4..e8480a2 100644
--- a/enigma/src/main/java/cuchaz/enigma/translation/representation/AccessFlags.java
+++ b/enigma/src/main/java/cuchaz/enigma/translation/representation/AccessFlags.java
@@ -39,10 +39,6 @@ public class AccessFlags {
                 return (flags & Opcodes.ACC_ENUM) != 0;
         }
 
-        public boolean isRecord() {
-                return (flags & Opcodes.ACC_RECORD) != 0;
-        }
-
         public boolean isBridge() {
                 return (flags & Opcodes.ACC_BRIDGE) != 0;
         }
diff --git a/enigma/src/main/java/cuchaz/enigma/translation/representation/entry/ClassDefEntry.java b/enigma/src/main/java/cuchaz/enigma/translation/representation/entry/ClassDefEntry.java
index 237c93d..ab5a422 100644
--- a/enigma/src/main/java/cuchaz/enigma/translation/representation/entry/ClassDefEntry.java
+++ b/enigma/src/main/java/cuchaz/enigma/translation/representation/entry/ClassDefEntry.java
@@ -75,6 +75,10 @@ public class ClassDefEntry extends ClassEntry implements DefEntry<ClassEntry> {
                 return interfaces;
         }
 
+        public boolean isRecord() {
+                return superClass.getName().equals("java/lang/Record");
+        }
+
         @Override
         public TranslateResult<ClassDefEntry> extendedTranslate(Translator translator, @Nonnull EntryMapping mapping) {
                 Signature translatedSignature = translator.translate(signature);