lndhub.go/database/models/user.go

package models

import (
	"github.com/dgrijalva/jwt-go"
	"github.com/labstack/echo/v4"
	"net/http"
	"time"
)

// User : User Model
type User struct {
	ID           uint `gorm:"primary_key"`
	Email        string
	Login        string
	Password     string
	RefreshToken string
	AccessToken  string
	UpdatedAt    time.Time
	CreatedAt    time.Time
}

// GenerateToken : Generate Token
func (u *User) GenerateAccessToken(c echo.Context, user *User) error {
	if u.ID == user.ID {
		token := jwt.NewWithClaims(jwt.SigningMethodHS256, jwt.MapClaims{
			"id": u.ID,
		})

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

		return c.JSON(http.StatusOK, map[string]string{
			u.AccessToken: t,
		})
	}

	return echo.ErrUnauthorized
}