| /* |
| * 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.*; |
| import com.intellij.codeInsight.intention.impl.BaseIntentionAction; |
| import com.intellij.ide.util.MemberChooser; |
| import com.intellij.openapi.application.ApplicationManager; |
| import com.intellij.openapi.command.undo.UndoUtil; |
| import com.intellij.openapi.diagnostic.Logger; |
| import com.intellij.openapi.editor.Editor; |
| import com.intellij.openapi.editor.ScrollType; |
| import com.intellij.openapi.project.Project; |
| import com.intellij.openapi.ui.DialogWrapper; |
| import com.intellij.psi.*; |
| import com.intellij.psi.codeStyle.CodeStyleManager; |
| import com.intellij.psi.codeStyle.JavaCodeStyleManager; |
| import com.intellij.psi.javadoc.PsiDocComment; |
| import com.intellij.psi.util.PsiUtil; |
| import com.intellij.psi.util.TypeConversionUtil; |
| import com.intellij.util.IncorrectOperationException; |
| import org.jetbrains.annotations.NotNull; |
| |
| import java.util.ArrayList; |
| import java.util.List; |
| |
| |
| /** |
| * @author ven |
| */ |
| public class CreateConstructorMatchingSuperFix extends BaseIntentionAction { |
| private static final Logger LOG = Logger.getInstance("com.intellij.codeInsight.daemon.impl.quickfix.CreateConstructorMatchingSuperFix"); |
| |
| private final PsiClass myClass; |
| |
| public CreateConstructorMatchingSuperFix(@NotNull PsiClass aClass) { |
| myClass = aClass; |
| } |
| |
| @Override |
| @NotNull |
| public String getFamilyName() { |
| return QuickFixBundle.message("create.constructor.matching.super"); |
| } |
| |
| @Override |
| public boolean isAvailable(@NotNull Project project, Editor editor, PsiFile file) { |
| if (!myClass.isValid() || !myClass.getManager().isInProject(myClass)) return false; |
| setText(QuickFixBundle.message("create.constructor.matching.super")); |
| return true; |
| } |
| |
| @Override |
| public void invoke(@NotNull final Project project, final Editor editor, PsiFile file) { |
| if (!FileModificationService.getInstance().prepareFileForWrite(myClass.getContainingFile())) return; |
| PsiClass baseClass = myClass.getSuperClass(); |
| LOG.assertTrue(baseClass != null); |
| PsiSubstitutor substitutor = TypeConversionUtil.getSuperClassSubstitutor(baseClass, myClass, PsiSubstitutor.EMPTY); |
| List<PsiMethodMember> baseConstructors = new ArrayList<PsiMethodMember>(); |
| PsiMethod[] baseConstrs = baseClass.getConstructors(); |
| for (PsiMethod baseConstr : baseConstrs) { |
| if (PsiUtil.isAccessible(baseConstr, myClass, myClass)) baseConstructors.add(new PsiMethodMember(baseConstr, substitutor)); |
| } |
| |
| chooseConstructor2Delegate(project, editor, substitutor, baseConstructors, baseConstrs, myClass); |
| } |
| |
| public static void chooseConstructor2Delegate(final Project project, |
| final Editor editor, |
| PsiSubstitutor substitutor, |
| List<PsiMethodMember> baseConstructors, |
| PsiMethod[] baseConstrs, |
| final PsiClass targetClass) { |
| PsiMethodMember[] constructors = baseConstructors.toArray(new PsiMethodMember[baseConstructors.size()]); |
| if (constructors.length == 0) { |
| constructors = new PsiMethodMember[baseConstrs.length]; |
| for (int i = 0; i < baseConstrs.length; i++) { |
| constructors[i] = new PsiMethodMember(baseConstrs[i], substitutor); |
| } |
| } |
| |
| LOG.assertTrue(constructors.length >=1); // Otherwise we won't have been messing with all this stuff |
| boolean isCopyJavadoc = true; |
| if (constructors.length > 1 && !ApplicationManager.getApplication().isUnitTestMode()) { |
| MemberChooser<PsiMethodMember> chooser = new MemberChooser<PsiMethodMember>(constructors, false, true, project); |
| chooser.setTitle(QuickFixBundle.message("super.class.constructors.chooser.title")); |
| chooser.show(); |
| if (chooser.getExitCode() != DialogWrapper.OK_EXIT_CODE) return; |
| constructors = chooser.getSelectedElements(new PsiMethodMember[0]); |
| isCopyJavadoc = chooser.isCopyJavadoc(); |
| } |
| |
| final PsiMethodMember[] constructors1 = constructors; |
| final boolean isCopyJavadoc1 = isCopyJavadoc; |
| ApplicationManager.getApplication().runWriteAction ( |
| new Runnable() { |
| @Override |
| public void run() { |
| try { |
| if (targetClass.getLBrace() == null) { |
| PsiClass psiClass = JavaPsiFacade.getInstance(targetClass.getProject()).getElementFactory().createClass("X"); |
| targetClass.addRangeAfter(psiClass.getLBrace(), psiClass.getRBrace(), targetClass.getLastChild()); |
| } |
| JVMElementFactory factory = JVMElementFactories.getFactory(targetClass.getLanguage(), project); |
| CodeStyleManager formatter = CodeStyleManager.getInstance(project); |
| PsiMethod derived = null; |
| for (PsiMethodMember candidate : constructors1) { |
| PsiMethod base = candidate.getElement(); |
| derived = GenerateMembersUtil.substituteGenericMethod(base, candidate.getSubstitutor(), targetClass); |
| |
| if (!isCopyJavadoc1) { |
| final PsiDocComment docComment = derived.getDocComment(); |
| if (docComment != null) { |
| docComment.delete(); |
| } |
| } |
| |
| final String targetClassName = targetClass.getName(); |
| LOG.assertTrue(targetClassName != null, targetClass); |
| derived.setName(targetClassName); |
| |
| ConstructorBodyGenerator generator = ConstructorBodyGenerator.INSTANCE.forLanguage(derived.getLanguage()); |
| if (generator != null) { |
| StringBuilder buffer = new StringBuilder(); |
| generator.start(buffer, derived.getName(), PsiParameter.EMPTY_ARRAY); |
| generator.generateSuperCallIfNeeded(buffer, derived.getParameterList().getParameters()); |
| generator.finish(buffer); |
| PsiMethod stub = factory.createMethodFromText(buffer.toString(), targetClass); |
| derived.getBody().replace(stub.getBody()); |
| } |
| derived = (PsiMethod)formatter.reformat(derived); |
| derived = (PsiMethod)JavaCodeStyleManager.getInstance(project).shortenClassReferences(derived); |
| PsiGenerationInfo<PsiMethod> info = OverrideImplementUtil.createGenerationInfo(derived); |
| info.insert(targetClass, null, true); |
| derived = info.getPsiMember(); |
| } |
| if (derived != null) { |
| editor.getCaretModel().moveToOffset(derived.getTextRange().getStartOffset()); |
| editor.getScrollingModel().scrollToCaret(ScrollType.RELATIVE); |
| } |
| } |
| catch (IncorrectOperationException e) { |
| LOG.error(e); |
| } |
| |
| UndoUtil.markPsiFileForUndo(targetClass.getContainingFile()); |
| } |
| } |
| ); |
| } |
| |
| @Override |
| public boolean startInWriteAction() { |
| return false; |
| } |
| } |