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.psi;
018    
019    import org.jetbrains.annotations.NotNull;
020    import org.jetbrains.annotations.Nullable;
021    
022    public class JetPsiUnparsingUtils {
023        private JetPsiUnparsingUtils() {
024        }
025    
026        @NotNull
027        public static String toIf(@Nullable JetExpression condition, @Nullable JetExpression thenExpression, @Nullable JetExpression elseExpression) {
028            return toIf(
029                    JetPsiUtil.getText(condition),
030                    JetPsiUtil.getText(thenExpression),
031                    elseExpression != null ? elseExpression.getText() : null
032            );
033        }
034    
035        @NotNull
036        public static String toIf(@NotNull String condition, @NotNull String thenExpression, @Nullable String elseExpression) {
037            return "if " + parenthesizeTextIfNeeded(condition) + " " + thenExpression + (elseExpression != null ? " else " + elseExpression : "");
038        }
039    
040        @NotNull
041        public static String toBinaryExpression(@Nullable JetExpression left, @NotNull String op, @Nullable JetElement right) {
042            return toBinaryExpression(JetPsiUtil.getText(left), op, JetPsiUtil.getText(right));
043        }
044    
045        @NotNull
046        public static String toBinaryExpression(@NotNull String left, @NotNull String op, @NotNull String right) {
047            return left + " " + op + " " + right;
048        }
049    
050        @NotNull
051        public static String parenthesizeIfNeeded(@Nullable JetExpression expression) {
052            String text = JetPsiUtil.getText(expression);
053    
054            return (expression instanceof JetParenthesizedExpression ||
055                    expression instanceof JetConstantExpression ||
056                    expression instanceof JetSimpleNameExpression ||
057                    expression instanceof JetDotQualifiedExpression)
058                   ? text : "(" + text + ")";
059        }
060    
061        @NotNull
062        public static String parenthesizeTextIfNeeded(@NotNull String expressionText) {
063            return (expressionText.startsWith("(") && expressionText.endsWith(")")) ? expressionText : "(" + expressionText + ")";
064        }
065    }