From 0a1afc875423eebc443347eb002ac87ec0016e4d Mon Sep 17 00:00:00 2001
From: JohnPaulTaylorII <johnpaultaylorii@gmail.com>
Date: Mon, 14 Mar 2022 16:03:37 -0400
Subject: [PATCH] [fixes #3133] Additional bits setting when creating type
 annotations

---
 .../eclipse/handlers/EclipseHandlerUtil.java   | 18 +++++++++++-------
 .../handlers/EclipseSingularsRecipes.java      |  7 ++++---
 .../handlers/HandleEqualsAndHashCode.java      | 10 +++++++---
 .../lombok/eclipse/handlers/HandleWithBy.java  |  2 +-
 .../singulars/EclipseGuavaSingularizer.java    |  2 +-
 .../EclipseJavaUtilListSetSingularizer.java    |  2 +-
 .../EclipseJavaUtilMapSingularizer.java        |  2 +-
 7 files changed, 26 insertions(+), 17 deletions(-)

diff --git a/src/core/lombok/eclipse/handlers/EclipseHandlerUtil.java b/src/core/lombok/eclipse/handlers/EclipseHandlerUtil.java
index c313cf5139..cda31e2c64 100644
--- a/src/core/lombok/eclipse/handlers/EclipseHandlerUtil.java
+++ b/src/core/lombok/eclipse/handlers/EclipseHandlerUtil.java
@@ -2601,25 +2601,25 @@ public static void createRelevantNonNullAnnotation(EclipseNode typeNode, MethodD
 		applyAnnotationToMethodDecl(typeNode, mth, lib.getNonNullAnnotation(), lib.isTypeUse());
 	}
 	
-	public static void createRelevantNullableAnnotation(EclipseNode typeNode, Argument arg) {
+	public static void createRelevantNullableAnnotation(EclipseNode typeNode, Argument arg, MethodDeclaration mth) {
 		NullAnnotationLibrary lib = typeNode.getAst().readConfiguration(ConfigurationKeys.ADD_NULL_ANNOTATIONS);
 		if (lib == null) return;
 		
-		applyAnnotationToVarDecl(typeNode, arg, lib.getNullableAnnotation(), lib.isTypeUse());
+		applyAnnotationToVarDecl(typeNode, arg, mth, lib.getNullableAnnotation(), lib.isTypeUse());
 	}
 	
-	public static void createRelevantNullableAnnotation(EclipseNode typeNode, Argument arg, List<NullAnnotationLibrary> applied) {
+	public static void createRelevantNullableAnnotation(EclipseNode typeNode, Argument arg, MethodDeclaration mth, List<NullAnnotationLibrary> applied) {
 		NullAnnotationLibrary lib = typeNode.getAst().readConfiguration(ConfigurationKeys.ADD_NULL_ANNOTATIONS);
 		if (lib == null || applied.contains(lib)) return;
 		
-		applyAnnotationToVarDecl(typeNode, arg, lib.getNullableAnnotation(), lib.isTypeUse());
+		applyAnnotationToVarDecl(typeNode, arg, mth, lib.getNullableAnnotation(), lib.isTypeUse());
 	}
 	
-	public static void createRelevantNonNullAnnotation(EclipseNode typeNode, Argument arg) {
+	public static void createRelevantNonNullAnnotation(EclipseNode typeNode, Argument arg, MethodDeclaration mth) {
 		NullAnnotationLibrary lib = typeNode.getAst().readConfiguration(ConfigurationKeys.ADD_NULL_ANNOTATIONS);
 		if (lib == null) return;
 		
-		applyAnnotationToVarDecl(typeNode, arg, lib.getNonNullAnnotation(), lib.isTypeUse());
+		applyAnnotationToVarDecl(typeNode, arg, mth, lib.getNonNullAnnotation(), lib.isTypeUse());
 	}
 	
 	private static void applyAnnotationToMethodDecl(EclipseNode typeNode, MethodDeclaration mth, String annType, boolean typeUse) {
@@ -2653,9 +2653,10 @@ private static void applyAnnotationToMethodDecl(EclipseNode typeNode, MethodDecl
 			}
 			a[0] = ann;
 			mth.returnType.annotations[len - 1] = a;
+			mth.bits |= Eclipse.HasTypeAnnotations;
 		}
 	}
-	private static void applyAnnotationToVarDecl(EclipseNode typeNode, Argument arg, String annType, boolean typeUse) {
+	private static void applyAnnotationToVarDecl(EclipseNode typeNode, Argument arg, MethodDeclaration mth, String annType, boolean typeUse) {
 		if (annType == null) return;
 		
 		int partCount = 1;
@@ -2686,6 +2687,9 @@ private static void applyAnnotationToVarDecl(EclipseNode typeNode, Argument arg,
 			}
 			a[0] = ann;
 			arg.type.annotations[len - 1] = a;
+			arg.type.bits |= Eclipse.HasTypeAnnotations;
+			arg.bits |= Eclipse.HasTypeAnnotations;
+			mth.bits |= Eclipse.HasTypeAnnotations;
 		}
 	}
 	
diff --git a/src/core/lombok/eclipse/handlers/EclipseSingularsRecipes.java b/src/core/lombok/eclipse/handlers/EclipseSingularsRecipes.java
index 998c1274e1..54f1a55135 100755
--- a/src/core/lombok/eclipse/handlers/EclipseSingularsRecipes.java
+++ b/src/core/lombok/eclipse/handlers/EclipseSingularsRecipes.java
@@ -44,6 +44,7 @@
 import org.eclipse.jdt.internal.compiler.ast.IfStatement;
 import org.eclipse.jdt.internal.compiler.ast.IntLiteral;
 import org.eclipse.jdt.internal.compiler.ast.MessageSend;
+import org.eclipse.jdt.internal.compiler.ast.MethodDeclaration;
 import org.eclipse.jdt.internal.compiler.ast.NullLiteral;
 import org.eclipse.jdt.internal.compiler.ast.OperatorIds;
 import org.eclipse.jdt.internal.compiler.ast.ParameterizedQualifiedTypeReference;
@@ -430,7 +431,7 @@ protected static Reference getBuilderReference(String builderVariable) {
 			}
 		}
 		
-		protected void nullBehaviorize(EclipseNode typeNode, SingularData data, List<Statement> statements, Argument arg) {
+		protected void nullBehaviorize(EclipseNode typeNode, SingularData data, List<Statement> statements, Argument arg, MethodDeclaration md) {
 			boolean ignoreNullCollections = data.isIgnoreNullCollections();
 			
 			if (ignoreNullCollections) {
@@ -439,11 +440,11 @@ protected void nullBehaviorize(EclipseNode typeNode, SingularData data, List<Sta
 				b.statements = statements.toArray(new Statement[statements.size()]);
 				statements.clear();
 				statements.add(new IfStatement(isNotNull, b, 0, 0));
-				EclipseHandlerUtil.createRelevantNullableAnnotation(typeNode, arg);
+				EclipseHandlerUtil.createRelevantNullableAnnotation(typeNode, arg, md);
 				return;
 			}
 			
-			EclipseHandlerUtil.createRelevantNonNullAnnotation(typeNode, arg);
+			EclipseHandlerUtil.createRelevantNonNullAnnotation(typeNode, arg, md);
 			Statement nullCheck = EclipseHandlerUtil.generateNullCheck(null, data.getPluralName(), typeNode, "%s cannot be null");
 			statements.add(0, nullCheck);
 		}
diff --git a/src/core/lombok/eclipse/handlers/HandleEqualsAndHashCode.java b/src/core/lombok/eclipse/handlers/HandleEqualsAndHashCode.java
index 940612e775..b0e0e526b0 100755
--- a/src/core/lombok/eclipse/handlers/HandleEqualsAndHashCode.java
+++ b/src/core/lombok/eclipse/handlers/HandleEqualsAndHashCode.java
@@ -642,13 +642,17 @@ public MethodDeclaration createEquals(EclipseNode type, Collection<Included<Ecli
 		method.bodyStart = method.declarationSourceStart = method.sourceStart = source.sourceStart;
 		method.bodyEnd = method.declarationSourceEnd = method.sourceEnd = source.sourceEnd;
 		QualifiedTypeReference objectRef = new QualifiedTypeReference(TypeConstants.JAVA_LANG_OBJECT, new long[] { p, p, p });
-		if (onParamTypeNullable != null) objectRef.annotations = new Annotation[][] {null, null, onParamTypeNullable};
+		if (onParamTypeNullable != null) {
+			objectRef.annotations = new Annotation[][] {null, null, onParamTypeNullable};
+			objectRef.bits |= Eclipse.HasTypeAnnotations;
+			method.bits |= Eclipse.HasTypeAnnotations;
+		}
 		setGeneratedBy(objectRef, source);
 		method.arguments = new Argument[] {new Argument(new char[] { 'o' }, 0, objectRef, Modifier.FINAL)};
 		method.arguments[0].sourceStart = pS; method.arguments[0].sourceEnd = pE;
 		if (!onParam.isEmpty() || onParamNullable != null)
 			method.arguments[0].annotations = copyAnnotations(source, onParam.toArray(new Annotation[0]), onParamNullable);
-		EclipseHandlerUtil.createRelevantNullableAnnotation(type, method.arguments[0], applied);
+		EclipseHandlerUtil.createRelevantNullableAnnotation(type, method.arguments[0], method, applied);
 		setGeneratedBy(method.arguments[0], source);
 		
 		List<Statement> statements = new ArrayList<Statement>();
@@ -898,7 +902,7 @@ public MethodDeclaration createCanEqual(EclipseNode type, ASTNode source, List<A
 		method.arguments = new Argument[] {new Argument(otherName, 0, objectRef, Modifier.FINAL)};
 		method.arguments[0].sourceStart = pS; method.arguments[0].sourceEnd = pE;
 		if (!onParam.isEmpty()) method.arguments[0].annotations = onParam.toArray(new Annotation[0]);
-		EclipseHandlerUtil.createRelevantNullableAnnotation(type, method.arguments[0]);
+		EclipseHandlerUtil.createRelevantNullableAnnotation(type, method.arguments[0], method);
 		setGeneratedBy(method.arguments[0], source);
 		
 		SingleNameReference otherRef = new SingleNameReference(otherName, p);
diff --git a/src/core/lombok/eclipse/handlers/HandleWithBy.java b/src/core/lombok/eclipse/handlers/HandleWithBy.java
index 5ab3cf81cf..bfba91d441 100644
--- a/src/core/lombok/eclipse/handlers/HandleWithBy.java
+++ b/src/core/lombok/eclipse/handlers/HandleWithBy.java
@@ -373,7 +373,7 @@ public MethodDeclaration createWithBy(TypeDeclaration parent, EclipseNode fieldN
 			method.statements = statements.toArray(new Statement[0]);
 		}
 		
-		createRelevantNonNullAnnotation(sourceNode, param);
+		createRelevantNonNullAnnotation(sourceNode, param, method);
 		createRelevantNonNullAnnotation(fieldNode, method);
 		
 		method.traverse(new SetGeneratedByVisitor(source), parent.scope);
diff --git a/src/core/lombok/eclipse/handlers/singulars/EclipseGuavaSingularizer.java b/src/core/lombok/eclipse/handlers/singulars/EclipseGuavaSingularizer.java
index 47822ff302..a43e033188 100755
--- a/src/core/lombok/eclipse/handlers/singulars/EclipseGuavaSingularizer.java
+++ b/src/core/lombok/eclipse/handlers/singulars/EclipseGuavaSingularizer.java
@@ -207,7 +207,7 @@ void generatePluralMethod(CheckerFrameworkVersion cfv, boolean deprecate, TypeRe
 		paramType = addTypeArgs(getTypeArgumentsCount(), true, builderType, paramType, data.getTypeArgs());
 		Argument param = new Argument(data.getPluralName(), 0, paramType, ClassFileConstants.AccFinal);
 		
-		nullBehaviorize(builderType, data, statements, param);
+		nullBehaviorize(builderType, data, statements, param, md);
 		
 		if (returnStatement != null) statements.add(returnStatement);
 		
diff --git a/src/core/lombok/eclipse/handlers/singulars/EclipseJavaUtilListSetSingularizer.java b/src/core/lombok/eclipse/handlers/singulars/EclipseJavaUtilListSetSingularizer.java
index fbde30219f..810a7878ec 100755
--- a/src/core/lombok/eclipse/handlers/singulars/EclipseJavaUtilListSetSingularizer.java
+++ b/src/core/lombok/eclipse/handlers/singulars/EclipseJavaUtilListSetSingularizer.java
@@ -184,7 +184,7 @@ void generatePluralMethod(CheckerFrameworkVersion cfv, boolean deprecate, TypeRe
 		paramType = addTypeArgs(1, true, builderType, paramType, data.getTypeArgs());
 		Argument param = new Argument(data.getPluralName(), 0, paramType, ClassFileConstants.AccFinal);
 		
-		nullBehaviorize(builderType, data, statements, param);
+		nullBehaviorize(builderType, data, statements, param, md);
 		if (returnStatement != null) statements.add(returnStatement);
 		
 		md.statements = statements.toArray(new Statement[0]);
diff --git a/src/core/lombok/eclipse/handlers/singulars/EclipseJavaUtilMapSingularizer.java b/src/core/lombok/eclipse/handlers/singulars/EclipseJavaUtilMapSingularizer.java
index 859cce9472..16ed2d44c5 100755
--- a/src/core/lombok/eclipse/handlers/singulars/EclipseJavaUtilMapSingularizer.java
+++ b/src/core/lombok/eclipse/handlers/singulars/EclipseJavaUtilMapSingularizer.java
@@ -316,7 +316,7 @@ private void generatePluralMethod(CheckerFrameworkVersion cfv, boolean deprecate
 		paramType = addTypeArgs(2, true, builderType, paramType, data.getTypeArgs());
 		Argument param = new Argument(data.getPluralName(), 0, paramType, ClassFileConstants.AccFinal);
 		
-		nullBehaviorize(builderType, data, statements, param);
+		nullBehaviorize(builderType, data, statements, param, md);
 		
 		if (returnStatement != null) statements.add(returnStatement);