001    /*
002     * Copyright 2010-2016 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.js.translate.initializer;
018    
019    import com.google.dart.compiler.backend.js.ast.JsStatement;
020    import org.jetbrains.annotations.NotNull;
021    import org.jetbrains.kotlin.js.translate.context.TranslationContext;
022    import org.jetbrains.kotlin.js.translate.general.Translation;
023    import org.jetbrains.kotlin.js.translate.general.TranslatorVisitor;
024    import org.jetbrains.kotlin.js.translate.utils.JsAstUtils;
025    import org.jetbrains.kotlin.psi.*;
026    
027    import static org.jetbrains.kotlin.js.translate.general.Translation.translateAsStatementAndMergeInBlockIfNeeded;
028    import static org.jetbrains.kotlin.js.translate.initializer.InitializerUtils.generateInitializerForDelegate;
029    import static org.jetbrains.kotlin.js.translate.initializer.InitializerUtils.generateInitializerForProperty;
030    import static org.jetbrains.kotlin.js.translate.utils.BindingUtils.getPropertyDescriptor;
031    
032    public final class InitializerVisitor extends TranslatorVisitor<Void> {
033        @Override
034        protected Void emptyResult(@NotNull TranslationContext context) {
035            return null;
036        }
037    
038        @Override
039        public final Void visitProperty(@NotNull KtProperty property, @NotNull TranslationContext context) {
040            KtExpression initializer = property.getInitializer();
041            if (initializer != null) {
042                JsStatement statement = generateInitializerForProperty(context, getPropertyDescriptor(context.bindingContext(), property),
043                                                                       Translation.translateAsExpression(initializer, context));
044                if (!JsAstUtils.isEmptyStatement(statement)) {
045                    context.addStatementsToCurrentBlock(JsAstUtils.flattenStatement(statement));
046                }
047            }
048    
049            JsStatement delegate = generateInitializerForDelegate(context, property);
050            if (delegate != null) {
051                context.addStatementToCurrentBlock(delegate);
052            }
053    
054            return null;
055        }
056    
057        @Override
058        public Void visitAnonymousInitializer(@NotNull KtAnonymousInitializer initializer, @NotNull TranslationContext context) {
059            KtExpression initializerBody = initializer.getBody();
060            if (initializerBody != null) {
061                context.addStatementsToCurrentBlock(JsAstUtils.flattenStatement(
062                        translateAsStatementAndMergeInBlockIfNeeded(initializerBody, context)));
063            }
064            return null;
065        }
066    
067        @Override
068        // Not interested in other types of declarations, they do not contain initializers.
069        public Void visitDeclaration(@NotNull KtDeclaration expression, @NotNull TranslationContext context) {
070            return null;
071        }
072    
073        @Override
074        public Void visitSecondaryConstructor(@NotNull KtSecondaryConstructor constructor, TranslationContext data) {
075            return null;
076        }
077    }