54 lines
1.1 KiB
Go
54 lines
1.1 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"log/slog"
|
||
|
|
"os"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/kamikazechaser/common/logg"
|
||
|
|
"github.com/knadh/koanf/parsers/toml"
|
||
|
|
"github.com/knadh/koanf/providers/env"
|
||
|
|
"github.com/knadh/koanf/providers/file"
|
||
|
|
"github.com/knadh/koanf/v2"
|
||
|
|
)
|
||
|
|
|
||
|
|
func initLogger() *slog.Logger {
|
||
|
|
loggOpts := logg.LoggOpts{
|
||
|
|
FormatType: logg.Logfmt,
|
||
|
|
LogLevel: slog.LevelInfo,
|
||
|
|
}
|
||
|
|
|
||
|
|
if os.Getenv("DEBUG") != "" {
|
||
|
|
loggOpts.LogLevel = slog.LevelDebug
|
||
|
|
}
|
||
|
|
|
||
|
|
if os.Getenv("DEV") != "" {
|
||
|
|
loggOpts.LogLevel = slog.LevelDebug
|
||
|
|
loggOpts.FormatType = logg.Human
|
||
|
|
}
|
||
|
|
|
||
|
|
return logg.NewLogg(loggOpts)
|
||
|
|
}
|
||
|
|
|
||
|
|
func initConfig() *koanf.Koanf {
|
||
|
|
var (
|
||
|
|
ko = koanf.New(".")
|
||
|
|
)
|
||
|
|
|
||
|
|
confFile := file.Provider(confFlag)
|
||
|
|
if err := ko.Load(confFile, toml.Parser()); err != nil {
|
||
|
|
lo.Error("could not parse configuration file", "error", err)
|
||
|
|
os.Exit(1)
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := ko.Load(env.Provider("INDEXER_", ".", func(s string) string {
|
||
|
|
return strings.ReplaceAll(strings.ToLower(
|
||
|
|
strings.TrimPrefix(s, "INDEXER_")), "__", ".")
|
||
|
|
}), nil); err != nil {
|
||
|
|
lo.Error("could not override config from env vars", "error", err)
|
||
|
|
os.Exit(1)
|
||
|
|
}
|
||
|
|
|
||
|
|
return ko
|
||
|
|
}
|