71 lines
1.0 KiB
Go
71 lines
1.0 KiB
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
|
|
"github.com/tncardoso/gocurses"
|
|
)
|
|
|
|
var logReady chan struct{}
|
|
|
|
func appendPop(max int, a []string, v ...string) []string {
|
|
if len(a) < max {
|
|
return append(a, v...)
|
|
} else {
|
|
for i := 0; i < len(v); i++ {
|
|
for j := 1; j < len(a); j++ {
|
|
a[j-1] = a[j]
|
|
}
|
|
}
|
|
return append(a[:max-len(v)], v...)
|
|
}
|
|
}
|
|
|
|
type Logger struct {
|
|
visible []string
|
|
all []byte
|
|
}
|
|
|
|
func newLogger() *Logger {
|
|
l := &Logger{}
|
|
initCurses(l)
|
|
return l
|
|
}
|
|
|
|
func (l *Logger) Write(p []byte) (int, error) {
|
|
row, _ := gocurses.Getmaxyx()
|
|
l.visible = appendPop(row-1, l.visible, string(p))
|
|
draw(l.visible)
|
|
|
|
l.all = append(l.all, p...)
|
|
|
|
return len(p), nil
|
|
}
|
|
|
|
func (l *Logger) Close() {
|
|
os.Mkdir("log", 0777)
|
|
|
|
os.Rename("log/latest.txt", "log/last.txt")
|
|
os.WriteFile("log/latest.txt", l.all, 0666)
|
|
}
|
|
|
|
func LogReady() <-chan struct{} {
|
|
if logReady == nil {
|
|
logReady = make(chan struct{})
|
|
}
|
|
|
|
return logReady
|
|
}
|
|
|
|
func init() {
|
|
l := newLogger()
|
|
log.SetOutput(l)
|
|
|
|
go func() {
|
|
for logReady == nil {
|
|
}
|
|
close(logReady)
|
|
}()
|
|
}
|