mirror of
https://github.com/AlistGo/alist.git
synced 2025-04-21 20:18:47 +08:00
6024e8d832
* refactor:separate the setting method from the db package to the op package and add the cache * refactor:separate the meta method from the db package to the op package * fix:setting not load database data * refactor:separate the user method from the db package to the op package * refactor:remove user JoinPath error * fix:op package user cache * refactor:fs package list method * fix:tile virtual paths (close #2743) * Revert "refactor:remove user JoinPath error" This reverts commit 4e20daaf9e700da047000d4fd4900abbe05c3848. * clean path directly may lead to unknown behavior * fix: The path of the meta passed in must be prefix of reqPath * chore: rename all virtualPath to mountPath * fix: `getStoragesByPath` and `GetStorageVirtualFilesByPath` is_sub_path: /a/b isn't subpath of /a/bc * fix: don't save setting if hook error Co-authored-by: Noah Hsu <i@nn.ci>
116 lines
2.4 KiB
Go
116 lines
2.4 KiB
Go
package op
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/Xhofe/go-cache"
|
|
"github.com/alist-org/alist/v3/internal/db"
|
|
"github.com/alist-org/alist/v3/internal/errs"
|
|
"github.com/alist-org/alist/v3/internal/model"
|
|
"github.com/alist-org/alist/v3/pkg/singleflight"
|
|
"github.com/alist-org/alist/v3/pkg/utils"
|
|
)
|
|
|
|
var userCache = cache.NewMemCache(cache.WithShards[*model.User](2))
|
|
var userG singleflight.Group[*model.User]
|
|
var guestUser *model.User
|
|
var adminUser *model.User
|
|
|
|
func GetAdmin() (*model.User, error) {
|
|
if adminUser == nil {
|
|
user, err := db.GetUserByRole(model.ADMIN)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
adminUser = user
|
|
}
|
|
return adminUser, nil
|
|
}
|
|
|
|
func GetGuest() (*model.User, error) {
|
|
if guestUser == nil {
|
|
user, err := db.GetUserByRole(model.GUEST)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
guestUser = user
|
|
}
|
|
return guestUser, nil
|
|
}
|
|
|
|
func GetUserByRole(role int) (*model.User, error) {
|
|
return db.GetUserByRole(role)
|
|
}
|
|
|
|
func GetUserByName(username string) (*model.User, error) {
|
|
if username == "" {
|
|
return nil, errs.EmptyUsername
|
|
}
|
|
if user, ok := userCache.Get(username); ok {
|
|
return user, nil
|
|
}
|
|
user, err, _ := userG.Do(username, func() (*model.User, error) {
|
|
_user, err := db.GetUserByName(username)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
userCache.Set(username, _user, cache.WithEx[*model.User](time.Hour))
|
|
return _user, nil
|
|
})
|
|
return user, err
|
|
}
|
|
|
|
func GetUserById(id uint) (*model.User, error) {
|
|
return db.GetUserById(id)
|
|
}
|
|
|
|
func GetUsers(pageIndex, pageSize int) (users []model.User, count int64, err error) {
|
|
return db.GetUsers(pageIndex, pageSize)
|
|
}
|
|
|
|
func CreateUser(u *model.User) error {
|
|
u.BasePath = utils.FixAndCleanPath(u.BasePath)
|
|
return db.CreateUser(u)
|
|
}
|
|
|
|
func DeleteUserById(id uint) error {
|
|
old, err := db.GetUserById(id)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if old.IsAdmin() || old.IsGuest() {
|
|
return errs.DeleteAdminOrGuest
|
|
}
|
|
userCache.Del(old.Username)
|
|
return db.DeleteUserById(id)
|
|
}
|
|
|
|
func UpdateUser(u *model.User) error {
|
|
old, err := db.GetUserById(u.ID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if u.IsAdmin() {
|
|
adminUser = nil
|
|
}
|
|
if u.IsGuest() {
|
|
guestUser = nil
|
|
}
|
|
userCache.Del(old.Username)
|
|
u.BasePath = utils.FixAndCleanPath(u.BasePath)
|
|
return db.UpdateUser(u)
|
|
}
|
|
|
|
func Cancel2FAByUser(u *model.User) error {
|
|
u.OtpSecret = ""
|
|
return UpdateUser(u)
|
|
}
|
|
|
|
func Cancel2FAById(id uint) error {
|
|
user, err := db.GetUserById(id)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return Cancel2FAByUser(user)
|
|
}
|