mirror of https://github.com/Xhofe/alist
53 lines
1.4 KiB
Go
53 lines
1.4 KiB
Go
package data
|
|
|
|
// initRoles creates the default admin and guest roles if missing.
|
|
// These roles are essential and must not be modified or removed.
|
|
|
|
import (
|
|
"github.com/alist-org/alist/v3/internal/model"
|
|
"github.com/alist-org/alist/v3/internal/op"
|
|
"github.com/alist-org/alist/v3/pkg/utils"
|
|
"github.com/pkg/errors"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func initRoles() {
|
|
guestRole, err := op.GetRoleByName("guest")
|
|
if err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
guestRole = &model.Role{
|
|
ID: uint(model.GUEST),
|
|
Name: "guest",
|
|
Description: "Guest",
|
|
PermissionScopes: []model.PermissionEntry{
|
|
{Path: "/", Permission: 0},
|
|
},
|
|
}
|
|
if err := op.CreateRole(guestRole); err != nil {
|
|
utils.Log.Fatalf("[init role] Failed to create guest role: %v", err)
|
|
}
|
|
} else {
|
|
utils.Log.Fatalf("[init role] Failed to get guest role: %v", err)
|
|
}
|
|
}
|
|
|
|
_, err = op.GetRoleByName("admin")
|
|
if err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
adminRole := &model.Role{
|
|
ID: uint(model.ADMIN),
|
|
Name: "admin",
|
|
Description: "Administrator",
|
|
PermissionScopes: []model.PermissionEntry{
|
|
{Path: "/", Permission: 0xFFFF},
|
|
},
|
|
}
|
|
if err := op.CreateRole(adminRole); err != nil {
|
|
utils.Log.Fatalf("[init role] Failed to create admin role: %v", err)
|
|
}
|
|
} else {
|
|
utils.Log.Fatalf("[init role] Failed to get admin role: %v", err)
|
|
}
|
|
}
|
|
}
|