Handle UI link

LykosAI · Oct 25, 2024 · 754eeb2 · 754eeb2
1 parent 68d20e6
commit 754eeb2
Showing 1 changed file with 21 additions and 2 deletions.
diff --git a/StabilityMatrix.Core/Models/Packages/FluxGym.cs b/StabilityMatrix.Core/Models/Packages/FluxGym.cs
@@ -1,4 +1,4 @@
-using System.Text.RegularExpressions;
+using System.Text.RegularExpressions;
 using StabilityMatrix.Core.Attributes;
 using StabilityMatrix.Core.Helper;
 using StabilityMatrix.Core.Helper.Cache;
@@ -150,9 +150,28 @@ public override async Task RunPackage(
     )
     {
         await SetupVenv(installLocation).ConfigureAwait(false);
+
+        void HandleConsoleOutput(ProcessOutput s)
+        {
+            onConsoleOutput?.Invoke(s);
+
+            if (s.Text.Contains("Running on local URL", StringComparison.OrdinalIgnoreCase))
+            {
+                var regex = new Regex(@"(https?:\/\/)([^:\s]+):(\d+)");
+                var match = regex.Match(s.Text);
+
+                if (match.Success)
+                {
+                    Console.WriteLine($"AAAAAA {match.Value}");
+                    WebUrl = match.Value;
+                }
+                OnStartupComplete(WebUrl);
+            }
+        }
+
         VenvRunner.RunDetached(
             [Path.Combine(installLocation, options.Command ?? LaunchCommand), ..options.Arguments],
-            onConsoleOutput,
+            HandleConsoleOutput,
             OnExit
         );
     }