88 lines
1.6 KiB
Go
88 lines
1.6 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
"sync"
|
|
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
var lock sync.Mutex
|
|
|
|
type serverConfig struct {
|
|
Port int `yaml:"port"`
|
|
DownloadPath string `yaml:"downloadPath"`
|
|
DownloaderPath string `yaml:"downloaderPath"`
|
|
RequireAuth bool `yaml:"require_auth"`
|
|
Username string `yaml:"username"`
|
|
Password string `yaml:"password"`
|
|
QueueSize int `yaml:"queue_size"`
|
|
SessionFilePath string `yaml:"session_file_path"`
|
|
}
|
|
|
|
type config struct {
|
|
cfg serverConfig
|
|
}
|
|
|
|
func (c *config) LoadFromFile(filename string) (serverConfig, error) {
|
|
fd, err := os.Open(filename)
|
|
if err != nil {
|
|
return serverConfig{}, err
|
|
}
|
|
|
|
if err := yaml.NewDecoder(fd).Decode(&c.cfg); err != nil {
|
|
return serverConfig{}, err
|
|
}
|
|
|
|
return c.cfg, nil
|
|
}
|
|
|
|
func (c *config) GetConfig() serverConfig {
|
|
return c.cfg
|
|
}
|
|
|
|
func (c *config) SetPort(port int) {
|
|
c.cfg.Port = port
|
|
}
|
|
|
|
func (c *config) DownloadPath(path string) {
|
|
c.cfg.DownloadPath = path
|
|
}
|
|
|
|
func (c *config) DownloaderPath(path string) {
|
|
c.cfg.DownloaderPath = path
|
|
}
|
|
|
|
func (c *config) RequireAuth(value bool) {
|
|
c.cfg.RequireAuth = value
|
|
}
|
|
|
|
func (c *config) Username(username string) {
|
|
c.cfg.Username = username
|
|
}
|
|
|
|
func (c *config) Password(password string) {
|
|
c.cfg.Password = password
|
|
}
|
|
|
|
func (c *config) QueueSize(size int) {
|
|
c.cfg.QueueSize = size
|
|
}
|
|
|
|
func (c *config) SessionFilePath(path string) {
|
|
c.cfg.SessionFilePath = path
|
|
}
|
|
|
|
var instance *config
|
|
|
|
func Instance() *config {
|
|
if instance == nil {
|
|
lock.Lock()
|
|
defer lock.Unlock()
|
|
if instance == nil {
|
|
instance = &config{serverConfig{}}
|
|
}
|
|
}
|
|
return instance
|
|
}
|