Web Driver Auto-update

# =====================================================
# Microsoft Edge WebDriver Auto-Update
# Updates the driver when the first three version
# components do not match the installed Edge browser.
# =====================================================

# ---------- CONFIGURATION ----------

$driverFolder = "C:\Work\scripts\edgeDriver"
$driverExe = Join-Path $driverFolder "msedgedriver.exe"
$tempZip = Join-Path $env:TEMP "edgedriver.zip"
$tempExtract = Join-Path $env:TEMP "edgedriver-extract"

$edgePaths = @(
    "C:\Program Files (x86)\Microsoft\Edge\Application\msedge.exe",
    "C:\Program Files\Microsoft\Edge\Application\msedge.exe"
)

# ---------- FIND MICROSOFT EDGE ----------

$edgeExe = $edgePaths |
    Where-Object { Test-Path $_ } |
    Select-Object -First 1

if (-not $edgeExe) {
    Write-Host "ERROR: Microsoft Edge was not found."
    exit 1
}

# ---------- GET BROWSER VERSION ----------

$browserVersion = ((Get-Item $edgeExe).VersionInfo.ProductVersion).Trim()
$browserParts = $browserVersion.Split(".")

if ($browserParts.Count -lt 3) {
    Write-Host "ERROR: Unable to parse the Microsoft Edge version."
    exit 1
}

$browserCompatibilityVersion = ($browserParts[0..2] -join ".")

Write-Host "Installed Edge version: $browserVersion"

# ---------- GET DRIVER VERSION ----------

$driverVersion = $null
$driverCompatibilityVersion = $null

if (Test-Path $driverExe) {
    try {
        $driverOutput = & $driverExe --version
        $driverVersion = [regex]::Match(
            $driverOutput,
            '\d+\.\d+\.\d+\.\d+'
        ).Value

        if ($driverVersion) {
            $driverParts = $driverVersion.Split(".")
            $driverCompatibilityVersion = ($driverParts[0..2] -join ".")
            Write-Host "Installed Edge WebDriver version: $driverVersion"
        }
    }
    catch {
        Write-Host "Unable to determine the installed WebDriver version."
    }
}
else {
    Write-Host "Edge WebDriver was not found."
}

# ---------- COMPARE VERSIONS ----------

if (
    $driverCompatibilityVersion -and
    $browserCompatibilityVersion -eq $driverCompatibilityVersion
) {
    Write-Host "SUCCESS: Edge and Edge WebDriver are compatible."
    Write-Host "No update is required."
    exit 0
}

Write-Host "A WebDriver version mismatch was detected."
Write-Host "Downloading the matching Edge WebDriver..."

# ---------- PREPARE UPDATE ----------

New-Item -ItemType Directory -Force -Path $driverFolder | Out-Null

Get-Process msedgedriver -ErrorAction SilentlyContinue |
    Stop-Process -Force

foreach ($path in @($tempZip, $tempExtract)) {
    if (Test-Path $path) {
        Remove-Item $path -Recurse -Force
    }
}

$downloadUrl = "https://msedgedriver.microsoft.com/$browserVersion/edgedriver_win64.zip"

Write-Host "Download URL: $downloadUrl"

# ---------- DOWNLOAD ----------

try {
    Invoke-WebRequest -Uri $downloadUrl -OutFile $tempZip -ErrorAction Stop
}
catch {
    Write-Host "ERROR: Unable to download Edge WebDriver."
    Write-Host $_.Exception.Message
    exit 1
}

# ---------- EXTRACT ----------

try {
    Expand-Archive `
        -Path $tempZip `
        -DestinationPath $tempExtract `
        -Force `
        -ErrorAction Stop
}
catch {
    Write-Host "ERROR: Unable to extract Edge WebDriver."
    Write-Host $_.Exception.Message
    exit 1
}

$downloadedDriver = Join-Path $tempExtract "msedgedriver.exe"

if (-not (Test-Path $downloadedDriver)) {
    Write-Host "ERROR: msedgedriver.exe was not found in the archive."
    exit 1
}

# ---------- REPLACE DRIVER ----------

try {
    Copy-Item $downloadedDriver $driverExe -Force -ErrorAction Stop
}
catch {
    Write-Host "ERROR: Unable to replace the existing Edge WebDriver."
    Write-Host $_.Exception.Message
    exit 1
}

# ---------- CLEAN UP ----------

foreach ($path in @($tempZip, $tempExtract)) {
    if (Test-Path $path) {
        Remove-Item $path -Recurse -Force
    }
}

# ---------- VERIFY ----------

try {
    $newDriverOutput = & $driverExe --version
    $newDriverVersion = [regex]::Match(
        $newDriverOutput,
        '\d+\.\d+\.\d+\.\d+'
    ).Value

    $newDriverParts = $newDriverVersion.Split(".")
    $newDriverCompatibilityVersion = ($newDriverParts[0..2] -join ".")

    Write-Host "Updated Edge WebDriver version: $newDriverVersion"

    if ($newDriverCompatibilityVersion -eq $browserCompatibilityVersion) {
        Write-Host "SUCCESS: Edge WebDriver was updated successfully."
        exit 0
    }

    Write-Host "ERROR: Edge and Edge WebDriver remain incompatible."
    exit 1
}
catch {
    Write-Host "ERROR: Unable to validate the updated Edge WebDriver."
    Write-Host $_.Exception.Message
    exit 1
}