Ensure website is loaded before download

scripts/Common/Scripts/BrowserAutomation.ps1

@@ -90,6 +90,12 @@ $null = New-Module {
     
             $browser = [OpenQA.Selenium.Firefox.FirefoxDriver]::new($options);
             $browser.Navigate().GoToUrl($URL);
+
+            while (-not ($browser.ExecuteScript("return document.readyState;") -eq "complete")) {
+                Start-Sleep 0.1;
+            }
+
+            try {
                 $null = & $downloadAction -Browser $browser;
         
                 while (& $downloadChecker) {
@@ -103,6 +109,10 @@ $null = New-Module {
                 Remove-Item -Recurse $dir;
                 $result;
             }
+            catch {
+                throw $Error;
+            }
+        }
     }
 
     <#
@@ -130,7 +140,24 @@ $null = New-Module {
                 [OpenQA.Selenium.Firefox.FirefoxDriver] $Browser
             )
 
+            $selector = [OpenQA.Selenium.By]::CssSelector($ButtonSelector);
+            [OpenQA.Selenium.IWebElement] $element = $null;
+
+            for ($i = 0; $i -lt 5; $i++) {
+                $element = $Browser.FindElement($selector);
+
+                if ($element) {
+                    break;
+                } else {
+                    Start-Sleep 1;
+                }
+            }
+
+            if ($element) {
                 $Browser.FindElement([OpenQA.Selenium.By]::CssSelector($ButtonSelector)).Click();
+            } else {
+                throw "Unable to find download button!";
+            }
         };
     }
 };

scripts/Windows/OS/Install.ps1

@@ -373,11 +373,11 @@ $null = New-Module {
                                 if ($install) {
                                     Install-ChocoPackage `
                                         k-litecodecpackmega `
-                                        jellyfin-media-player `
                                         vlc `
                                         ;
 
                                     Remove-DesktopIcon "VLC*";
+                                    Install-ChocoPackage jellyfin-media-player -ArgumentList "--install-args","/norestart"
                                     Install-WingetPackage Ytmdesktop.Ytmdesktop;
                                     Remove-DesktopIcon "Youtube Music*";
                                 }
@@ -415,16 +415,20 @@ $null = New-Module {
                                     Install-ChocoPackage `
                                         goggalaxy `
                                         epicgameslauncher `
-                                        steam `
                                         rayman-controlpanel `
                                         ppsspp `
                                         ;
 
                                     Remove-DesktopIcon "*Epic Games*";
-                                    Remove-DesktopIcon "*Steam*";
                                     Remove-DesktopIcon "*PPSSPP *-Bit*";
 
-                                    Install-ChocoPackage ubisoft-connect -ArgumentList "--ignore-checksums";
+                                    Install-ChocoPackage `
+                                        steam `
+                                        ubisoft-connect `
+                                        -ArgumentList "--ignore-checksums" `
+                                        ;
+
+                                    Remove-DesktopIcon "*Steam*"; 
                                     Remove-DesktopIcon "*Ubisoft Connect*";
 
                                     Install-WingetPackage ElectronicArts.EADesktop;