56 lines
1.5 KiB
Go
56 lines
1.5 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/rs/zerolog"
|
|
)
|
|
|
|
var (
|
|
// CurrentLogFile is used for accessing the location of the currently used log file across packages.
|
|
CurrentLogFile string
|
|
logFile *os.File
|
|
logDir string
|
|
logger zerolog.Logger
|
|
)
|
|
|
|
// StartLogger instantiates an instance of our zerolog loggger so we can hook it in our main package.
|
|
// While this does return a logger, it should not be used for additional retrievals of the logger. Use GetLogger()
|
|
func StartLogger() zerolog.Logger {
|
|
logDir = snek.GetString("logger.directory")
|
|
if !strings.HasSuffix(logDir, "/") {
|
|
logDir += "/"
|
|
}
|
|
if err := os.MkdirAll(logDir, os.ModePerm); err != nil {
|
|
println("cannot create log directory: " + logDir + "(" + err.Error() + ")")
|
|
os.Exit(1)
|
|
}
|
|
|
|
logFileName := "HellPot"
|
|
|
|
if snek.GetBool("logger.use_date_filename") {
|
|
tn := strings.ReplaceAll(time.Now().Format(time.RFC822), " ", "_")
|
|
tn = strings.ReplaceAll(logFileName, ":", "-")
|
|
logFileName = logFileName + "_" + tn
|
|
}
|
|
|
|
CurrentLogFile = logDir + logFileName + ".log"
|
|
|
|
if logFile, err = os.OpenFile(CurrentLogFile, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0o666); err != nil {
|
|
println("cannot create log file: " + err.Error())
|
|
os.Exit(1)
|
|
}
|
|
|
|
multi := zerolog.MultiLevelWriter(zerolog.ConsoleWriter{NoColor: NoColor, Out: os.Stdout}, logFile)
|
|
logger = zerolog.New(multi).With().Timestamp().Logger()
|
|
return logger
|
|
}
|
|
|
|
// GetLogger retrieves our global logger object
|
|
func GetLogger() *zerolog.Logger {
|
|
// future logic here
|
|
return &logger
|
|
}
|