Stylistic fixes

[govpn.git] / src / cypherpunks.ru / govpn / logger.go

diff --git a/src/cypherpunks.ru/govpn/logger.go b/src/cypherpunks.ru/govpn/logger.go
index f38a245f08a7e56c2e136bf9e41e42a1a6186f6b..02dc2ae911f541976e648904f84d02e4971d210c 100644 (file)
--- a/src/cypherpunks.ru/govpn/logger.go
+++ b/src/cypherpunks.ru/govpn/logger.go
@@ -1,3 +1,5 @@
+// +build !windows
+
 /*
 GoVPN -- simple secure free software virtual private network daemon
 Copyright (C) 2014-2017 Sergey Matveev <stargrave@stargrave.org>
@@ -20,37 +22,96 @@ along with this program.  If not, see <http://www.gnu.org/licenses/>.
 package govpn
 
 import (
-       "log"
-       "log/syslog"
-)
+       "bytes"
+       "fmt"
+       logsyslog "log/syslog"
+       "os"
+       "sort"
+       "time"
 
-var (
-       sysloger *log.Logger
+       "github.com/Sirupsen/logrus"
+       "github.com/Sirupsen/logrus/hooks/syslog"
+       "github.com/pkg/errors"
 )
 
-// Enable logging to syslog, instead of default stdout log.
-func SyslogEnable() {
+// syslogFormatter is a syslog friendly formatter
+type syslogFormatter struct{}
+
+// Format converts a log entry into a list of bytes
+func (sf *syslogFormatter) Format(entry *logrus.Entry) ([]byte, error) {
+       var buf bytes.Buffer
        var err error
-       sysloger, err = syslog.NewLogger(syslog.LOG_INFO, 0)
-       if err != nil {
-               log.Fatalln(err)
+       keys := make([]string, 0, len(entry.Data))
+       for k := range entry.Data {
+               keys = append(keys, k)
+       }
+       sort.Strings(keys)
+
+       for _, k := range keys {
+               if _, err = buf.WriteString(fmt.Sprintf("[%s]%+v ", k, entry.Data[k])); err != nil {
+                       return nil, errors.Wrapf(err, "buf.WriteString %s", k)
+               }
+       }
+       if _, err = buf.WriteString(entry.Message); err != nil {
+               return nil, errors.Wrap(err, "buf.WriteString message")
        }
+       return buf.Bytes(), nil
 }
 
-// Call either syslog-related logger.Printf if SyslogEnabled,
-// default log.Printf otherwise.
-func Printf(f string, v ...interface{}) {
-       if sysloger == nil {
-               log.Printf(f, v...)
+// NewLogger returns a logger for specified level. Syslog or Windows
+// Events can be turned on.
+func NewLogger(level string, syslog bool) (*logrus.Logger, error) {
+       var logger *logrus.Logger
+       logLevel, err := logrus.ParseLevel(level)
+       if err != nil {
+               return nil, errors.Wrap(err, "logrus.ParseLevel")
+       }
+
+       if syslog {
+               syslogHook, err := logrus_syslog.NewSyslogHook("", "", logsyslog.LOG_INFO, "GoVPN")
+               if err != nil {
+                       return nil, errors.Wrap(err, "logrus_syslog.NewSyslogHook")
+               }
+               logger = &logrus.Logger{
+                       Formatter: &syslogFormatter{},
+                       Hooks:     make(logrus.LevelHooks),
+               }
+               logger.Hooks.Add(syslogHook)
        } else {
-               sysloger.Printf(f, v...)
+               logger = &logrus.Logger{
+                       Formatter: &logrus.TextFormatter{
+                               ForceColors:      true,
+                               DisableColors:    false,
+                               DisableTimestamp: false,
+                               FullTimestamp:    true,
+                               TimestampFormat:  time.RFC3339Nano,
+                               DisableSorting:   false,
+                       },
+                       Hooks: make(logrus.LevelHooks),
+               }
+       }
+       logger.Out = os.Stderr
+       logger.Level = logLevel
+       logger.WithFields(logrus.Fields{
+               "version": VersionGet(),
+               "level":   logLevel.String(),
+       }).Info("Initialize logging")
+       return logger, nil
+}
+
+// ExtendLogFields adds batch of log Fields to existing fields ones
+func ExtendLogFields(input *logrus.Fields, add logrus.Fields) {
+       i := *input
+       for k, v := range add {
+               i[k] = v
        }
 }
 
-// Call both default log.Printf and syslog-related one.
-func BothPrintf(f string, v ...interface{}) {
-       log.Printf(f, v...)
-       if sysloger != nil {
-               sysloger.Printf(f, v...)
+// MergeLogFields combines multiple log fields into a single one
+func MergeLogFields(fields ...logrus.Fields) (output logrus.Fields) {
+       output = logrus.Fields{}
+       for _, f := range fields {
+               ExtendLogFields(&output, f)
        }
+       return
 }