Kavita/Kavita.Database/Seed.cs

using System;
using System.Collections.Generic;
using System.Globalization;
using System.IO;
using System.Linq;
using System.Reflection;
using System.Text.Json;
using System.Threading.Tasks;
using Kavita.API.Database;
using Kavita.API.Repositories;
using Kavita.API.Services;
using Kavita.Common;
using Kavita.Common.EnvironmentInfo;
using Kavita.Models;
using Kavita.Models.Constants;
using Kavita.Models.DTOs.Settings;
using Kavita.Models.Entities;
using Kavita.Models.Entities.Enums;
using Kavita.Models.Entities.MetadataMatching;
using Kavita.Models.Entities.User;
using Microsoft.AspNetCore.Identity;
using Microsoft.EntityFrameworkCore;

namespace Kavita.Database;

public static class Seed
{

    public static async Task SeedRoles(RoleManager<AppRole> roleManager)
    {
        var roles = typeof(PolicyConstants)
            .GetFields(BindingFlags.Public | BindingFlags.Static)
            .Where(f => f.FieldType == typeof(string))
            .ToDictionary(f => f.Name,
                f => (string) f.GetValue(null)!).Values
            .Select(policyName => new AppRole() {Name = policyName})
            .ToList();

        foreach (var role in roles)
        {
            var exists = await roleManager.RoleExistsAsync(role.Name!);
            if (!exists)
            {
                await roleManager.CreateAsync(role);
            }
        }
    }

    public static async Task SeedThemes(IDataContext context)
    {
        await context.Database.EnsureCreatedAsync();

        foreach (var theme in Defaults.DefaultThemes)
        {
            var existing = await context.SiteTheme.FirstOrDefaultAsync(s => s.Name.Equals(theme.Name));
            if (existing == null)
            {
                await context.SiteTheme.AddAsync(theme);
            }
        }

        await context.SaveChangesAsync();
    }

    public static async Task SeedFonts(IDataContext context)
    {
        await context.Database.EnsureCreatedAsync();

        foreach (var font in Defaults.DefaultFonts)
        {
            var existing = await context.EpubFont.FirstOrDefaultAsync(f => f.Name.Equals(font.Name));
            if (existing == null)
            {
                await context.EpubFont.AddAsync(font);
            }
        }

        await context.SaveChangesAsync();
    }

    public static async Task SeedDefaultStreams(IUnitOfWork unitOfWork)
    {
        var allUsers = await unitOfWork.UserRepository.GetAllUsersAsync(AppUserIncludes.DashboardStreams);
        foreach (var user in allUsers)
        {
            if (user.DashboardStreams.Count != 0) continue;
            user.DashboardStreams ??= [];
            foreach (var defaultStream in Defaults.DefaultStreams)
            {
                var newStream = new AppUserDashboardStream
                {
                    Name = defaultStream.Name,
                    IsProvided = defaultStream.IsProvided,
                    Order = defaultStream.Order,
                    StreamType = defaultStream.StreamType,
                    Visible = defaultStream.Visible,
                };

                user.DashboardStreams.Add(newStream);
            }
            unitOfWork.UserRepository.Update(user);
            await unitOfWork.CommitAsync();
        }
    }

    public static async Task SeedDefaultSideNavStreams(IUnitOfWork unitOfWork)
    {
        var allUsers = await unitOfWork.UserRepository.GetAllUsersAsync(AppUserIncludes.SideNavStreams);
        foreach (var user in allUsers)
        {
            user.SideNavStreams ??= [];
            foreach (var defaultStream in Defaults.DefaultSideNavStreams)
            {
                if (user.SideNavStreams.Any(s => s.Name == defaultStream.Name && s.StreamType == defaultStream.StreamType)) continue;
                var newStream = new AppUserSideNavStream()
                {
                    Name = defaultStream.Name,
                    IsProvided = defaultStream.IsProvided,
                    Order = defaultStream.Order,
                    StreamType = defaultStream.StreamType,
                    Visible = defaultStream.Visible,
                };

                user.SideNavStreams.Add(newStream);
            }
            unitOfWork.UserRepository.Update(user);
            await unitOfWork.CommitAsync();
        }
    }

    public static async Task SeedDefaultHighlightSlots(IUnitOfWork unitOfWork)
    {
        var allUsers = await unitOfWork.UserRepository.GetAllUsersAsync(AppUserIncludes.UserPreferences);
        foreach (var user in allUsers)
        {
            if (user.UserPreferences.BookReaderHighlightSlots.Any()) break;

            user.UserPreferences.BookReaderHighlightSlots = Defaults.DefaultHighlightSlots.ToList();
            unitOfWork.UserRepository.Update(user);
        }
        await unitOfWork.CommitAsync();
    }

    public static async Task SeedSettings(IDataContext context, IDirectoryService directoryService)
    {
        await context.Database.EnsureCreatedAsync();
        Defaults.DefaultSettings = [
            ..new List<ServerSetting>()
            {
                new() {Key = ServerSettingKey.CacheDirectory, Value = directoryService.CacheDirectory},
                new() {Key = ServerSettingKey.TaskScan, Value = "daily"},
                new() {Key = ServerSettingKey.TaskBackup, Value = "daily"},
                new() {Key = ServerSettingKey.TaskCleanup, Value = "daily"},
                new() {Key = ServerSettingKey.LoggingLevel, Value = "Debug"},
                new()
                {
                    Key = ServerSettingKey.BackupDirectory, Value = Path.GetFullPath(directoryService.BackupDirectory)
                },
                new()
                {
                    Key = ServerSettingKey.Port, Value = Configuration.DefaultHttpPort + string.Empty
                }, // Not used from DB, but DB is sync with appSettings.json
                new() {
                    Key = ServerSettingKey.IpAddresses, Value = Configuration.DefaultIpAddresses
                }, // Not used from DB, but DB is sync with appSettings.json
                new() {Key = ServerSettingKey.AllowStatCollection, Value = "true"},
                new() {Key = ServerSettingKey.EnableOpds, Value = "true"},
                new() {Key = ServerSettingKey.BaseUrl, Value = "/"},
                new() {Key = ServerSettingKey.InstallId, Value = HashUtil.AnonymousToken()},
                new() {Key = ServerSettingKey.InstallVersion, Value = BuildInfo.Version.ToString()},
                new() {Key = ServerSettingKey.BookmarkDirectory, Value = directoryService.BookmarkDirectory},
                new() {Key = ServerSettingKey.TotalBackups, Value = "30"},
                new() {Key = ServerSettingKey.TotalLogs, Value = "30"},
                new() {Key = ServerSettingKey.EnableFolderWatching, Value = "false"},
                new() {Key = ServerSettingKey.HostName, Value = string.Empty},
                new() {Key = ServerSettingKey.EncodeMediaAs, Value = nameof(EncodeFormat.PNG)},
                new() {Key = ServerSettingKey.LicenseKey, Value = string.Empty},
                new() {Key = ServerSettingKey.OnDeckProgressDays, Value = "30"},
                new() {Key = ServerSettingKey.OnDeckUpdateDays, Value = "7"},
                new() {Key = ServerSettingKey.CoverImageSize, Value = nameof(CoverImageSize.Default)},
                new() {Key = ServerSettingKey.PdfRenderResolution, Value = nameof(PdfRenderResolution.Default)},
                new() {
                    Key = ServerSettingKey.CacheSize, Value = Configuration.DefaultCacheMemory + string.Empty
                }, // Not used from DB, but DB is sync with appSettings.json
                new() { Key = ServerSettingKey.OidcConfiguration, Value = JsonSerializer.Serialize(new OidcConfigDto())},

                new() {Key = ServerSettingKey.EmailHost, Value = string.Empty},
                new() {Key = ServerSettingKey.EmailPort, Value = string.Empty},
                new() {Key = ServerSettingKey.EmailAuthPassword, Value = string.Empty},
                new() {Key = ServerSettingKey.EmailAuthUserName, Value = string.Empty},
                new() {Key = ServerSettingKey.EmailSenderAddress, Value = string.Empty},
                new() {Key = ServerSettingKey.EmailSenderDisplayName, Value = string.Empty},
                new() {Key = ServerSettingKey.EmailEnableSsl, Value = "true"},
                new() {Key = ServerSettingKey.EmailSizeLimit, Value = 26_214_400 + string.Empty},
                new() {Key = ServerSettingKey.EmailCustomizedTemplates, Value = "false"},
                new() {Key = ServerSettingKey.FirstInstallVersion, Value = BuildInfo.Version.ToString()},
                new() {Key = ServerSettingKey.FirstInstallDate, Value = DateTime.UtcNow.ToString(CultureInfo.InvariantCulture)},
                new() {Key = ServerSettingKey.StatsApiHits, Value = "0"},
            }.ToArray()
        ];

        foreach (var defaultSetting in Defaults.DefaultSettings)
        {
            var existing = await context.ServerSetting.FirstOrDefaultAsync(s => s.Key == defaultSetting.Key);
            if (existing == null)
            {
                await context.ServerSetting.AddAsync(defaultSetting);
            }
        }

        await context.SaveChangesAsync();

        // Port, IpAddresses and LoggingLevel are managed in appSettings.json. Update the DB values to match
        (await context.ServerSetting.FirstAsync(s => s.Key == ServerSettingKey.Port)).Value =
            Configuration.Port + string.Empty;
        (await context.ServerSetting.FirstAsync(s => s.Key == ServerSettingKey.IpAddresses)).Value =
            Configuration.IpAddresses;
        (await context.ServerSetting.FirstAsync(s => s.Key == ServerSettingKey.CacheDirectory)).Value =
            directoryService.CacheDirectory + string.Empty;
        (await context.ServerSetting.FirstAsync(s => s.Key == ServerSettingKey.BackupDirectory)).Value =
            directoryService.BackupDirectory + string.Empty;
        (await context.ServerSetting.FirstAsync(s => s.Key == ServerSettingKey.CacheSize)).Value =
            Configuration.CacheSize + string.Empty;

        await SetOidcSettingsFromDisk(context);


        await context.SaveChangesAsync();
    }

    public static async Task SetOidcSettingsFromDisk(IDataContext context)
    {
        var oidcSettingEntry = await context.ServerSetting
            .FirstOrDefaultAsync(setting => setting.Key == ServerSettingKey.OidcConfiguration);

        var storedOidcSettings = JsonSerializer.Deserialize<OidcConfigDto>(oidcSettingEntry!.Value)!;

        var diskOidcSettings = Configuration.OidcSettings;

        storedOidcSettings.Authority = diskOidcSettings.Authority;
        storedOidcSettings.ClientId = diskOidcSettings.ClientId;
        storedOidcSettings.Secret = diskOidcSettings.Secret;
        storedOidcSettings.CustomScopes = diskOidcSettings.CustomScopes;

        oidcSettingEntry.Value = JsonSerializer.Serialize(storedOidcSettings);
    }

    public static async Task SeedMetadataSettings(IDataContext context)
    {
        await context.Database.EnsureCreatedAsync();

        var existing = await context.MetadataSettings.FirstOrDefaultAsync();
        if (existing == null)
        {
            existing = new MetadataSettings()
            {
                Enabled = true,
                EnablePeople = true,
                EnableRelationships = true,
                EnableSummary = true,
                EnablePublicationStatus = true,
                EnableStartDate = true,
                EnableTags = false,
                EnableGenres = true,
                EnableLocalizedName = false,
                FirstLastPeopleNaming = true,
                EnableCoverImage = true,
                EnableChapterTitle = false,
                EnableChapterSummary = true,
                EnableChapterPublisher = true,
                EnableChapterCoverImage = false,
                EnableChapterReleaseDate = true,
                PersonRoles = [PersonRole.Writer, PersonRole.CoverArtist, PersonRole.Character]
            };
            await context.MetadataSettings.AddAsync(existing);
        }


        await context.SaveChangesAsync();

    }
}