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