roman.nikolsky
134 lines
2.9 KiB
Go
134 lines
2.9 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
type Config struct {
|
|
// Matrix
|
|
MatrixHomeserver string
|
|
MatrixUserID string
|
|
MatrixAccessToken string
|
|
MatrixDeviceID string
|
|
MatrixEncryption bool
|
|
|
|
// Search & Download
|
|
SearchLimit int
|
|
SearchTimeout time.Duration
|
|
DownloadTimeout time.Duration
|
|
AudioFormat string
|
|
AudioQualityPreview int
|
|
AudioQualityFinal int
|
|
PreviewTTL time.Duration
|
|
AllowedUsers []string
|
|
|
|
// Proxy
|
|
ProxyEnabled bool
|
|
ProxyURL string
|
|
|
|
// Paths
|
|
MusicDir string
|
|
MusicSubdir string
|
|
TmpDir string
|
|
YtdlpPath string
|
|
|
|
// Logging
|
|
LogLevel string
|
|
|
|
// HTTP
|
|
ListenAddr string
|
|
}
|
|
|
|
func Load() *Config {
|
|
return &Config{
|
|
MatrixHomeserver: getEnv("MATRIX_HOMESERVER", ""),
|
|
MatrixUserID: getEnv("MATRIX_USER_ID", ""),
|
|
MatrixAccessToken: getEnv("MATRIX_ACCESS_TOKEN", ""),
|
|
MatrixDeviceID: getEnv("MATRIX_DEVICE_ID", "YTDLBOT"),
|
|
MatrixEncryption: getEnvBool("MATRIX_ENCRYPTION", false),
|
|
|
|
SearchLimit: getEnvInt("SEARCH_LIMIT", 10),
|
|
SearchTimeout: getEnvDuration("SEARCH_TIMEOUT", 10*time.Second),
|
|
DownloadTimeout: getEnvDuration("DOWNLOAD_TIMEOUT", 120*time.Second),
|
|
AudioFormat: getEnv("AUDIO_FORMAT", "opus"),
|
|
AudioQualityPreview: getEnvInt("AUDIO_QUALITY_PREVIEW", 5),
|
|
AudioQualityFinal: getEnvInt("AUDIO_QUALITY_FINAL", 0),
|
|
PreviewTTL: getEnvDuration("PREVIEW_TTL", 10*time.Minute),
|
|
AllowedUsers: getEnvSlice("ALLOWED_USERS", nil),
|
|
|
|
ProxyEnabled: getEnvBool("PROXY_ENABLED", true),
|
|
ProxyURL: getEnv("PROXY_URL", "socks5://localhost:1080"),
|
|
|
|
MusicDir: getEnv("MUSIC_DIR", "/music"),
|
|
MusicSubdir: getEnv("MUSIC_SUBDIR", "ytdlp"),
|
|
TmpDir: getEnv("TMP_DIR", "/tmp/ytdlp-previews"),
|
|
YtdlpPath: getEnv("YTDLP_PATH", "yt-dlp"),
|
|
|
|
LogLevel: getEnv("LOG_LEVEL", "INFO"),
|
|
ListenAddr: getEnv("LISTEN_ADDR", ":8080"),
|
|
}
|
|
}
|
|
|
|
func (c *Config) MusicCollectionDir() string {
|
|
return strings.TrimRight(c.MusicDir, "/") + "/" + c.MusicSubdir
|
|
}
|
|
|
|
func getEnv(key, fallback string) string {
|
|
if v := os.Getenv(key); v != "" {
|
|
return v
|
|
}
|
|
return fallback
|
|
}
|
|
|
|
func getEnvInt(key string, fallback int) int {
|
|
v := os.Getenv(key)
|
|
if v == "" {
|
|
return fallback
|
|
}
|
|
n, err := strconv.Atoi(v)
|
|
if err != nil {
|
|
return fallback
|
|
}
|
|
return n
|
|
}
|
|
|
|
func getEnvBool(key string, fallback bool) bool {
|
|
v := os.Getenv(key)
|
|
if v == "" {
|
|
return fallback
|
|
}
|
|
switch strings.ToLower(v) {
|
|
case "true", "1", "yes":
|
|
return true
|
|
default:
|
|
return false
|
|
}
|
|
}
|
|
|
|
func getEnvDuration(key string, fallback time.Duration) time.Duration {
|
|
v := os.Getenv(key)
|
|
if v == "" {
|
|
return fallback
|
|
}
|
|
d, err := time.ParseDuration(v)
|
|
if err != nil {
|
|
return fallback
|
|
}
|
|
return d
|
|
}
|
|
|
|
func getEnvSlice(key string, fallback []string) []string {
|
|
v := os.Getenv(key)
|
|
if v == "" {
|
|
return fallback
|
|
}
|
|
parts := strings.Split(v, ",")
|
|
for i := range parts {
|
|
parts[i] = strings.TrimSpace(parts[i])
|
|
}
|
|
return parts
|
|
}
|