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.PackageViewDescriptor;
024    import org.jetbrains.jet.lang.psi.*;
025    import org.jetbrains.k2js.translate.context.TranslationContext;
026    import org.jetbrains.k2js.translate.utils.ErrorReportingUtils;
027    
028    import static org.jetbrains.k2js.translate.general.Translation.translateAsExpression;
029    import static org.jetbrains.k2js.translate.utils.BindingUtils.getDescriptorForReferenceExpression;
030    import static org.jetbrains.k2js.translate.utils.PsiUtils.getNotNullSimpleNameSelector;
031    import static org.jetbrains.k2js.translate.utils.PsiUtils.getSelector;
032    
033    public final class QualifiedExpressionTranslator {
034    
035        private QualifiedExpressionTranslator() {
036        }
037    
038        @NotNull
039        public static AccessTranslator getAccessTranslator(@NotNull JetQualifiedExpression expression,
040                                                           @NotNull TranslationContext context) {
041            JsExpression receiver = translateReceiver(expression, context);
042            return VariableAccessTranslator.newInstance(context, getNotNullSimpleNameSelector(expression), receiver);
043        }
044    
045        @NotNull
046        public static JsExpression translateQualifiedExpression(@NotNull JetQualifiedExpression expression,
047                                                                @NotNull TranslationContext context) {
048            JsExpression receiver = translateReceiver(expression, context);
049            JetExpression selector = getSelector(expression);
050            return dispatchToCorrectTranslator(receiver, selector, context);
051        }
052    
053        @NotNull
054        private static JsExpression dispatchToCorrectTranslator(
055                @Nullable JsExpression receiver,
056                @NotNull JetExpression selector,
057                @NotNull TranslationContext context
058        ) {
059            if (ReferenceTranslator.canBePropertyAccess(selector, context)) {
060                assert selector instanceof JetSimpleNameExpression : "Selectors for properties must be simple names.";
061                return VariableAccessTranslator.newInstance(context, (JetSimpleNameExpression)selector, receiver).translateAsGet();
062            }
063            if (selector instanceof JetCallExpression) {
064                return invokeCallExpressionTranslator(receiver, selector, context);
065            }
066            //TODO: never get there
067            if (selector instanceof JetSimpleNameExpression) {
068                return ReferenceTranslator.translateSimpleNameWithQualifier((JetSimpleNameExpression) selector, receiver, context);
069            }
070            throw new AssertionError("Unexpected qualified expression: " + selector.getText());
071        }
072    
073        @NotNull
074        private static JsExpression invokeCallExpressionTranslator(
075                @Nullable JsExpression receiver,
076                @NotNull JetExpression selector,
077                @NotNull TranslationContext context
078        ) {
079            try {
080                return CallExpressionTranslator.translate((JetCallExpression) selector, receiver, context);
081            } catch (RuntimeException e) {
082                throw  ErrorReportingUtils.reportErrorWithLocation(selector, e);
083            }
084        }
085    
086        @Nullable
087        private static JsExpression translateReceiver(@NotNull JetQualifiedExpression expression,
088                                                      @NotNull TranslationContext context) {
089            JetExpression receiverExpression = expression.getReceiverExpression();
090            if (isFullQualifierForExpression(receiverExpression, context)) {
091                return null;
092            }
093            return translateAsExpression(receiverExpression, context);
094        }
095    
096        //TODO: prove correctness
097        private static boolean isFullQualifierForExpression(@Nullable JetExpression receiverExpression, @NotNull TranslationContext context) {
098            if (receiverExpression == null) {
099                return false;
100            }
101            if (receiverExpression instanceof JetReferenceExpression) {
102                DeclarationDescriptor descriptorForReferenceExpression =
103                    getDescriptorForReferenceExpression(context.bindingContext(), (JetReferenceExpression)receiverExpression);
104                if (descriptorForReferenceExpression instanceof PackageViewDescriptor) {
105                    return true;
106                }
107            }
108            if (receiverExpression instanceof JetQualifiedExpression) {
109                return isFullQualifierForExpression(((JetQualifiedExpression)receiverExpression).getSelectorExpression(), context);
110            }
111            return false;
112        }
113    }