luet/pkg/logger/logger.go
Ettore Di Giacinto ccdca66420
Small fixes and enhance output
Fixes a small issue with the Spinner which prevented to update the
spinner style. Also respect the temp folders of the output when
extracting the rootfs.
2019-11-14 20:22:49 +01:00

105 lines
1.8 KiB
Go

package logger
import (
"fmt"
"os"
"sync"
"time"
. "github.com/logrusorgru/aurora"
"github.com/briandowns/spinner"
)
var s *spinner.Spinner
var m = &sync.Mutex{}
var enabled = false
func Spinner(i int) {
m.Lock()
defer m.Unlock()
if i > 43 {
i = 43
}
if s == nil {
s = spinner.New(spinner.CharSets[i], 100*time.Millisecond) // Build our new spinner
}
s.UpdateCharSet(spinner.CharSets[i])
enabled = true
s.Start() // Start the spinner
}
func SpinnerText(suffix, prefix string) {
m.Lock()
defer m.Unlock()
if s == nil {
s = spinner.New(spinner.CharSets[22], 100*time.Millisecond) // Build our new spinner
}
s.Suffix = Bold(Magenta(suffix)).BgBlack().String()
s.Prefix = Bold(Cyan(prefix)).String()
}
func SpinnerStop() {
m.Lock()
defer m.Unlock()
if s != nil {
s.Stop()
}
enabled = false
}
func msg(level string, msg ...interface{}) {
var message string
for _, m := range msg {
message += " " + fmt.Sprintf("%v", m)
}
var levelMsg string
switch level {
case "warning":
levelMsg = Bold(Yellow("🌩 " + message)).BgBlack().String()
case "debug":
levelMsg = White(message).BgBlack().String()
case "info":
levelMsg = Bold(White(message)).BgBlack().String()
case "error":
levelMsg = Bold(Red("💣 " + message + "🔥")).BgBlack().String()
}
if enabled {
SpinnerText(levelMsg, "")
return
}
cmd := []interface{}{}
for _, f := range msg {
cmd = append(cmd, f)
}
m.Lock()
defer m.Unlock()
fmt.Println(levelMsg)
//fmt.Println(cmd...)
}
func Warning(mess ...interface{}) {
msg("warning", mess...)
}
func Debug(mess ...interface{}) {
msg("debug", mess...)
}
func Info(mess ...interface{}) {
msg("info", mess...)
}
func Error(mess ...interface{}) {
msg("error", mess...)
}
func Fatal(mess ...interface{}) {
Error(mess)
os.Exit(1)
}