001    /*
002     * Copyright 2010-2015 JetBrains s.r.o.
003     *
004     * Licensed under the Apache License, Version 2.0 (the "License");
005     * you may not use this file except in compliance with the License.
006     * You may obtain a copy of the License at
007     *
008     * http://www.apache.org/licenses/LICENSE-2.0
009     *
010     * Unless required by applicable law or agreed to in writing, software
011     * distributed under the License is distributed on an "AS IS" BASIS,
012     * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013     * See the License for the specific language governing permissions and
014     * limitations under the License.
015     */
016    
017    package org.jetbrains.kotlin;
018    
019    import com.intellij.psi.tree.IElementType;
020    import com.intellij.psi.tree.IFileElementType;
021    import org.jetbrains.kotlin.idea.KotlinLanguage;
022    import org.jetbrains.kotlin.psi.*;
023    import org.jetbrains.kotlin.psi.stubs.elements.KtStubElementTypes;
024    
025    public interface KtNodeTypes {
026        IFileElementType KT_FILE = new IFileElementType(KotlinLanguage.INSTANCE);
027    
028        IElementType CLASS     = KtStubElementTypes.CLASS;
029        IElementType FUN       = KtStubElementTypes.FUNCTION;
030        IElementType PROPERTY  = KtStubElementTypes.PROPERTY;
031        IElementType DESTRUCTURING_DECLARATION = new KtNodeType("DESTRUCTURING_DECLARATION", KtDestructuringDeclaration.class);
032        IElementType DESTRUCTURING_DECLARATION_ENTRY = new KtNodeType("DESTRUCTURING_DECLARATION_ENTRY", KtDestructuringDeclarationEntry.class);
033    
034        //KtNodeType TYPEDEF    = new KtNodeType("TYPEDEF", KtTypedef.class);
035        IElementType OBJECT_DECLARATION = KtStubElementTypes.OBJECT_DECLARATION;
036    
037        IElementType ENUM_ENTRY            = KtStubElementTypes.ENUM_ENTRY;
038        IElementType CLASS_INITIALIZER = KtStubElementTypes.CLASS_INITIALIZER;
039        IElementType SCRIPT_INITIALIZER    = new KtNodeType("SCRIPT_INITIALIZER", KtScriptInitializer.class);
040        IElementType SECONDARY_CONSTRUCTOR  = KtStubElementTypes.SECONDARY_CONSTRUCTOR;
041        IElementType PRIMARY_CONSTRUCTOR    = KtStubElementTypes.PRIMARY_CONSTRUCTOR;
042    
043        IElementType TYPE_PARAMETER_LIST                 = KtStubElementTypes.TYPE_PARAMETER_LIST;
044        IElementType TYPE_PARAMETER                      = KtStubElementTypes.TYPE_PARAMETER;
045        IElementType SUPER_TYPE_LIST                     = KtStubElementTypes.SUPER_TYPE_LIST;
046        IElementType DELEGATED_SUPER_TYPE_ENTRY          = KtStubElementTypes.DELEGATED_SUPER_TYPE_ENTRY;
047        IElementType SUPER_TYPE_CALL_ENTRY               = KtStubElementTypes.SUPER_TYPE_CALL_ENTRY;
048        IElementType SUPER_TYPE_ENTRY                    = KtStubElementTypes.SUPER_TYPE_ENTRY;
049        KtNodeType PROPERTY_DELEGATE                    = new KtNodeType("PROPERTY_DELEGATE", KtPropertyDelegate.class);
050        IElementType CONSTRUCTOR_CALLEE                  = KtStubElementTypes.CONSTRUCTOR_CALLEE;
051        IElementType VALUE_PARAMETER_LIST               = KtStubElementTypes.VALUE_PARAMETER_LIST;
052        IElementType VALUE_PARAMETER                    = KtStubElementTypes.VALUE_PARAMETER;
053    
054        IElementType CLASS_BODY                         = KtStubElementTypes.CLASS_BODY;
055        IElementType IMPORT_LIST                        = KtStubElementTypes.IMPORT_LIST;
056        IElementType FILE_ANNOTATION_LIST               = KtStubElementTypes.FILE_ANNOTATION_LIST;
057        IElementType IMPORT_DIRECTIVE                   = KtStubElementTypes.IMPORT_DIRECTIVE;
058        IElementType MODIFIER_LIST                      = KtStubElementTypes.MODIFIER_LIST;
059        IElementType ANNOTATION                         = KtStubElementTypes.ANNOTATION;
060        IElementType ANNOTATION_ENTRY                   = KtStubElementTypes.ANNOTATION_ENTRY;
061        IElementType ANNOTATION_TARGET                  = KtStubElementTypes.ANNOTATION_TARGET;
062    
063        IElementType TYPE_ARGUMENT_LIST                 = KtStubElementTypes.TYPE_ARGUMENT_LIST;
064        KtNodeType VALUE_ARGUMENT_LIST                 = new KtNodeType("VALUE_ARGUMENT_LIST", KtValueArgumentList.class);
065        KtNodeType VALUE_ARGUMENT                      = new KtNodeType("VALUE_ARGUMENT", KtValueArgument.class);
066        KtNodeType LAMBDA_ARGUMENT                     = new KtNodeType("LAMBDA_ARGUMENT", KtLambdaArgument.class);
067        KtNodeType VALUE_ARGUMENT_NAME                 = new KtNodeType("VALUE_ARGUMENT_NAME", KtValueArgumentName.class);
068        IElementType TYPE_REFERENCE                     = KtStubElementTypes.TYPE_REFERENCE;
069    
070        IElementType USER_TYPE                = KtStubElementTypes.USER_TYPE;
071        IElementType DYNAMIC_TYPE             = KtStubElementTypes.DYNAMIC_TYPE;
072        IElementType FUNCTION_TYPE            = KtStubElementTypes.FUNCTION_TYPE;
073        IElementType FUNCTION_TYPE_RECEIVER   = KtStubElementTypes.FUNCTION_TYPE_RECEIVER;
074        KtNodeType SELF_TYPE     = new KtNodeType("SELF_TYPE", KtSelfType.class);
075        IElementType NULLABLE_TYPE            = KtStubElementTypes.NULLABLE_TYPE;
076        IElementType TYPE_PROJECTION          = KtStubElementTypes.TYPE_PROJECTION;
077    
078        // TODO: review
079        IElementType PROPERTY_ACCESSOR       = KtStubElementTypes.PROPERTY_ACCESSOR;
080        IElementType INITIALIZER_LIST        = KtStubElementTypes.INITIALIZER_LIST;
081        IElementType TYPE_CONSTRAINT_LIST    = KtStubElementTypes.TYPE_CONSTRAINT_LIST;
082        IElementType TYPE_CONSTRAINT         = KtStubElementTypes.TYPE_CONSTRAINT;
083    
084        IElementType CONSTRUCTOR_DELEGATION_CALL = new KtNodeType.KtLeftBoundNodeType("CONSTRUCTOR_DELEGATION_CALL", KtConstructorDelegationCall.class);
085        KtNodeType CONSTRUCTOR_DELEGATION_REFERENCE = new KtNodeType.KtLeftBoundNodeType("CONSTRUCTOR_DELEGATION_REFERENCE", KtConstructorDelegationReferenceExpression.class);
086    
087        // TODO: Not sure if we need separate NT for each kind of constants
088        KtNodeType NULL               = new KtNodeType("NULL", KtConstantExpression.class);
089        KtNodeType BOOLEAN_CONSTANT   = new KtNodeType("BOOLEAN_CONSTANT", KtConstantExpression.class);
090        KtNodeType FLOAT_CONSTANT     = new KtNodeType("FLOAT_CONSTANT", KtConstantExpression.class);
091        KtNodeType CHARACTER_CONSTANT = new KtNodeType("CHARACTER_CONSTANT", KtConstantExpression.class);
092        KtNodeType INTEGER_CONSTANT   = new KtNodeType("INTEGER_CONSTANT", KtConstantExpression.class);
093    
094        KtNodeType STRING_TEMPLATE    = new KtNodeType("STRING_TEMPLATE", KtStringTemplateExpression.class);
095        KtNodeType LONG_STRING_TEMPLATE_ENTRY = new KtNodeType("LONG_STRING_TEMPLATE_ENTRY", KtBlockStringTemplateEntry.class);
096        KtNodeType SHORT_STRING_TEMPLATE_ENTRY = new KtNodeType("SHORT_STRING_TEMPLATE_ENTRY", KtSimpleNameStringTemplateEntry.class);
097        KtNodeType LITERAL_STRING_TEMPLATE_ENTRY = new KtNodeType("LITERAL_STRING_TEMPLATE_ENTRY", KtLiteralStringTemplateEntry.class);
098        KtNodeType ESCAPE_STRING_TEMPLATE_ENTRY = new KtNodeType("ESCAPE_STRING_TEMPLATE_ENTRY", KtEscapeStringTemplateEntry.class);
099    
100        KtNodeType PARENTHESIZED             = new KtNodeType("PARENTHESIZED", KtParenthesizedExpression.class);
101        KtNodeType RETURN                    = new KtNodeType("RETURN", KtReturnExpression.class);
102        KtNodeType THROW                     = new KtNodeType("THROW", KtThrowExpression.class);
103        KtNodeType CONTINUE                  = new KtNodeType("CONTINUE", KtContinueExpression.class);
104        KtNodeType BREAK                     = new KtNodeType("BREAK", KtBreakExpression.class);
105        KtNodeType IF                        = new KtNodeType("IF", KtIfExpression.class);
106        KtNodeType CONDITION                 = new KtNodeType("CONDITION", KtContainerNode.class);
107        KtNodeType THEN                      = new KtNodeType("THEN", KtContainerNode.class);
108        KtNodeType ELSE                      = new KtNodeType("ELSE", KtContainerNode.class);
109        KtNodeType TRY                       = new KtNodeType("TRY", KtTryExpression.class);
110        KtNodeType CATCH                     = new KtNodeType("CATCH", KtCatchClause.class);
111        KtNodeType FINALLY                   = new KtNodeType("FINALLY", KtFinallySection.class);
112        KtNodeType FOR                       = new KtNodeType("FOR", KtForExpression.class);
113        KtNodeType WHILE                     = new KtNodeType("WHILE", KtWhileExpression.class);
114        KtNodeType DO_WHILE                  = new KtNodeType("DO_WHILE", KtDoWhileExpression.class);
115        KtNodeType LOOP_RANGE                = new KtNodeType("LOOP_RANGE", KtContainerNode.class);
116        KtNodeType BODY                      = new KtNodeType("BODY", KtContainerNode.class);
117        KtNodeType BLOCK                     = new KtNodeType("BLOCK", KtBlockExpression.class);
118        KtNodeType LAMBDA_EXPRESSION         = new KtNodeType("LAMBDA_EXPRESSION", KtLambdaExpression.class);
119        KtNodeType FUNCTION_LITERAL          = new KtNodeType("FUNCTION_LITERAL", KtFunctionLiteral.class);
120        KtNodeType ANNOTATED_EXPRESSION      = new KtNodeType("ANNOTATED_EXPRESSION", KtAnnotatedExpression.class);
121    
122        IElementType REFERENCE_EXPRESSION     = KtStubElementTypes.REFERENCE_EXPRESSION;
123        IElementType ENUM_ENTRY_SUPERCLASS_REFERENCE_EXPRESSION = KtStubElementTypes.ENUM_ENTRY_SUPERCLASS_REFERENCE_EXPRESSION;
124        KtNodeType OPERATION_REFERENCE       = new KtNodeType("OPERATION_REFERENCE", KtOperationReferenceExpression.class);
125        KtNodeType LABEL                     = new KtNodeType("LABEL", KtLabelReferenceExpression.class);
126    
127        KtNodeType LABEL_QUALIFIER           = new KtNodeType("LABEL_QUALIFIER", KtContainerNode.class);
128    
129        KtNodeType THIS_EXPRESSION           = new KtNodeType("THIS_EXPRESSION", KtThisExpression.class);
130        KtNodeType SUPER_EXPRESSION          = new KtNodeType("SUPER_EXPRESSION", KtSuperExpression.class);
131        KtNodeType BINARY_EXPRESSION         = new KtNodeType("BINARY_EXPRESSION", KtBinaryExpression.class);
132        KtNodeType BINARY_WITH_TYPE          = new KtNodeType("BINARY_WITH_TYPE", KtBinaryExpressionWithTypeRHS.class);
133        KtNodeType IS_EXPRESSION = new KtNodeType("IS_EXPRESSION", KtIsExpression.class); // TODO:
134        KtNodeType PREFIX_EXPRESSION         = new KtNodeType("PREFIX_EXPRESSION", KtPrefixExpression.class);
135        KtNodeType POSTFIX_EXPRESSION        = new KtNodeType("POSTFIX_EXPRESSION", KtPostfixExpression.class);
136        KtNodeType LABELED_EXPRESSION         = new KtNodeType("LABELED_EXPRESSION", KtLabeledExpression.class);
137        KtNodeType CALL_EXPRESSION           = new KtNodeType("CALL_EXPRESSION", KtCallExpression.class);
138        KtNodeType ARRAY_ACCESS_EXPRESSION   = new KtNodeType("ARRAY_ACCESS_EXPRESSION", KtArrayAccessExpression.class);
139        KtNodeType INDICES                   = new KtNodeType("INDICES", KtContainerNode.class);
140        IElementType DOT_QUALIFIED_EXPRESSION = KtStubElementTypes.DOT_QUALIFIED_EXPRESSION;
141        KtNodeType CALLABLE_REFERENCE_EXPRESSION = new KtNodeType("CALLABLE_REFERENCE_EXPRESSION", KtCallableReferenceExpression.class);
142        KtNodeType CLASS_LITERAL_EXPRESSION  = new KtNodeType("CLASS_LITERAL_EXPRESSION", KtClassLiteralExpression.class);
143        KtNodeType SAFE_ACCESS_EXPRESSION    = new KtNodeType("SAFE_ACCESS_EXPRESSION", KtSafeQualifiedExpression.class);
144    
145        KtNodeType OBJECT_LITERAL            = new KtNodeType("OBJECT_LITERAL", KtObjectLiteralExpression.class);
146    
147        KtNodeType WHEN                      = new KtNodeType("WHEN", KtWhenExpression.class);
148        KtNodeType WHEN_ENTRY                = new KtNodeType("WHEN_ENTRY", KtWhenEntry.class);
149    
150        KtNodeType WHEN_CONDITION_IN_RANGE   = new KtNodeType("WHEN_CONDITION_IN_RANGE", KtWhenConditionInRange.class);
151        KtNodeType WHEN_CONDITION_IS_PATTERN = new KtNodeType("WHEN_CONDITION_IS_PATTERN", KtWhenConditionIsPattern.class);
152        KtNodeType WHEN_CONDITION_EXPRESSION = new KtNodeType("WHEN_CONDITION_WITH_EXPRESSION", KtWhenConditionWithExpression.class);
153    
154        IElementType PACKAGE_DIRECTIVE = KtStubElementTypes.PACKAGE_DIRECTIVE;
155    
156        IElementType SCRIPT = KtStubElementTypes.SCRIPT;
157    
158        IFileElementType TYPE_CODE_FRAGMENT = new KtTypeCodeFragmentType();
159        IFileElementType EXPRESSION_CODE_FRAGMENT = new KtExpressionCodeFragmentType();
160        IFileElementType BLOCK_CODE_FRAGMENT = new KtBlockCodeFragmentType();
161    }