forked from vikunja/vikunja
103 lines
3.4 KiB
Go
103 lines
3.4 KiB
Go
// Vikunja is a todo-list application to facilitate your life.
|
|
// Copyright 2018 Vikunja and contributors. All rights reserved.
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
package log
|
|
|
|
import (
|
|
"github.com/op/go-logging"
|
|
"github.com/spf13/viper"
|
|
"io"
|
|
"log"
|
|
"os"
|
|
"time"
|
|
)
|
|
|
|
// ErrFmt holds the format for all the console logging
|
|
const ErrFmt = `${time_rfc3339_nano}: ${level} ` + "\t" + `▶ ${prefix} ${short_file}:${line}`
|
|
|
|
// WebFmt holds the format for all logging related to web requests
|
|
const WebFmt = `${time_rfc3339_nano}: WEB ` + "\t" + `▶ ${remote_ip} ${id} ${method} ${status} ${uri} ${latency_human} - ${user_agent}`
|
|
|
|
// Fmt is the general log format
|
|
const Fmt = `%{color}%{time:` + time.RFC3339Nano + `}: %{level}` + "\t" + `▶ %{shortpkg}/%{shortfunc} %{id:03x}%{color:reset} %{message}`
|
|
|
|
// Log is the handler for the logger
|
|
var Log = logging.MustGetLogger("vikunja")
|
|
|
|
// InitLogger initializes the global log handler
|
|
func InitLogger() {
|
|
if !viper.GetBool("log.enabled") {
|
|
// Disable all logging when loggin in general is disabled, overwriting everything a user might have set.
|
|
viper.Set("log.errors", "off")
|
|
viper.Set("log.standard", "off")
|
|
viper.Set("log.database", "off")
|
|
viper.Set("log.http", "off")
|
|
viper.Set("log.echo", "off")
|
|
return
|
|
}
|
|
|
|
if viper.GetString("log.errors") == "file" || viper.GetString("log.standard") == "file" {
|
|
err := os.Mkdir(viper.GetString("log.path"), 0744)
|
|
if err != nil && !os.IsExist(err) {
|
|
log.Fatal("Could not create log folder: ", err.Error())
|
|
}
|
|
}
|
|
|
|
var logBackends []logging.Backend
|
|
|
|
// We define our two backends
|
|
if viper.GetString("log.standard") != "off" {
|
|
stdWriter := GetLogWriter("standard")
|
|
stdBackend := logging.NewLogBackend(stdWriter, "", 0)
|
|
|
|
// Set the standard backend
|
|
logBackends = append(logBackends, logging.NewBackendFormatter(stdBackend, logging.MustStringFormatter(Fmt+"\n")))
|
|
}
|
|
|
|
if viper.GetString("log.error") != "off" {
|
|
errWriter := GetLogWriter("error")
|
|
errBackend := logging.NewLogBackend(errWriter, "", 0)
|
|
|
|
// Only warnings and more severe messages should go to the error backend
|
|
errBackendLeveled := logging.AddModuleLevel(errBackend)
|
|
errBackendLeveled.SetLevel(logging.WARNING, "")
|
|
logBackends = append(logBackends, errBackendLeveled)
|
|
}
|
|
|
|
// Set our backends
|
|
logging.SetBackend(logBackends...)
|
|
}
|
|
|
|
// GetLogWriter returns the writer to where the normal log goes, depending on the config
|
|
func GetLogWriter(logfile string) (writer io.Writer) {
|
|
writer = os.Stderr // Set the default case to prevent nil pointer panics
|
|
switch viper.GetString("log." + logfile) {
|
|
case "file":
|
|
f, err := os.OpenFile(viper.GetString("log.path")+"/"+logfile+".log", os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
writer = f
|
|
break
|
|
case "stderr":
|
|
writer = os.Stderr
|
|
case "stdout":
|
|
default:
|
|
writer = os.Stdout
|
|
}
|
|
return
|
|
}
|