org.openrewrite.java.internal.VariableDeclsToString Maven / Gradle / Ivy

/*
 * Copyright 2020 the original author or authors.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 

 * https://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.openrewrite.java.internal;

import org.openrewrite.java.tree.J;

import java.util.stream.Collectors;

public class VariableDeclsToString {
    public static String toString(J.VariableDecls v) {
        return VARIABLE_PRINTER.visit(v);
    }

    private static final PrintJava VARIABLE_PRINTER = new PrintJava() {
        @Override
        public String visitMultiVariable(J.VariableDecls multiVariable) {
            String modifiers = visitModifiers(multiVariable.getModifiers()).trim();
            String varargs = multiVariable.getVarargs() == null ? "" : "...";

            return (modifiers.isEmpty() ? "" : modifiers + " ") +
                    (multiVariable.getTypeExpr() == null ? "" : multiVariable.getTypeExpr().printTrimmed() + " ") +
                    multiVariable.getDimensionsBeforeName().stream()
                            .map(d -> "[]")
                            .reduce("", this::reduce) +
                    varargs +
                    multiVariable.getVars().stream()
                            .map(J.VariableDecls.NamedVar::getSimpleName)
                            .collect(Collectors.joining(", "));
        }
    };
}