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.NamespaceDescriptor;
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            PropertyAccessTranslator result =
043                PropertyAccessTranslator.newInstance(getNotNullSimpleNameSelector(expression), receiver,
044                                                     CallType.getCallTypeForQualifiedExpression(expression), context);
045            result.setCallType(CallType.getCallTypeForQualifiedExpression(expression));
046            return result;
047        }
048    
049        @NotNull
050        public static JsExpression translateQualifiedExpression(@NotNull JetQualifiedExpression expression,
051                                                                @NotNull TranslationContext context) {
052            JsExpression receiver = translateReceiver(expression, context);
053            JetExpression selector = getSelector(expression);
054            CallType callType = CallType.getCallTypeForQualifiedExpression(expression);
055            return dispatchToCorrectTranslator(receiver, selector, callType, context);
056        }
057    
058        @NotNull
059        private static JsExpression dispatchToCorrectTranslator(
060                @Nullable JsExpression receiver,
061                @NotNull JetExpression selector,
062                @NotNull CallType callType,
063                @NotNull TranslationContext context
064        ) {
065            if (PropertyAccessTranslator.canBePropertyGetterCall(selector, context)) {
066                assert selector instanceof JetSimpleNameExpression : "Selectors for properties must be simple names.";
067                return PropertyAccessTranslator.translateAsPropertyGetterCall
068                    ((JetSimpleNameExpression)selector, receiver, callType, context);
069            }
070            if (selector instanceof JetCallExpression) {
071                return invokeCallExpressionTranslator(receiver, selector, callType, context);
072            }
073            //TODO: never get there
074            if (selector instanceof JetSimpleNameExpression) {
075                return ReferenceTranslator.translateSimpleName((JetSimpleNameExpression)selector, context);
076            }
077            throw new AssertionError("Unexpected qualified expression: " + selector.getText());
078        }
079    
080        @NotNull
081        private static JsExpression invokeCallExpressionTranslator(@Nullable JsExpression receiver,
082                @NotNull JetExpression selector,
083                @NotNull CallType callType,
084                @NotNull TranslationContext context) {
085            try {
086                return CallExpressionTranslator.translate((JetCallExpression) selector, receiver, callType, context);
087            } catch (RuntimeException e) {
088                throw  ErrorReportingUtils.reportErrorWithLocation(selector, e);
089            }
090        }
091    
092        @Nullable
093        private static JsExpression translateReceiver(@NotNull JetQualifiedExpression expression,
094                                                      @NotNull TranslationContext context) {
095            JetExpression receiverExpression = expression.getReceiverExpression();
096            if (isFullQualifierForExpression(receiverExpression, context)) {
097                return null;
098            }
099            return translateAsExpression(receiverExpression, context);
100        }
101    
102        //TODO: prove correctness
103        private static boolean isFullQualifierForExpression(@Nullable JetExpression receiverExpression, @NotNull TranslationContext context) {
104            if (receiverExpression == null) {
105                return false;
106            }
107            if (receiverExpression instanceof JetReferenceExpression) {
108                DeclarationDescriptor descriptorForReferenceExpression =
109                    getDescriptorForReferenceExpression(context.bindingContext(), (JetReferenceExpression)receiverExpression);
110                if (descriptorForReferenceExpression instanceof NamespaceDescriptor) {
111                    return true;
112                }
113            }
114            if (receiverExpression instanceof JetQualifiedExpression) {
115                return isFullQualifierForExpression(((JetQualifiedExpression)receiverExpression).getSelectorExpression(), context);
116            }
117            return false;
118        }
119    }