Set logging level for Logrus and TDlib from config

1 files changed, 29 insertions, 0 deletions

diff --git a/log.go b/log.go
new file mode 100644
index 0000000..b287815
--- /dev/null
+++ b/log.go
@@ -0,0 +1,29 @@
+package main
+
+import (
+	log "github.com/sirupsen/logrus"
+)
+
+var logConstants = map[string]log.Level{
+	":fatal":   log.FatalLevel,
+	":error":   log.ErrorLevel,
+	":warn":    log.WarnLevel,
+	":info":    log.InfoLevel,
+	":debug":   log.DebugLevel,
+	":verbose": log.TraceLevel,
+	":all":     log.TraceLevel,
+}
+
+func stringToLogConstant(c string) log.Level {
+	level, ok := logConstants[c]
+	if !ok {
+		level = log.FatalLevel
+	}
+
+	return level
+}
+
+// SetLogrusLevel sets Logrus logging level from a string
+func SetLogrusLevel(level string) {
+	log.SetLevel(stringToLogConstant(level))
+}