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.tree.TokenSet;
022import com.intellij.psi.util.PsiTreeUtil;
023import org.jetbrains.annotations.NotNull;
024import org.jetbrains.annotations.Nullable;
025import org.jetbrains.jet.JetNodeTypes;
026import org.jetbrains.jet.lexer.JetTokens;
027
028import java.util.List;
029
030public class JetClassBody extends JetElementImpl implements JetDeclarationContainer {
031    public JetClassBody(@NotNull ASTNode node) {
032        super(node);
033    }
034
035    @Override
036    @NotNull
037    public List<JetDeclaration> getDeclarations() {
038        return PsiTreeUtil.getChildrenOfTypeAsList(this, JetDeclaration.class);
039    }
040
041    @Override
042    public void accept(@NotNull JetVisitorVoid visitor) {
043        visitor.visitClassBody(this);
044    }
045
046    @Override
047    public <R, D> R accept(@NotNull JetVisitor<R, D> visitor, D data) {
048        return visitor.visitClassBody(this, data);
049    }
050
051    @NotNull
052    public List<JetClassInitializer> getAnonymousInitializers() {
053        return findChildrenByType(JetNodeTypes.ANONYMOUS_INITIALIZER);
054    }
055
056    @NotNull
057    public List<JetProperty> getProperties() {
058        return findChildrenByType(JetNodeTypes.PROPERTY);
059    }
060
061    @Nullable
062    public JetClassObject getClassObject() {
063        return (JetClassObject) findChildByType(JetNodeTypes.CLASS_OBJECT);
064    }
065
066    @Nullable
067    public PsiElement getRBrace() {
068        ASTNode[] children = getNode().getChildren(TokenSet.create(JetTokens.RBRACE));
069        return children.length == 1 ? children[0].getPsi() : null;
070    }
071
072    @Nullable
073    public PsiElement getLBrace() {
074        ASTNode[] children = getNode().getChildren(TokenSet.create(JetTokens.LBRACE));
075        return children.length == 1 ? children[0].getPsi() : null;
076    }
077}