X-Git-Url: http://www.git.cypherpunks.ru/?a=blobdiff_plain;f=log.go;h=ea81c149375b491fb1642f9dc2693e2047041f24;hb=HEAD;hp=335ca53c3d0a146079286a4c4b04caed91c6f450;hpb=bc7701e7a4f95cee680e0736ec3e68a8b0b5c09f;p=goredo.git diff --git a/log.go b/log.go index 335ca53..ea81c14 100644 --- a/log.go +++ b/log.go @@ -1,69 +1,149 @@ -/* -goredo -- redo implementation on pure Go -Copyright (C) 2020 Sergey Matveev - -This program is free software: you can redistribute it and/or modify -it under the terms of the GNU General Public License as published by -the Free Software Foundation, version 3 of the License. - -This program is distributed in the hope that it will be useful, -but WITHOUT ANY WARRANTY; without even the implied warranty of -MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the -GNU General Public License for more details. - -You should have received a copy of the GNU General Public License -along with this program. If not, see . -*/ +// goredo -- djb's redo implementation on pure Go +// Copyright (C) 2020-2024 Sergey Matveev +// +// This program is free software: you can redistribute it and/or modify +// it under the terms of the GNU General Public License as published by +// the Free Software Foundation, version 3 of the License. +// +// This program is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +// GNU General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this program. If not, see . // Logging facilities package main import ( + "bytes" + "flag" "fmt" "os" "strings" + "sync" + + "golang.org/x/term" ) const ( - CReset = "\033[0m" - CBold = "\033[1m" - CBlack = "\033[30;1m" - CRed = "\033[31;1m" - CGreen = "\033[32;1m" - CYellow = "\033[33;1m" - CBlue = "\033[34;1m" - CMagenta = "\033[35;1m" - CCyan = "\033[36;1m" - CWhite = "\033[37;1m" - - CDebug = CYellow - CRedo = CGreen - CWait = CBlue - CLock = CCyan - CErr = CRed - CWarn = CMagenta - CJS = CWhite - CNone = "NONE" + EnvLevel = "REDO_LEVEL" + EnvNoProgress = "REDO_NO_PROGRESS" + EnvDebug = "REDO_DEBUG" + EnvLogWait = "REDO_LOG_WAIT" + EnvLogLock = "REDO_LOG_LOCK" + EnvLogPid = "REDO_LOG_PID" + EnvLogJS = "REDO_LOG_JS" + EnvNoColor = "NO_COLOR" ) var ( - NoColor bool - Debug bool - LogWait bool - LogLock bool - LogJS bool - MyPid int + Level = 0 + NoColor bool + NoProgress bool + Debug bool + LogWait bool + LogLock bool + LogJS bool + MyPID int + + CDebug string + CRedo string + CWait string + CLock string + CErr string + CWarn string + CJS string + CReset string + CNone = "NONE" + + flagDebug = flag.Bool("d", false, fmt.Sprintf("enable debug logging (%s=1)", EnvDebug)) + flagNoProgress *bool + flagLogWait *bool + flagLogLock *bool + flagLogPid *bool + flagLogJS *bool + + LogMutex sync.Mutex + KeyEraseLine string + LastLoggedTgt string ) -func trace(level, format string, args ...interface{}) { +func init() { + var b bytes.Buffer + t := term.NewTerminal(&b, "") + CDebug = string(t.Escape.Yellow) + CRedo = string(t.Escape.Green) + CWait = string(t.Escape.Blue) + CLock = string(t.Escape.Cyan) + CErr = string(t.Escape.Red) + CWarn = string(t.Escape.Magenta) + CJS = string(t.Escape.White) + CReset = string(t.Escape.Reset) + KeyEraseLine = fmt.Sprintf("%s[K", CReset[0:1]) + + cmdName := CmdName() + if !(cmdName == CmdNameRedo || cmdName == CmdNameRedoIfchange) { + return + } + flagNoProgress = flag.Bool("no-progress", false, + fmt.Sprintf("no progress printing (%s=1), also implies -no-status", EnvNoProgress)) + flagLogWait = flag.Bool("log-wait", false, + fmt.Sprintf("enable wait messages logging (%s=1)", EnvLogWait)) + flagLogLock = flag.Bool("log-lock", false, + fmt.Sprintf("enable lock messages logging (%s=1)", EnvLogLock)) + flagLogPid = flag.Bool("log-pid", false, + fmt.Sprintf("append PIDs (%s=1)", EnvLogPid)) + flagLogJS = flag.Bool("log-js", false, + fmt.Sprintf("enable jobserver messages logging (%s=1)", EnvLogJS)) +} + +func erasedStatus(s, end string) string { + if NoProgress || NoColor { + return s + end + } + return s + KeyEraseLine + end +} + +func withPrependedTgt(s string) { + if s[0] != '[' { + stderrWrite(erasedStatus(s, "\n")) + return + } + i := strings.IndexByte(s, ']') + if i == -1 { + stderrWrite(s) + return + } + tgt, s := s[1:i], s[i+1:] + if tgt != LastLoggedTgt { + LastLoggedTgt = tgt + tgt = "redo " + tgt + " ..." + if MyPID != 0 { + tgt = fmt.Sprintf("[%d] %s", MyPID, tgt) + } + stderrWrite(erasedStatus(colourize(CDebug, tgt), "\n")) + } + stderrWrite(erasedStatus(s, "\n")) +} + +func stderrWrite(s string) { + LogMutex.Lock() + os.Stderr.WriteString(s) + LogMutex.Unlock() +} + +func tracef(level, format string, args ...interface{}) { var p string - if MyPid != 0 { - p = fmt.Sprintf("[%d] ", MyPid) + if MyPID != 0 { + p = fmt.Sprintf("[%d] ", MyPID) } switch level { case CNone: - os.Stderr.WriteString(StderrPrefix + p + fmt.Sprintf(format+"\n", args...)) + p = erasedStatus(StderrPrefix+p+fmt.Sprintf(format, args...), "\n") + stderrWrite(p) return case CDebug: if !Debug { @@ -76,6 +156,9 @@ func trace(level, format string, args ...interface{}) { } p += "wait " case CRedo: + if NoProgress { + return + } p += "redo " case CLock: if !(LogLock || Debug) { @@ -94,7 +177,8 @@ func trace(level, format string, args ...interface{}) { } msg := fmt.Sprintf(format, args...) msg = StderrPrefix + colourize(level, p+strings.Repeat(". ", Level)+msg) - os.Stderr.WriteString(msg + "\n") + msg = erasedStatus(msg, "\n") + stderrWrite(msg) } func colourize(colour, s string) string {