Merge branch 'ypankratyev/duplicate_extension_in_register_quickfix'
[idea/community.git] / plugins / devkit / src / inspections / quickfix / RegisterExtensionFixProvider.java
1 /*
2  * Copyright 2000-2014 JetBrains s.r.o.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 package org.jetbrains.idea.devkit.inspections.quickfix;
17
18 import com.intellij.codeInsight.intention.IntentionAction;
19 import com.intellij.codeInspection.GlobalInspectionTool;
20 import com.intellij.codeInspection.InspectionEP;
21 import com.intellij.codeInspection.LocalInspectionEP;
22 import com.intellij.codeInspection.LocalInspectionTool;
23 import com.intellij.codeInspection.reference.UnusedDeclarationFixProvider;
24 import com.intellij.psi.PsiClass;
25 import com.intellij.psi.PsiElement;
26 import com.intellij.psi.PsiIdentifier;
27 import com.intellij.psi.util.InheritanceUtil;
28 import org.jetbrains.annotations.NotNull;
29 import org.jetbrains.idea.devkit.util.ExtensionPointCandidate;
30 import org.jetbrains.idea.devkit.util.ExtensionPointLocator;
31
32 import java.util.Set;
33
34 public class RegisterExtensionFixProvider implements UnusedDeclarationFixProvider {
35
36   @NotNull
37   @Override
38   public IntentionAction[] getQuickFixes(@NotNull PsiElement element) {
39     if (!(element instanceof PsiIdentifier)) return IntentionAction.EMPTY_ARRAY;
40     PsiElement parent = element.getParent();
41     if (!(parent instanceof PsiClass)) return IntentionAction.EMPTY_ARRAY;
42
43     PsiClass psiClass = (PsiClass)parent;
44     if (InheritanceUtil.isInheritor(psiClass, LocalInspectionTool.class.getName())) {
45       return new IntentionAction[]{new RegisterInspectionFix(psiClass, LocalInspectionEP.LOCAL_INSPECTION)};
46     }
47     if (InheritanceUtil.isInheritor(psiClass, GlobalInspectionTool.class.getName())) {
48       return new IntentionAction[]{new RegisterInspectionFix(psiClass, InspectionEP.GLOBAL_INSPECTION)};
49     }
50
51     ExtensionPointLocator extensionPointLocator = new ExtensionPointLocator(psiClass);
52     Set<ExtensionPointCandidate> candidateList = extensionPointLocator.findSuperCandidates();
53     if (!candidateList.isEmpty()) {
54       return new IntentionAction[]{new RegisterExtensionFix(psiClass, candidateList)};
55     }
56     return IntentionAction.EMPTY_ARRAY;
57   }
58 }