feat: add ValidationMiddleware to RouteModuleConfig

internal/handlers/common.go

@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"net/http"
 	"net/url"
+	"reflect"
 	"strconv"
 	"strings"
 	"time"
@@ -290,3 +291,24 @@ func HandleServiceError(w http.ResponseWriter, err error, defaultMsg string, def
 	SendErrorResponse(w, defaultMsg, defaultCode)
 	return false
 }
+
+func GetValidatedDTO[T any](r *http.Request) (*T, bool) {
+	dtoVal := middleware.GetValidatedDTOFromContext(r.Context())
+	if dtoVal == nil {
+		return nil, false
+	}
+	dto, ok := dtoVal.(*T)
+	return dto, ok
+}
+
+func WithValidation[T any](validationMiddleware func(http.Handler) http.Handler, handler http.HandlerFunc) http.HandlerFunc {
+	if validationMiddleware == nil {
+		return handler
+	}
+	var zero T
+	dtoType := reflect.TypeOf(zero)
+	return func(w http.ResponseWriter, r *http.Request) {
+		ctx := middleware.SetDTOTypeInContext(r.Context(), dtoType)
+		validationMiddleware(handler).ServeHTTP(w, r.WithContext(ctx))
+	}
+}

internal/handlers/routes.go

@@ -13,9 +13,10 @@ type RouteModule interface {
 }
 
 type RouteModuleConfig struct {
-	AuthService      middleware.TokenVerifier
-	GeneralRateLimit func(chi.Router) chi.Router
-	AuthRateLimit    func(chi.Router) chi.Router
-	CSRFMiddleware   func(http.Handler) http.Handler
-	AuthMiddleware   func(http.Handler) http.Handler
+	AuthService          middleware.TokenVerifier
+	GeneralRateLimit     func(chi.Router) chi.Router
+	AuthRateLimit        func(chi.Router) chi.Router
+	CSRFMiddleware       func(http.Handler) http.Handler
+	AuthMiddleware       func(http.Handler) http.Handler
+	ValidationMiddleware func(http.Handler) http.Handler
 }

internal/server/router.go

@@ -71,8 +71,9 @@ func NewRouter(cfg RouterConfig) http.Handler {
 		AuthRateLimit: func(r chi.Router) chi.Router {
 			return r.With(middleware.AuthRateLimitMiddlewareWithLimit(cfg.RateLimitConfig.AuthLimit))
 		},
-		CSRFMiddleware: middleware.CSRFMiddleware(),
-		AuthMiddleware: middleware.NewAuth(cfg.AuthService),
+		CSRFMiddleware:       middleware.CSRFMiddleware(),
+		AuthMiddleware:       middleware.NewAuth(cfg.AuthService),
+		ValidationMiddleware: middleware.ValidationMiddleware(),
 	}
 
 	if cfg.PageHandler != nil {