Files
memes-telegram-integration/config.go
2024-08-27 00:06:39 +08:00

38 lines
762 B
Go

package memogram
import (
"os"
"path"
"github.com/caarlos0/env"
"github.com/joho/godotenv"
"github.com/pkg/errors"
)
type Config struct {
ServerAddr string `env:"SERVER_ADDR,required"`
BotToken string `env:"BOT_TOKEN,required"`
Data string `env:"DATA"`
}
func getConfigFromEnv() (*Config, error) {
envFileName := ".env"
if _, err := os.Stat(envFileName); err == nil {
err := godotenv.Load(envFileName)
if err != nil {
panic(err.Error())
}
}
config := Config{}
if err := env.Parse(&config); err != nil {
return nil, errors.Wrap(err, "invalid configuration")
}
if config.Data == "" {
// Default to `data.txt` if not specified.
config.Data = "data.txt"
}
config.Data = path.Join(".", config.Data)
return &config, nil
}