feat: add matrix category support

penguin-statistics · Aug 29, 2022 · 86fb8c7 · 86fb8c7
1 parent c8c3bb3
commit 86fb8c7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/internal/server/httpserver/error.go b/internal/server/httpserver/error.go
@@ -75,6 +75,11 @@ func ErrorHandler(ctx *fiber.Ctx, err error) error {
 		return HandleCustomError(ctx, pgerr.ErrInvalidReq)
 	}
 
+	if e, ok := err.(*pgerr.PenguinError); ok {
+		// Use custom error handler if it's a custom error
+		return HandleCustomError(ctx, e)
+	}
+
 	// must be an unexpected runtime error then
 	log.Error().
 		Stack().

diff --git a/internal/util/rekuest/validate.go b/internal/util/rekuest/validate.go
@@ -175,7 +175,7 @@ func ValidServer(ctx *fiber.Ctx, server string) error {
 
 func ValidCategory(ctx *fiber.Ctx, category string) error {
 	type request struct {
-		Category string `validate:"oneof=automated manual"`
+		Category string `validate:"oneof=all automated manual"`
 	}
 
 	if err := ValidStruct(ctx, request{category}); err != nil {