wabbajack/Wabbajack.App/ServiceExtensions.cs

using System;
using System.Net.Http;
using System.Reactive.Disposables;
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.Networking.NexusApi;
using Wabbajack.Paths;
using Wabbajack.Paths.IO;
using Wabbajack.Services.OSIntegrated;
using SettingsView = Wabbajack.App.Screens.SettingsView;

namespace Wabbajack.App
{
    public static class ServiceExtensions
    {
        public static IServiceCollection AddAppServices(this IServiceCollection services)
        {
            services.AddAllSingleton<ILoggerProvider, LoggerProvider>();
            services.AddSingleton<MessageBus>();
            services.AddSingleton<MainWindow>();
            services.AddSingleton<BrowseViewModel>();

            services.AddTransient<BrowseItemViewModel>();

            services.AddTransient<LogViewModel>();

            services.AddDTOConverters();
            services.AddDTOSerializer();
            services.AddSingleton<ModeSelectionViewModel>();
            services.AddTransient<FileSelectionBoxViewModel>();
            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<InstallationStateManager>();
            services.AddSingleton<HttpClient>();
            
            services.AddAllSingleton<IReceiverMarker, StandardInstallationViewModel>();
            services.AddAllSingleton<IReceiverMarker, InstallConfigurationViewModel>();
            services.AddAllSingleton<IReceiverMarker, CompilerConfigurationViewModel>();
            services.AddAllSingleton<IReceiverMarker, MainWindowViewModel>();
            services.AddAllSingleton<IReceiverMarker, SettingsViewModel>();
            services.AddAllSingleton<IReceiverMarker, NexusLoginViewModel>();
            services.AddAllSingleton<IReceiverMarker, LoversLabOAuthLoginViewModel>();
            services.AddAllSingleton<IReceiverMarker, VectorPlexusOAuthLoginViewModel>();
            services.AddAllSingleton<IReceiverMarker, CompilationViewModel>();
            services.AddAllSingleton<IReceiverMarker, 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")
            });

            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 CefAppImpl app;
        private static Timer messagePump;
        private const int messagePumpDelay = 10;
        private static void App_FrameworkInitialized(object? sender, EventArgs e)
        {
            if (CefNetApplication.Instance.UsesExternalMessageLoop)
            {
                messagePump = new Timer(_ => Dispatcher.UIThread.Post(CefApi.DoMessageLoopWork), null, messagePumpDelay, messagePumpDelay);
            }
        }
    }
}