#155 Fix RemoveMappersFix to use new 2023.3 RemoveUnusedVariableFix i…

…ntention action
mapstruct · Dec 16, 2023 · b120326 · b120326
1 parent b393e6d
commit b120326
Showing 1 changed file with 38 additions and 1 deletion.
diff --git a/src/main/java/org/mapstruct/intellij/inspection/WrongUsageOfMappersFactoryInspection.java b/src/main/java/org/mapstruct/intellij/inspection/WrongUsageOfMappersFactoryInspection.java
@@ -5,9 +5,13 @@
  */
 package org.mapstruct.intellij.inspection;
 
+import java.lang.reflect.InvocationTargetException;
+import java.lang.reflect.Method;
+
 import com.intellij.codeInsight.AnnotationUtil;
 import com.intellij.codeInsight.daemon.impl.quickfix.RemoveUnusedVariableFix;
 import com.intellij.codeInsight.intention.AddAnnotationPsiFix;
+import com.intellij.codeInsight.intention.IntentionAction;
 import com.intellij.codeInsight.intention.QuickFixFactory;
 import com.intellij.codeInspection.IntentionWrapper;
 import com.intellij.codeInspection.LocalQuickFix;
@@ -44,6 +48,20 @@ public class WrongUsageOfMappersFactoryInspection extends InspectionBase {
         "getMapper"
     ).parameterTypes( CommonClassNames.JAVA_LANG_CLASS );
 
+    private static final Method AS_INTENTION_ACTION_METHOD;
+
+    static {
+        Method asIntentionActionMethod;
+        try {
+            asIntentionActionMethod = RemoveMappersFix.class.getMethod( "asIntentAction" );
+        }
+        catch ( NoSuchMethodException e ) {
+            asIntentionActionMethod = null;
+        }
+
+        AS_INTENTION_ACTION_METHOD = asIntentionActionMethod;
+    }
+
     @NotNull
     @Override
     PsiElementVisitor buildVisitorInternal(@NotNull ProblemsHolder holder, boolean isOnTheFly) {
@@ -145,8 +163,27 @@ public String getFamilyName() {
     private static LocalQuickFix createRemoveMappersFix(@NotNull PsiMethodCallExpression methodCallExpression) {
         PsiElement parent = methodCallExpression.getParent();
         if ( parent instanceof PsiVariable ) {
+            RemoveMappersFix fix = new RemoveMappersFix( (PsiVariable) parent );
+            IntentionAction action = null;
+            if ( fix instanceof IntentionAction ) {
+                action = (IntentionAction) fix;
+            } else if ( AS_INTENTION_ACTION_METHOD != null ) {
+                try {
+                    Object intentionAction = AS_INTENTION_ACTION_METHOD.invoke( fix );
+                    if ( intentionAction instanceof IntentionAction ) {
+                        action = (IntentionAction) intentionAction;
+                    }
+                }
+                catch ( IllegalAccessException | InvocationTargetException e ) {
+                    action = null;
+                }
+            }
+
+            if ( action == null ) {
+                return null;
+            }
             return IntentionWrapper.wrapToQuickFix(
-                new RemoveMappersFix( (PsiVariable) parent ),
+                action,
                 methodCallExpression.getContainingFile()
             );
         }