package main

import (
    "github.com/sirupsen/logrus"
    "github.com/spf13/viper"
)

var logger *logrus.Logger

func initLogger() {
    logger = &logrus.Logger{
        Out:          logrus.StandardLogger().Out,
        Formatter:    &logrus.TextFormatter{},
        Level:        logrus.InfoLevel,
        ReportCaller: true,
    }
    logLevel := viper.GetString("log.level")
    switch logLevel {
    case "debug":
        logger.SetLevel(logrus.DebugLevel)
    case "info":
        logger.SetLevel(logrus.InfoLevel)
    case "warn":
        logger.SetLevel(logrus.WarnLevel)
    case "error":
        logger.SetLevel(logrus.ErrorLevel)
    default:
        logger.SetLevel(logrus.InfoLevel)
    }
}