NightMate updates

makefile

@@ -45,6 +45,7 @@ help: ## Display this help screen
 
 mockgen:
 	mockgen -source=pkg/repository/interface/user.go -destination=pkg/repository/mockrepo/mock_user.go
+	mockgen -source=pkg/repository/interface/refresh_token.go -destination=pkg/repository/mockrepo/refresh_token.go
 	mockgen -source=pkg/usecase/interface/user.go -destination=pkg/usecase/mockusecase/mock_user.go
 	mockgen -source=pkg/usecase/interface/cart.go -destination=pkg/usecase/mockusecase/mock_cart.go
-	mockgen -source=pkg/usecase/interface/wallet.go -destination=pkg/usecase/mockusecase/mock_wallet.go
+	mockgen -source=pkg/usecase/interface/wallet.go -destination=pkg/usecase/mockusecase/mock_wallet.go

pkg/db/connection.go

@@ -36,6 +36,7 @@ func ConnectDatabase(cfg config.Config) (*gorm.DB, error) {
 		&domain.OfferTable{},
 		&domain.Images{},
 		&domain.AdminRefreshToken{},
+		&domain.UserRefreshToken{},
 	)
 
 	return db, dbErr

pkg/repository/refresh_token.go

@@ -33,16 +33,28 @@ func (c *RefreshTokenDataBase) AdminFindRefreshToken(ctx context.Context, adminI
 }
 
 func (c *RefreshTokenDataBase) UserRefreshTokenAdd(token string, userID uint) error {
-	sql := `INSERT INTO user_refresh_tokens (refresh_token,user_id)VALUES($1,$2)`
+	var check bool
+	tokenExist := `SELECT EXISTS(select * from user_refresh_tokens where users_id=$1)`
+	if err := c.DB.Raw(tokenExist, userID).Scan(&check).Error; err != nil {
+		return err
+	}
+	if check {
+		updateQuery := `update user_refresh_tokens SET refresh_token=$1 where users_id=$2`
+		if err := c.DB.Exec(updateQuery, token, userID).Error; err != nil {
+			return err
+		}
+		return nil
+	}
+	sql := `INSERT INTO user_refresh_tokens (refresh_token,users_id)VALUES($1,$2)`
 	if err := c.DB.Exec(sql, token, userID).Error; err != nil {
-		return fmt.Errorf("failed to add refresh token")
+		return err
 	}
 	return nil
 }
 
 func (c *RefreshTokenDataBase) UserFindRefreshToken(ctx context.Context, userID uint) (string, error) {
 	var token string
-	sql := `SELECT refresh_token FROM user_refresh_tokens where user_id = $1`
+	sql := `SELECT refresh_token FROM user_refresh_tokens where users_id = $1`
 	if err := c.DB.Raw(sql, userID).Scan(&token).Error; err != nil {
 		return "", err
 	}