Merge pull request #224 from mingyaulee/patch-1

Fallback latest version in powershell script
nanasess · Sep 17, 2023 · cac7048 · cac7048
2 parents bc67901 + 0d6f1ea
commit cac7048
Showing 1 changed file with 28 additions and 12 deletions.
diff --git a/lib/setup-chromedriver.ps1 b/lib/setup-chromedriver.ps1
@@ -2,6 +2,7 @@ Param(
     [string]$version
 )
 
+$json_url = "https://googlechromelabs.github.io/chrome-for-testing/known-good-versions-with-downloads.json"
 $chrome_fullversion = (Get-Item "C:\Program Files\Google\Chrome\Application\chrome.exe").VersionInfo.FileVersion
 
 Write-Output "Chrome version: $chrome_fullversion"
@@ -13,25 +14,40 @@ else
 {
     $chrome_majorversion = $version.Split(".")[0]
 }
+
 if($chrome_majorversion -lt 115)
 {
     $response = Invoke-WebRequest "http://chromedriver.storage.googleapis.com/LATEST_RELEASE_$chrome_majorversion"
     $version = $response.Content
     Invoke-WebRequest "https://chromedriver.storage.googleapis.com/$version/chromedriver_win32.zip" -OutFile chromedriver_win32.zip
     Expand-Archive -Path chromedriver_win32.zip -DestinationPath C:\SeleniumWebDrivers\ChromeDriver -Force
     Remove-Item chromedriver_win32.zip
+    Return 0
 }
-else
+
+if([string]::IsNullOrEmpty($version))
+{
+    $version = $chrome_fullversion
+}
+
+$arch = "win32"
+Write-Output $arch
+
+Write-Output "Downloading $json_url..."
+$json = Invoke-WebRequest $json_url -UseBasicParsing | ConvertFrom-Json
+$url = $json | Select-Object -ExpandProperty versions | Where-Object { $_.version -eq $version } | Select-Object -ExpandProperty downloads | Select-Object -ExpandProperty chromedriver | Where-Object { $_.platform -eq $arch } | Select-Object -ExpandProperty url
+if (!$url)
 {
-    if([string]::IsNullOrEmpty($version))
-    {
-        $version = $chrome_fullversion
-    }
-    $arch = "win32"
-    Write-Output $arch
-    $url = Invoke-WebRequest "https://googlechromelabs.github.io/chrome-for-testing/known-good-versions-with-downloads.json" -UseBasicParsing | ConvertFrom-Json | Select-Object -ExpandProperty versions | Where-Object { $_.version -eq $version } | Select-Object -ExpandProperty downloads | Select-Object -ExpandProperty chromedriver | Where-Object { $_.platform -eq $arch } | Select-Object -ExpandProperty url
-    Invoke-WebRequest $url -OutFile chromedriver-win32.zip
-    Expand-Archive -Path chromedriver-win32.zip -Force
-    Move-Item -Path .\chromedriver-win32\chromedriver-win32\chromedriver.exe -Destination C:\SeleniumWebDrivers\ChromeDriver -Force
-    Remove-Item chromedriver-win32.zip
+    Write-Output "Falling back to latest version of ChromeDriver for $arch"
+    $version3 = $version.Substring(0, $version.LastIndexOf('.'))
+    Write-Output "VERSION3 = $version3"
+    $version = $json | ConvertFrom-Json | Select-Object -ExpandProperty versions | Where-Object { $_.version -like "$version3.*" } | Select-Object -Last 1 -ExpandProperty version
+    Write-Output "VERSION = $version"
+    $url = $json | Select-Object -ExpandProperty versions | Where-Object { $_.version -eq $version } | Select-Object -ExpandProperty downloads | Select-Object -ExpandProperty chromedriver | Where-Object { $_.platform -eq $arch } | Select-Object -ExpandProperty url
 }
+Write-Output "Installing ChromeDriver $version for $arch"
+Write-Output "Downloading $url..."
+Invoke-WebRequest $url -OutFile chromedriver-win32.zip
+Expand-Archive -Path chromedriver-win32.zip -Force
+Move-Item -Path .\chromedriver-win32\chromedriver-win32\chromedriver.exe -Destination C:\SeleniumWebDrivers\ChromeDriver -Force
+Remove-Item chromedriver-win32.zip