| /* |
| * Copyright 2000-2009 JetBrains s.r.o. |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| package com.intellij.codeInsight.daemon.impl.quickfix; |
| |
| import com.intellij.codeInsight.FileModificationService; |
| import com.intellij.codeInsight.daemon.QuickFixBundle; |
| import com.intellij.codeInsight.generation.GenerateMembersUtil; |
| import com.intellij.codeInspection.LocalQuickFixAndIntentionActionOnPsiElement; |
| import com.intellij.openapi.diagnostic.Logger; |
| import com.intellij.openapi.editor.Editor; |
| import com.intellij.openapi.project.Project; |
| import com.intellij.psi.*; |
| import com.intellij.psi.codeStyle.CodeStyleManager; |
| import com.intellij.psi.codeStyle.JavaCodeStyleManager; |
| import com.intellij.psi.util.MethodSignatureUtil; |
| import com.intellij.psi.util.PsiUtil; |
| import com.intellij.util.IncorrectOperationException; |
| import com.intellij.util.containers.ContainerUtil; |
| import org.jetbrains.annotations.NonNls; |
| import org.jetbrains.annotations.NotNull; |
| import org.jetbrains.annotations.Nullable; |
| |
| import java.util.ArrayList; |
| import java.util.List; |
| |
| public class AddMethodFix extends LocalQuickFixAndIntentionActionOnPsiElement { |
| private static final Logger LOG = Logger.getInstance("#com.intellij.codeInsight.daemon.impl.quickfix.AddMethodFix"); |
| |
| private final PsiMethod myMethodPrototype; |
| private String myText; |
| private final List<String> myExceptions = new ArrayList<String>(); |
| |
| public AddMethodFix(@NotNull PsiMethod methodPrototype, @NotNull PsiClass implClass) { |
| super(implClass); |
| myMethodPrototype = methodPrototype; |
| setText(QuickFixBundle.message("add.method.text", methodPrototype.getName(), implClass.getName())); |
| } |
| |
| public AddMethodFix(@NonNls @NotNull String methodText, @NotNull PsiClass implClass, @NotNull String... exceptions) { |
| this(createMethod(methodText, implClass), implClass); |
| ContainerUtil.addAll(myExceptions, exceptions); |
| } |
| |
| private static PsiMethod createMethod(final String methodText, final PsiClass implClass) { |
| try { |
| return JavaPsiFacade.getInstance(implClass.getProject()).getElementFactory().createMethodFromText(methodText, implClass); |
| } |
| catch (IncorrectOperationException e) { |
| LOG.error(e); |
| return null; |
| } |
| } |
| |
| private static PsiMethod reformat(Project project, PsiMethod result) throws IncorrectOperationException { |
| CodeStyleManager codeStyleManager = CodeStyleManager.getInstance(project); |
| result = (PsiMethod) codeStyleManager.reformat(result); |
| |
| JavaCodeStyleManager javaCodeStyleManager = JavaCodeStyleManager.getInstance(project); |
| result = (PsiMethod) javaCodeStyleManager.shortenClassReferences(result); |
| return result; |
| } |
| |
| protected void setText(@NotNull String text) { |
| myText = text; |
| } |
| |
| @NotNull |
| @Override |
| public String getText() { |
| return myText; |
| } |
| |
| @Override |
| @NotNull |
| public String getFamilyName() { |
| return QuickFixBundle.message("add.method.family"); |
| } |
| |
| @Override |
| public boolean isAvailable(@NotNull Project project, |
| @NotNull PsiFile file, |
| @NotNull PsiElement startElement, |
| @NotNull PsiElement endElement) { |
| final PsiClass myClass = (PsiClass)startElement; |
| |
| return myMethodPrototype.isValid() && |
| myClass.isValid() && |
| myClass.getManager().isInProject(myClass) && |
| myText != null && |
| MethodSignatureUtil.findMethodBySignature(myClass, myMethodPrototype, false) == null |
| ; |
| } |
| |
| @Override |
| public void invoke(@NotNull Project project, |
| @NotNull PsiFile file, |
| @Nullable("is null when called from inspection") Editor editor, |
| @NotNull PsiElement startElement, |
| @NotNull PsiElement endElement) { |
| final PsiClass myClass = (PsiClass)startElement; |
| if (!FileModificationService.getInstance().prepareFileForWrite(myClass.getContainingFile())) return; |
| PsiCodeBlock body; |
| if (myClass.isInterface() && (body = myMethodPrototype.getBody()) != null) body.delete(); |
| for (String exception : myExceptions) { |
| PsiUtil.addException(myMethodPrototype, exception); |
| } |
| PsiMethod method = (PsiMethod)myClass.add(myMethodPrototype); |
| method = (PsiMethod)method.replace(reformat(project, method)); |
| if (editor != null && method.getContainingFile() == file) { |
| GenerateMembersUtil.positionCaret(editor, method, true); |
| } |
| } |
| } |