pkg: create loglevel package

shellhub-io · Sep 19, 2022 · 067fc04 · 067fc04
1 parent f20e991
commit 067fc04
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/pkg/loglevel/loglevel.go b/pkg/loglevel/loglevel.go
@@ -0,0 +1,24 @@
+package loglevel
+
+import (
+	"os"
+
+	"github.com/sirupsen/logrus"
+)
+
+func SetLogLevel() {
+	level := logrus.InfoLevel
+
+	if env, ok := os.LookupEnv("SHELLHUB_ENV"); ok && env == "development" {
+		level = logrus.TraceLevel
+	}
+
+	if env, ok := os.LookupEnv("SHELLHUB_LOG_LEVEL"); ok {
+		if v, err := logrus.ParseLevel(env); err != nil {
+			level = v
+		}
+	}
+
+	logrus.WithField("log_level", level.String()).Info("Setting log level")
+	logrus.SetLevel(level)
+}