| /* |
| * Copyright 2000-2012 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.intention.IntentionAction; |
| 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.util.TypeConversionUtil; |
| import com.intellij.util.IncorrectOperationException; |
| import org.jetbrains.annotations.NotNull; |
| import org.jetbrains.annotations.Nullable; |
| |
| /** |
| * @author Danila Ponomarenko |
| */ |
| public class ReplaceWithListAccessFix implements IntentionAction { |
| private final PsiArrayAccessExpression myArrayAccessExpression; |
| |
| public ReplaceWithListAccessFix(@NotNull PsiArrayAccessExpression arrayAccessExpression) { |
| myArrayAccessExpression = arrayAccessExpression; |
| } |
| |
| @NotNull |
| @Override |
| public String getText() { |
| return QuickFixBundle.message("replace.with.list.access.text"); |
| } |
| |
| @NotNull |
| @Override |
| public String getFamilyName() { |
| return getText(); |
| } |
| |
| @Override |
| public boolean isAvailable(@NotNull Project project, Editor editor, PsiFile file) { |
| if (!myArrayAccessExpression.isValid()) return false; |
| if (!TypeConversionUtil.areTypesAssignmentCompatible(PsiType.INT, myArrayAccessExpression.getIndexExpression())) { |
| return false; |
| } |
| final PsiElement parent = myArrayAccessExpression.getParent(); |
| if (parent instanceof PsiAssignmentExpression) { |
| final PsiExpression lExpression = ((PsiAssignmentExpression)parent).getLExpression(); |
| if (lExpression.equals(myArrayAccessExpression) && !(parent.getParent() instanceof PsiExpressionStatement)) { |
| return false; |
| } |
| } |
| |
| final PsiExpression arrayExpression = myArrayAccessExpression.getArrayExpression(); |
| final PsiType type = arrayExpression.getType(); |
| final PsiType listType = createUtilListType(project); |
| |
| if (type == null || listType == null) return false; |
| |
| |
| return listType.isAssignableFrom(type); |
| } |
| |
| @Nullable |
| private PsiType createUtilListType(@NotNull Project project) { |
| final JavaPsiFacade facade = JavaPsiFacade.getInstance(project); |
| final PsiClass listClass = JavaPsiFacade.getInstance(project).findClass(CommonClassNames.JAVA_UTIL_LIST, myArrayAccessExpression.getResolveScope()); |
| |
| if (listClass == null) return null; |
| |
| final PsiElementFactory factory = facade.getElementFactory(); |
| return factory.createType(listClass); |
| } |
| |
| @Override |
| public void invoke(@NotNull Project project, Editor editor, PsiFile file) throws IncorrectOperationException { |
| if (!FileModificationService.getInstance().prepareFileForWrite(file)) return; |
| final PsiExpression arrayExpression = myArrayAccessExpression.getArrayExpression(); |
| final PsiExpression indexExpression = myArrayAccessExpression.getIndexExpression(); |
| |
| if (indexExpression == null) return; |
| |
| final PsiElementFactory factory = JavaPsiFacade.getInstance(project).getElementFactory(); |
| final CodeStyleManager codeStyleManager = CodeStyleManager.getInstance(project); |
| |
| final PsiElement parent = myArrayAccessExpression.getParent(); |
| if (parent instanceof PsiAssignmentExpression) { |
| final PsiAssignmentExpression assignmentExpression = (PsiAssignmentExpression)parent; |
| |
| final PsiExpression lExpression = assignmentExpression.getLExpression(); |
| final PsiExpression rExpression = assignmentExpression.getRExpression(); |
| if (lExpression.equals(myArrayAccessExpression) && parent.getParent() instanceof PsiExpressionStatement && rExpression != null) { |
| replaceWithSet(factory, codeStyleManager, arrayExpression, indexExpression, rExpression, assignmentExpression); |
| return; |
| } |
| } |
| replaceWithGet(factory, codeStyleManager, arrayExpression, indexExpression, myArrayAccessExpression); |
| } |
| |
| @NotNull |
| private static PsiElement replaceWithGet(@NotNull PsiElementFactory factory, |
| @NotNull CodeStyleManager codeStyleManager, |
| @NotNull PsiExpression arrayExpression, |
| @NotNull PsiExpression indexExpression, |
| @NotNull PsiElement anchor) { |
| |
| final PsiElement listAccess = factory.createExpressionFromText( |
| arrayExpression.getText() + ".get(" + indexExpression.getText() + ")", |
| anchor); |
| return anchor.replace(codeStyleManager.reformat(listAccess)); |
| } |
| |
| private static PsiElement replaceWithSet(@NotNull PsiElementFactory factory, |
| @NotNull CodeStyleManager codeStyleManager, |
| @NotNull PsiExpression arrayExpression, |
| @NotNull PsiExpression indexExpression, |
| @NotNull PsiExpression expression, |
| @NotNull PsiElement anchor) { |
| final PsiElement listAccess = factory.createExpressionFromText( |
| arrayExpression.getText() + ".set(" + indexExpression.getText() + "," + expression.getText() + ")", |
| anchor |
| ); |
| return anchor.replace(codeStyleManager.reformat(listAccess)); |
| } |
| |
| @Override |
| public boolean startInWriteAction() { |
| return true; |
| } |
| } |