feat: call "Finish" while app is exiting

OrgEleCho · Mar 27, 2024 · b35e792 · b35e792
1 parent b215d4b
commit b35e792
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 0 deletions.
diff --git a/src/CurvaLauncher/App.xaml.cs b/src/CurvaLauncher/App.xaml.cs
@@ -123,6 +123,15 @@ protected override void OnStartup(StartupEventArgs e)
             }
         }
 
+        protected override void OnExit(ExitEventArgs e)
+        {
+            _ = App.ServiceProvider
+                .GetRequiredService<PluginService>()
+                .FinishAllPlugins();
+
+            base.OnExit(e);
+        }
+
         private bool EnsureAppSingleton()
         {
             EventWaitHandle singletonEvent =

diff --git a/src/CurvaLauncher/Services/PluginService.cs b/src/CurvaLauncher/Services/PluginService.cs
@@ -152,4 +152,16 @@ public async Task ReloadAllPlugins()
                 await asyncPlugin.InitializeAsync();
         }
     }
+
+    [RelayCommand]
+    public async Task FinishAllPlugins()
+    {
+        foreach (var plugin in PluginInstances.Where(ins => ins.IsEnabled))
+        {
+            if (plugin.Plugin is ISyncPlugin syncPlugin)
+                syncPlugin.Finish();
+            else if (plugin.Plugin is IAsyncPlugin asyncPlugin)
+                await asyncPlugin.FinishAsync();
+        }
+    }
 }
diff --git a/src/Plugins/CurvaLauncher.Plugins.Everything/EverythingPlugin.cs b/src/Plugins/CurvaLauncher.Plugins.Everything/EverythingPlugin.cs
@@ -65,5 +65,10 @@ public override IEnumerable<IQueryResult> Query(string query)
                 weight -= float.Epsilon;
             }
         }
+
+        public override void Finish()
+        {
+
+        }
     }
 }