Merge pull request #47 from xushiwei/test

yap/ytest/auth/jwt

Author: xushiwei

go.mod

@@ -2,4 +2,7 @@ module github.com/goplus/yap
 
 go 1.18
 
-require github.com/qiniu/x v1.13.3-0.20240129201727-4013f184c5c7
+require (
+    github.com/qiniu/x v1.13.3-0.20240129201727-4013f184c5c7
+    github.com/golang-jwt/jwt/v5 v5.2.0
+)

go.sum

@@ -1,2 +1,4 @@
+github.com/golang-jwt/jwt/v5 v5.2.0 h1:d/ix8ftRUorsN+5eMIlF4T6J8CAt9rch3My2winC1Jw=
+github.com/golang-jwt/jwt/v5 v5.2.0/go.mod h1:pqrtFR0X4osieyHYxtmOUWsAWrfe1Q5UVIyoH402zdk=
 github.com/qiniu/x v1.13.3-0.20240129201727-4013f184c5c7 h1:rzpGloMVgWexXti9yRXMH8aAdjNy8rZrlAQdKvzZeMg=
 github.com/qiniu/x v1.13.3-0.20240129201727-4013f184c5c7/go.mod h1:INZ2TSWSJVWO/RuELQROERcslBwVgFG7MkTfEdaQz9E=

gop.mod

@@ -4,6 +4,8 @@ project _yap.gox App github.com/goplus/yap
 
 project _yapt.gox App github.com/goplus/yap/ytest
 class _yapt.gox Case
+import github.com/goplus/yap/ytest/auth/jwt
 
 project _ytest.gox App github.com/goplus/yap/ytest
 class _ytest.gox Case
+import github.com/goplus/yap/ytest/auth/jwt

ytest/auth/auth.go

@@ -24,3 +24,24 @@ import (
 type RTComposer interface {
 	Compose(base http.RoundTripper) http.RoundTripper
 }
+
+// -----------------------------------------------------------------------------
+
+type tokenRT struct {
+	rt    http.RoundTripper
+	token string
+}
+
+func (p *tokenRT) RoundTrip(req *http.Request) (*http.Response, error) {
+	req.Header.Set("Authorization", p.token)
+	return p.rt.RoundTrip(req)
+}
+
+func WithToken(rt http.RoundTripper, token string) http.RoundTripper {
+	return &tokenRT{
+		rt:    rt,
+		token: token,
+	}
+}
+
+// -----------------------------------------------------------------------------

ytest/auth/bearer/token.go

@@ -24,27 +24,12 @@ import (
 
 // -----------------------------------------------------------------------------
 
-type tokenRounderTripper struct {
-	rt    http.RoundTripper
-	token string
-}
-
-func (p *tokenRounderTripper) RoundTrip(req *http.Request) (*http.Response, error) {
-	req.Header.Set("Authorization", "Bearer "+p.token)
-	return p.rt.RoundTrip(req)
-}
-
-// -----------------------------------------------------------------------------
-
 type tokenAuth struct {
 	token string
 }
 
 func (p *tokenAuth) Compose(rt http.RoundTripper) http.RoundTripper {
-	return &tokenRounderTripper{
-		rt:    rt,
-		token: p.token,
-	}
+	return auth.WithToken(rt, "Bearer "+p.token)
 }
 
 // New creates an Authorization by specified token.

ytest/auth/jwt/jwt.go

@@ -0,0 +1,93 @@
+/*
+ * Copyright (c) 2024 The GoPlus Authors (goplus.org). All rights reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package jwt
+
+import (
+	"log"
+	"net/http"
+	"time"
+
+	"github.com/golang-jwt/jwt/v5"
+	"github.com/goplus/yap/ytest/auth"
+)
+
+// -----------------------------------------------------------------------------
+
+type Signaturer struct {
+	method jwt.SigningMethod
+	claims jwt.MapClaims
+	secret any
+}
+
+func (p *Signaturer) Set(k string, v any) *Signaturer {
+	p.claims[k] = v
+	return p
+}
+
+func (p *Signaturer) Audience(aud ...string) *Signaturer {
+	p.claims["aud"] = jwt.ClaimStrings(aud)
+	return p
+}
+
+func (p *Signaturer) Expiration(exp time.Time) *Signaturer {
+	p.claims["exp"] = jwt.NewNumericDate(exp)
+	return p
+}
+
+func (p *Signaturer) NotBefore(nbf time.Time) *Signaturer {
+	p.claims["nbf"] = jwt.NewNumericDate(nbf)
+	return p
+}
+
+func (p *Signaturer) IssuedAt(iat time.Time) *Signaturer {
+	p.claims["iat"] = jwt.NewNumericDate(iat)
+	return p
+}
+
+func (p *Signaturer) Compose(rt http.RoundTripper) http.RoundTripper {
+	token := jwt.NewWithClaims(p.method, p.claims)
+	raw, err := token.SignedString(p.secret)
+	if err != nil {
+		log.Panicln("jwt token.SignedString:", err)
+	}
+	return auth.WithToken(rt, "Bearer "+raw)
+}
+
+func newSign(method jwt.SigningMethod, secret any) *Signaturer {
+	return &Signaturer{
+		method: method,
+		claims: make(jwt.MapClaims),
+		secret: secret,
+	}
+}
+
+// HS256 creates a signing methods by using the HMAC-SHA256.
+func HS256(key []byte) *Signaturer {
+	return newSign(jwt.SigningMethodHS256, key)
+}
+
+// HS384 creates a signing methods by using the HMAC-SHA384.
+func HS384(key []byte) *Signaturer {
+	return newSign(jwt.SigningMethodHS384, key)
+}
+
+// HS512 creates a signing methods by using the HMAC-SHA512.
+func HS512(key []byte) *Signaturer {
+	return newSign(jwt.SigningMethodHS512, key)
+}
+
+// -----------------------------------------------------------------------------