nginx-ui/model/auth.go

package model

import (
	"github.com/0xJacky/Nginx-UI/settings"
	"github.com/dgrijalva/jwt-go"
	"time"
)

type Auth struct {
	Model

	Name     string `json:"name"`
	Password string `json:"-"`
}

type AuthToken struct {
	Token string `json:"token"`
}

type JWTClaims struct {
	Name string `json:"name"`
	jwt.StandardClaims
}

func GetUser(name string) (user Auth, err error) {
	err = db.Where("name = ?", name).First(&user).Error
	if err != nil {
		return Auth{}, err
	}
	return user, err
}

func DeleteToken(token string) error {
	return db.Where("token = ?", token).Delete(&AuthToken{}).Error
}

func CheckToken(token string) int64 {
	return db.Where("token = ?", token).Find(&AuthToken{}).RowsAffected
}

func GenerateJWT(name string) (string, error) {
	claims := JWTClaims{
		Name: name,
		StandardClaims: jwt.StandardClaims{
			ExpiresAt: time.Now().Add(24 * time.Hour).Unix(),
		},
	}
	unsignedToken := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
	signedToken, err := unsignedToken.SignedString([]byte(settings.ServerSettings.JwtSecret))
	if err != nil {
		return "", err
	}

	err = db.Create(&AuthToken{
		Token: signedToken,
	}).Error

	if err != nil {
		return "", err
	}

	return signedToken, err
}