waku-org/go-waku

package utils

import (
    "runtime/debug"
    "strings"

    logging "github.com/ipfs/go-log/v2"

    "go.uber.org/zap"
    "go.uber.org/zap/zapcore"
)

var log *zap.Logger
var messageLoggers map[string]*zap.Logger

// Logger creates a zap.Logger with some reasonable defaults
func Logger(name ...string) *zap.Logger {
    loggerName := "gowaku"
    if len(name) != 0 {
        loggerName = name[0]
    }

    if log == nil {
        InitLogger("console", "stdout", loggerName, zapcore.InfoLevel)
    }
    return log
}

// MessagesLogger returns a logger used for debug logging of sent/received messages
func MessagesLogger(prefix string) *zap.Logger {
    if messageLoggers == nil {
        messageLoggers = make(map[string]*zap.Logger)
    }
    logger := messageLoggers[prefix]
    if logger == nil {
        logger = logging.Logger(prefix + ".messages").Desugar()
        messageLoggers[prefix] = logger
    }

    return logger
}

// InitLogger initializes a global logger using an specific encoding
func InitLogger(encoding string, output string, name string, level zapcore.Level) {
    cfg := logging.GetConfig()
    cfg.Level = logging.LogLevel(level)

    if encoding == "json" {
        cfg.Format = logging.JSONOutput
    } else if encoding == "nocolor" {
        cfg.Format = logging.PlaintextOutput
    } else {
        cfg.Format = logging.ColorizedOutput
    }

    if output == "stdout" || output == "" {
        cfg.Stdout = true
        cfg.Stderr = false
    } else {
        if encoding == "console" {
            cfg.Format = logging.PlaintextOutput
        }
        cfg.Stdout = false
        cfg.Stderr = false

        outputParts := strings.Split(output, ":")
        if len(outputParts) == 2 {
            cfg.File = outputParts[1]
        } else {
            if len(outputParts) > 2 || outputParts[0] != "file" {
                panic("invalid output format")
            }
            cfg.File = "./waku.log"
        }
    }
    if cfg.Level == logging.LevelError {
        // Override default level setting
        cfg.Level = logging.LevelInfo
    }

    logging.SetupLogging(cfg)

    log = logging.Logger(name).Desugar()
}

func LogOnPanic() {
    if err := recover(); err != nil {
        Logger().Error("panic in goroutine", zap.Any("error", err), zap.String("stacktrace", string(debug.Stack())))
        panic(err)
    }
}