lndhub.go/lib/tokens/jwt.go

package tokens

import (
	"time"

	"github.com/bumi/lndhub.go/db/models"
	"github.com/bumi/lndhub.go/lib"
	"github.com/golang-jwt/jwt"
	"github.com/labstack/echo/v4"
	"github.com/labstack/echo/v4/middleware"
)

type jwtCustomClaims struct {
	ID int64 `json:"id"`

	jwt.StandardClaims
}

func Middleware() echo.MiddlewareFunc {
	config := middleware.JWTConfig{
		ContextKey: "UserJwt",
		SigningKey: []byte("secret"),
		SuccessHandler: func(c echo.Context) {
			ctx := c.(*lib.LndhubContext)
			token := ctx.Get("UserJwt").(*jwt.Token)
			claims := token.Claims.(jwt.MapClaims)
			userId := claims["id"].(float64)
			ctx.Set("UserId", userId)
		},
	}
	return middleware.JWTWithConfig(config)
}

// GenerateAccessToken : Generate Access Token
func GenerateAccessToken(u *models.User) (string, error) {
	claims := &jwtCustomClaims{
		u.ID,
		jwt.StandardClaims{
			// one week expiration
			ExpiresAt: time.Now().Add(time.Hour * 27 * 7).Unix(),
		},
	}
	token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)

	t, err := token.SignedString([]byte("secret"))
	if err != nil {
		return "", err
	}

	return t, nil
}

// GenerateRefreshToken : Generate Refresh Token
func GenerateRefreshToken(u *models.User) (string, error) {
	token := jwt.NewWithClaims(jwt.SigningMethodHS256, jwt.MapClaims{
		"id": u.ID,
	})

	t, err := token.SignedString([]byte("secret"))
	if err != nil {
		return "", err
	}

	return t, nil
}