Additional interface added for resolver fix

Author: konarshankar07

src/com/magento/idea/magento2plugin/inspections/php/fix/PhpImplementResolverClassQuickFix.java

@@ -7,24 +7,57 @@
 
 import com.intellij.codeInspection.LocalQuickFix;
 import com.intellij.codeInspection.ProblemDescriptor;
+import com.intellij.openapi.command.WriteCommandAction;
 import com.intellij.openapi.project.Project;
+import com.intellij.openapi.ui.DialogBuilder;
 import com.intellij.psi.PsiElement;
 import com.jetbrains.php.lang.psi.PhpPsiElementFactory;
 import com.jetbrains.php.lang.psi.elements.PhpClass;
 import com.magento.idea.magento2plugin.magento.files.GraphQlResolver;
 import org.jetbrains.annotations.NotNull;
+import javax.swing.*;
 
 public class PhpImplementResolverClassQuickFix implements LocalQuickFix {
+    public static final String FAMILY_NAME = "Implements Resolver interface";
+    public static final String DIALOG_TITLE = "Select one of the following interface";
     @NotNull
     @Override
     public String getFamilyName() {
-        return "Implements Resolver interface";
+        return FAMILY_NAME;
     }
 
     @Override
     public void applyFix(@NotNull Project project, @NotNull ProblemDescriptor descriptor) {
-        PsiElement correctInterface = PhpPsiElementFactory.createImplementsList(project, GraphQlResolver.RESOLVER_INTERFACE);
-        PhpClass graphQlResolverClass = (PhpClass) descriptor.getPsiElement().getParent();
-        graphQlResolverClass.getImplementsList().replace(correctInterface);
+        String[] expectedInterface = { GraphQlResolver.BATCH_RESOLVER_INTERFACE,
+                GraphQlResolver.BATCH_SERVICE_CONTRACT_RESOLVER_INTERFACE,
+                GraphQlResolver.RESOLVER_INTERFACE
+        };
+        final JComboBox expectedInterfaceDropdown = new JComboBox(expectedInterface);
+        DialogBuilder dialogBox = createDialogBox(expectedInterfaceDropdown, project);
+        if (dialogBox.showAndGet()) {
+            String getSelectedInterface = expectedInterfaceDropdown.getSelectedItem().toString();
+            PsiElement correctInterface = PhpPsiElementFactory.createImplementsList(project, getSelectedInterface);
+            PhpClass graphQlResolverClass = (PhpClass) descriptor.getPsiElement().getParent();
+            WriteCommandAction.writeCommandAction(project).run(() -> {
+                graphQlResolverClass.getImplementsList().replace(correctInterface);
+            });
+        }
+    }
+
+    private DialogBuilder createDialogBox(JComboBox selectedClass, Project project)
+    {
+        JPanel panel = new JPanel();
+        panel.add(selectedClass);
+        DialogBuilder builder = new DialogBuilder(project);
+        builder.setTitle(DIALOG_TITLE);
+        builder.setCenterPanel(panel);
+        builder.addOkAction();
+        builder.addCancelAction();
+        return builder;
+    }
+
+    @Override
+    public boolean startInWriteAction() {
+        return false;
     }
 }

src/com/magento/idea/magento2plugin/magento/files/GraphQlResolver.java

@@ -6,5 +6,7 @@
 
 public class GraphQlResolver {
     public static final String RESOLVER_INTERFACE = "\\Magento\\Framework\\GraphQl\\Query\\ResolverInterface";
+    public static final String BATCH_RESOLVER_INTERFACE = "\\Magento\\Framework\\GraphQl\\Query\\Resolver\\BatchResolverInterface";
+    public static final String BATCH_SERVICE_CONTRACT_RESOLVER_INTERFACE = "\\Magento\\Framework\\GraphQl\\Query\\Resolver\\BatchServiceContractResolverInterface";
     public static final String CLASS_ARGUMENT = "class";
 }