java/java-impl/src/com/intellij/codeInsight/intention/impl/AddSingleMemberStaticImportAction.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
  17 /*
  18  * @author ven
  19  */
  20 package com.intellij.codeInsight.intention.impl;
  21
  22 import com.intellij.codeInsight.CodeInsightBundle;
  23 import com.intellij.codeInsight.CodeInsightUtilBase;
  24 import com.intellij.codeInsight.intention.PsiElementBaseIntentionAction;
  25 import com.intellij.openapi.diagnostic.Logger;
  26 import com.intellij.openapi.editor.Editor;
  27 import com.intellij.openapi.project.Project;
  28 import com.intellij.openapi.util.Key;
  29 import com.intellij.psi.*;
  30 import com.intellij.psi.impl.source.tree.java.PsiReferenceExpressionImpl;
  31 import com.intellij.psi.util.PsiTreeUtil;
  32 import com.intellij.psi.util.PsiUtil;
  33 import com.intellij.util.IncorrectOperationException;
  34 import org.jetbrains.annotations.NotNull;
  35 import org.jetbrains.annotations.Nullable;
  36
  37 public class AddSingleMemberStaticImportAction extends PsiElementBaseIntentionAction {
  38   private static final Logger LOG = Logger.getInstance("#com.intellij.codeInsight.intention.impl.AddSingleMemberStaticImportAction");
  39   private static final Key<PsiElement> TEMP_REFERENT_USER_DATA = new Key<PsiElement>("TEMP_REFERENT_USER_DATA");
  40
  41   @NotNull
  42   public String getFamilyName() {
  43     return CodeInsightBundle.message("intention.add.single.member.static.import.family");
  44   }
  45
  46   public boolean isAvailable(@NotNull Project project, Editor editor, @Nullable PsiElement element) {
  47     if (element == null || !PsiUtil.isLanguageLevel5OrHigher(element)) return false;
  48     PsiFile file = element.getContainingFile();
  49     if (element instanceof PsiIdentifier && element.getParent() instanceof PsiReferenceExpression &&
  50         ((PsiReferenceExpression)element.getParent()).getQualifierExpression() != null) {
  51       PsiReferenceExpression refExpr = (PsiReferenceExpression)element.getParent();
  52       if (refExpr.getParameterList() != null &&
  53           refExpr.getParameterList().getFirstChild() != null) return false;
  54       PsiElement resolved = refExpr.resolve();
  55       if (resolved instanceof PsiMember &&
  56           ((PsiModifierListOwner)resolved).hasModifierProperty(PsiModifier.STATIC)) {
  57         PsiClass aClass = ((PsiMember)resolved).getContainingClass();
  58         if (aClass != null && !PsiTreeUtil.isAncestor(aClass, element, true)) {
  59           String qName = aClass.getQualifiedName();
  60           if (qName != null) {
  61             qName = qName + "." +refExpr.getReferenceName();
  62             if (file instanceof PsiJavaFile) {
  63               if (((PsiJavaFile)file).getImportList().findSingleImportStatement(refExpr.getReferenceName()) == null) {
  64                 setText(CodeInsightBundle.message("intention.add.single.member.static.import.text", qName));
  65                 return true;
  66               }
  67             }
  68           }
  69         }
  70       }
  71     }
  72
  73     return false;
  74   }
  75
  76
  77   public void invoke(@NotNull final Project project, final Editor editor, PsiFile file) throws IncorrectOperationException {
  78     if (!CodeInsightUtilBase.prepareFileForWrite(file)) return;
  79     PsiElement element = file.findElementAt(editor.getCaretModel().getOffset());
  80     final PsiReferenceExpression refExpr = (PsiReferenceExpression)element.getParent();
  81     final PsiElement resolved = refExpr.resolve();
  82
  83     file.accept(new JavaRecursiveElementWalkingVisitor() {
  84       @Override public void visitReferenceExpression(PsiReferenceExpression expression) {
  85         super.visitReferenceExpression(expression);
  86         if (refExpr.getReferenceName().equals(expression.getReferenceName())) {
  87           PsiElement resolved = expression.resolve();
  88           if (resolved != null) {
  89             expression.putUserData(TEMP_REFERENT_USER_DATA, resolved);
  90           }
  91         }
  92       }
  93     });
  94
  95     ((PsiReferenceExpressionImpl)refExpr).bindToElementViaStaticImport(((PsiMember)resolved).getContainingClass(), ((PsiNamedElement)resolved).getName(), ((PsiJavaFile)file).getImportList());
  96
  97     file.accept(new JavaRecursiveElementVisitor() {
  98       @Override public void visitReferenceExpression(PsiReferenceExpression expression) {
  99         if (expression.getParameterList() != null &&
 100             expression.getParameterList().getFirstChild() != null) return;
 101
 102         if (refExpr.getReferenceName().equals(expression.getReferenceName())) {
 103           if (!expression.isQualified()) {
 104             PsiElement referent = expression.getUserData(TEMP_REFERENT_USER_DATA);
 105
 106             if (referent instanceof PsiMember && referent != expression.resolve()) {
 107               PsiElementFactory factory = JavaPsiFacade.getInstance(expression.getProject()).getElementFactory();
 108               try {
 109                 PsiReferenceExpression copy = (PsiReferenceExpression)factory.createExpressionFromText("A." + expression.getReferenceName(), null);
 110                 expression = (PsiReferenceExpression)expression.replace(copy);
 111                 ((PsiReferenceExpression)expression.getQualifierExpression()).bindToElement(((PsiMember)referent).getContainingClass());
 112               }
 113               catch (IncorrectOperationException e) {
 114                 LOG.error (e);
 115               }
 116             }
 117             expression.putUserData(TEMP_REFERENT_USER_DATA, null);
 118           } else {
 119             if (expression.getQualifierExpression() instanceof PsiReferenceExpression) {
 120               PsiElement aClass = ((PsiReferenceExpression)expression.getQualifierExpression()).resolve();
 121               if (aClass == ((PsiMember)resolved).getContainingClass()) {
 122                 try {
 123                   expression.getQualifierExpression().delete();
 124                 }
 125                 catch (IncorrectOperationException e) {
 126                   LOG.error(e);
 127                 }
 128               }
 129             }
 130           }
 131           expression.putUserData(TEMP_REFERENT_USER_DATA, null);
 132         }
 133
 134         super.visitReferenceExpression(expression);
 135       }
 136     });
 137   }
 138 }