java/java-impl/src/com/intellij/codeInsight/daemon/impl/quickfix/MakeMethodConstructorFix.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.codeInsight.intention.IntentionAction;
  20 import com.intellij.codeInsight.CodeInsightUtilBase;
  21 import com.intellij.openapi.editor.Editor;
  22 import com.intellij.openapi.project.Project;
  23 import com.intellij.psi.PsiFile;
  24 import com.intellij.psi.PsiMethod;
  25 import com.intellij.util.IncorrectOperationException;
  26 import org.jetbrains.annotations.NotNull;
  27
  28 /**
  29  * @author Alexey Kudravtsev
  30  */
  31 public class MakeMethodConstructorFix implements IntentionAction {
  32   private final PsiMethod myMethod;
  33
  34   public MakeMethodConstructorFix(PsiMethod method) {
  35     myMethod = method;
  36   }
  37
  38   @NotNull
  39   public String getText() {
  40     return QuickFixBundle.message("convert.method.to.constructor");
  41   }
  42
  43   @NotNull
  44   public String getFamilyName() {
  45     return getText();
  46   }
  47
  48   public boolean isAvailable(@NotNull Project project, Editor editor, PsiFile file) {
  49     return myMethod.isValid() && myMethod.getReturnTypeElement() != null && myMethod.getManager().isInProject(myMethod);
  50   }
  51
  52   public void invoke(@NotNull Project project, Editor editor, PsiFile file) throws IncorrectOperationException {
  53     if (!CodeInsightUtilBase.preparePsiElementForWrite(myMethod)) return;
  54     myMethod.getReturnTypeElement().delete();
  55   }
  56
  57   public boolean startInWriteAction() {
  58     return true;
  59   }
  60 }