fix: 🐛 Really really fix it this time?

Ombi-app · Oct 15, 2021 · 543d36e · 543d36e
1 parent 3d3e8c9
commit 543d36e
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/Ombi/Program.cs b/src/Ombi/Program.cs
@@ -20,6 +20,7 @@
 using Microsoft.Extensions.Logging;
 using Ombi.Api.TheMovieDb.Models;
 using System.Net.Http;
+using System.Text.RegularExpressions;
 
 namespace Ombi
 {
@@ -269,17 +270,16 @@ await strat.ExecuteAsync(async () =>
                 var indexHtml = await File.ReadAllTextAsync(indexPath);
                 var sb = new StringBuilder(indexHtml);
 
-                var headPosition = indexHtml.IndexOf("<script");
-                var firstLinkPosition = indexHtml.IndexOf("<style type=");
+                const string scriptRegex = "<script.type=.+>window.+=.+;<.script>";
+                const string baseUrlRegex = "<base.href=\".+\">";
 
-                sb.Remove(headPosition, firstLinkPosition - headPosition - 6);
+                indexHtml = Regex.Replace(indexHtml, scriptRegex, $"<script type='text/javascript'>window[\"baseHref\"] = '{trimmedBaseUrl}';</script>");
 
-                sb.Insert(headPosition + 6,
-                    $"<script type='text/javascript'>window[\"baseHref\"] = '{trimmedBaseUrl}';</script><base href=\"{trimmedBaseUrl}/\">");
+                indexHtml = Regex.Replace(indexHtml, baseUrlRegex, $"<base href=\"{trimmedBaseUrl}/\">");
 
                 try
                 {
-                    await File.WriteAllTextAsync(indexPath, sb.ToString());
+                    await File.WriteAllTextAsync(indexPath, indexHtml);
                 }
                 catch (UnauthorizedAccessException)
                 {