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.*;
020 import gnu.trove.THashMap;
021 import org.jetbrains.annotations.NotNull;
022 import org.jetbrains.kotlin.descriptors.PackageFragmentDescriptor;
023 import org.jetbrains.kotlin.js.facade.exceptions.TranslationRuntimeException;
024 import org.jetbrains.kotlin.js.translate.context.Namer;
025 import org.jetbrains.kotlin.js.translate.context.TranslationContext;
026 import org.jetbrains.kotlin.js.translate.general.AbstractTranslator;
027 import org.jetbrains.kotlin.name.FqName;
028 import org.jetbrains.kotlin.psi.KtFile;
029 import org.jetbrains.kotlin.resolve.BindingContext;
030 import org.jetbrains.kotlin.resolve.BindingContextUtils;
031
032 import java.util.*;
033
034 import static com.google.dart.compiler.backend.js.ast.JsVars.JsVar;
035
036 public final class PackageDeclarationTranslator extends AbstractTranslator {
037 private final Iterable<KtFile> files;
038 private final Map<PackageFragmentDescriptor, PackageTranslator> packageFragmentToTranslator =
039 new LinkedHashMap<PackageFragmentDescriptor, PackageTranslator>();
040
041 public static List<JsStatement> translateFiles(@NotNull Collection<KtFile> files, @NotNull TranslationContext context) {
042 return new PackageDeclarationTranslator(files, context).translate();
043 }
044
045 private PackageDeclarationTranslator(@NotNull Iterable<KtFile> files, @NotNull TranslationContext context) {
046 super(context);
047
048 this.files = files;
049 }
050
051 @NotNull
052 private List<JsStatement> translate() {
053 // predictable order
054 Map<FqName, DefineInvocation> packageFqNameToDefineInvocation = new THashMap<FqName, DefineInvocation>();
055
056 for (KtFile file : files) {
057 PackageFragmentDescriptor packageFragment =
058 BindingContextUtils.getNotNull(context().bindingContext(), BindingContext.FILE_TO_PACKAGE_FRAGMENT, file);
059
060 PackageTranslator translator = packageFragmentToTranslator.get(packageFragment);
061 if (translator == null) {
062 createRootPackageDefineInvocationIfNeeded(packageFqNameToDefineInvocation);
063 translator = PackageTranslator.create(packageFragment, context());
064 packageFragmentToTranslator.put(packageFragment, translator);
065 }
066
067 try {
068 translator.translate(file);
069 }
070 catch (TranslationRuntimeException e) {
071 throw e;
072 }
073 catch (RuntimeException e) {
074 throw new TranslationRuntimeException(file, e);
075 }
076 catch (AssertionError e) {
077 throw new TranslationRuntimeException(file, e);
078 }
079 }
080
081 for (PackageTranslator translator : packageFragmentToTranslator.values()) {
082 translator.add(packageFqNameToDefineInvocation);
083 }
084
085 JsVars vars = new JsVars(true);
086 vars.addIfHasInitializer(getRootPackageDeclaration(packageFqNameToDefineInvocation.get(FqName.ROOT)));
087
088 return Collections.<JsStatement>singletonList(vars);
089 }
090
091 private void createRootPackageDefineInvocationIfNeeded(@NotNull Map<FqName, DefineInvocation> packageFqNameToDefineInvocation) {
092 if (!packageFqNameToDefineInvocation.containsKey(FqName.ROOT)) {
093 packageFqNameToDefineInvocation.put(
094 FqName.ROOT, DefineInvocation.create(FqName.ROOT, null, new JsObjectLiteral(true), context()));
095 }
096 }
097
098 private JsVar getRootPackageDeclaration(@NotNull DefineInvocation defineInvocation) {
099 JsExpression rootPackageVar = new JsInvocation(context().namer().rootPackageDefinitionMethodReference(), defineInvocation.asList());
100 return new JsVar(context().scope().declareName(Namer.getRootPackageName()), rootPackageVar);
101 }
102 }