scalapb.textformat.TextGenerator.scala Maven / Gradle / Ivy

Go to download
package scalapb.textformat

import scala.collection.mutable

class TextGenerator(singleLine: Boolean = true, escapeNonAscii: Boolean = true) {
  private val sb          = new mutable.StringBuilder()
  private var indentLevel = 0
  private var lineStart   = true

  private def maybeNewLine(): Unit = {
    if (lineStart) {
      if (!singleLine)
        sb.append(" " * (indentLevel * 2))
      else if (sb.nonEmpty) sb.append(' ')
    }
    ()
  }

  def add(s: String): TextGenerator = {
    maybeNewLine()
    sb.append(s)
    lineStart = false
    this
  }

  def addMaybeEscape(s: String): TextGenerator = {
    add(
      if (escapeNonAscii)
        TextFormatUtils.escapeText(s)
      else
        TextFormatUtils.escapeDoubleQuotesAndBackslashes(s).replace("\n", "\\n")
    )
  }

  def addNewLine(s: String): TextGenerator = {
    maybeNewLine()
    sb.append(s)
    if (!singleLine) {
      sb.append('\n')
    }
    lineStart = true
    this
  }

  def indent(): TextGenerator = {
    indentLevel += 1
    this
  }

  def outdent(): TextGenerator = {
    indentLevel -= 1
    this
  }

  def result(): String = sb.result()
}