683c1d83b7bce5ee2636d94757a6e1c28876bdd2
[idea/community.git] / plugins / groovy / src / org / jetbrains / plugins / groovy / lang / completion / filters / types / BuiltInTypeFilter.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 package org.jetbrains.plugins.groovy.lang.completion.filters.types;
18
19 import com.intellij.lang.ASTNode;
20 import com.intellij.psi.PsiElement;
21 import com.intellij.psi.filters.ElementFilter;
22 import org.jetbrains.annotations.NonNls;
23 import org.jetbrains.plugins.groovy.lang.completion.GroovyCompletionUtil;
24 import org.jetbrains.plugins.groovy.lang.lexer.GroovyTokenTypes;
25 import org.jetbrains.plugins.groovy.lang.psi.GrReferenceElement;
26 import org.jetbrains.plugins.groovy.lang.psi.GroovyFile;
27 import org.jetbrains.plugins.groovy.lang.psi.api.auxiliary.modifiers.GrModifierList;
28 import org.jetbrains.plugins.groovy.lang.psi.api.statements.arguments.GrArgumentList;
29 import org.jetbrains.plugins.groovy.lang.psi.api.statements.expressions.GrExpression;
30 import org.jetbrains.plugins.groovy.lang.psi.api.statements.params.GrParameter;
31 import org.jetbrains.plugins.groovy.lang.psi.api.toplevel.imports.GrImportStatement;
32 import org.jetbrains.plugins.groovy.lang.psi.api.toplevel.packaging.GrPackageDefinition;
33 import org.jetbrains.plugins.groovy.lang.psi.impl.PsiImplUtil;
34
35 /**
36  * @author ilyas
37  */
38 public class BuiltInTypeFilter implements ElementFilter {
39   public boolean isAcceptable(Object element, PsiElement context) {
40     PsiElement previous = PsiImplUtil.realPrevious(context.getParent().getPrevSibling());
41     if (previous != null &&
42         GroovyTokenTypes.mAT.equals(previous.getNode().getElementType())) {
43       return false;
44     }
45     if (GroovyCompletionUtil.asSimpleVariable(context) ||
46         GroovyCompletionUtil.asTypedMethod(context) ||
47         GroovyCompletionUtil.asVariableInBlock(context)) {
48       return true;
49     }
50     if ((context.getParent() instanceof GrParameter &&
51         ((GrParameter) context.getParent()).getTypeElementGroovy() == null) ||
52         context.getParent() instanceof GrReferenceElement &&
53             !(context.getParent().getParent() instanceof GrImportStatement) &&
54             !(context.getParent().getParent() instanceof GrPackageDefinition) &&
55             !(context.getParent().getParent() instanceof GrArgumentList)) {
56       PsiElement prevSibling = context.getPrevSibling();
57       if (context.getParent() instanceof GrReferenceElement && prevSibling != null && prevSibling.getNode() != null) {
58         ASTNode node = prevSibling.getNode();
59         return !GroovyTokenTypes.DOTS.contains(node.getElementType());
60       } else {
61         return !(previous != null &&
62             GroovyTokenTypes.mAT.equals(previous.getNode().getElementType()));
63       }
64     }
65     if (PsiImplUtil.realPrevious(context.getParent().getPrevSibling()) instanceof GrModifierList) {
66       return true;
67     }
68     if (PsiImplUtil.realPrevious(context.getPrevSibling()) instanceof GrModifierList) {
69       return true;
70     }
71     return context.getParent() instanceof GrExpression &&
72         context.getParent().getParent() instanceof GroovyFile &&
73         GroovyCompletionUtil.isNewStatement(context, false);
74   }
75
76   public boolean isClassAcceptable(Class hintClass) {
77     return true;
78   }
79
80   @NonNls
81   public String toString() {
82     return "built-in-types keywords filter";
83   }
84
85 }