116 lines
2.5 KiB
Go
116 lines
2.5 KiB
Go
package bolt
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
|
|
storm "github.com/asdine/storm/v3"
|
|
|
|
"github.com/gtsteffaniak/filebrowser/backend/errors"
|
|
"github.com/gtsteffaniak/filebrowser/backend/settings"
|
|
"github.com/gtsteffaniak/filebrowser/backend/users"
|
|
"github.com/gtsteffaniak/filebrowser/backend/utils"
|
|
)
|
|
|
|
type usersBackend struct {
|
|
db *storm.DB
|
|
}
|
|
|
|
func (st usersBackend) GetBy(i interface{}) (user *users.User, err error) {
|
|
user = &users.User{}
|
|
|
|
var arg string
|
|
var val interface{}
|
|
switch i := i.(type) {
|
|
case uint:
|
|
val = i
|
|
arg = "ID"
|
|
case int:
|
|
val = uint(i)
|
|
case string:
|
|
arg = "Username"
|
|
val = i
|
|
default:
|
|
return nil, errors.ErrInvalidDataType
|
|
}
|
|
|
|
err = st.db.One(arg, val, user)
|
|
|
|
if err != nil {
|
|
if err == storm.ErrNotFound {
|
|
return nil, errors.ErrNotExist
|
|
}
|
|
return nil, err
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func (st usersBackend) Gets() ([]*users.User, error) {
|
|
var allUsers []*users.User
|
|
err := st.db.All(&allUsers)
|
|
if err == storm.ErrNotFound {
|
|
return nil, errors.ErrNotExist
|
|
}
|
|
if err != nil {
|
|
return allUsers, err
|
|
}
|
|
return allUsers, err
|
|
}
|
|
|
|
func (st usersBackend) Update(user *users.User, fields ...string) error {
|
|
if len(fields) == 0 {
|
|
return st.Save(user)
|
|
}
|
|
|
|
val := reflect.ValueOf(user).Elem()
|
|
|
|
for _, field := range fields {
|
|
// Capitalize the first letter (you can adjust this based on your field naming convention)
|
|
correctedField := utils.CapitalizeFirst(field)
|
|
|
|
userField := val.FieldByName(correctedField)
|
|
if !userField.IsValid() {
|
|
return fmt.Errorf("invalid field: %s", field)
|
|
}
|
|
if !userField.CanSet() {
|
|
return fmt.Errorf("cannot update unexported field: %s", field)
|
|
}
|
|
|
|
val := userField.Interface()
|
|
if err := st.db.UpdateField(user, correctedField, val); err != nil {
|
|
return fmt.Errorf("Error updating user field: %s, error: %v", correctedField, err.Error())
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (st usersBackend) Save(user *users.User) error {
|
|
if len(user.Password) < settings.Config.Auth.Methods.PasswordAuth.MinLength {
|
|
return fmt.Errorf("password must be at least %d characters long", settings.Config.Auth.Methods.PasswordAuth.MinLength)
|
|
}
|
|
pass, err := users.HashPwd(user.Password)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
user.Password = pass
|
|
err = st.db.Save(user)
|
|
if err == storm.ErrAlreadyExists {
|
|
return errors.ErrExist
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (st usersBackend) DeleteByID(id uint) error {
|
|
return st.db.DeleteStruct(&users.User{ID: id})
|
|
}
|
|
|
|
func (st usersBackend) DeleteByUsername(username string) error {
|
|
user, err := st.GetBy(username)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return st.db.DeleteStruct(user)
|
|
}
|