mirror of
https://github.com/navidrome/navidrome.git
synced 2026-01-03 06:15:22 +00:00
eaf7795716
* feat(cli): add user administration * clean go.mod, address comments * fix lint, I hope * bump compilation timeoit in adapter_media_agent_test * address initial comments * feedback 2 * update user commands to use context to allow proper cancellation Signed-off-by: Deluan <deluan@navidrome.org> * enforce admin user requirement in context for command execution Signed-off-by: Deluan <deluan@navidrome.org> --------- Signed-off-by: Deluan <deluan@navidrome.org> Co-authored-by: Deluan <deluan@navidrome.org>
43 lines
1.0 KiB
Go
43 lines
1.0 KiB
Go
package cmd
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/navidrome/navidrome/core/auth"
|
|
"github.com/navidrome/navidrome/db"
|
|
"github.com/navidrome/navidrome/log"
|
|
"github.com/navidrome/navidrome/model"
|
|
"github.com/navidrome/navidrome/model/request"
|
|
"github.com/navidrome/navidrome/persistence"
|
|
)
|
|
|
|
func getAdminContext(ctx context.Context) (model.DataStore, context.Context) {
|
|
sqlDB := db.Db()
|
|
ds := persistence.New(sqlDB)
|
|
ctx = auth.WithAdminUser(ctx, ds)
|
|
u, _ := request.UserFrom(ctx)
|
|
if !u.IsAdmin {
|
|
log.Fatal(ctx, "There must be at least one admin user to run this command.")
|
|
}
|
|
return ds, ctx
|
|
}
|
|
|
|
func getUser(ctx context.Context, id string, ds model.DataStore) (*model.User, error) {
|
|
user, err := ds.User(ctx).FindByUsername(id)
|
|
|
|
if err != nil && !errors.Is(err, model.ErrNotFound) {
|
|
return nil, fmt.Errorf("finding user by name: %w", err)
|
|
}
|
|
|
|
if errors.Is(err, model.ErrNotFound) {
|
|
user, err = ds.User(ctx).Get(id)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("finding user by id: %w", err)
|
|
}
|
|
}
|
|
|
|
return user, nil
|
|
}
|