node build fixed

seanime-2.9.10/internal/util/crashlog/crashlog.go

@@ -0,0 +1,148 @@
+package crashlog
+
+import (
+	"bytes"
+	"context"
+	"fmt"
+	"github.com/rs/zerolog"
+	"github.com/samber/mo"
+	"io"
+	"os"
+	"path/filepath"
+	"seanime/internal/util"
+	"sync"
+	"time"
+)
+
+// Global variable that continuously records logs from specific programs and writes them to a file when something unexpected happens.
+
+type CrashLogger struct {
+	//logger    *zerolog.Logger
+	//logBuffer *bytes.Buffer
+	//mu        sync.Mutex
+	logDir mo.Option[string]
+}
+
+type CrashLoggerArea struct {
+	name       string
+	logger     *zerolog.Logger
+	logBuffer  *bytes.Buffer
+	mu         sync.Mutex
+	ctx        context.Context
+	cancelFunc context.CancelFunc
+}
+
+var GlobalCrashLogger = NewCrashLogger()
+
+// NewCrashLogger creates a new CrashLogger instance.
+func NewCrashLogger() *CrashLogger {
+
+	//var logBuffer bytes.Buffer
+	//
+	//fileOutput := zerolog.ConsoleWriter{
+	//	Out:           &logBuffer,
+	//	TimeFormat:    time.DateTime,
+	//	FormatMessage: util.ZerologFormatMessageSimple,
+	//	FormatLevel:   util.ZerologFormatLevelSimple,
+	//	NoColor:       true,
+	//}
+	//
+	//multi := zerolog.MultiLevelWriter(fileOutput)
+	//logger := zerolog.New(multi).With().Timestamp().Logger()
+
+	return &CrashLogger{
+		//logger:    &logger,
+		//logBuffer: &logBuffer,
+		//mu:        sync.Mutex{},
+		logDir: mo.None[string](),
+	}
+}
+
+func (c *CrashLogger) SetLogDir(dir string) {
+	c.logDir = mo.Some(dir)
+}
+
+// InitArea creates a new CrashLoggerArea instance.
+// This instance can be used to log crashes in a specific area.
+func (c *CrashLogger) InitArea(area string) *CrashLoggerArea {
+
+	var logBuffer bytes.Buffer
+
+	fileOutput := zerolog.ConsoleWriter{
+		Out:         &logBuffer,
+		TimeFormat:  time.DateTime,
+		FormatLevel: util.ZerologFormatLevelSimple,
+		NoColor:     true,
+	}
+
+	multi := zerolog.MultiLevelWriter(fileOutput)
+	logger := zerolog.New(multi).With().Timestamp().Logger()
+
+	//ctx, cancelFunc := context.WithCancel(context.Background())
+
+	return &CrashLoggerArea{
+		logger:    &logger,
+		name:      area,
+		logBuffer: &logBuffer,
+		mu:        sync.Mutex{},
+		//ctx:        ctx,
+		//cancelFunc: cancelFunc,
+	}
+}
+
+// Stdout returns the CrashLoggerArea's log buffer so that it can be used as a writer.
+//
+//	Example:
+//		crashLogger := crashlog.GlobalCrashLogger.InitArea("ffmpeg")
+//		defer crashLogger.Close()
+//
+//		cmd.Stdout = crashLogger.Stdout()
+func (a *CrashLoggerArea) Stdout() io.Writer {
+	return a.logBuffer
+}
+
+func (a *CrashLoggerArea) LogError(msg string) {
+	a.logger.Error().Msg(msg)
+}
+
+func (a *CrashLoggerArea) LogErrorf(format string, args ...interface{}) {
+	a.logger.Error().Msgf(format, args...)
+}
+
+func (a *CrashLoggerArea) LogInfof(format string, args ...interface{}) {
+	a.logger.Info().Msgf(format, args...)
+}
+
+// Close should be always called using defer when a new area is created
+//
+//	logArea := crashlog.GlobalCrashLogger.InitArea("ffmpeg")
+//	defer logArea.Close()
+func (a *CrashLoggerArea) Close() {
+	a.logBuffer.Reset()
+	//a.cancelFunc()
+}
+
+func (c *CrashLogger) WriteAreaLogToFile(area *CrashLoggerArea) {
+	logDir, found := c.logDir.Get()
+	if !found {
+		return
+	}
+
+	// e.g. crash-ffmpeg-2021-09-01_15-04-05.log
+	logFilePath := filepath.Join(logDir, fmt.Sprintf("crash-%s-%s.log", area.name, time.Now().Format("2006-01-02_15-04-05")))
+
+	// Create file
+	logFile, err := os.OpenFile(logFilePath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0664)
+	if err != nil {
+		fmt.Printf("Failed to open log file: %s\n", logFilePath)
+		return
+	}
+	defer logFile.Close()
+
+	area.mu.Lock()
+	defer area.mu.Unlock()
+	if _, err := area.logBuffer.WriteTo(logFile); err != nil {
+		fmt.Printf("Failed to write crash log buffer to file for %s\n", area.name)
+	}
+	area.logBuffer.Reset()
+}