feat: add env vars to disable render endpoints

BLOCKY_DISABLE_GLB/PNG/GIF/MP4 return 403 when set to true
This commit is contained in:
devilreef 2026-01-23 01:36:43 +06:00
parent ce511e1a85
commit f8947fdcc4
5 changed files with 110 additions and 7 deletions

View file

@ -0,0 +1,35 @@
package api
import (
"encoding/json"
"net/http"
"blockyserver/internal/config"
)
// EndpointGuard creates middleware that returns 403 if endpoint is disabled
func EndpointGuard(enabled bool, endpointName string) func(http.Handler) http.Handler {
return func(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
if !enabled {
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(http.StatusForbidden)
json.NewEncoder(w).Encode(ErrorResponse{
Error: endpointName + " endpoint is disabled",
})
return
}
next.ServeHTTP(w, r)
})
}
}
// NewEndpointGuards creates guards for all render endpoints based on config
func NewEndpointGuards(cfg *config.EndpointConfig) map[string]func(http.Handler) http.Handler {
return map[string]func(http.Handler) http.Handler{
"glb": EndpointGuard(cfg.GLBEnabled, "/render/glb"),
"png": EndpointGuard(cfg.PNGEnabled, "/render/png"),
"gif": EndpointGuard(cfg.GIFEnabled, "/render/gif"),
"mp4": EndpointGuard(cfg.MP4Enabled, "/render/mp4"),
}
}