java/java-impl/src/com/intellij/codeInsight/daemon/impl/quickfix/CreateAbstractMethodFromUsageFix.java

   1 /*
   2  * Copyright 2000-2009 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 com.intellij.codeInsight.daemon.impl.quickfix;
  17
  18 import com.intellij.codeInsight.daemon.QuickFixBundle;
  19 import com.intellij.psi.*;
  20 import com.intellij.psi.util.PsiUtil;
  21 import org.jetbrains.annotations.NotNull;
  22
  23 import java.util.ArrayList;
  24 import java.util.List;
  25
  26 public class CreateAbstractMethodFromUsageFix extends CreateMethodFromUsageFix {
  27   public CreateAbstractMethodFromUsageFix(PsiMethodCallExpression methodCall) {
  28     super(methodCall);
  29   }
  30
  31   @Override
  32   protected String getDisplayString(String name) {
  33     return QuickFixBundle.message("create.abstract.method.from.usage.text", name);
  34   }
  35
  36   @NotNull
  37   @Override
  38   protected List<PsiClass> getTargetClasses(PsiElement element) {
  39     List<PsiClass> result = new ArrayList<PsiClass>();
  40     PsiReferenceExpression expr = getMethodCall().getMethodExpression();
  41     for (PsiClass each : super.getTargetClasses(element)) {
  42       if (PsiUtil.isAbstractClass(each) && !each.isInterface() && !shouldCreateStaticMember(expr, each)) result.add(each);
  43     }
  44     return result;
  45   }
  46
  47   @Override
  48   protected String getVisibility(PsiClass parentClass, PsiClass targetClass) {
  49     String result = super.getVisibility(parentClass, targetClass);
  50     return PsiModifier.PUBLIC.equals(result) ? result : PsiModifier.PROTECTED;
  51   }
  52
  53   @Override
  54   protected boolean shouldBeAbstract(PsiClass targetClass) {
  55     return true;
  56   }
  57 }