com.ossuminc.riddl.utils.StringBuildingPrintStream.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of riddl-utils_3 Show documentation

Various utilities used throughout riddl libraries

There is a newer version: 0.52.1

/*
 * Copyright 2019 Ossum, Inc.
 *
 * SPDX-License-Identifier: Apache-2.0
 */

package com.ossuminc.riddl.utils

import java.io.OutputStream
import java.io.PrintStream
import scala.collection.mutable
import scala.scalajs.js.annotation.JSExportTopLevel

/** A PrintStream that captures any output into a String */
final class StringBuildingPrintStream private (out: OutputStream, stringBuilder: mutable.StringBuilder)
    extends PrintStream(out) {

  /** Returns a String of all output written to this PrintStream */
  def mkString(): String = stringBuilder.mkString
}

@JSExportTopLevel("StringBuildingPrintStream")
object StringBuildingPrintStream {

  /** Returns a new StringBuildingPrintStream */
  def apply(): StringBuildingPrintStream = {
    val stringBuilder = new mutable.StringBuilder
    val out = new OutputStream {
      override def write(b: Int): Unit = stringBuilder.append(b.toChar)
    }
    new StringBuildingPrintStream(out, stringBuilder)
  }
}