test: added Validator middleware powered by reflection

Author: ralvarezdev

go.mod

@@ -6,12 +6,12 @@ require (
 	github.com/ralvarezdev/go-flags v0.3.1
 	github.com/ralvarezdev/go-jwt v0.4.8
 	github.com/ralvarezdev/go-logger v0.4.5
-	github.com/ralvarezdev/go-validator v0.5.21
+	github.com/ralvarezdev/go-reflect v0.2.1
+	github.com/ralvarezdev/go-validator v0.5.22
 )
 
 require (
 	github.com/golang-jwt/jwt/v5 v5.2.1 // indirect
-	github.com/ralvarezdev/go-reflect v0.1.0 // indirect
 	github.com/ralvarezdev/go-strings v0.1.6 // indirect
 	golang.org/x/crypto v0.32.0 // indirect
 )

go.sum

@@ -6,11 +6,11 @@ github.com/ralvarezdev/go-jwt v0.4.8 h1:ugZ0Lyfen7QzmgweLnJynya3yih8rPev48Xjj7HY
 github.com/ralvarezdev/go-jwt v0.4.8/go.mod h1:RGGbn72uT5av5Hlo50H4xSXg6chmnpecl/a2yGAFNzQ=
 github.com/ralvarezdev/go-logger v0.4.5 h1:Lw7DdZUb4YSQDA7J9nv79soojw1eFHVVzO8wdE9d+Qg=
 github.com/ralvarezdev/go-logger v0.4.5/go.mod h1:vq47onxpED19/o3dh5/YbBKK0eZhqJENPgACYkqNNi8=
-github.com/ralvarezdev/go-reflect v0.1.0 h1:q0CypQURgOgxkLHilYVFazYi3nlSxNdkjSoYpuhFYOA=
-github.com/ralvarezdev/go-reflect v0.1.0/go.mod h1:jv3EqQfUixWtNeJd/Rnzf0IDcYZ1z3GApRdAGQHdHZQ=
+github.com/ralvarezdev/go-reflect v0.2.1 h1:Q200FruGvc8INRRpnmvK+iRKTnCslRAWnwGAj1fXMak=
+github.com/ralvarezdev/go-reflect v0.2.1/go.mod h1:jv3EqQfUixWtNeJd/Rnzf0IDcYZ1z3GApRdAGQHdHZQ=
 github.com/ralvarezdev/go-strings v0.1.6 h1:NWI+XCipWaOo0vhSNpZz+7Eb4igk/GXD5RaFsk4ZPqY=
 github.com/ralvarezdev/go-strings v0.1.6/go.mod h1:8sFOqmPJpqzS7bTjf91EzUCITnwpmkfifwY80GxV5r8=
-github.com/ralvarezdev/go-validator v0.5.21 h1:Y2ndLcDUlgjHNG6pVSu4poE30OvWAGnKdtU+Gu9bBPg=
-github.com/ralvarezdev/go-validator v0.5.21/go.mod h1:q24kfA3gLlcYkMo/W6Ip7JEpWtZGEJBN6Rvn0N1IOF0=
+github.com/ralvarezdev/go-validator v0.5.22 h1:mjvWKwhQyqQ6FwJUWIPfsywjOK0nMYf7HKDYAojh1xo=
+github.com/ralvarezdev/go-validator v0.5.22/go.mod h1:q24kfA3gLlcYkMo/W6Ip7JEpWtZGEJBN6Rvn0N1IOF0=
 golang.org/x/crypto v0.32.0 h1:euUpcYgM8WcP71gNpTqQCn6rC2t6ULUPiOzfWaXVVfc=
 golang.org/x/crypto v0.32.0/go.mod h1:ZnnJkOaASj8g0AjIduWNlq2NRxL0PlBrbKVyZ6V/Ugc=

http/context/constants.go

@@ -0,0 +1,6 @@
+package context
+
+const (
+	// CtxBodyKey is the context key for the body
+	CtxBodyKey = "body"
+)

http/context/context.go

@@ -0,0 +1,22 @@
+package context
+
+import (
+	"context"
+	"net/http"
+)
+
+// SetCtxBody sets the body in the context
+func SetCtxBody(r *http.Request, body interface{}) *http.Request {
+	ctx := context.WithValue(r.Context(), CtxBodyKey, body)
+	return r.WithContext(ctx)
+}
+
+// GetCtxBody tries to get the body from the context
+func GetCtxBody(r *http.Request) (interface{}, error) {
+	// Get the token claims from the context
+	value := r.Context().Value(CtxBodyKey)
+	if value == nil {
+		return nil, ErrMissingBodyInContext
+	}
+	return value, nil
+}

http/context/errors.go

@@ -0,0 +1,9 @@
+package context
+
+import (
+	"errors"
+)
+
+var (
+	ErrMissingBodyInContext = errors.New("missing body in context")
+)

http/middleware/validator/errors.go

@@ -0,0 +1,10 @@
+package validator
+
+import (
+	"errors"
+)
+
+var (
+	ErrNilValidateFn     = errors.New("validate function is nil")
+	ErrInvalidValidateFn = errors.New("validate function is invalid")
+)

http/middleware/validator/middleware.go

@@ -0,0 +1,96 @@
+package validator
+
+import (
+	"context"
+	gonethttpctx "github.com/ralvarezdev/go-net/http/context"
+	gonethttphandler "github.com/ralvarezdev/go-net/http/handler"
+	goreflect "github.com/ralvarezdev/go-reflect"
+	govalidatorstructmapper "github.com/ralvarezdev/go-validator/struct/mapper"
+	"net/http"
+)
+
+// Middleware struct
+type Middleware struct {
+	handler   gonethttphandler.Handler
+	generator govalidatorstructmapper.Generator
+}
+
+// NewMiddleware creates a new Middleware instance
+func NewMiddleware(
+	handler gonethttphandler.Handler,
+	generator govalidatorstructmapper.Generator,
+) (*Middleware, error) {
+	// Check if the handler or the generator is nil
+	if handler == nil {
+		return nil, gonethttphandler.ErrNilHandler
+	}
+	if generator == nil {
+		return nil, govalidatorstructmapper.ErrNilGenerator
+	}
+
+	return &Middleware{
+		handler: handler,
+	}, nil
+}
+
+// Validate validates the request body and stores it in the context
+func (m *Middleware) Validate(
+	body interface{},
+	createValidateFn interface{},
+) func(next http.Handler) http.Handler {
+	// Get the type of the body
+	bodyType := goreflect.GetTypeOf(body)
+
+	// Create the mapper
+	mapper, err := m.generator.NewMapper(body)
+	if err != nil {
+		panic(err)
+	}
+
+	return func(next http.Handler) http.Handler {
+		return http.HandlerFunc(
+			func(w http.ResponseWriter, r *http.Request) {
+				// Get a new instance of the body
+				dest := goreflect.NewInstanceFromType(bodyType)
+
+				// Get the validate function
+				results, err := goreflect.CallFunction(
+					createValidateFn,
+					dest,
+					mapper,
+				)
+				if err != nil {
+					panic(err)
+				}
+				validateFn := results[0]
+
+				// Parse the validate function
+				if validateFn == nil {
+					panic(ErrNilValidateFn)
+				}
+				parsedValidateFn, ok := validateFn.(func() (interface{}, error))
+				if !ok {
+					panic(ErrInvalidValidateFn)
+				}
+
+				// Decode the request body
+				if !m.handler.Parse(
+					w,
+					r,
+					dest,
+					parsedValidateFn,
+				) {
+					return
+				}
+
+				// Store the validated body in the context
+				ctx := context.WithValue(
+					r.Context(),
+					gonethttpctx.CtxBodyKey,
+					dest,
+				)
+				next.ServeHTTP(w, r.WithContext(ctx))
+			},
+		)
+	}
+}

http/middleware/validator/validator.go

@@ -0,0 +1,12 @@
+package validator
+
+import (
+	"net/http"
+)
+
+// Validator interface
+type Validator interface {
+	Validate(
+		createValidateFn func() (interface{}, func() (interface{}, error)),
+	) func(next http.Handler) http.Handler
+}