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.lang.ASTNode;
020 import com.intellij.lang.Language;
021 import com.intellij.lang.PsiBuilder;
022 import com.intellij.lang.PsiBuilderFactory;
023 import com.intellij.lexer.Lexer;
024 import com.intellij.openapi.project.Project;
025 import com.intellij.psi.tree.IElementType;
026 import com.intellij.psi.tree.IErrorCounterReparseableElementType;
027 import com.intellij.psi.tree.IFileElementType;
028 import org.jetbrains.annotations.Nullable;
029 import org.jetbrains.kotlin.idea.KotlinLanguage;
030 import org.jetbrains.kotlin.lexer.KotlinLexer;
031 import org.jetbrains.kotlin.lexer.KtTokens;
032 import org.jetbrains.kotlin.parsing.KotlinParser;
033 import org.jetbrains.kotlin.psi.*;
034 import org.jetbrains.kotlin.psi.stubs.elements.KtStubElementTypes;
035
036 public interface KtNodeTypes {
037 IFileElementType KT_FILE = new IFileElementType(KotlinLanguage.INSTANCE);
038
039 IElementType CLASS = KtStubElementTypes.CLASS;
040 IElementType FUN = KtStubElementTypes.FUNCTION;
041 IElementType PROPERTY = KtStubElementTypes.PROPERTY;
042 IElementType DESTRUCTURING_DECLARATION = new KtNodeType("DESTRUCTURING_DECLARATION", KtDestructuringDeclaration.class);
043 IElementType DESTRUCTURING_DECLARATION_ENTRY = new KtNodeType("DESTRUCTURING_DECLARATION_ENTRY", KtDestructuringDeclarationEntry.class);
044
045 IElementType OBJECT_DECLARATION = KtStubElementTypes.OBJECT_DECLARATION;
046 IElementType TYPEALIAS = KtStubElementTypes.TYPEALIAS;
047
048 IElementType ENUM_ENTRY = KtStubElementTypes.ENUM_ENTRY;
049 IElementType CLASS_INITIALIZER = KtStubElementTypes.CLASS_INITIALIZER;
050 IElementType SCRIPT_INITIALIZER = new KtNodeType("SCRIPT_INITIALIZER", KtScriptInitializer.class);
051 IElementType SECONDARY_CONSTRUCTOR = KtStubElementTypes.SECONDARY_CONSTRUCTOR;
052 IElementType PRIMARY_CONSTRUCTOR = KtStubElementTypes.PRIMARY_CONSTRUCTOR;
053
054 IElementType TYPE_PARAMETER_LIST = KtStubElementTypes.TYPE_PARAMETER_LIST;
055 IElementType TYPE_PARAMETER = KtStubElementTypes.TYPE_PARAMETER;
056 IElementType SUPER_TYPE_LIST = KtStubElementTypes.SUPER_TYPE_LIST;
057 IElementType DELEGATED_SUPER_TYPE_ENTRY = KtStubElementTypes.DELEGATED_SUPER_TYPE_ENTRY;
058 IElementType SUPER_TYPE_CALL_ENTRY = KtStubElementTypes.SUPER_TYPE_CALL_ENTRY;
059 IElementType SUPER_TYPE_ENTRY = KtStubElementTypes.SUPER_TYPE_ENTRY;
060 KtNodeType PROPERTY_DELEGATE = new KtNodeType("PROPERTY_DELEGATE", KtPropertyDelegate.class);
061 IElementType CONSTRUCTOR_CALLEE = KtStubElementTypes.CONSTRUCTOR_CALLEE;
062 IElementType VALUE_PARAMETER_LIST = KtStubElementTypes.VALUE_PARAMETER_LIST;
063 IElementType VALUE_PARAMETER = KtStubElementTypes.VALUE_PARAMETER;
064
065 IElementType CLASS_BODY = KtStubElementTypes.CLASS_BODY;
066 IElementType IMPORT_LIST = KtStubElementTypes.IMPORT_LIST;
067 IElementType FILE_ANNOTATION_LIST = KtStubElementTypes.FILE_ANNOTATION_LIST;
068 IElementType IMPORT_DIRECTIVE = KtStubElementTypes.IMPORT_DIRECTIVE;
069 IElementType MODIFIER_LIST = KtStubElementTypes.MODIFIER_LIST;
070 IElementType ANNOTATION = KtStubElementTypes.ANNOTATION;
071 IElementType ANNOTATION_ENTRY = KtStubElementTypes.ANNOTATION_ENTRY;
072 IElementType ANNOTATION_TARGET = KtStubElementTypes.ANNOTATION_TARGET;
073
074 IElementType TYPE_ARGUMENT_LIST = KtStubElementTypes.TYPE_ARGUMENT_LIST;
075 KtNodeType VALUE_ARGUMENT_LIST = new KtNodeType("VALUE_ARGUMENT_LIST", KtValueArgumentList.class);
076 KtNodeType VALUE_ARGUMENT = new KtNodeType("VALUE_ARGUMENT", KtValueArgument.class);
077 KtNodeType LAMBDA_ARGUMENT = new KtNodeType("LAMBDA_ARGUMENT", KtLambdaArgument.class);
078 KtNodeType VALUE_ARGUMENT_NAME = new KtNodeType("VALUE_ARGUMENT_NAME", KtValueArgumentName.class);
079 IElementType TYPE_REFERENCE = KtStubElementTypes.TYPE_REFERENCE;
080
081 IElementType USER_TYPE = KtStubElementTypes.USER_TYPE;
082 IElementType DYNAMIC_TYPE = KtStubElementTypes.DYNAMIC_TYPE;
083 IElementType FUNCTION_TYPE = KtStubElementTypes.FUNCTION_TYPE;
084 IElementType FUNCTION_TYPE_RECEIVER = KtStubElementTypes.FUNCTION_TYPE_RECEIVER;
085 KtNodeType SELF_TYPE = new KtNodeType("SELF_TYPE", KtSelfType.class);
086 IElementType NULLABLE_TYPE = KtStubElementTypes.NULLABLE_TYPE;
087 IElementType TYPE_PROJECTION = KtStubElementTypes.TYPE_PROJECTION;
088
089 // TODO: review
090 IElementType PROPERTY_ACCESSOR = KtStubElementTypes.PROPERTY_ACCESSOR;
091 IElementType INITIALIZER_LIST = KtStubElementTypes.INITIALIZER_LIST;
092 IElementType TYPE_CONSTRAINT_LIST = KtStubElementTypes.TYPE_CONSTRAINT_LIST;
093 IElementType TYPE_CONSTRAINT = KtStubElementTypes.TYPE_CONSTRAINT;
094
095 IElementType CONSTRUCTOR_DELEGATION_CALL = new KtNodeType.KtLeftBoundNodeType("CONSTRUCTOR_DELEGATION_CALL", KtConstructorDelegationCall.class);
096 KtNodeType CONSTRUCTOR_DELEGATION_REFERENCE = new KtNodeType.KtLeftBoundNodeType("CONSTRUCTOR_DELEGATION_REFERENCE", KtConstructorDelegationReferenceExpression.class);
097
098 // TODO: Not sure if we need separate NT for each kind of constants
099 KtNodeType NULL = new KtNodeType("NULL", KtConstantExpression.class);
100 KtNodeType BOOLEAN_CONSTANT = new KtNodeType("BOOLEAN_CONSTANT", KtConstantExpression.class);
101 KtNodeType FLOAT_CONSTANT = new KtNodeType("FLOAT_CONSTANT", KtConstantExpression.class);
102 KtNodeType CHARACTER_CONSTANT = new KtNodeType("CHARACTER_CONSTANT", KtConstantExpression.class);
103 KtNodeType INTEGER_CONSTANT = new KtNodeType("INTEGER_CONSTANT", KtConstantExpression.class);
104
105 KtNodeType STRING_TEMPLATE = new KtNodeType("STRING_TEMPLATE", KtStringTemplateExpression.class);
106 KtNodeType LONG_STRING_TEMPLATE_ENTRY = new KtNodeType("LONG_STRING_TEMPLATE_ENTRY", KtBlockStringTemplateEntry.class);
107 KtNodeType SHORT_STRING_TEMPLATE_ENTRY = new KtNodeType("SHORT_STRING_TEMPLATE_ENTRY", KtSimpleNameStringTemplateEntry.class);
108 KtNodeType LITERAL_STRING_TEMPLATE_ENTRY = new KtNodeType("LITERAL_STRING_TEMPLATE_ENTRY", KtLiteralStringTemplateEntry.class);
109 KtNodeType ESCAPE_STRING_TEMPLATE_ENTRY = new KtNodeType("ESCAPE_STRING_TEMPLATE_ENTRY", KtEscapeStringTemplateEntry.class);
110
111 KtNodeType PARENTHESIZED = new KtNodeType("PARENTHESIZED", KtParenthesizedExpression.class);
112 KtNodeType RETURN = new KtNodeType("RETURN", KtReturnExpression.class);
113 KtNodeType THROW = new KtNodeType("THROW", KtThrowExpression.class);
114 KtNodeType CONTINUE = new KtNodeType("CONTINUE", KtContinueExpression.class);
115 KtNodeType BREAK = new KtNodeType("BREAK", KtBreakExpression.class);
116 KtNodeType IF = new KtNodeType("IF", KtIfExpression.class);
117 KtNodeType CONDITION = new KtNodeType("CONDITION", KtContainerNode.class);
118 KtNodeType THEN = new KtNodeType("THEN", KtContainerNodeForControlStructureBody.class);
119 KtNodeType ELSE = new KtNodeType("ELSE", KtContainerNodeForControlStructureBody.class);
120 KtNodeType TRY = new KtNodeType("TRY", KtTryExpression.class);
121 KtNodeType CATCH = new KtNodeType("CATCH", KtCatchClause.class);
122 KtNodeType FINALLY = new KtNodeType("FINALLY", KtFinallySection.class);
123 KtNodeType FOR = new KtNodeType("FOR", KtForExpression.class);
124 KtNodeType WHILE = new KtNodeType("WHILE", KtWhileExpression.class);
125 KtNodeType DO_WHILE = new KtNodeType("DO_WHILE", KtDoWhileExpression.class);
126 KtNodeType LOOP_RANGE = new KtNodeType("LOOP_RANGE", KtContainerNode.class);
127 KtNodeType BODY = new KtNodeType("BODY", KtContainerNodeForControlStructureBody.class);
128 KtNodeType BLOCK = new KtNodeType("BLOCK", KtBlockExpression.class);
129
130 IElementType LAMBDA_EXPRESSION = new IErrorCounterReparseableElementType("LAMBDA_EXPRESSION", KotlinLanguage.INSTANCE) {
131 @Override
132 public ASTNode parseContents(ASTNode chameleon) {
133 Project project = chameleon.getPsi().getProject();
134 PsiBuilder builder = PsiBuilderFactory.getInstance().createBuilder(project, chameleon, null, KotlinLanguage.INSTANCE, chameleon.getChars());
135 return KotlinParser.parseLambdaExpression(builder).getFirstChildNode();
136 }
137
138 @Nullable
139 @Override
140 public ASTNode createNode(CharSequence text) {
141 return new KtLambdaExpression(text);
142 }
143
144 @Override
145 public int getErrorsCount(CharSequence seq, Language fileLanguage, Project project) {
146 Lexer lexer = new KotlinLexer();
147
148 lexer.start(seq);
149 if (lexer.getTokenType() != KtTokens.LBRACE) return IErrorCounterReparseableElementType.FATAL_ERROR;
150 lexer.advance();
151 int balance = 1;
152 while (true) {
153 IElementType type = lexer.getTokenType();
154 if (type == null) break;
155 if (balance == 0) {
156 return IErrorCounterReparseableElementType.FATAL_ERROR;
157 }
158 if (type == KtTokens.LBRACE) {
159 balance++;
160 }
161 else if (type == KtTokens.RBRACE) {
162 balance--;
163 }
164 lexer.advance();
165 }
166 return balance;
167 }
168 };
169
170
171 KtNodeType FUNCTION_LITERAL = new KtNodeType("FUNCTION_LITERAL", KtFunctionLiteral.class);
172 KtNodeType ANNOTATED_EXPRESSION = new KtNodeType("ANNOTATED_EXPRESSION", KtAnnotatedExpression.class);
173
174 IElementType REFERENCE_EXPRESSION = KtStubElementTypes.REFERENCE_EXPRESSION;
175 IElementType ENUM_ENTRY_SUPERCLASS_REFERENCE_EXPRESSION = KtStubElementTypes.ENUM_ENTRY_SUPERCLASS_REFERENCE_EXPRESSION;
176 KtNodeType OPERATION_REFERENCE = new KtNodeType("OPERATION_REFERENCE", KtOperationReferenceExpression.class);
177 KtNodeType LABEL = new KtNodeType("LABEL", KtLabelReferenceExpression.class);
178
179 KtNodeType LABEL_QUALIFIER = new KtNodeType("LABEL_QUALIFIER", KtContainerNode.class);
180
181 KtNodeType THIS_EXPRESSION = new KtNodeType("THIS_EXPRESSION", KtThisExpression.class);
182 KtNodeType SUPER_EXPRESSION = new KtNodeType("SUPER_EXPRESSION", KtSuperExpression.class);
183 KtNodeType BINARY_EXPRESSION = new KtNodeType("BINARY_EXPRESSION", KtBinaryExpression.class);
184 KtNodeType BINARY_WITH_TYPE = new KtNodeType("BINARY_WITH_TYPE", KtBinaryExpressionWithTypeRHS.class);
185 KtNodeType IS_EXPRESSION = new KtNodeType("IS_EXPRESSION", KtIsExpression.class); // TODO:
186 KtNodeType PREFIX_EXPRESSION = new KtNodeType("PREFIX_EXPRESSION", KtPrefixExpression.class);
187 KtNodeType POSTFIX_EXPRESSION = new KtNodeType("POSTFIX_EXPRESSION", KtPostfixExpression.class);
188 KtNodeType LABELED_EXPRESSION = new KtNodeType("LABELED_EXPRESSION", KtLabeledExpression.class);
189 KtNodeType CALL_EXPRESSION = new KtNodeType("CALL_EXPRESSION", KtCallExpression.class);
190 KtNodeType ARRAY_ACCESS_EXPRESSION = new KtNodeType("ARRAY_ACCESS_EXPRESSION", KtArrayAccessExpression.class);
191 KtNodeType INDICES = new KtNodeType("INDICES", KtContainerNode.class);
192 IElementType DOT_QUALIFIED_EXPRESSION = KtStubElementTypes.DOT_QUALIFIED_EXPRESSION;
193 KtNodeType CALLABLE_REFERENCE_EXPRESSION = new KtNodeType("CALLABLE_REFERENCE_EXPRESSION", KtCallableReferenceExpression.class);
194 KtNodeType CLASS_LITERAL_EXPRESSION = new KtNodeType("CLASS_LITERAL_EXPRESSION", KtClassLiteralExpression.class);
195 KtNodeType SAFE_ACCESS_EXPRESSION = new KtNodeType("SAFE_ACCESS_EXPRESSION", KtSafeQualifiedExpression.class);
196
197 KtNodeType OBJECT_LITERAL = new KtNodeType("OBJECT_LITERAL", KtObjectLiteralExpression.class);
198
199 KtNodeType WHEN = new KtNodeType("WHEN", KtWhenExpression.class);
200 KtNodeType WHEN_ENTRY = new KtNodeType("WHEN_ENTRY", KtWhenEntry.class);
201
202 KtNodeType WHEN_CONDITION_IN_RANGE = new KtNodeType("WHEN_CONDITION_IN_RANGE", KtWhenConditionInRange.class);
203 KtNodeType WHEN_CONDITION_IS_PATTERN = new KtNodeType("WHEN_CONDITION_IS_PATTERN", KtWhenConditionIsPattern.class);
204 KtNodeType WHEN_CONDITION_EXPRESSION = new KtNodeType("WHEN_CONDITION_WITH_EXPRESSION", KtWhenConditionWithExpression.class);
205
206 IElementType PACKAGE_DIRECTIVE = KtStubElementTypes.PACKAGE_DIRECTIVE;
207
208 IElementType SCRIPT = KtStubElementTypes.SCRIPT;
209
210 IFileElementType TYPE_CODE_FRAGMENT = new KtTypeCodeFragmentType();
211 IFileElementType EXPRESSION_CODE_FRAGMENT = new KtExpressionCodeFragmentType();
212 IFileElementType BLOCK_CODE_FRAGMENT = new KtBlockCodeFragmentType();
213 }