49 lines
1.2 KiB
Go
49 lines
1.2 KiB
Go
|
package utils
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"github.com/caarlos0/env"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Config struct {
|
||
|
|
E621APIKey string `env:"e621_API_KEY"`
|
||
|
|
E621Username string `env:"e621_USERNAME"`
|
||
|
|
Neo4jURL string `env:"NEO4J_URL"`
|
||
|
|
Neo4jUsername string `env:"NEO4J_USERNAME"`
|
||
|
|
Neo4jPassword string `env:"NEO4J_PASSWORD"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// LoadConfig loads the configuration from environment variables
|
||
|
|
func LoadConfig() (*Config, error) {
|
||
|
|
config := &Config{}
|
||
|
|
if err := env.Parse(config); err != nil {
|
||
|
|
return nil, fmt.Errorf("error parsing configuration: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := ValidateConfig(config); err != nil {
|
||
|
|
return nil, fmt.Errorf("configuration validation failed: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return config, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// ValidateConfig checks if all required fields in the configuration are set
|
||
|
|
func ValidateConfig(config *Config) error {
|
||
|
|
if config.E621APIKey == "" {
|
||
|
|
return fmt.Errorf("e621_API_KEY is not set")
|
||
|
|
}
|
||
|
|
if config.E621Username == "" {
|
||
|
|
return fmt.Errorf("e621_USERNAME is not set")
|
||
|
|
}
|
||
|
|
if config.Neo4jURL == "" {
|
||
|
|
return fmt.Errorf("NEO4J_URL is not set")
|
||
|
|
}
|
||
|
|
if config.Neo4jUsername == "" {
|
||
|
|
return fmt.Errorf("NEO4J_USERNAME is not set")
|
||
|
|
}
|
||
|
|
if config.Neo4jPassword == "" {
|
||
|
|
return fmt.Errorf("NEO4J_PASSWORD is not set")
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|