001/*
002 * Copyright 2010-2013 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
017package org.jetbrains.jet.lang.resolve.java.kt;
018
019import com.intellij.psi.PsiAnnotation;
020import com.intellij.psi.PsiMethod;
021import org.jetbrains.annotations.NotNull;
022import org.jetbrains.annotations.Nullable;
023import org.jetbrains.jet.lang.resolve.java.JvmStdlibNames;
024import org.jetbrains.jet.lang.resolve.java.resolver.JavaAnnotationResolver;
025
026public class JetMethodAnnotation extends PsiAnnotationWithFlags {
027    private static final JetMethodAnnotation NULL_ANNOTATION = new JetMethodAnnotation(null);
028    static {
029        NULL_ANNOTATION.checkInitialized();
030    }
031
032    private String typeParameters;
033    private String returnType;
034    private String propertyType;
035
036    private JetMethodAnnotation(@Nullable PsiAnnotation psiAnnotation) {
037        super(psiAnnotation);
038    }
039
040    @Override
041    protected void initialize() {
042        super.initialize();
043        typeParameters = getStringAttribute(JvmStdlibNames.JET_METHOD_TYPE_PARAMETERS_FIELD, "");
044        returnType = getStringAttribute(JvmStdlibNames.JET_METHOD_RETURN_TYPE_FIELD, "");
045        propertyType = getStringAttribute(JvmStdlibNames.JET_METHOD_PROPERTY_TYPE_FIELD, "");
046    }
047
048    public int kind() {
049        return flags() & JvmStdlibNames.FLAG_METHOD_KIND_MASK;
050    }
051
052    @NotNull
053    public String typeParameters() {
054        checkInitialized();
055        return typeParameters;
056    }
057
058    @NotNull
059    public String returnType() {
060        checkInitialized();
061        return returnType;
062    }
063
064    @NotNull
065    public String propertyType() {
066        checkInitialized();
067        return propertyType;
068    }
069
070    public static JetMethodAnnotation get(PsiMethod psiMethod) {
071        PsiAnnotation annotation =
072                JavaAnnotationResolver.findOwnAnnotation(psiMethod, JvmStdlibNames.JET_METHOD.getFqName().asString());
073        return annotation != null ? new JetMethodAnnotation(annotation) : NULL_ANNOTATION;
074    }
075}