111 lines
2.2 KiB
Go
111 lines
2.2 KiB
Go
package settings
|
|
|
|
import (
|
|
"github.com/filebrowser/filebrowser/v2/errors"
|
|
"github.com/filebrowser/filebrowser/v2/rules"
|
|
"github.com/filebrowser/filebrowser/v2/users"
|
|
)
|
|
|
|
// StorageBackend is a settings storage backend.
|
|
type StorageBackend interface {
|
|
Get() (*Settings, error)
|
|
Save(*Settings) error
|
|
GetServer() (*Server, error)
|
|
SaveServer(*Server) error
|
|
}
|
|
|
|
// Storage is a settings storage.
|
|
type Storage struct {
|
|
back StorageBackend
|
|
}
|
|
|
|
// NewStorage creates a settings storage from a backend.
|
|
func NewStorage(back StorageBackend) *Storage {
|
|
return &Storage{back: back}
|
|
}
|
|
|
|
// Get returns the settings for the current instance.
|
|
func (s *Storage) Get() (*Settings, error) {
|
|
set, err := s.back.Get()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if set.UserHomeBasePath == "" {
|
|
set.UserHomeBasePath = DefaultUsersHomeBasePath
|
|
}
|
|
if set.Tus == (Tus{}) {
|
|
set.Tus = Tus{
|
|
ChunkSize: DefaultTusChunkSize,
|
|
RetryCount: DefaultTusRetryCount,
|
|
}
|
|
}
|
|
return set, nil
|
|
}
|
|
|
|
var defaultEvents = []string{
|
|
"save",
|
|
"copy",
|
|
"rename",
|
|
"upload",
|
|
"delete",
|
|
}
|
|
|
|
// Save saves the settings for the current instance.
|
|
func (s *Storage) Save(set *Settings) error {
|
|
if len(set.Key) == 0 {
|
|
return errors.ErrEmptyKey
|
|
}
|
|
|
|
if set.Defaults.Locale == "" {
|
|
set.Defaults.Locale = "en"
|
|
}
|
|
|
|
if set.Defaults.Commands == nil {
|
|
set.Defaults.Commands = []string{}
|
|
}
|
|
|
|
if set.Defaults.ViewMode == "" {
|
|
set.Defaults.ViewMode = users.MosaicViewMode
|
|
}
|
|
|
|
if set.Rules == nil {
|
|
set.Rules = []rules.Rule{}
|
|
}
|
|
|
|
if set.Shell == nil {
|
|
set.Shell = []string{}
|
|
}
|
|
|
|
if set.Commands == nil {
|
|
set.Commands = map[string][]string{}
|
|
}
|
|
|
|
for _, event := range defaultEvents {
|
|
if _, ok := set.Commands["before_"+event]; !ok {
|
|
set.Commands["before_"+event] = []string{}
|
|
}
|
|
|
|
if _, ok := set.Commands["after_"+event]; !ok {
|
|
set.Commands["after_"+event] = []string{}
|
|
}
|
|
}
|
|
|
|
err := s.back.Save(set)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// GetServer wraps StorageBackend.GetServer.
|
|
func (s *Storage) GetServer() (*Server, error) {
|
|
return s.back.GetServer()
|
|
}
|
|
|
|
// SaveServer wraps StorageBackend.SaveServer and adds some verification.
|
|
func (s *Storage) SaveServer(ser *Server) error {
|
|
ser.Clean()
|
|
return s.back.SaveServer(ser)
|
|
}
|