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
017 package org.jetbrains.k2js.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.jet.lang.descriptors.*;
023 import org.jetbrains.jet.lang.psi.*;
024 import org.jetbrains.jet.lang.resolve.name.Name;
025 import org.jetbrains.jet.lang.types.JetType;
026 import org.jetbrains.k2js.translate.context.Namer;
027 import org.jetbrains.k2js.translate.context.TranslationContext;
028 import org.jetbrains.k2js.translate.general.Translation;
029 import org.jetbrains.k2js.translate.general.TranslatorVisitor;
030 import org.jetbrains.k2js.translate.initializer.ClassInitializerTranslator;
031 import org.jetbrains.k2js.translate.utils.BindingUtils;
032 import org.jetbrains.k2js.translate.utils.JsAstUtils;
033 import org.jetbrains.k2js.translate.utils.TranslationUtils;
034
035 import java.util.Collection;
036 import java.util.List;
037
038 import static org.jetbrains.k2js.translate.initializer.InitializerUtils.createPropertyInitializer;
039 import static org.jetbrains.k2js.translate.utils.BindingUtils.*;
040
041 public class DeclarationBodyVisitor extends TranslatorVisitor<Void> {
042 protected final List<JsPropertyInitializer> result;
043 protected final List<JsPropertyInitializer> staticResult;
044 protected final List<JsPropertyInitializer> enumEntryList = new SmartList<JsPropertyInitializer>();
045
046 public DeclarationBodyVisitor() {
047 this(new SmartList<JsPropertyInitializer>(), new SmartList<JsPropertyInitializer>());
048 }
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 public Void visitClass(@NotNull JetClass expression, @NotNull TranslationContext context) {
066 return null;
067 }
068
069 @Override
070 public Void visitEnumEntry(
071 final JetEnumEntry enumEntry, TranslationContext data
072 ) {
073 JsExpression jsEnumEntryCreation;
074 ClassDescriptor descriptor = getClassDescriptor(data.bindingContext(), enumEntry);
075 Collection<JetType> supertypes = descriptor.getTypeConstructor().getSupertypes();
076 if (enumEntry.getBody() != null || supertypes.size() > 1) {
077 jsEnumEntryCreation = ClassTranslator.generateClassCreation(enumEntry, descriptor, data);
078 } else {
079 assert supertypes.size() == 1 : "Simple Enum entry must have one supertype";
080 jsEnumEntryCreation = new ClassInitializerTranslator(enumEntry, data).generateEnumEntryInstanceCreation(supertypes.iterator().next());
081 }
082 Named named = new Named() {
083 @NotNull
084 @Override
085 public Name getName() {
086 String name = enumEntry.getName();
087 assert name != null : "Enum entry name must be not null";
088 return Name.identifier(name);
089 }
090 };
091 enumEntryList.add(new JsPropertyInitializer(data.nameToLiteral(named), jsEnumEntryCreation));
092 return null;
093 }
094
095 @Override
096 public Void visitClassObject(
097 JetClassObject classObject, TranslationContext context
098 ) {
099 JetObjectDeclaration declaration = classObject.getObjectDeclaration();
100 assert declaration != null : "Declaration for class object must be not null";
101 ClassDescriptor descriptor = getClassDescriptor(context.bindingContext(), declaration);
102 JsExpression value = ClassTranslator.generateClassCreation(declaration, descriptor, context);
103
104 JsFunction fun = TranslationUtils.simpleReturnFunction(context.getScopeForDescriptor(descriptor), value);
105 staticResult.add(createPropertyInitializer(Namer.getNamedForClassObjectInitializer(), fun, context));
106 return null;
107 }
108
109 @Override
110 public Void visitNamedFunction(@NotNull JetNamedFunction expression, @NotNull TranslationContext context) {
111 FunctionDescriptor descriptor = getFunctionDescriptor(context.bindingContext(), expression);
112 if (descriptor.getModality() == Modality.ABSTRACT) {
113 return null;
114 }
115
116 JsPropertyInitializer methodAsPropertyInitializer = Translation.functionTranslator(expression, context).translateAsMethod();
117 if (context.isEcma5()) {
118 JsExpression methodBodyExpression = methodAsPropertyInitializer.getValueExpr();
119 methodAsPropertyInitializer.setValueExpr(JsAstUtils.createPropertyDataDescriptor(descriptor, methodBodyExpression));
120 }
121 result.add(methodAsPropertyInitializer);
122 return null;
123 }
124
125 @Override
126 public Void visitProperty(@NotNull JetProperty expression, @NotNull TranslationContext context) {
127 PropertyDescriptor propertyDescriptor = BindingUtils.getPropertyDescriptor(context.bindingContext(), expression);
128 PropertyTranslator.translateAccessors(propertyDescriptor, expression, result, context);
129 return null;
130 }
131
132 @Override
133 public Void visitObjectDeclarationName(@NotNull JetObjectDeclarationName expression,
134 @NotNull TranslationContext context) {
135 if (!context.isEcma5()) {
136 PropertyTranslator
137 .translateAccessors(getPropertyDescriptorForObjectDeclaration(context.bindingContext(), expression), result, context);
138 }
139
140 return null;
141 }
142
143 @Override
144 public Void visitAnonymousInitializer(@NotNull JetClassInitializer expression, @NotNull TranslationContext context) {
145 // parsed it in initializer visitor => no additional actions are needed
146 return null;
147 }
148 }