zio.logging.LogColor.scala Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2019-2024 John A. De Goes and the ZIO Contributors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package zio.logging

import zio.{ Chunk, Config }

import scala.io.AnsiColor

final case class LogColor private (private[logging] val ansi: String) extends AnyVal

object LogColor {
  val RED: LogColor     = LogColor(AnsiColor.RED)
  val BLUE: LogColor    = LogColor(AnsiColor.BLUE)
  val YELLOW: LogColor  = LogColor(AnsiColor.YELLOW)
  val CYAN: LogColor    = LogColor(AnsiColor.CYAN)
  val GREEN: LogColor   = LogColor(AnsiColor.GREEN)
  val MAGENTA: LogColor = LogColor(AnsiColor.MAGENTA)
  val WHITE: LogColor   = LogColor(AnsiColor.WHITE)
  val RESET: LogColor   = LogColor(AnsiColor.RESET)

  private[logging] val logColorMapping: Map[String, LogColor] = Map(
    "RED"     -> LogColor.RED,
    "BLUE"    -> LogColor.BLUE,
    "YELLOW"  -> LogColor.YELLOW,
    "CYAN"    -> LogColor.CYAN,
    "GREEN"   -> LogColor.GREEN,
    "MAGENTA" -> LogColor.MAGENTA,
    "WHITE"   -> LogColor.WHITE,
    "RESET"   -> LogColor.RESET
  )

  private[logging] def logColorValue(value: String): Either[Config.Error.InvalidData, LogColor] =
    logColorMapping.get(value.toUpperCase) match {
      case Some(v) => Right(v)
      case None    => Left(Config.Error.InvalidData(Chunk.empty, s"Expected a LogColor, but found ${value}"))
    }

  val config: Config[LogColor] = Config.string.mapOrFail(logColorValue)
}