rework service disposal

2023-01-31 19:52:12 +01:00
parent ef5e77017f
commit 2ca3e639b4
24 changed files with 299 additions and 265 deletions
--- a/MareSynchronos/MarePlugin.cs
+++ b/MareSynchronos/MarePlugin.cs
@@ -0,0 +1,156 @@
+using Dalamud.Game.Command;
+using Dalamud.Plugin;
+using Dalamud.Interface.ImGuiFileDialog;
+using MareSynchronos.Managers;
+using MareSynchronos.WebAPI;
+using Dalamud.Interface.Windowing;
+using MareSynchronos.UI;
+using MareSynchronos.Utils;
+using MareSynchronos.FileCache;
+using MareSynchronos.MareConfiguration;
+using MareSynchronos.Mediator;
+using Microsoft.Extensions.DependencyInjection;
+
+namespace MareSynchronos;
+
+public class MarePlugin : MediatorSubscriberBase, IDisposable
+{
+    private readonly ServiceProvider _serviceProvider;
+    private const string _commandName = "/mare";
+    private IServiceScope? _runtimeServiceScope;
+
+    public MarePlugin(ServiceProvider serviceProvider, MareMediator mediator) : base(mediator)
+    {
+        _serviceProvider = serviceProvider;
+
+        mediator.Subscribe<SwitchToMainUiMessage>(this, (_) => Task.Run(WaitForPlayerAndLaunchCharacterManager));
+        mediator.Subscribe<DalamudLoginMessage>(this, (_) => DalamudUtilOnLogIn());
+        mediator.Subscribe<DalamudLogoutMessage>(this, (_) => DalamudUtilOnLogOut());
+
+        serviceProvider.GetRequiredService<SettingsUi>();
+        serviceProvider.GetRequiredService<CompactUi>();
+        serviceProvider.GetRequiredService<GposeUi>();
+        serviceProvider.GetRequiredService<IntroUi>();
+        serviceProvider.GetRequiredService<DownloadUi>();
+        serviceProvider.GetRequiredService<NotificationService>();
+    }
+
+    public override void Dispose()
+    {
+        base.Dispose();
+
+        _serviceProvider.GetRequiredService<CommandManager>().RemoveHandler(_commandName);
+
+        _runtimeServiceScope?.Dispose();
+        _serviceProvider.Dispose();
+
+        Logger.Debug("Shut down");
+    }
+
+    private void DalamudUtilOnLogIn()
+    {
+        Logger.Debug("Client login");
+
+        var pi = _serviceProvider.GetRequiredService<DalamudPluginInterface>();
+        pi.UiBuilder.Draw += Draw;
+        pi.UiBuilder.OpenConfigUi += OpenUi;
+        _serviceProvider.GetRequiredService<CommandManager>().AddHandler(_commandName, new CommandInfo(OnCommand)
+        {
+            HelpMessage = "Opens the Mare Synchronos UI",
+        });
+
+        if (!_serviceProvider.GetRequiredService<ConfigurationService>().Current.HasValidSetup()
+            || !_serviceProvider.GetRequiredService<ServerConfigurationManager>().HasValidConfig())
+        {
+            _serviceProvider.GetRequiredService<MareMediator>().Publish(new SwitchToIntroUiMessage());
+            return;
+        }
+
+        _serviceProvider.GetRequiredService<PeriodicFileScanner>().StartScan();
+        Task.Run(WaitForPlayerAndLaunchCharacterManager);
+    }
+
+    private void DalamudUtilOnLogOut()
+    {
+        Logger.Debug("Client logout");
+        _runtimeServiceScope?.Dispose();
+        var pi = _serviceProvider.GetRequiredService<DalamudPluginInterface>();
+        pi.UiBuilder.Draw -= Draw;
+        pi.UiBuilder.OpenConfigUi -= OpenUi;
+        _serviceProvider.GetRequiredService<CommandManager>().RemoveHandler(_commandName);
+    }
+
+    private async Task WaitForPlayerAndLaunchCharacterManager()
+    {
+        var dalamudUtil = _serviceProvider.GetRequiredService<DalamudUtil>();
+        while (!dalamudUtil.IsPlayerPresent)
+        {
+            await Task.Delay(100).ConfigureAwait(false);
+        }
+
+        try
+        {
+            Logger.Debug("Launching Managers");
+
+            _runtimeServiceScope?.Dispose();
+            _runtimeServiceScope = _serviceProvider.CreateScope();
+            _runtimeServiceScope.ServiceProvider.GetRequiredService<TransientResourceManager>();
+            _runtimeServiceScope.ServiceProvider.GetRequiredService<PlayerManager>();
+            _runtimeServiceScope.ServiceProvider.GetRequiredService<OnlinePlayerManager>();
+        }
+        catch (Exception ex)
+        {
+            Logger.Warn(ex.Message);
+        }
+    }
+
+    private void Draw()
+    {
+        _serviceProvider.GetRequiredService<WindowSystem>().Draw();
+        _serviceProvider.GetRequiredService<FileDialogManager>().Draw();
+    }
+
+    private void OnCommand(string command, string args)
+    {
+        var splitArgs = args.ToLowerInvariant().Trim().Split(" ", StringSplitOptions.RemoveEmptyEntries);
+
+        if (splitArgs == null || splitArgs.Length == 0)
+        {
+            // Interpret this as toggling the UI
+            OpenUi();
+            return;
+        }
+
+        if (string.Equals(splitArgs[0], "toggle", StringComparison.OrdinalIgnoreCase))
+        {
+            var serverConfigurationManager = _serviceProvider.GetRequiredService<ServerConfigurationManager>();
+            if (serverConfigurationManager.CurrentServer == null) return;
+            var fullPause = splitArgs.Length > 1 ? splitArgs[1] switch
+            {
+                "on" => false,
+                "off" => true,
+                _ => !serverConfigurationManager.CurrentServer.FullPause,
+            } : !serverConfigurationManager.CurrentServer.FullPause;
+
+            if (fullPause != serverConfigurationManager.CurrentServer.FullPause)
+            {
+                serverConfigurationManager.CurrentServer.FullPause = fullPause;
+                serverConfigurationManager.Save();
+                _ = _serviceProvider.GetRequiredService<ApiController>().CreateConnections();
+            }
+        }
+        else if (string.Equals(splitArgs[0], "gpose", StringComparison.OrdinalIgnoreCase))
+        {
+            _serviceProvider.GetRequiredService<GposeUi>().Toggle();
+        }
+    }
+
+    private void OpenUi()
+    {
+
+        if (_serviceProvider.GetRequiredService<ConfigurationService>().Current.HasValidSetup())
+            _serviceProvider.GetRequiredService<CompactUi>().Toggle();
+        else
+            _serviceProvider.GetRequiredService<IntroUi>().Toggle();
+    }
+}