Kavita/Kavita.Models/Builders/AppUserBuilder.cs

using System.Linq;
using Kavita.Models.Entities;
using Kavita.Models.Entities.Enums;
using Kavita.Models.Entities.User;

namespace Kavita.Models.Builders;

public class AppUserBuilder : IEntityBuilder<AppUser>
{
    private readonly AppUser _appUser;
    public AppUser Build() => _appUser;

    public AppUserBuilder(string username, string email, SiteTheme? theme = null)
    {
        _appUser = new AppUser()
        {
            UserName = username,
            Email = email,
            UserPreferences = new AppUserPreferences
            {
                Theme = theme ?? Defaults.DefaultThemes.First(),
                Locale = "en"
            },
            ReadingLists = [],
            Bookmarks = [],
            Libraries = [],
            Ratings = [],
            Progresses = [],
            Devices = [],
            Id = 0,
            DashboardStreams = [],
            SideNavStreams = [],
            ReadingProfiles = [],
            AuthKeys = Defaults.CreateDefaultAuthKeys()
        };
    }

    public AppUserBuilder WithLibrary(Library library, bool createSideNavStream = false)
    {
        _appUser.Libraries.Add(library);
        if (!createSideNavStream) return this;

        if (library.Id != 0 && _appUser.SideNavStreams.Any(s => s.LibraryId == library.Id)) return this;
        _appUser.SideNavStreams.Add(new AppUserSideNavStream()
        {
            Name = library.Name,
            IsProvided = false,
            Visible = true,
            LibraryId = library.Id,
            StreamType = SideNavStreamType.Library,
            Order = _appUser.SideNavStreams.Max(s => s.Order) + 1,
        });

        return this;
    }


    public AppUserBuilder WithLocale(string locale)
    {
        _appUser.UserPreferences.Locale = locale;
        return this;
    }

    public AppUserBuilder WithRole(string role)
    {
        _appUser.UserRoles ??= [];
        _appUser.UserRoles.Add(new AppUserRole() {Role = new AppRole() {Name = role}});
        return this;
    }

    public AppUserBuilder WithIdentityProvider(IdentityProvider identityProvider)
    {
        _appUser.IdentityProvider = identityProvider;
        return this;
    }
}