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 MULTI_VARIABLE_DECLARATION  = new KtNodeType("MULTI_VARIABLE_DECLARATION", KtMultiDeclaration.class);
032        IElementType MULTI_VARIABLE_DECLARATION_ENTRY = new KtNodeType("MULTI_VARIABLE_DECLARATION_ENTRY", KtMultiDeclarationEntry.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 DELEGATION_SPECIFIER_LIST           = KtStubElementTypes.DELEGATION_SPECIFIER_LIST;
046        IElementType DELEGATOR_BY                        = KtStubElementTypes.DELEGATOR_BY;
047        IElementType DELEGATOR_SUPER_CALL                = KtStubElementTypes.DELEGATOR_SUPER_CALL;
048        IElementType DELEGATOR_SUPER_CLASS               = KtStubElementTypes.DELEGATOR_SUPER_CLASS;
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 FUNCTION_LITERAL_ARGUMENT           = new KtNodeType("FUNCTION_LITERAL_ARGUMENT", KtFunctionLiteralArgument.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("CONSTRUCTOR_DELEGATION_CALL", KtConstructorDelegationCall.class);
085        KtNodeType CONSTRUCTOR_DELEGATION_REFERENCE =
086                new KtNodeType("CONSTRUCTOR_DELEGATION_REFERENCE", KtConstructorDelegationReferenceExpression.class);
087    
088        // TODO: Not sure if we need separate NT for each kind of constants
089        KtNodeType NULL               = new KtNodeType("NULL", KtConstantExpression.class);
090        KtNodeType BOOLEAN_CONSTANT   = new KtNodeType("BOOLEAN_CONSTANT", KtConstantExpression.class);
091        KtNodeType FLOAT_CONSTANT     = new KtNodeType("FLOAT_CONSTANT", KtConstantExpression.class);
092        KtNodeType CHARACTER_CONSTANT = new KtNodeType("CHARACTER_CONSTANT", KtConstantExpression.class);
093        KtNodeType INTEGER_CONSTANT   = new KtNodeType("INTEGER_CONSTANT", KtConstantExpression.class);
094    
095        KtNodeType STRING_TEMPLATE    = new KtNodeType("STRING_TEMPLATE", KtStringTemplateExpression.class);
096        KtNodeType LONG_STRING_TEMPLATE_ENTRY = new KtNodeType("LONG_STRING_TEMPLATE_ENTRY", KtBlockStringTemplateEntry.class);
097        KtNodeType SHORT_STRING_TEMPLATE_ENTRY = new KtNodeType("SHORT_STRING_TEMPLATE_ENTRY", KtSimpleNameStringTemplateEntry.class);
098        KtNodeType LITERAL_STRING_TEMPLATE_ENTRY = new KtNodeType("LITERAL_STRING_TEMPLATE_ENTRY", KtLiteralStringTemplateEntry.class);
099        KtNodeType ESCAPE_STRING_TEMPLATE_ENTRY = new KtNodeType("ESCAPE_STRING_TEMPLATE_ENTRY", KtEscapeStringTemplateEntry.class);
100    
101        KtNodeType PARENTHESIZED             = new KtNodeType("PARENTHESIZED", KtParenthesizedExpression.class);
102        KtNodeType RETURN                    = new KtNodeType("RETURN", KtReturnExpression.class);
103        KtNodeType THROW                     = new KtNodeType("THROW", KtThrowExpression.class);
104        KtNodeType CONTINUE                  = new KtNodeType("CONTINUE", KtContinueExpression.class);
105        KtNodeType BREAK                     = new KtNodeType("BREAK", KtBreakExpression.class);
106        KtNodeType IF                        = new KtNodeType("IF", KtIfExpression.class);
107        KtNodeType CONDITION                 = new KtNodeType("CONDITION", KtContainerNode.class);
108        KtNodeType THEN                      = new KtNodeType("THEN", KtContainerNode.class);
109        KtNodeType ELSE                      = new KtNodeType("ELSE", KtContainerNode.class);
110        KtNodeType TRY                       = new KtNodeType("TRY", KtTryExpression.class);
111        KtNodeType CATCH                     = new KtNodeType("CATCH", KtCatchClause.class);
112        KtNodeType FINALLY                   = new KtNodeType("FINALLY", KtFinallySection.class);
113        KtNodeType FOR                       = new KtNodeType("FOR", KtForExpression.class);
114        KtNodeType WHILE                     = new KtNodeType("WHILE", KtWhileExpression.class);
115        KtNodeType DO_WHILE                  = new KtNodeType("DO_WHILE", KtDoWhileExpression.class);
116        KtNodeType LOOP_RANGE                = new KtNodeType("LOOP_RANGE", KtContainerNode.class);
117        KtNodeType BODY                      = new KtNodeType("BODY", KtContainerNode.class);
118        KtNodeType BLOCK                     = new KtNodeType("BLOCK", KtBlockExpression.class);
119        KtNodeType FUNCTION_LITERAL_EXPRESSION = new KtNodeType("FUNCTION_LITERAL_EXPRESSION", KtFunctionLiteralExpression.class);
120        KtNodeType FUNCTION_LITERAL          = new KtNodeType("FUNCTION_LITERAL", KtFunctionLiteral.class);
121        KtNodeType ANNOTATED_EXPRESSION      = new KtNodeType("ANNOTATED_EXPRESSION", KtAnnotatedExpression.class);
122    
123        IElementType REFERENCE_EXPRESSION     = KtStubElementTypes.REFERENCE_EXPRESSION;
124        IElementType ENUM_ENTRY_SUPERCLASS_REFERENCE_EXPRESSION = KtStubElementTypes.ENUM_ENTRY_SUPERCLASS_REFERENCE_EXPRESSION;
125        KtNodeType OPERATION_REFERENCE       = new KtNodeType("OPERATION_REFERENCE", KtOperationReferenceExpression.class);
126        KtNodeType LABEL                     = new KtNodeType("LABEL", KtLabelReferenceExpression.class);
127    
128        KtNodeType LABEL_QUALIFIER           = new KtNodeType("LABEL_QUALIFIER", KtContainerNode.class);
129    
130        KtNodeType THIS_EXPRESSION           = new KtNodeType("THIS_EXPRESSION", KtThisExpression.class);
131        KtNodeType SUPER_EXPRESSION          = new KtNodeType("SUPER_EXPRESSION", KtSuperExpression.class);
132        KtNodeType BINARY_EXPRESSION         = new KtNodeType("BINARY_EXPRESSION", KtBinaryExpression.class);
133        KtNodeType BINARY_WITH_TYPE          = new KtNodeType("BINARY_WITH_TYPE", KtBinaryExpressionWithTypeRHS.class);
134        KtNodeType IS_EXPRESSION = new KtNodeType("IS_EXPRESSION", KtIsExpression.class); // TODO:
135        KtNodeType PREFIX_EXPRESSION         = new KtNodeType("PREFIX_EXPRESSION", KtPrefixExpression.class);
136        KtNodeType POSTFIX_EXPRESSION        = new KtNodeType("POSTFIX_EXPRESSION", KtPostfixExpression.class);
137        KtNodeType LABELED_EXPRESSION         = new KtNodeType("LABELED_EXPRESSION", KtLabeledExpression.class);
138        KtNodeType CALL_EXPRESSION           = new KtNodeType("CALL_EXPRESSION", KtCallExpression.class);
139        KtNodeType ARRAY_ACCESS_EXPRESSION   = new KtNodeType("ARRAY_ACCESS_EXPRESSION", KtArrayAccessExpression.class);
140        KtNodeType INDICES                   = new KtNodeType("INDICES", KtContainerNode.class);
141        IElementType DOT_QUALIFIED_EXPRESSION = KtStubElementTypes.DOT_QUALIFIED_EXPRESSION;
142        KtNodeType CALLABLE_REFERENCE_EXPRESSION = new KtNodeType("CALLABLE_REFERENCE_EXPRESSION", KtCallableReferenceExpression.class);
143        KtNodeType CLASS_LITERAL_EXPRESSION  = new KtNodeType("CLASS_LITERAL_EXPRESSION", KtClassLiteralExpression.class);
144        KtNodeType SAFE_ACCESS_EXPRESSION    = new KtNodeType("SAFE_ACCESS_EXPRESSION", KtSafeQualifiedExpression.class);
145    
146        KtNodeType OBJECT_LITERAL            = new KtNodeType("OBJECT_LITERAL", KtObjectLiteralExpression.class);
147        KtNodeType ROOT_PACKAGE = new KtNodeType("ROOT_PACKAGE", KtRootPackageExpression.class);
148    
149        KtNodeType WHEN                      = new KtNodeType("WHEN", KtWhenExpression.class);
150        KtNodeType WHEN_ENTRY                = new KtNodeType("WHEN_ENTRY", KtWhenEntry.class);
151    
152        KtNodeType WHEN_CONDITION_IN_RANGE   = new KtNodeType("WHEN_CONDITION_IN_RANGE", KtWhenConditionInRange.class);
153        KtNodeType WHEN_CONDITION_IS_PATTERN = new KtNodeType("WHEN_CONDITION_IS_PATTERN", KtWhenConditionIsPattern.class);
154        KtNodeType WHEN_CONDITION_EXPRESSION = new KtNodeType("WHEN_CONDITION_WITH_EXPRESSION", KtWhenConditionWithExpression.class);
155    
156        IElementType PACKAGE_DIRECTIVE = KtStubElementTypes.PACKAGE_DIRECTIVE;
157    
158        IElementType SCRIPT = KtStubElementTypes.SCRIPT;
159    
160        IFileElementType TYPE_CODE_FRAGMENT = new KtTypeCodeFragmentType();
161        IFileElementType EXPRESSION_CODE_FRAGMENT = new KtExpressionCodeFragmentType();
162        IFileElementType BLOCK_CODE_FRAGMENT = new KtBlockCodeFragmentType();
163    }