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.declaration;
018
019 import com.google.dart.compiler.backend.js.ast.*;
020 import com.intellij.util.SmartList;
021 import org.jetbrains.annotations.NotNull;
022 import org.jetbrains.annotations.Nullable;
023 import org.jetbrains.kotlin.descriptors.*;
024 import org.jetbrains.kotlin.js.translate.context.TranslationContext;
025 import org.jetbrains.kotlin.js.translate.general.Translation;
026 import org.jetbrains.kotlin.js.translate.general.TranslatorVisitor;
027 import org.jetbrains.kotlin.js.translate.initializer.ClassInitializerTranslator;
028 import org.jetbrains.kotlin.js.translate.utils.BindingUtils;
029 import org.jetbrains.kotlin.js.translate.utils.TranslationUtils;
030 import org.jetbrains.kotlin.psi.*;
031 import org.jetbrains.kotlin.types.KotlinType;
032
033 import java.util.ArrayList;
034 import java.util.List;
035
036 import static org.jetbrains.kotlin.js.translate.utils.BindingUtils.getClassDescriptor;
037 import static org.jetbrains.kotlin.js.translate.utils.BindingUtils.getFunctionDescriptor;
038 import static org.jetbrains.kotlin.js.translate.utils.JsDescriptorUtils.getSupertypesWithoutFakes;
039
040 public class DeclarationBodyVisitor extends TranslatorVisitor<Void> {
041 protected final List<JsPropertyInitializer> result;
042 private final List<JsPropertyInitializer> staticResult;
043 private final List<JsPropertyInitializer> enumEntryList = new SmartList<JsPropertyInitializer>();
044
045 @NotNull
046 private final JsScope scope;
047
048 @Nullable
049 private List<JsStatement> initializerStatements;
050
051 public DeclarationBodyVisitor(
052 @NotNull List<JsPropertyInitializer> result, @NotNull List<JsPropertyInitializer> staticResult,
053 @NotNull JsScope scope
054 ) {
055 this.result = result;
056 this.staticResult = staticResult;
057 this.scope = scope;
058 }
059
060 @NotNull
061 public List<JsPropertyInitializer> getResult() {
062 return result;
063 }
064
065 public List<JsPropertyInitializer> getEnumEntryList() {
066 return enumEntryList;
067 }
068
069 @Override
070 protected Void emptyResult(@NotNull TranslationContext context) {
071 return null;
072 }
073
074 @Override
075 public Void visitClassOrObject(@NotNull KtClassOrObject declaration, TranslationContext context) {
076 staticResult.addAll(ClassTranslator.translate(declaration, context).getProperties());
077
078 if (declaration instanceof KtObjectDeclaration) {
079 KtObjectDeclaration objectDeclaration = (KtObjectDeclaration) declaration;
080 if (objectDeclaration.isCompanion()) {
081 DeclarationDescriptor descriptor = BindingUtils.getDescriptorForElement(context.bindingContext(), declaration);
082 addInitializerStatement(context.getQualifiedReference(descriptor).makeStmt());
083 }
084 }
085
086 return null;
087 }
088
089 @Override
090 public Void visitEnumEntry(@NotNull KtEnumEntry enumEntry, TranslationContext data) {
091 ClassDescriptor descriptor = getClassDescriptor(data.bindingContext(), enumEntry);
092 List<KotlinType> supertypes = getSupertypesWithoutFakes(descriptor);
093 if (enumEntry.getBody() != null || supertypes.size() > 1) {
094 enumEntryList.addAll(ClassTranslator.translate(enumEntry, data).getProperties());
095 } else {
096 assert supertypes.size() == 1 : "Simple Enum entry must have one supertype";
097 JsExpression jsEnumEntryCreation = new ClassInitializerTranslator(enumEntry, data)
098 .generateEnumEntryInstanceCreation(supertypes.get(0));
099 jsEnumEntryCreation = TranslationUtils.simpleReturnFunction(data.scope(), jsEnumEntryCreation);
100 enumEntryList.add(new JsPropertyInitializer(data.getNameForDescriptor(descriptor).makeRef(), jsEnumEntryCreation));
101 }
102 return null;
103 }
104
105 @Override
106 public Void visitNamedFunction(@NotNull KtNamedFunction expression, TranslationContext context) {
107 FunctionDescriptor descriptor = getFunctionDescriptor(context.bindingContext(), expression);
108 if (descriptor.getModality() == Modality.ABSTRACT) {
109 return null;
110 }
111
112 context = context.newDeclaration(descriptor, context.getDefinitionPlace());
113 JsPropertyInitializer methodAsPropertyInitializer = Translation.functionTranslator(expression, context).translateAsMethod();
114 result.add(methodAsPropertyInitializer);
115 return null;
116 }
117
118 @Override
119 public Void visitProperty(@NotNull KtProperty expression, TranslationContext context) {
120 PropertyDescriptor propertyDescriptor = BindingUtils.getPropertyDescriptor(context.bindingContext(), expression);
121 context = context.newDeclaration(propertyDescriptor, context.getDefinitionPlace());
122 PropertyTranslatorKt.translateAccessors(propertyDescriptor, expression, result, context);
123 return null;
124 }
125
126 @Override
127 public Void visitAnonymousInitializer(@NotNull KtAnonymousInitializer expression, TranslationContext context) {
128 // parsed it in initializer visitor => no additional actions are needed
129 return null;
130 }
131
132 @Override
133 public Void visitSecondaryConstructor(@NotNull KtSecondaryConstructor constructor, TranslationContext data) {
134 return null;
135 }
136
137 private void addInitializerStatement(@NotNull JsStatement statement) {
138 if (initializerStatements == null) {
139 initializerStatements = new ArrayList<JsStatement>();
140 JsFunction initializerFunction = new JsFunction(scope, new JsBlock(initializerStatements), "class initializer");
141 staticResult.add(new JsPropertyInitializer(new JsNameRef("object_initializer$"), initializerFunction));
142 }
143 initializerStatements.add(statement);
144 }
145 }