java/java-impl/src/com/intellij/codeInsight/completion/PreferEnumConstantsWeigher.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.codeInsight.completion;
  17
  18 import com.intellij.codeInsight.lookup.LookupElement;
  19 import com.intellij.psi.*;
  20 import com.intellij.psi.util.PsiTreeUtil;
  21 import org.jetbrains.annotations.NotNull;
  22
  23 /**
  24  * @author peter
  25 */
  26 public class PreferEnumConstantsWeigher extends CompletionWeigher {
  27
  28   public Comparable weigh(@NotNull final LookupElement item, final CompletionLocation location) {
  29     if (item.getObject() instanceof PsiEnumConstant) return 2;
  30
  31     final CompletionParameters parameters = location.getCompletionParameters();
  32     if (!(parameters.getOriginalFile() instanceof PsiJavaFile)) return 1;
  33
  34     if (PsiKeyword.TRUE.equals(item.getLookupString()) || PsiKeyword.FALSE.equals(item.getLookupString())) {
  35       boolean inReturn = PsiTreeUtil.getParentOfType(parameters.getPosition(), PsiReturnStatement.class, false, PsiMember.class) != null;
  36       return inReturn ? 2 : 0;
  37     }
  38
  39     return 1;
  40   }
  41 }