commonMain.com.copperleaf.kudzu.util.printAst.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kudzu-core-js Show documentation

A monadic (I think...) recursive-descent parser written in Kotlin

There is a newer version: 5.3.0

package com.copperleaf.kudzu.util

import com.copperleaf.kudzu.node.Node
import com.copperleaf.kudzu.node.NonTerminalNode
import com.copperleaf.kudzu.node.TerminalNode

fun Node.printAst(currentIndent: Int): String {
    return when (this) {
        is TerminalNode -> printTerminalNodeAst(currentIndent)
        is NonTerminalNode -> printNonTerminalNodeAst(currentIndent)
    }
}

fun TerminalNode.printTerminalNodeAst(currentIndent: Int): String {
    return "${indent(currentIndent)}($astNodeName: '$text')"
}

fun NonTerminalNode.printNonTerminalNodeAst(currentIndent: Int): String {
    val childrenPrinted = if (children.isNotEmpty()) {
        children
            .map { it.printAst(currentIndent + 2) }
            .joinToString(separator = "\n")
    } else {
        "${indent(currentIndent + 2)}(empty)"
    }

    return "${indent(currentIndent)}($astNodeName:\n" +
        childrenPrinted +
        "\n" +
        "${indent(currentIndent)})"
}

fun indent(currentIndent: Int): String {
    return (0 until currentIndent).map { " " }.joinToString(separator = "")
}