-
Notifications
You must be signed in to change notification settings - Fork 0
/
jwt.go
50 lines (39 loc) · 1.35 KB
/
jwt.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
package utils
import (
"github.com/golang-jwt/jwt/v5"
"github.com/google/uuid"
"log"
"os"
)
type UserClaims struct {
Id uuid.UUID `json:"id"`
Email string `json:"email"`
Password string `json:"password"`
jwt.RegisteredClaims
}
func init() {
err := os.Setenv("TOKEN_SECRET", "TOKEN_SECRET")
if err != nil {
log.Fatal("fail to set token in the os env")
}
}
func NewAccessToken(claims UserClaims) (string, error) {
accessToken := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
return accessToken.SignedString([]byte(os.Getenv("TOKEN_SECRET")))
}
func NewRefreshToken(claims jwt.RegisteredClaims) (string, error) {
refreshToken := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
return refreshToken.SignedString([]byte(os.Getenv("TOKEN_SECRET")))
}
func ParseAccessToken(accessToken string) *UserClaims {
parsedAccessToken, _ := jwt.ParseWithClaims(accessToken, &UserClaims{}, func(token *jwt.Token) (interface{}, error) {
return []byte(os.Getenv("TOKEN_SECRET")), nil
})
return parsedAccessToken.Claims.(*UserClaims)
}
func ParseRefreshToken(refreshToken string) *jwt.RegisteredClaims {
parsedRefreshToken, _ := jwt.ParseWithClaims(refreshToken, &jwt.RegisteredClaims{}, func(token *jwt.Token) (interface{}, error) {
return []byte(os.Getenv("TOKEN_SECRET")), nil
})
return parsedRefreshToken.Claims.(*jwt.RegisteredClaims)
}