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.JsBinaryOperation;
020    import com.google.dart.compiler.backend.js.ast.JsBinaryOperator;
021    import com.google.dart.compiler.backend.js.ast.JsBlock;
022    import com.google.dart.compiler.backend.js.ast.JsExpression;
023    import org.jetbrains.annotations.NotNull;
024    import org.jetbrains.kotlin.js.translate.context.TranslationContext;
025    import org.jetbrains.kotlin.js.translate.reference.AccessTranslator;
026    import org.jetbrains.kotlin.lexer.KtToken;
027    import org.jetbrains.kotlin.psi.KtBinaryExpression;
028    import org.jetbrains.kotlin.psi.KtSimpleNameExpression;
029    import org.jetbrains.kotlin.types.expressions.OperatorConventions;
030    
031    import static org.jetbrains.kotlin.js.translate.utils.PsiUtils.getOperationToken;
032    import static org.jetbrains.kotlin.js.translate.utils.PsiUtils.isAssignment;
033    import static org.jetbrains.kotlin.js.translate.utils.TranslationUtils.translateRightExpression;
034    
035    public final class IntrinsicAssignmentTranslator extends AssignmentTranslator {
036        private JsExpression right;
037        private AccessTranslator accessTranslator;
038    
039        @NotNull
040        public static JsExpression doTranslate(@NotNull KtBinaryExpression expression,
041                                               @NotNull TranslationContext context) {
042            return (new IntrinsicAssignmentTranslator(expression, context)).translate();
043        }
044    
045        private IntrinsicAssignmentTranslator(@NotNull KtBinaryExpression expression,
046                                              @NotNull TranslationContext context) {
047            super(expression, context);
048    
049            JsBlock rightBlock = new JsBlock();
050            right = translateRightExpression(context, expression, rightBlock);
051            accessTranslator = createAccessTranslator(expression.getLeft(), !rightBlock.isEmpty());
052            context.addStatementsToCurrentBlockFrom(rightBlock);
053        }
054    
055        @NotNull
056        private JsExpression translate() {
057            if (isAssignment(getOperationToken(expression))) {
058                return translateAsPlainAssignment();
059            }
060            return translateAsAssignmentOperation();
061        }
062    
063        @NotNull
064        private JsExpression translateAsAssignmentOperation() {
065            if (expression.getLeft() instanceof KtSimpleNameExpression) {
066                return translateAsPlainAssignmentOperation();
067            }
068            return translateAsAssignToCounterpart();
069        }
070    
071        @NotNull
072        private JsExpression translateAsAssignToCounterpart() {
073            JsBinaryOperator operator = getCounterpartOperator();
074            JsBinaryOperation counterpartOperation =
075                    new JsBinaryOperation(operator, accessTranslator.translateAsGet(), right);
076            return accessTranslator.translateAsSet(counterpartOperation);
077        }
078    
079        @NotNull
080        private JsBinaryOperator getCounterpartOperator() {
081            KtToken assignmentOperationToken = getOperationToken(expression);
082            assert OperatorConventions.ASSIGNMENT_OPERATIONS.containsKey(assignmentOperationToken);
083            KtToken counterpartToken = OperatorConventions.ASSIGNMENT_OPERATION_COUNTERPARTS.get(assignmentOperationToken);
084            assert OperatorTable.hasCorrespondingBinaryOperator(counterpartToken) :
085                    "Unsupported token encountered: " + counterpartToken.toString();
086            return OperatorTable.getBinaryOperator(counterpartToken);
087        }
088    
089        @NotNull
090        private JsExpression translateAsPlainAssignmentOperation() {
091            JsBinaryOperator operator = getAssignmentOperator();
092            return new JsBinaryOperation(operator, accessTranslator.translateAsGet(), right);
093        }
094    
095        @NotNull
096        private JsBinaryOperator getAssignmentOperator() {
097            KtToken token = getOperationToken(expression);
098            assert OperatorConventions.ASSIGNMENT_OPERATIONS.containsKey(token);
099            assert OperatorTable.hasCorrespondingBinaryOperator(token) :
100                    "Unsupported token encountered: " + token.toString();
101            return OperatorTable.getBinaryOperator(token);
102        }
103    
104        @NotNull
105        private JsExpression translateAsPlainAssignment() {
106            return accessTranslator.translateAsSet(right);
107        }
108    }