Merge pull request #175 from konarshankar07/additional-fix-added-for-grapql-resolver--task-169

#169 :- Quick fix for Resolver inspection adds implement directive to the next line

Author: Vitaliy

resources/magento2/inspection.properties

@@ -1,3 +1,5 @@
 inspection.plugin.duplicateInSameFile=The plugin name already used in this file. For more details see Inspection Description.
 inspection.plugin.duplicateInOtherPlaces=The plugin name "{0}" for targeted "{1}" class is already used in the module "{2}" ({3} scope). For more details see Inspection Description.
-inspection.graphql.resolver.mustImplement=Class must implements any of the following interfaces: \\Magento\\Framework\\GraphQl\\Query\\ResolverInterface, \\Magento\\Framework\\GraphQl\\Query\\Resolver\\BatchResolverInterface, \\Magento\\Framework\\GraphQl\\Query\\Resolver\\BatchServiceContractResolverInterface
+inspection.graphql.resolver.mustImplement=Class must implements any of the following interfaces: \\Magento\\Framework\\GraphQl\\Query\\ResolverInterface, \\Magento\\Framework\\GraphQl\\Query\\Resolver\\BatchResolverInterface, \\Magento\\Framework\\GraphQl\\Query\\Resolver\\BatchServiceContractResolverInterface
+inspection.graphql.resolver.fix.family=Implement Resolver interface
+inspection.graphql.resolver.fix.title=Select one of the following interface

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

@@ -7,24 +7,67 @@
 
 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.refactoring.extract.extractInterface.PhpExtractInterfaceProcessor;
 import com.jetbrains.php.lang.psi.PhpPsiElementFactory;
 import com.jetbrains.php.lang.psi.elements.PhpClass;
+import com.magento.idea.magento2plugin.bundles.InspectionBundle;
 import com.magento.idea.magento2plugin.magento.files.GraphQlResolver;
 import org.jetbrains.annotations.NotNull;
+import javax.swing.*;
 
 public class PhpImplementResolverClassQuickFix implements LocalQuickFix {
+    private InspectionBundle inspectionBundle = new InspectionBundle();
+
     @NotNull
     @Override
     public String getFamilyName() {
-        return "Implements Resolver interface";
+        return inspectionBundle.message("inspection.graphql.resolver.fix.family");
     }
 
     @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);
+        ((JLabel)expectedInterfaceDropdown.getRenderer()).setHorizontalAlignment(SwingConstants.CENTER);
+        DialogBuilder dialogBox = createDialogBox(expectedInterfaceDropdown, project);
+        if (dialogBox.showAndGet()) {
+            String getSelectedInterface = expectedInterfaceDropdown.getSelectedItem().toString();
+            final PsiElement erroredElement = descriptor.getPsiElement();
+            final PsiElement correctInterface = PhpPsiElementFactory.createImplementsList(project, getSelectedInterface);
+            final PhpClass graphQlResolverClass = (PhpClass) erroredElement.getParent();
+            String[] implementedInterfaceNames = graphQlResolverClass.getInterfaceNames();
+            WriteCommandAction.runWriteCommandAction(project, () -> {
+                if (implementedInterfaceNames.length == 0) {
+                    PhpExtractInterfaceProcessor.addImplementClause(project, graphQlResolverClass, getSelectedInterface);
+                } else {
+                    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(inspectionBundle.message("inspection.graphql.resolver.fix.title"));
+        builder.setCenterPanel(panel);
+        builder.addOkAction();
+        builder.addCancelAction();
+        return builder;
+    }
+
+    @Override
+    public boolean startInWriteAction() {
+        return false;
     }
 }