Key auth validator now accepts context

Signed-off-by: Vishal Rana <vr@labstack.com>

middleware/key_auth.go

@@ -32,7 +32,7 @@ type (
 	}
 
 	// KeyAuthValidator defines a function to validate KeyAuth credentials.
-	KeyAuthValidator func(string) bool
+	KeyAuthValidator func(string, echo.Context) bool
 
 	keyExtractor func(echo.Context) (string, error)
 )
@@ -94,7 +94,7 @@ func KeyAuthWithConfig(config KeyAuthConfig) echo.MiddlewareFunc {
 			if err != nil {
 				return echo.NewHTTPError(http.StatusBadRequest, err.Error())
 			}
-			if config.Validator(key) {
+			if config.Validator(key, c) {
 				return next(c)
 			}
 

middleware/key_auth_test.go

@@ -15,7 +15,7 @@ func TestKeyAuth(t *testing.T) {
 	res := httptest.NewRecorder()
 	c := e.NewContext(req, res)
 	config := KeyAuthConfig{
-		Validator: func(key string) bool {
+		Validator: func(key string, c echo.Context) bool {
 			return key == "valid-key"
 		},
 	}