Add KeyfuncCtx method to interface (#120)

MicahParks · Mar 25, 2024 · 8e0299c · 8e0299c
1 parent b9d1823
commit 8e0299c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/keyfunc.go b/keyfunc.go
@@ -20,7 +20,7 @@ var (
 // github.com/MicahParks/jwkset as a JWK Set storage.
 type Keyfunc interface {
 	Keyfunc(token *jwt.Token) (any, error)
-	KeyfuncCtx(ctx context.Context) func(token *jwt.Token) (any, error)
+	KeyfuncCtx(ctx context.Context) jwt.Keyfunc
 	Storage() jwkset.Storage
 }
 
@@ -113,7 +113,7 @@ func NewJWKSetJSON(raw json.RawMessage) (Keyfunc, error) {
 	return New(options)
 }
 
-func (k keyfunc) KeyfuncCtx(ctx context.Context) func(token *jwt.Token) (any, error) {
+func (k keyfunc) KeyfuncCtx(ctx context.Context) jwt.Keyfunc {
 	return func(token *jwt.Token) (any, error) {
 		kidInter, ok := token.Header[jwkset.HeaderKID]
 		if !ok {