From 92d42b3c53077b705673859c73fdd62c2111f42d Mon Sep 17 00:00:00 2001
From: Ionite <dev@ionite.io>
Date: Wed, 3 Apr 2024 20:32:59 -0400
Subject: [PATCH] PreOutputActions support for ModuleApplyStepEventArgs

---
 .../Models/Inference/ModuleApplyStepEventArgs.cs       | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/StabilityMatrix.Avalonia/Models/Inference/ModuleApplyStepEventArgs.cs b/StabilityMatrix.Avalonia/Models/Inference/ModuleApplyStepEventArgs.cs
index f4bd13fb..487d39b7 100644
--- a/StabilityMatrix.Avalonia/Models/Inference/ModuleApplyStepEventArgs.cs
+++ b/StabilityMatrix.Avalonia/Models/Inference/ModuleApplyStepEventArgs.cs
@@ -28,6 +28,16 @@ public class ModuleApplyStepEventArgs : EventArgs
 
     public List<(string SourcePath, string DestinationRelativePath)> FilesToTransfer { get; init; } = [];
 
+    public List<Action<ModuleApplyStepEventArgs>> PreOutputActions { get; init; } = [];
+
+    public void InvokeAllPreOutputActions()
+    {
+        foreach (var action in PreOutputActions)
+        {
+            action(this);
+        }
+    }
+
     /// <summary>
     /// Creates a new <see cref="ModuleApplyStepEventArgs"/> with the given <see cref="ComfyNodeBuilder"/>.
     /// </summary>