X-Git-Url: http://www.git.cypherpunks.ru/?p=goredo.git;a=blobdiff_plain;f=log.go;h=f4abdd078f41fc9b5e5aae1ee2dc2bd66633ba7a;hp=8669a642f98c1561e50643b01f987f2c09250c57;hb=b4eefdd675c9aef9ff8bd1089d031ee05733195b;hpb=f15fe27971bcf37a06bb9542ea1907f11444515a

diff --git a/log.go b/log.go
index 8669a64..f4abdd0 100644
--- 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,63 +20,121 @@ 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 (
-	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
-	CDone  = 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
-	LogDone 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
 )
 
-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 {
 			return
 		}
 		p += "dbg  "
+	case CWait:
+		if !(LogWait || Debug) {
+			return
+		}
+		p += "wait "
 	case CRedo:
-		p += "redo "
-	case CDone:
-		if !(LogDone || Debug) {
+		if NoProgress {
 			return
 		}
-		p += "done "
+		p += "redo "
 	case CLock:
 		if !(LogLock || Debug) {
 			return
@@ -94,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 {