doobie.util.pretty.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of doobie-core_3 Show documentation

Pure functional JDBC layer for Scala.

There is a newer version: 1.0.0-RC6

// Copyright (c) 2013-2020 Rob Norris and Contributors
// This software is licensed under the MIT License (MIT).
// For more information see LICENSE or https://opensource.org/licenses/MIT

package doobie.util

import cats.Monoid
import scala.annotation.tailrec
import scala.Predef.*

/** Some functions for pretty-printing. */
object pretty {

  /** A block of text that can be aligned with other blocks. */
  final case class Block(lines: List[String]) {

    lazy val width = lines.foldLeft(0)((n, s) => n max s.length)
    lazy val height = lines.length

    def leftOf(other: Block): Block =
      leftOfP(other, "")

    def leftOf1(other: Block): Block =
      leftOfP(other, " ")

    def leftOf2(other: Block): Block =
      leftOfP(other, "  ")

    def leftOfP(other: Block, padding: String): Block = {
      val h = height max other.height
      val ls = lines.iterator.map(_.padTo(width, ' ')).padTo(h, " " * width)
      val os = other.lines.iterator.padTo(h, "")
      Block(ls.zip(os).map { case (l, o) => l + padding + o }.toList)
    }

    def padLeft(padding: String): Block = Block.empty.leftOfP(this, padding)

    def above(other: Block): Block =
      Block(lines ++ other.lines)

    def above1(other: Block): Block =
      Block(lines ++ ("" :: other.lines))

    override def toString: String =
      lines.mkString("\n")

    def trimLeft(n: Int): Block =
      Block(lines.map(_ drop n))

    def wrap(cols: Int): Block =
      Block(lines.flatMap(pretty.wrap(cols)))
  }

  object Block {
    val empty = Block(Nil)
    def fromString(s: String) = Block(List(s))
    def fromLines(s: String) = Block(s.linesIterator.toList)
    def fromErrorMsgLines(e: Throwable) = Block(Option(e.getMessage).toList.flatMap(_.linesIterator.toList))

    implicit val BlockMonoid: Monoid[Block] = new Monoid[Block] {
      def empty = Block.empty
      def combine(a: Block, b: Block) = a.above(b)
    }
  }

  def wrap(cols: Int)(s: String): List[String] = {
    @tailrec
    def go(ws: List[String], accum: List[String]): List[String] =
      ws match {
        case Nil => accum.reverse
        case w :: ws =>
          accum match {
            case Nil => go(ws, List(w))
            case l :: ls =>
              val l0 = l + " " + w
              if (l0.length > cols) go(ws, w :: accum)
              else go(ws, l0 :: ls)
          }
      }

    go(s.split(" +").toList, Nil)
  }

}