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.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.jet.lang.descriptors.DeclarationDescriptor;
023    import org.jetbrains.jet.lang.descriptors.FunctionDescriptor;
024    import org.jetbrains.jet.lang.descriptors.PropertyDescriptor;
025    import org.jetbrains.jet.lang.descriptors.VariableDescriptor;
026    import org.jetbrains.jet.lang.psi.JetExpression;
027    import org.jetbrains.jet.lang.psi.JetQualifiedExpression;
028    import org.jetbrains.jet.lang.psi.JetSimpleNameExpression;
029    import org.jetbrains.k2js.translate.context.TranslationContext;
030    
031    import static org.jetbrains.jet.lang.psi.JetPsiUtil.isBackingFieldReference;
032    import static org.jetbrains.k2js.translate.utils.BindingUtils.getDescriptorForReferenceExpression;
033    import static org.jetbrains.k2js.translate.utils.PsiUtils.getSelectorAsSimpleName;
034    
035    public final class ReferenceTranslator {
036    
037        private ReferenceTranslator() {
038        }
039    
040        @NotNull
041        public static JsExpression translateSimpleName(@NotNull JetSimpleNameExpression expression,
042                @NotNull TranslationContext context) {
043            return getAccessTranslator(expression, context).translateAsGet();
044        }
045    
046        @NotNull
047        public static JsExpression translateAsFQReference(@NotNull DeclarationDescriptor referencedDescriptor,
048                @NotNull TranslationContext context) {
049            JsExpression alias = context.getAliasForDescriptor(referencedDescriptor);
050            return alias != null ? alias : context.getQualifiedReference(referencedDescriptor);
051        }
052    
053        @NotNull
054        public static JsExpression translateAsLocalNameReference(@NotNull DeclarationDescriptor descriptor,
055                @NotNull TranslationContext context) {
056            if (descriptor instanceof FunctionDescriptor || descriptor instanceof VariableDescriptor) {
057                JsExpression alias = context.getAliasForDescriptor(descriptor);
058                if (alias != null) {
059                    return alias;
060                }
061            }
062            return context.getNameForDescriptor(descriptor).makeRef();
063        }
064    
065        @NotNull
066        public static AccessTranslator getAccessTranslator(@NotNull JetSimpleNameExpression referenceExpression,
067                @NotNull TranslationContext context) {
068            return getAccessTranslator(referenceExpression, null, context);
069        }
070    
071        @NotNull
072        public static AccessTranslator getAccessTranslator(@NotNull JetSimpleNameExpression referenceExpression,
073                @Nullable JsExpression receiver,
074                @NotNull TranslationContext context) {
075            if (isBackingFieldReference(referenceExpression)) {
076                return BackingFieldAccessTranslator.newInstance(referenceExpression, context);
077            }
078            if (canBePropertyAccess(referenceExpression, context)) {
079                return VariableAccessTranslator.newInstance(context, referenceExpression, receiver);
080            }
081            if (ClassObjectAccessTranslator.isClassObjectReference(referenceExpression, context)) {
082                return ClassObjectAccessTranslator.newInstance(referenceExpression, context);
083            }
084            return ReferenceAccessTranslator.newInstance(referenceExpression, context);
085        }
086    
087        public static boolean canBePropertyAccess(@NotNull JetExpression expression, @NotNull TranslationContext context) {
088            JetSimpleNameExpression simpleNameExpression = null;
089            if (expression instanceof JetQualifiedExpression) {
090                simpleNameExpression = getSelectorAsSimpleName((JetQualifiedExpression) expression);
091            }
092            else if (expression instanceof JetSimpleNameExpression) {
093                simpleNameExpression = (JetSimpleNameExpression) expression;
094            }
095            return simpleNameExpression != null &&
096                   (getDescriptorForReferenceExpression(context.bindingContext(), simpleNameExpression) instanceof PropertyDescriptor);
097        }
098    
099    }