Redundant @documentencoding

[goredo.git] / log.go

diff --git a/log.go b/log.go
index f4fa2ca9c65f647b8aa1d48d71232984b76bed90..f4abdd078f41fc9b5e5aae1ee2dc2bd66633ba7a 100644 (file)
--- a/log.go
+++ b/log.go
@@ -1,6 +1,6 @@
 /*
-goredo -- redo implementation on pure Go
-Copyright (C) 2020 Sergey Matveev <stargrave@stargrave.org>
+goredo -- djb's redo implementation on pure Go
+Copyright (C) 2020-2023 Sergey Matveev <stargrave@stargrave.org>
 
 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
@@ -20,66 +20,105 @@ along with this program.  If not, see <http://www.gnu.org/licenses/>.
 package main
 
 import (
+       "bytes"
        "flag"
        "fmt"
        "os"
        "strings"
+       "sync"
+
+       "golang.org/x/term"
 )
 
 const (
-       EnvLevel   = "REDO_LEVEL"
-       EnvDebug   = "REDO_DEBUG"
-       EnvLogWait = "REDO_LOG_WAIT"
-       EnvLogLock = "REDO_LOG_LOCK"
-       EnvLogPid  = "REDO_LOG_PID"
-       EnvLogJS   = "REDO_LOG_JS"
-       EnvNoColor = "NO_COLOR"
-
-       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 (
-       Level   = 0
-       NoColor bool
-       Debug   bool
-       LogWait bool
-       LogLock bool
-       LogJS   bool
-       MyPid   int
-
-       flagDebug   = flag.Bool("debug", false, "enable debug logging (REDO_DEBUG=1)")
-       flagLogWait = flag.Bool("log-wait", false, "enable wait messages logging (REDO_LOG_WAIT=1)")
-       flagLogLock = flag.Bool("log-lock", false, "enable lock messages logging (REDO_LOG_LOCK=1)")
-       flagLogPid  = flag.Bool("log-pid", false, "append PIDs (REDO_LOG_PID=1)")
-       flagLogJS   = flag.Bool("log-js", false, "enable jobserver messages logging (REDO_LOG_JS=1)")
+       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
 )
 
-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 tracef(level, format string, args ...interface{}) {
        var p string
        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")
+               LogMutex.Lock()
+               os.Stderr.WriteString(p)
+               LogMutex.Unlock()
                return
        case CDebug:
                if !Debug {
@@ -92,6 +131,9 @@ func trace(level, format string, args ...interface{}) {
                }
                p += "wait "
        case CRedo:
+               if NoProgress {
+                       return
+               }
                p += "redo "
        case CLock:
                if !(LogLock || Debug) {
@@ -110,7 +152,10 @@ 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")
+       LogMutex.Lock()
+       os.Stderr.WriteString(msg)
+       LogMutex.Unlock()
 }
 
 func colourize(colour, s string) string {