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.operation;
018    
019    import com.google.dart.compiler.backend.js.ast.JsExpression;
020    import org.jetbrains.annotations.NotNull;
021    import org.jetbrains.jet.lang.descriptors.FunctionDescriptor;
022    import org.jetbrains.jet.lang.psi.JetBinaryExpression;
023    import org.jetbrains.k2js.translate.context.TranslationContext;
024    import org.jetbrains.k2js.translate.reference.CallBuilder;
025    import org.jetbrains.k2js.translate.utils.BindingUtils;
026    
027    public final class OverloadedAssignmentTranslator extends AssignmentTranslator {
028    
029        @NotNull
030        public static JsExpression doTranslate(@NotNull JetBinaryExpression expression,
031                @NotNull TranslationContext context) {
032            return (new OverloadedAssignmentTranslator(expression, context)).translate();
033        }
034    
035        @NotNull
036        private final FunctionDescriptor operationDescriptor;
037    
038        private OverloadedAssignmentTranslator(@NotNull JetBinaryExpression expression,
039                @NotNull TranslationContext context) {
040            super(expression, context);
041            FunctionDescriptor functionDescriptor =
042                    BindingUtils.getFunctionDescriptorForOperationExpression(context.bindingContext(), expression);
043            assert functionDescriptor != null : "";
044            this.operationDescriptor = functionDescriptor;
045        }
046    
047        @NotNull
048        private JsExpression translate() {
049            if (isVariableReassignment) {
050                return reassignment();
051            }
052            return overloadedMethodInvocation();
053        }
054    
055        @NotNull
056        private JsExpression reassignment() {
057            return accessTranslator.translateAsSet(overloadedMethodInvocation());
058        }
059    
060        @NotNull
061        private JsExpression overloadedMethodInvocation() {
062            return CallBuilder.build(context())
063                    .descriptor(operationDescriptor)
064                    .receiver(accessTranslator.translateAsGet())
065                    .args(right)
066                    .translate();
067        }
068    }