exception.UnexpectedColumnsException.scala Maven / Gradle / Ivy

Go to download
// Copyright (c) 2018-2021 by Rob Norris
// This software is licensed under the MIT License (MIT).
// For more information see LICENSE or https://opensource.org/licenses/MIT

package skunk.exception

import cats.syntax.foldable._
import skunk.Command
import skunk.util.Text
import skunk.data.TypedRowDescription
import skunk.data.TypedRowDescription.Field

case class UnexpectedRowsException(
  command: Command[_],
  rd:      TypedRowDescription,
) extends SkunkException(
  sql       = Some(command.sql),
  message   = "Statement returns data.",
  hint      = Some(s"This ${framed("command")} returns rows and should be a ${framed("query")}."),
  sqlOrigin = Some(command.origin),
) {

  import Text.green

  private def describe(f: Field): List[Text] =
    List(green(f.name), Text(f.tpe.name))

  private def columns: String =
    s"""|The unexpected output columns are
        |
        |  ${Text.grid(rd.fields.map(describe)).intercalate(Text("\n|  ")).render}
        |
        |""".stripMargin

  override def sections: List[String] =
    super.sections :+ columns

}