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.operation;
018    
019    import com.google.dart.compiler.backend.js.ast.JsExpression;
020    import org.jetbrains.annotations.NotNull;
021    import org.jetbrains.kotlin.descriptors.FunctionDescriptor;
022    import org.jetbrains.kotlin.js.translate.callTranslator.CallTranslator;
023    import org.jetbrains.kotlin.js.translate.context.TranslationContext;
024    import org.jetbrains.kotlin.psi.JetBinaryExpression;
025    import org.jetbrains.kotlin.resolve.calls.callUtil.CallUtilPackage;
026    import org.jetbrains.kotlin.resolve.calls.model.ResolvedCall;
027    
028    public final class OverloadedAssignmentTranslator extends AssignmentTranslator {
029    
030        @NotNull
031        public static JsExpression doTranslate(@NotNull JetBinaryExpression expression,
032                @NotNull TranslationContext context) {
033            return (new OverloadedAssignmentTranslator(expression, context)).translate();
034        }
035    
036        @NotNull
037        private final ResolvedCall<? extends FunctionDescriptor> resolvedCall;
038    
039        private OverloadedAssignmentTranslator(@NotNull JetBinaryExpression expression,
040                @NotNull TranslationContext context) {
041            super(expression, context);
042            resolvedCall = CallUtilPackage.getFunctionResolvedCallWithAssert(expression, context.bindingContext());
043        }
044    
045        @NotNull
046        private JsExpression translate() {
047            if (isVariableReassignment) {
048                return reassignment();
049            }
050            return overloadedMethodInvocation();
051        }
052    
053        @NotNull
054        private JsExpression reassignment() {
055            return accessTranslator.translateAsSet(overloadedMethodInvocation());
056        }
057    
058        @NotNull
059        private JsExpression overloadedMethodInvocation() {
060            return CallTranslator.translate(context(), resolvedCall, accessTranslator.translateAsGet());
061        }
062    }