java/java-impl/src/com/intellij/psi/impl/source/PsiLabelReference.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.psi.impl.source;
  17
  18 import com.intellij.openapi.util.TextRange;
  19 import com.intellij.psi.*;
  20 import com.intellij.psi.impl.PsiImplUtil;
  21 import com.intellij.util.IncorrectOperationException;
  22
  23 import java.util.ArrayList;
  24 import java.util.List;
  25
  26 import org.jetbrains.annotations.NotNull;
  27
  28 /**
  29  * Created by IntelliJ IDEA.
  30  * User: ik
  31  * Date: 06.05.2003
  32  * Time: 23:32:45
  33  * To change this template use Options | File Templates.
  34  */
  35 public class PsiLabelReference implements PsiReference{
  36   private final PsiStatement myStatement;
  37   private PsiIdentifier myIdentifier;
  38
  39   public PsiLabelReference(PsiStatement stat, PsiIdentifier identifier){
  40     myStatement = stat;
  41     myIdentifier = identifier;
  42   }
  43
  44   public PsiElement getElement(){
  45     return myStatement;
  46   }
  47
  48   public TextRange getRangeInElement(){
  49     final int parent = myIdentifier.getStartOffsetInParent();
  50     return new TextRange(parent, myIdentifier.getTextLength() + parent);
  51   }
  52
  53     public PsiElement resolve(){
  54       final String label = myIdentifier.getText();
  55       if(label == null) return null;
  56       PsiElement context = myStatement;
  57       while(context != null){
  58         if(context instanceof PsiLabeledStatement){
  59           final PsiLabeledStatement statement = (PsiLabeledStatement) context;
  60           if(label.equals(statement.getName()))
  61             return statement;
  62         }
  63         context = context.getContext();
  64       }
  65       return null;
  66     }
  67
  68     public String getCanonicalText(){
  69       return getElement().getText();
  70     }
  71
  72     public PsiElement handleElementRename(String newElementName) throws IncorrectOperationException{
  73       myIdentifier = (PsiIdentifier) PsiImplUtil.setName(myIdentifier, newElementName);
  74       return myIdentifier;
  75     }
  76
  77     public PsiElement bindToElement(@NotNull PsiElement element) throws IncorrectOperationException{
  78       if(element instanceof PsiLabeledStatement){
  79         myIdentifier = (PsiIdentifier) PsiImplUtil.setName(myIdentifier, ((PsiLabeledStatement)element).getName());
  80         return myIdentifier;
  81       }
  82       throw new IncorrectOperationException("Can't bind not to labeled statement");
  83     }
  84
  85     public boolean isReferenceTo(PsiElement element){
  86       return resolve() == element;
  87     }
  88
  89     public Object[] getVariants(){
  90       final List result = new ArrayList();
  91       PsiElement context = myStatement;
  92       while(context != null){
  93         if(context instanceof PsiLabeledStatement){
  94           result.add(context);
  95         }
  96         context = context.getContext();
  97       }
  98       return result.toArray();
  99     }
 100
 101   public boolean isSoft(){
 102     return false;
 103   }
 104 }