Merge pull request #15 from minusno0708/feat/validate-token-exp

トークンの有効期限による認証を追加

domain/token.go

@@ -53,3 +53,16 @@ func (t *Token) UUID() string {
 func (t *Token) UserID() string {
 	return t.value.Claims.(jwt.MapClaims)["user_id"].(string)
 }
+
+func (t *Token) Exp() int64 {
+	exp, ok := t.value.Claims.(jwt.MapClaims)["exp"].(int64)
+	if !ok {
+		exp = int64(t.value.Claims.(jwt.MapClaims)["exp"].(float64))
+	}
+	return exp
+
+}
+
+func (t *Token) IsExpired() bool {
+	return time.Now().Unix() > int64(t.Exp())
+}

domain/token_test.go

@@ -1,6 +1,8 @@
 package domain
 
-import "testing"
+import (
+	"testing"
+)
 
 func TestDomainToken(t *testing.T) {
 	userID := "test_user"
@@ -24,3 +26,20 @@ func TestDomainToken(t *testing.T) {
 		t.Error("UUID is empty")
 	}
 }
+
+func TestDomainTokenExpired(t *testing.T) {
+	userID := "test_user"
+	token := NewToken(userID)
+	if token.IsExpired() {
+		t.Error("Token is expired")
+	}
+
+	tokenString, err := token.ToString()
+	if err != nil {
+		t.Error("Error while generating token")
+	}
+	parsedToken, _ := ParseToken(tokenString)
+	if parsedToken.IsExpired() {
+		t.Error("Token is expired")
+	}
+}

usecase/token.go

@@ -43,6 +43,10 @@ func (tu tokenUseCase) ValidateToken(tokenString string) (string, error) {
 		return "", err
 	}
 
+	if token.IsExpired() {
+		return "", errors.New("token is expired")
+	}
+
 	userID := token.UserID()
 	tokenUUID, err := tu.tokenRepository.ValidateToken(userID)
 	if err != nil {