wabbajack/Wabbajack.App/ServiceExtensions.cs

using System;
using System.Net.Http;
using System.Threading;
using System.Threading.Tasks;
using Avalonia.Threading;
using CefNet;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;
using Wabbajack.App.Controls;
using Wabbajack.App.Interfaces;
using Wabbajack.App.Messages;
using Wabbajack.App.Models;
using Wabbajack.App.Screens;
using Wabbajack.App.Services;
using Wabbajack.App.Utilities;
using Wabbajack.App.ViewModels;
using Wabbajack.App.Views;
using Wabbajack.Downloaders.Interfaces;
using Wabbajack.DTOs;
using Wabbajack.DTOs.DownloadStates;
using Wabbajack.DTOs.JsonConverters;
using Wabbajack.Paths.IO;
using Wabbajack.Services.OSIntegrated;

namespace Wabbajack.App;

public static class ServiceExtensions
{
    private const int messagePumpDelay = 10;


    private static CefAppImpl app;
    private static Timer messagePump;

    public static IServiceCollection AddAppServices(this IServiceCollection services)
    {
        services.AddAllSingleton<ILoggerProvider, LoggerProvider>();
        services.AddSingleton<MainWindow>();
        services.AddSingleton<BrowseViewModel>();
        services.AddTransient<BrowseItemViewModel>();
        services.AddTransient<LogViewModel>();

        services.AddTransient<InstalledListViewModel>();

        services.AddDTOConverters();
        services.AddDTOSerializer();
        services.AddSingleton<ModeSelectionViewModel>();
        services.AddTransient<FileSelectionBoxViewModel>();
        services.AddSingleton<IScreenView, ErrorPageView>();
        services.AddSingleton<IScreenView, LogScreenView>();
        services.AddSingleton<IScreenView, ModeSelectionView>();
        services.AddSingleton<IScreenView, InstallConfigurationView>();
        services.AddSingleton<IScreenView, CompilerConfigurationView>();
        services.AddSingleton<IScreenView, StandardInstallationView>();
        services.AddSingleton<IScreenView, CompilationView>();
        services.AddSingleton<IScreenView, SettingsView>();
        services.AddSingleton<IScreenView, BrowseView>();
        services.AddSingleton<IScreenView, LauncherView>();
        services.AddSingleton<IScreenView, PlaySelectView>();
        services.AddSingleton<ImageCache>();

        services.AddSingleton<InstallationStateManager>();
        services.AddSingleton<HttpClient>();

        services.AddSingleton<LogScreenViewModel>();
        services.AddSingleton<PlaySelectViewModel>();
        services.AddSingleton<ErrorPageViewModel>();
        services.AddSingleton<StandardInstallationViewModel>();
        services.AddSingleton<InstallConfigurationViewModel>();
        services.AddSingleton<CompilerConfigurationViewModel>();
        services.AddSingleton<MainWindowViewModel>();
        services.AddSingleton<SettingsViewModel>();
        services.AddSingleton<NexusLoginViewModel>();
        services.AddSingleton<LoversLabOAuthLoginViewModel>();
        services.AddSingleton<VectorPlexusOAuthLoginViewModel>();
        services.AddSingleton<CompilationViewModel>();
        services.AddSingleton<LauncherViewModel>();

        // Services
        services.AddAllSingleton<IDownloader, IDownloader<Manual>, ManualDownloader>();

        var resources = KnownFolders.EntryPoint;
        services.AddSingleton(s => new CefSettings
        {
            NoSandbox = true,
            PersistSessionCookies = true,
            MultiThreadedMessageLoop = false,
            WindowlessRenderingEnabled = true,
            ExternalMessagePump = true,
            LocalesDirPath = resources.Combine("locales").ToString(),
            ResourcesDirPath = resources.ToString(),
            UserAgent = "",
            CachePath = KnownFolders.WabbajackAppLocal.Combine("cef_cache").ToString()
        });

        services.AddSingleton(s => new Configuration
        {
            EncryptedDataLocation = KnownFolders.WabbajackAppLocal.Combine("encrypted"),
            ModListsDownloadLocation = KnownFolders.EntryPoint.Combine("downloaded_mod_lists"),
            SavedSettingsLocation = KnownFolders.WabbajackAppLocal.Combine("saved_settings"),
            LogLocation = KnownFolders.EntryPoint.Combine("logs"),
            ImageCacheLocation = KnownFolders.WabbajackAppLocal.Combine("image_cache")
        });

        services.AddSingleton<SettingsManager>();

        services.AddSingleton(s =>
        {
            App.FrameworkInitialized += App_FrameworkInitialized;

            var app = new CefAppImpl();
            app.ScheduleMessagePumpWorkCallback = OnScheduleMessagePumpWork;

            app.CefProcessMessageReceived += App_CefProcessMessageReceived;
            app.Initialize(resources.ToString(), s.GetService<CefSettings>());


            return app;
        });

        services.AddOSIntegrated();
        return services;
    }

    private static async void OnScheduleMessagePumpWork(long delayMs)
    {
        await Task.Delay((int) delayMs);
        Dispatcher.UIThread.Post(CefApi.DoMessageLoopWork);
    }

    private static void App_CefProcessMessageReceived(object? sender, CefProcessMessageReceivedEventArgs e)
    {
        var msg = e.Name;
    }

    private static void App_FrameworkInitialized(object? sender, EventArgs e)
    {
        if (CefNetApplication.Instance.UsesExternalMessageLoop)
            messagePump = new Timer(_ => Dispatcher.UIThread.Post(CefApi.DoMessageLoopWork), null, messagePumpDelay,
                messagePumpDelay);
    }
}