nginx-proxy-manager/backend/internal/entity/auth/model.go

package auth

import (
	"npm/internal/database"
	"npm/internal/entity"

	"github.com/rotisserie/eris"
	"golang.org/x/crypto/bcrypt"
)

const (
	// TypePassword is the Password Type
	TypePassword = "password"
)

// Model is the model
type Model struct {
	entity.ModelBase
	UserID uint   `json:"user_id" gorm:"column:user_id"`
	Type   string `json:"type" gorm:"column:type;default:password"`
	Secret string `json:"secret,omitempty" gorm:"column:secret"`
}

// TableName overrides the table name used by gorm
func (Model) TableName() string {
	return "auth"
}

// LoadByID will load from an ID
func (m *Model) LoadByID(id int) error {
	db := database.GetDB()
	result := db.First(&m, id)
	return result.Error
}

// LoadByUserIDType will load from an ID
func (m *Model) LoadByUserIDType(userID int, authType string) error {
	db := database.GetDB()
	result := db.
		Where("user_id = ?", userID).
		Where("type = ?", authType).
		First(&m)
	return result.Error
}

/*
// Touch will update model's timestamp(s)
func (m *Model) Touch(created bool) {
	var d types.DBDate
	d.Time = time.Now()
	if created {
		m.CreatedOn = d
	}
	m.ModifiedOn = d
}
*/

// Save will save this model to the DB
func (m *Model) Save() error {
	db := database.GetDB()
	// todo: touch? not sure that save does this or not?
	result := db.Save(m)
	return result.Error
}

// SetPassword will generate a hashed password based on given string
func (m *Model) SetPassword(password string) error {
	hash, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.MinCost+2)
	if err != nil {
		return err
	}

	m.Type = TypePassword
	m.Secret = string(hash)

	return nil
}

// ValidateSecret will check if a given secret matches the encrypted secret
func (m *Model) ValidateSecret(secret string) error {
	if m.Type != TypePassword {
		return eris.New("Could not validate Secret, auth type is not a Password")
	}

	err := bcrypt.CompareHashAndPassword([]byte(m.Secret), []byte(secret))
	if err != nil {
		return eris.New("Invalid Password")
	}

	return nil
}