mirror of
https://github.com/rls-moe/nyx
synced 2024-09-27 16:03:45 +02:00
29 lines
613 B
Go
29 lines
613 B
Go
package middle
|
|
|
|
import (
|
|
"context"
|
|
"go.rls.moe/nyx/config"
|
|
"net/http"
|
|
)
|
|
|
|
func ConfigCtx(config *config.Config) func(http.Handler) http.Handler {
|
|
return func(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if !config.IsHostNameValid(r.Host) {
|
|
return
|
|
}
|
|
|
|
r = r.WithContext(context.WithValue(r.Context(), configKey, config))
|
|
next.ServeHTTP(w, r)
|
|
})
|
|
}
|
|
}
|
|
|
|
func GetConfig(r *http.Request) *config.Config {
|
|
val := r.Context().Value(configKey)
|
|
if val == nil {
|
|
panic("Config Middleware not configured")
|
|
}
|
|
return val.(*config.Config)
|
|
}
|