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.reference;
018
019 import com.google.dart.compiler.backend.js.ast.JsExpression;
020 import org.jetbrains.annotations.NotNull;
021 import org.jetbrains.annotations.Nullable;
022 import org.jetbrains.kotlin.descriptors.*;
023 import org.jetbrains.kotlin.js.translate.context.Namer;
024 import org.jetbrains.kotlin.js.translate.context.TranslationContext;
025 import org.jetbrains.kotlin.js.translate.utils.AnnotationsUtils;
026 import org.jetbrains.kotlin.js.translate.utils.BindingUtils;
027 import org.jetbrains.kotlin.psi.KtExpression;
028 import org.jetbrains.kotlin.psi.KtQualifiedExpression;
029 import org.jetbrains.kotlin.psi.KtSimpleNameExpression;
030 import org.jetbrains.kotlin.resolve.DescriptorUtils;
031
032 import static org.jetbrains.kotlin.js.translate.utils.BindingUtils.getDescriptorForReferenceExpression;
033 import static org.jetbrains.kotlin.js.translate.utils.JsAstUtils.setQualifier;
034 import static org.jetbrains.kotlin.js.translate.utils.PsiUtils.getSelectorAsSimpleName;
035 import static org.jetbrains.kotlin.psi.KtPsiUtil.isBackingFieldReference;
036
037 public final class ReferenceTranslator {
038
039 private ReferenceTranslator() {
040 }
041
042 @NotNull
043 public static JsExpression translateSimpleName(@NotNull KtSimpleNameExpression expression,
044 @NotNull TranslationContext context) {
045 return getAccessTranslator(expression, context).translateAsGet();
046 }
047
048 @NotNull
049 public static JsExpression translateSimpleNameWithQualifier(
050 @NotNull KtSimpleNameExpression expression,
051 @Nullable JsExpression qualifier,
052 @NotNull TranslationContext context
053 ) {
054 JsExpression simpleName = translateSimpleName(expression, context);
055
056 // Ignore qualifier if expression is EnumEntry or companion object reference and always use FQ name.
057 DeclarationDescriptor descriptor = BindingUtils.getDescriptorForReferenceExpression(context.bindingContext(), expression);
058 //TODO: should go away when objects inside classes are supported
059 if (DescriptorUtils.isCompanionObject(descriptor) && !AnnotationsUtils.isNativeObject(descriptor)) {
060 return simpleName;
061 }
062 if (descriptor instanceof ClassDescriptor) {
063 ClassDescriptor entryClass = (ClassDescriptor) descriptor;
064 if (entryClass.getKind() == ClassKind.ENUM_ENTRY && !AnnotationsUtils.isNativeObject(entryClass)) {
065 DeclarationDescriptor enumClass = entryClass.getContainingDeclaration();
066 qualifier = Namer.getCompanionObjectAccessor(translateAsFQReference(enumClass, context));
067 }
068 }
069
070 if (qualifier != null) { // TODO: hack for nested Object
071 setQualifier(simpleName, qualifier);
072 }
073
074 return simpleName;
075 }
076
077 @NotNull
078 public static JsExpression translateAsFQReference(@NotNull DeclarationDescriptor referencedDescriptor,
079 @NotNull TranslationContext context) {
080 JsExpression alias = context.getAliasForDescriptor(referencedDescriptor);
081 return alias != null ? alias : context.getQualifiedReference(referencedDescriptor);
082 }
083
084 @NotNull
085 public static JsExpression translateAsLocalNameReference(@NotNull DeclarationDescriptor descriptor,
086 @NotNull TranslationContext context) {
087 if (descriptor instanceof FunctionDescriptor || descriptor instanceof VariableDescriptor) {
088 JsExpression alias = context.getAliasForDescriptor(descriptor);
089 if (alias != null) {
090 return alias;
091 }
092 }
093 return context.getNameForDescriptor(descriptor).makeRef();
094 }
095
096 @NotNull
097 public static AccessTranslator getAccessTranslator(@NotNull KtSimpleNameExpression referenceExpression,
098 @NotNull TranslationContext context) {
099 return getAccessTranslator(referenceExpression, null, context);
100 }
101
102 @NotNull
103 public static AccessTranslator getAccessTranslator(@NotNull KtSimpleNameExpression referenceExpression,
104 @Nullable JsExpression receiver,
105 @NotNull TranslationContext context) {
106 if (isBackingFieldReference(getDescriptorForReferenceExpression(context.bindingContext(), referenceExpression))) {
107 return BackingFieldAccessTranslator.newInstance(referenceExpression, context);
108 }
109 if (canBePropertyAccess(referenceExpression, context)) {
110 return VariableAccessTranslator.newInstance(context, referenceExpression, receiver);
111 }
112 if (CompanionObjectAccessTranslator.isCompanionObjectReference(referenceExpression, context)) {
113 return CompanionObjectAccessTranslator.newInstance(referenceExpression, context);
114 }
115 return ReferenceAccessTranslator.newInstance(referenceExpression, context);
116 }
117
118 public static boolean canBePropertyAccess(@NotNull KtExpression expression, @NotNull TranslationContext context) {
119 KtSimpleNameExpression simpleNameExpression = null;
120 if (expression instanceof KtQualifiedExpression) {
121 simpleNameExpression = getSelectorAsSimpleName((KtQualifiedExpression) expression);
122 }
123 else if (expression instanceof KtSimpleNameExpression) {
124 simpleNameExpression = (KtSimpleNameExpression) expression;
125 }
126
127 if (simpleNameExpression == null) return false;
128
129 DeclarationDescriptor descriptor = getDescriptorForReferenceExpression(context.bindingContext(), simpleNameExpression);
130
131 // Skip ValueParameterDescriptor because sometime we can miss resolved call for it, e.g. when set something to delegated property.
132 return descriptor instanceof VariableDescriptor && !(descriptor instanceof ValueParameterDescriptor);
133 }
134
135 }