50 lines
1.0 KiB
Go
50 lines
1.0 KiB
Go
package chatlog
|
|
|
|
import (
|
|
stdlog "log"
|
|
"os"
|
|
"path/filepath"
|
|
"time"
|
|
|
|
"github.com/sjzar/chatlog/pkg/util"
|
|
|
|
"github.com/rs/zerolog"
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var (
|
|
Debug bool
|
|
Console bool
|
|
)
|
|
|
|
func initLog(cmd *cobra.Command, args []string) {
|
|
zerolog.SetGlobalLevel(zerolog.InfoLevel)
|
|
|
|
if Debug {
|
|
zerolog.SetGlobalLevel(zerolog.DebugLevel)
|
|
}
|
|
|
|
log.Logger = log.Output(zerolog.ConsoleWriter{Out: os.Stderr, TimeFormat: time.RFC3339})
|
|
stdlog.SetOutput(os.Stderr)
|
|
}
|
|
|
|
func initTuiLog(cmd *cobra.Command, args []string) {
|
|
logpath := util.DefaultWorkDir("")
|
|
util.PrepareDir(logpath)
|
|
|
|
logFile, err := os.OpenFile(filepath.Join(logpath, "chatlog.log"), os.O_CREATE|os.O_WRONLY|os.O_APPEND, os.ModePerm)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
log.Logger = log.Output(zerolog.ConsoleWriter{Out: logFile, NoColor: true, TimeFormat: time.RFC3339})
|
|
logrus.SetOutput(logFile)
|
|
stdlog.SetOutput(logFile)
|
|
|
|
if Debug {
|
|
zerolog.SetGlobalLevel(zerolog.DebugLevel)
|
|
}
|
|
}
|