mirror of
https://github.com/NotAShelf/goblin.git
synced 2024-11-01 11:01:17 +00:00
45 lines
1.2 KiB
Go
45 lines
1.2 KiB
Go
package config
|
|
|
|
import (
|
|
"github.com/spf13/viper"
|
|
"log"
|
|
)
|
|
|
|
type AppConfig struct {
|
|
Port string
|
|
Private bool
|
|
TemplateDir string
|
|
LogDir string
|
|
PasteDir string
|
|
Expire int // Add an expiration setting
|
|
}
|
|
|
|
func LoadConfig() (*AppConfig, error) {
|
|
var config AppConfig
|
|
|
|
viper.SetConfigName("config") // name of the config file (without extension)
|
|
viper.AddConfigPath(".") // look in the current directory for the config file
|
|
viper.AutomaticEnv() // read in environment variables that match
|
|
|
|
viper.SetDefault("Port", "8080")
|
|
viper.SetDefault("Private", false)
|
|
viper.SetDefault("TemplateDir", "templates")
|
|
viper.SetDefault("LogDir", "logs")
|
|
viper.SetDefault("PasteDir", "pastes")
|
|
viper.SetDefault("Expire", 24) // in hours
|
|
viper.SetDefault("Metrics", false)
|
|
|
|
// read the configuration file if it's present
|
|
// goblin can also be configured through command lines during runtime
|
|
if err := viper.ReadInConfig(); err != nil {
|
|
log.Printf("Failed to read config file: %v\n", err)
|
|
}
|
|
|
|
// Unmarshal the configuration into the AppConfig struct
|
|
if err := viper.Unmarshal(&config); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &config, nil
|
|
}
|