1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
package internal
import (
"fmt"
"os"
"strconv"
"strings"
)
// Default configuration values.
const (
DefaultPort = 8080
DefaultMediaRoot = "./media"
DefaultDBPath = "data.db"
DefaultMaxUploadSizeMB = 100
DefaultSessionTimeoutHours = 24
DefaultGCIntervalMinutes = 30
DefaultShareDefaultExpiryDays = 7
DefaultLogLevel = "info"
DefaultSecureCookies = true
)
// Config holds all application configuration loaded from environment variables.
type Config struct {
Port int
MediaRoot string
DBPath string
MaxUploadSizeMB int
SessionTimeoutHours int
GCIntervalMinutes int
ShareDefaultExpiryDays int
LogLevel string
SecureCookies bool
}
// envInt reads an integer environment variable, validates it with the given check,
// and sets the field via the provided setter. If the variable is unset, the setter
// is not called and the default remains.
func envInt(name string, check func(int) error, set func(int)) error {
if v := os.Getenv(name); v != "" {
n, err := strconv.Atoi(v)
if err != nil {
return fmt.Errorf("invalid %s: %w", name, err)
}
if err := check(n); err != nil {
return fmt.Errorf("invalid %s: %w", name, err)
}
set(n)
}
return nil
}
// envString reads a string environment variable, trims space, and sets the field
// via the provided setter if the variable is non-empty.
func envString(name string, set func(string)) {
if v := os.Getenv(name); v != "" {
set(strings.TrimSpace(v))
}
}
// LoadConfig reads configuration from environment variables and returns
// a populated Config. Unset variables use the package defaults.
func LoadConfig() (*Config, error) {
cfg := &Config{
Port: DefaultPort,
MediaRoot: DefaultMediaRoot,
DBPath: DefaultDBPath,
MaxUploadSizeMB: DefaultMaxUploadSizeMB,
SessionTimeoutHours: DefaultSessionTimeoutHours,
GCIntervalMinutes: DefaultGCIntervalMinutes,
ShareDefaultExpiryDays: DefaultShareDefaultExpiryDays,
LogLevel: DefaultLogLevel,
SecureCookies: DefaultSecureCookies,
}
validLevels := map[string]struct{}{
"debug": {},
"info": {},
"warn": {},
"error": {},
}
if err := envInt("PORT", func(n int) error {
if n < 1 || n > 65535 {
return fmt.Errorf("must be between 1 and 65535, got %d", n)
}
return nil
}, func(n int) { cfg.Port = n }); err != nil {
return nil, err
}
envString("MEDIA_ROOT", func(s string) { cfg.MediaRoot = s })
envString("DB_PATH", func(s string) { cfg.DBPath = s })
if err := envInt("MAX_UPLOAD_SIZE_MB", func(n int) error {
if n < 1 {
return fmt.Errorf("must be >= 1, got %d", n)
}
return nil
}, func(n int) { cfg.MaxUploadSizeMB = n }); err != nil {
return nil, err
}
if err := envInt("SESSION_TIMEOUT_HOURS", func(n int) error {
if n < 1 {
return fmt.Errorf("must be >= 1, got %d", n)
}
return nil
}, func(n int) { cfg.SessionTimeoutHours = n }); err != nil {
return nil, err
}
if err := envInt("GC_INTERVAL_MINUTES", func(n int) error {
if n < 1 {
return fmt.Errorf("must be >= 1, got %d", n)
}
return nil
}, func(n int) { cfg.GCIntervalMinutes = n }); err != nil {
return nil, err
}
if err := envInt("SHARE_DEFAULT_EXPIRY_DAYS", func(n int) error {
if n < 1 {
return fmt.Errorf("must be >= 1, got %d", n)
}
return nil
}, func(n int) { cfg.ShareDefaultExpiryDays = n }); err != nil {
return nil, err
}
if v := os.Getenv("LOG_LEVEL"); v != "" {
level := strings.ToLower(strings.TrimSpace(v))
if _, ok := validLevels[level]; !ok {
return nil, fmt.Errorf("invalid LOG_LEVEL: must be one of debug, info, warn, error, got %q", level)
}
cfg.LogLevel = level
}
if v := os.Getenv("SECURE_COOKIES"); v != "" {
b, err := strconv.ParseBool(strings.TrimSpace(v))
if err != nil {
return nil, fmt.Errorf("invalid SECURE_COOKIES: %w", err)
}
cfg.SecureCookies = b
}
return cfg, nil
}
|
