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.psi;
018
019import com.intellij.lang.ASTNode;
020import com.intellij.psi.PsiElement;
021import com.intellij.psi.util.PsiTreeUtil;
022import org.jetbrains.annotations.NotNull;
023import org.jetbrains.annotations.Nullable;
024import org.jetbrains.jet.JetNodeTypes;
025import org.jetbrains.jet.lexer.JetTokens;
026
027import java.util.Collections;
028import java.util.List;
029
030public class JetPropertyAccessor extends JetDeclarationImpl
031        implements JetDeclarationWithBody, JetModifierListOwner, JetWithExpressionInitializer {
032    public JetPropertyAccessor(@NotNull ASTNode node) {
033        super(node);
034    }
035
036    @Override
037    public void accept(@NotNull JetVisitorVoid visitor) {
038        visitor.visitPropertyAccessor(this);
039    }
040
041    @Override
042    public <R, D> R accept(@NotNull JetVisitor<R, D> visitor, D data) {
043        return visitor.visitPropertyAccessor(this, data);
044    }
045
046    public boolean isSetter() {
047        return findChildByType(JetTokens.SET_KEYWORD) != null;
048    }
049
050    public boolean isGetter() {
051        return findChildByType(JetTokens.GET_KEYWORD) != null;
052    }
053
054    @Nullable
055    public JetParameter getParameter() {
056        JetParameterList parameterList = (JetParameterList) findChildByType(JetNodeTypes.VALUE_PARAMETER_LIST);
057        if (parameterList == null) return null;
058        List<JetParameter> parameters = parameterList.getParameters();
059        if (parameters.isEmpty()) return null;
060        return parameters.get(0);
061    }
062
063    @NotNull
064    @Override
065    public List<JetParameter> getValueParameters() {
066        JetParameter parameter = getParameter();
067        if (parameter == null) {
068            return Collections.emptyList();
069        }
070        return Collections.singletonList(parameter);
071    }
072
073    @Nullable
074    @Override
075    public JetExpression getBodyExpression() {
076        return findChildByClass(JetExpression.class);
077    }
078
079    @Override
080    public boolean hasBlockBody() {
081        return findChildByType(JetTokens.EQ) == null;
082    }
083
084    @Override
085    public boolean hasDeclaredReturnType() {
086        return true;
087    }
088
089    @NotNull
090    @Override
091    public JetElement asElement() {
092        return this;
093    }
094
095    @Nullable
096    public JetTypeReference getReturnTypeReference() {
097        return findChildByClass(JetTypeReference.class);
098    }
099
100    @NotNull
101    public PsiElement getNamePlaceholder() {
102        PsiElement get = findChildByType(JetTokens.GET_KEYWORD);
103        if (get != null) {
104            return get;
105        }
106        return findChildByType(JetTokens.SET_KEYWORD);
107    }
108
109    @Nullable
110    @Override
111    public JetExpression getInitializer() {
112        return PsiTreeUtil.getNextSiblingOfType(findChildByType(JetTokens.EQ), JetExpression.class);
113    }
114}