-
Notifications
You must be signed in to change notification settings - Fork 0
/
jwt.go
69 lines (56 loc) · 1.59 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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
package service
import (
"errors"
"time"
"github.com/hadihammurabi/belajar-go-graphql/config"
"github.com/hadihammurabi/belajar-go-graphql/internal/app/entity"
"github.com/sarulabs/di"
jwt "github.com/dgrijalva/jwt-go"
)
// JWTService struct
type JWTService struct {
Config *config.JWTConfig
UserService entity.UserService
TokenService entity.TokenService
}
// NewJWTService func
func NewJWTService(ioc di.Container) entity.JWTService {
jwtConfig := ioc.Get("config").(*(config.Config)).JWT
return &JWTService{
Config: jwtConfig,
UserService: NewUserService(ioc),
TokenService: NewTokenService(ioc),
}
}
// Create func
func (jwtService JWTService) Create(userData *entity.User) (*entity.Token, error) {
claims := &jwt.StandardClaims{}
claims.ExpiresAt = time.Now().Add(time.Hour * 3).Unix()
token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
t, err := token.SignedString([]byte(jwtService.Config.Secret))
if err != nil {
return nil, errors.New("token generation fail")
}
expToTime := time.Unix(claims.ExpiresAt, 0)
tokenCreated, err := jwtService.TokenService.Create(&entity.Token{
UserID: userData.ID,
Token: t,
ExpiredAt: &expToTime,
})
if err != nil {
return nil, errors.New("token generation fail")
}
return tokenCreated, nil
}
// GetUser func
func (jwtService JWTService) GetUser(token string) (*entity.User, error) {
tokenData, err := jwtService.TokenService.FindByToken(token)
if err != nil {
return nil, err
}
user, err := jwtService.UserService.FindByID(tokenData.UserID)
if err != nil {
return nil, err
}
return user, nil
}