org.backuity.puppet.Logger.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of puppet-module-installer_2.11 Show documentation

puppet-module-installer

There is a newer version: 2.2.1

Show newest version

package org.backuity.puppet

import org.backuity.puppet.LogLevel.LogLevel

trait Logger {
  def debug(s: String)
  def info(msg: String)
  def warn(msg: String)
  def error(msg: String)
}

object LogLevel extends Enumeration {
  type LogLevel = Value
  val Error = Value(0)
  val Warn = Value(1)
  val Info = Value(2)
  val Debug = Value(3)
}

object Logger {
  class AnsiConsole(logLevel: LogLevel) extends Console(logLevel) {
    import AnsiFormatter._

    override def error(msg: String): Unit = {
      super.error(ansi"\red{$msg}")
    }

    override def warn(msg: String): Unit = {
      super.warn(ansi"\yellow{$msg}")
    }

    override def debug(msg: String): Unit = {
      super.debug(ansi"\blue{$msg}")
    }
  }

  class Console(logLevel : LogLevel) extends Logger {

    override def debug(msg: String): Unit ={
      if( logLevel >= LogLevel.Debug ) {
        println(msg)
      }
    }

    override def info(msg: String): Unit = {
      if( logLevel >= LogLevel.Info ) {
        println(msg)
      }
    }

    override def warn(msg: String): Unit = {
      if( logLevel >= LogLevel.Warn ) {
        println(msg)
      }
    }

    override def error(msg: String): Unit = {
      println(msg)
    }
  }
}