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.reference;
018    
019    import com.google.dart.compiler.backend.js.ast.JsExpression;
020    import org.jetbrains.annotations.NotNull;
021    import org.jetbrains.kotlin.descriptors.DeclarationDescriptor;
022    import org.jetbrains.kotlin.descriptors.FunctionDescriptor;
023    import org.jetbrains.kotlin.descriptors.ValueParameterDescriptor;
024    import org.jetbrains.kotlin.descriptors.VariableDescriptor;
025    import org.jetbrains.kotlin.js.translate.context.TranslationContext;
026    import org.jetbrains.kotlin.psi.KtExpression;
027    import org.jetbrains.kotlin.psi.KtQualifiedExpression;
028    import org.jetbrains.kotlin.psi.KtSimpleNameExpression;
029    
030    import static org.jetbrains.kotlin.js.translate.utils.BindingUtils.getDescriptorForReferenceExpression;
031    import static org.jetbrains.kotlin.js.translate.utils.PsiUtils.getSelectorAsSimpleName;
032    import static org.jetbrains.kotlin.psi.KtPsiUtil.isBackingFieldReference;
033    
034    public final class ReferenceTranslator {
035    
036        private ReferenceTranslator() {
037        }
038    
039        @NotNull
040        public static JsExpression translateSimpleName(@NotNull KtSimpleNameExpression expression, @NotNull TranslationContext context) {
041            return getAccessTranslator(expression, context).translateAsGet();
042        }
043    
044        @NotNull
045        public static JsExpression translateAsFQReference(@NotNull DeclarationDescriptor referencedDescriptor,
046                @NotNull TranslationContext context) {
047            JsExpression alias = context.getAliasForDescriptor(referencedDescriptor);
048            return alias != null ? alias : context.getQualifiedReference(referencedDescriptor);
049        }
050    
051        @NotNull
052        public static JsExpression translateAsLocalNameReference(@NotNull DeclarationDescriptor descriptor,
053                @NotNull TranslationContext context) {
054            if (descriptor instanceof FunctionDescriptor || descriptor instanceof VariableDescriptor) {
055                JsExpression alias = context.getAliasForDescriptor(descriptor);
056                if (alias != null) {
057                    return alias;
058                }
059            }
060            return context.getQualifiedReference(descriptor);
061        }
062    
063        @NotNull
064        public static AccessTranslator getAccessTranslator(@NotNull KtSimpleNameExpression referenceExpression,
065                @NotNull TranslationContext context) {
066            if (isBackingFieldReference(getDescriptorForReferenceExpression(context.bindingContext(), referenceExpression))) {
067                return BackingFieldAccessTranslator.newInstance(referenceExpression, context);
068            }
069            if (canBePropertyAccess(referenceExpression, context)) {
070                return VariableAccessTranslator.newInstance(context, referenceExpression, null);
071            }
072            if (CompanionObjectIntrinsicAccessTranslator.isCompanionObjectReference(referenceExpression, context)) {
073                return CompanionObjectIntrinsicAccessTranslator.newInstance(referenceExpression, context);
074            }
075            return ReferenceAccessTranslator.newInstance(referenceExpression, context);
076        }
077    
078        public static boolean canBePropertyAccess(@NotNull KtExpression expression, @NotNull TranslationContext context) {
079            KtSimpleNameExpression simpleNameExpression = null;
080            if (expression instanceof KtQualifiedExpression) {
081                simpleNameExpression = getSelectorAsSimpleName((KtQualifiedExpression) expression);
082            }
083            else if (expression instanceof KtSimpleNameExpression) {
084                simpleNameExpression = (KtSimpleNameExpression) expression;
085            }
086    
087            if (simpleNameExpression == null) return false;
088    
089            DeclarationDescriptor descriptor = getDescriptorForReferenceExpression(context.bindingContext(), simpleNameExpression);
090    
091            // Skip ValueParameterDescriptor because sometime we can miss resolved call for it, e.g. when set something to delegated property.
092            return descriptor instanceof VariableDescriptor && !(descriptor instanceof ValueParameterDescriptor);
093        }
094    
095    }