io.github.simplifier_ag.scala.spdf.ParamShow.scala Maven / Gradle / Ivy

Go to download
package io.github.simplifier_ag.scala.spdf

/**
 * Renders a parameter of type T to a sequence of strings that will be
 * appended to the command line.
 */
trait ParamShow[T] {
  def show(name: String, value: T): Iterable[String]
}

object ParamShow {

  implicit object StringParamShow extends ParamShow[String] {
    override def show(name: String, value: String): Iterable[String] =
      formatParam(name, Some(value))
  }

  implicit object BooleanParamShow extends ParamShow[Boolean] {
    override def show(name: String, value: Boolean): Iterable[String] =
      if (value) {
        formatParam(name, None)
      } else {
        Iterable.empty
      }
  }

  implicit object OptionBooleanParamShow extends ParamShow[Option[Boolean]] {
    override def show(name: String, valueOpt: Option[Boolean]): Iterable[String] =
      valueOpt.toIterable.flatMap {
        formatParam(name, _)
      }
  }

  implicit object IntParamShow extends ParamShow[Int] {
    override def show(name: String, value: Int): Iterable[String] =
      formatParam(name, Some(value.toString))
  }

  implicit object FloatParamShow extends ParamShow[Float] {
    override def show(name: String, value: Float): Iterable[String] =
      formatParam(name, Some("%.2f".format(value)))
  }

  implicit object PageOrientationParamShow extends ParamShow[PageOrientation] {
    override def show(name: String, value: PageOrientation): Iterable[String] =
      formatParam(name, Some(value.value))
  }

  implicit object ErrorHandlingParamShow extends ParamShow[ErrorHandling] {
    override def show(name: String, value: ErrorHandling): Iterable[String] =
      formatParam(name, Some(value.value))
  }

  implicit object LogLevelParamShow extends ParamShow[LogLevel] {
    override def show(name: String, value: LogLevel): Iterable[String] =
      formatParam(name, Some(value.value))
  }

  implicit object RepeatableStringParamShow extends ParamShow[Seq[String]] {
    override def show(name: String, value: Seq[String]): Iterable[String] =
      value.flatMap(stringValue => formatParam(name, Some(stringValue)))
  }

  implicit object RepeatableStringMapParamShow extends ParamShow[Map[String, String]] {
    override def show(name: String, value: Map[String, String]): Iterable[String] =
      value.flatMap { case (key, strVal) =>
        formatParam(name, Some(s"$key $strVal"))
      }
  }

  private def formatParam(name: String, value: Option[String]): Iterable[String] =
    Seq(Some("--" + name), value).flatten

  private def formatParam(name: String, value: Boolean): Iterable[String] = if (value) {
    Some("--" + name)
  } else {
    Some("--no-" + name)
  }

}