org.jetbrains.kotlin.fir.renderer.FirPrinter.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kotlin-compiler Show documentation

the Kotlin compiler

There is a newer version: 2.1.0-RC

/*
 * Copyright 2010-2022 JetBrains s.r.o. and Kotlin Programming Language contributors.
 * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
 */

package org.jetbrains.kotlin.fir.renderer

import org.jetbrains.kotlin.fir.FirElement
import org.jetbrains.kotlin.utils.Printer

open class FirPrinter(builder: StringBuilder) {
    private val printer = Printer(builder)

    private var lineBeginning = true

    fun print(vararg objects: Any) {
        if (lineBeginning) {
            lineBeginning = false
            printer.print(*objects)
        } else {
            printer.printWithNoIndent(*objects)
        }
    }

    fun println(vararg objects: Any) {
        print(*objects)
        printer.printlnWithNoIndent()
        lineBeginning = true
    }

    internal fun pushIndent() {
        printer.pushIndent()
    }

    internal fun popIndent() {
        printer.popIndent()
    }

    fun newLine() {
        println()
    }

    fun renderInBraces(leftBrace: String = "{", rightBrace: String = "}", f: () -> Unit) {
        println(" $leftBrace")
        pushIndent()
        f()
        popIndent()
        println(rightBrace)
    }

    internal fun renderSeparated(elements: List, visitor: FirRenderer.Visitor) {
        for ((index, element) in elements.withIndex()) {
            if (index > 0) {
                print(", ")
            }
            element.accept(visitor)
        }
    }

    internal fun renderSeparatedWithNewlines(elements: List, visitor: FirRenderer.Visitor) {
        for ((index, element) in elements.withIndex()) {
            if (index > 0) {
                print(",")
                newLine()
            }
            element.accept(visitor)
        }
    }

    override fun toString(): String {
        return printer.toString()
    }
}