From fb585fabe6048959e617c20fe49f79f068172121 Mon Sep 17 00:00:00 2001
From: Yavolte <Yavolte@nobla.cn>
Date: Mon, 30 Jun 2025 11:25:55 +0800
Subject: [PATCH] use jwt claims

---
 middleware/auth/jwt.go | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/middleware/auth/jwt.go b/middleware/auth/jwt.go
index b49ba0d..f71951f 100644
--- a/middleware/auth/jwt.go
+++ b/middleware/auth/jwt.go
@@ -57,9 +57,16 @@ func WithAllow(paths ...string) Option {
 	}
 }
 
-func WithClaims(claims reflect.Type) Option {
+func WithClaims(claims any) Option {
 	return func(o *options) {
-		o.claims = claims
+		if tv, ok := claims.(reflect.Type); ok {
+			o.claims = tv
+		} else {
+			o.claims = reflect.TypeOf(claims)
+			if o.claims.Kind() == reflect.Ptr {
+				o.claims = o.claims.Elem()
+			}
+		}
 	}
 }