| /* |
| * Copyright 2000-2013 JetBrains s.r.o. |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package org.jetbrains.plugins.groovy.lang.parser.parsing.types; |
| |
| import com.intellij.lang.PsiBuilder; |
| import org.jetbrains.plugins.groovy.GroovyBundle; |
| import org.jetbrains.plugins.groovy.lang.lexer.TokenSets; |
| import org.jetbrains.plugins.groovy.lang.parser.GroovyElementTypes; |
| import org.jetbrains.plugins.groovy.lang.parser.parsing.util.ParserUtils; |
| |
| import static org.jetbrains.plugins.groovy.lang.parser.parsing.statements.typeDefinitions.ReferenceElement.ReferenceElementResult; |
| import static org.jetbrains.plugins.groovy.lang.parser.parsing.statements.typeDefinitions.ReferenceElement.ReferenceElementResult.FAIL; |
| import static org.jetbrains.plugins.groovy.lang.parser.parsing.statements.typeDefinitions.ReferenceElement.ReferenceElementResult.REF_WITH_TYPE_PARAMS; |
| import static org.jetbrains.plugins.groovy.lang.parser.parsing.statements.typeDefinitions.ReferenceElement.parseReferenceElement; |
| |
| /** |
| * @autor: Dmitry.Krasilschikov |
| * @date: 16.03.2007 |
| */ |
| public class TypeSpec implements GroovyElementTypes { |
| public static ReferenceElementResult parse(PsiBuilder builder) { |
| return parse(builder, false, true); //allow lower and upper case first letter |
| } |
| |
| public static ReferenceElementResult parse(PsiBuilder builder, boolean isUpper, final boolean expressionPossible) { |
| if (TokenSets.BUILT_IN_TYPES.contains(builder.getTokenType())) { |
| return parseBuiltInType(builder); |
| } |
| if (builder.getTokenType() == mIDENT) { |
| return parseClassType(builder, isUpper, expressionPossible); |
| } |
| return FAIL; |
| } |
| |
| /** |
| * For built-in types |
| * |
| * @param builder |
| * @return |
| */ |
| public static ReferenceElementResult parseBuiltInType(PsiBuilder builder) { |
| PsiBuilder.Marker arrMarker = builder.mark(); |
| ParserUtils.eatElement(builder, BUILT_IN_TYPE); |
| if (mLBRACK.equals(builder.getTokenType())) { |
| declarationBracketsParse(builder, arrMarker); |
| } else { |
| arrMarker.drop(); |
| } |
| return REF_WITH_TYPE_PARAMS; |
| } |
| |
| |
| /** |
| * For array definitions |
| * |
| * @param builder |
| * @param marker |
| */ |
| private static void declarationBracketsParse(PsiBuilder builder, PsiBuilder.Marker marker) { |
| ParserUtils.getToken(builder, mLBRACK); |
| ParserUtils.getToken(builder, mRBRACK, GroovyBundle.message("rbrack.expected")); |
| PsiBuilder.Marker newMarker = marker.precede(); |
| marker.done(ARRAY_TYPE); |
| if (mLBRACK.equals(builder.getTokenType())) { |
| declarationBracketsParse(builder, newMarker); |
| } else { |
| newMarker.drop(); |
| } |
| } |
| |
| private static ReferenceElementResult parseClassType(PsiBuilder builder, boolean isUpper, final boolean expressionPossible) { |
| PsiBuilder.Marker arrMarker = builder.mark(); |
| PsiBuilder.Marker typeElementMarker = builder.mark(); |
| |
| final ReferenceElementResult result = parseReferenceElement(builder, isUpper, expressionPossible); |
| if (result == FAIL) { |
| typeElementMarker.drop(); |
| arrMarker.rollbackTo(); |
| return FAIL; |
| } |
| |
| typeElementMarker.done(CLASS_TYPE_ELEMENT); |
| |
| if (ParserUtils.lookAhead(builder, mLBRACK, mRBRACK)) { |
| declarationBracketsParse(builder, arrMarker); |
| return REF_WITH_TYPE_PARAMS; |
| } else { |
| arrMarker.drop(); |
| return result; |
| } |
| } |
| |
| /** |
| * ******************************************************************************************************** |
| * **************** Strict type specification parsing |
| * ******************************************************************************************************** |
| */ |
| |
| /** |
| * Strict parsing |
| * |
| * @param builder |
| * @return |
| */ |
| public static ReferenceElementResult parseStrict(PsiBuilder builder, boolean expressionPossible) { |
| if (TokenSets.BUILT_IN_TYPES.contains(builder.getTokenType())) { |
| return parseBuiltInTypeStrict(builder); |
| } |
| else if (builder.getTokenType() == mIDENT) { |
| return parseClassOrInterfaceTypeStrict(builder, expressionPossible); |
| } |
| return FAIL; |
| } |
| |
| |
| /** |
| * For built-in types |
| * |
| * @param builder |
| * @return |
| */ |
| private static ReferenceElementResult parseBuiltInTypeStrict(PsiBuilder builder) { |
| PsiBuilder.Marker arrMarker = builder.mark(); |
| ParserUtils.eatElement(builder, BUILT_IN_TYPE); |
| if (mLBRACK.equals(builder.getTokenType())) { |
| return declarationBracketsParseStrict(builder, arrMarker) ? REF_WITH_TYPE_PARAMS : FAIL; |
| } else { |
| arrMarker.drop(); |
| return REF_WITH_TYPE_PARAMS; |
| } |
| } |
| |
| |
| /** |
| * Strict parsing of array definitions |
| * |
| * @param builder |
| * @param marker |
| */ |
| private static boolean declarationBracketsParseStrict(PsiBuilder builder, PsiBuilder.Marker marker) { |
| ParserUtils.getToken(builder, mLBRACK); |
| if (!ParserUtils.getToken(builder, mRBRACK, GroovyBundle.message("rbrack.expected"))) { |
| marker.rollbackTo(); |
| return false; |
| } |
| PsiBuilder.Marker newMarker = marker.precede(); |
| marker.done(ARRAY_TYPE); |
| if (mLBRACK.equals(builder.getTokenType())) { |
| return declarationBracketsParseStrict(builder, newMarker); |
| } else { |
| newMarker.drop(); |
| return true; |
| } |
| } |
| |
| |
| /** |
| * Strict class type parsing |
| * |
| * @param builder |
| * @return |
| */ |
| private static ReferenceElementResult parseClassOrInterfaceTypeStrict(PsiBuilder builder, boolean expressionPossible) { |
| PsiBuilder.Marker arrMarker = builder.mark(); |
| PsiBuilder.Marker typeElementMarker = builder.mark(); |
| |
| final ReferenceElementResult result = parseReferenceElement(builder, false, expressionPossible); |
| if (result == FAIL) { |
| typeElementMarker.drop(); |
| arrMarker.rollbackTo(); |
| return result; |
| } |
| |
| typeElementMarker.done(CLASS_TYPE_ELEMENT); |
| |
| if (mLBRACK.equals(builder.getTokenType())) { |
| return declarationBracketsParseStrict(builder, arrMarker) ? REF_WITH_TYPE_PARAMS : FAIL; |
| } |
| else { |
| arrMarker.drop(); |
| return result; |
| } |
| } |
| } |