#Requires -Version 5.1
<#
    JbTecWiz Support Centre -- generated fix script

    Fault : Display driver stopped responding and has recovered -- TDR, event 4101
    Fix   : Establish which driver, and get to a known-good one
    Source: https://jbtecwiz.com/support/win-hw-tdr

    Run as     : Elevated PowerShell
    Expect     : 45 minutes
    Risk       : medium
    Reversible : yes

    WHEN THIS IS THE RIGHT FIX
    It happens at idle or since an update. Start here.

    HOW TO UNDO IT
    Reinstall the version noted in step two.

    This script walks the fix one step at a time and asks before each
    one. Steps with no command are things you do yourself -- it prints
    those and waits. Run with -DryRun to print without executing.

    --------------------------------------------------------------------
    NO WARRANTY - USE AT YOUR OWN RISK

    This script is provided by JbTecWiz as-is and with no warranty of any
    kind, express or implied. You run it entirely at your own risk.

    JbTecWiz accepts no liability for any loss or damage arising from its
    use, including but not limited to data loss, downtime, or configuration
    changes that turn out to be wrong for your system.

    You are responsible for reading this script before running it, for
    satisfying yourself that it suits the machine in front of you, and for
    having a working backup first. Some steps cannot be undone.
    --------------------------------------------------------------------
#>

[CmdletBinding()]
param(
    # Print every step and command without running anything.
    [switch]$DryRun,
    # Do not ask before each step. Read the script first if you use this.
    [switch]$Unattended
)

$ErrorActionPreference = 'Stop'
$script:Failed = 0

function Write-Rule { param([string]$Text)
    Write-Host ''
    Write-Host ('-' * 70) -ForegroundColor DarkGray
    if ($Text) { Write-Host $Text -ForegroundColor Cyan }
}

function Show-Prose { param([string]$Text, [string]$Colour = "Gray")
    if (-not $Text) { return }
    $words = $Text -split "\s+"; $line = "  "
    foreach ($w in $words) {
        if (($line.Length + $w.Length) -gt 74) { Write-Host $line -ForegroundColor $Colour; $line = "  " }
        $line += "$w "
    }
    if ($line.Trim()) { Write-Host $line -ForegroundColor $Colour }
}

function Invoke-Step {
    param(
        [int]$Number, [string]$Do, [string]$Why,
        [scriptblock]$Command, [switch]$Manual, [string]$Shell = "powershell"
    )
    Write-Rule "  Step $Number of 5"
    Show-Prose $Do "White"
    if ($Why) { Write-Host ""; Show-Prose $Why "DarkGray" }

    if ($Manual) {
        Write-Host ''
        Write-Host '  -> Do this yourself, then press Enter to carry on.' -ForegroundColor Yellow
        if (-not $Unattended -and -not $DryRun) { [void](Read-Host) }
        return
    }

    Write-Host ''
    foreach ($l in ($Command.ToString().Trim() -split "`n")) {
        Write-Host ("    " + $l.Trim()) -ForegroundColor Green
    }
    Write-Host ''

    if ($DryRun) { Write-Host "  (dry run -- not executed)" -ForegroundColor DarkGray; return }
    if (-not $Unattended) {
        $a = Read-Host "  Run this step? [Y]es / [S]kip / [Q]uit"
        if ($a -match "^[Qq]") { Write-Host "  Stopped at your request."; exit 0 }
        if ($a -match "^[Ss]") { Write-Host "  Skipped." -ForegroundColor DarkGray; return }
    }
    try { & $Command }
    catch {
        $script:Failed++
        Write-Host ("  Step $Number failed: " + $_.Exception.Message) -ForegroundColor Red
        Show-Prose "The rest of the fix may depend on this. Read the write-up before carrying on." "Red"
        if (-not $Unattended) {
            $c = Read-Host "  Carry on anyway? [y/N]"
            if ($c -notmatch "^[Yy]") { exit 1 }
        }
    }
}

Write-Rule
Write-Host '  Display driver stopped responding and has recovered -- TDR, event 4101' -ForegroundColor White
Write-Host '  Establish which driver, and get to a known-good one' -ForegroundColor Cyan
Write-Host ''
Write-Host '  Risk: medium   Reversible   45 minutes'
Write-Host ''
Show-Prose 'No warranty. Use at your own risk - JbTecWiz accepts no liability. Read it before you run it, and have a backup.' 'DarkYellow'
Write-Rule

if (-not $Unattended -and -not $DryRun) {
    $go = Read-Host '  Ready? [y/N]'
    if ($go -notmatch "^[Yy]") { Write-Host "  Nothing was changed."; exit 0 }
}

Invoke-Step -Number 1 -Do 'Read the events. The message names the driver file, which is the whole diagnosis.' -Command {
    Get-WinEvent -LogName System -MaxEvents 200 | Where-Object { $_.Id -eq 4101 -or $_.ProviderName -like '*Display*' } | Format-Table TimeCreated,Id,Message -Wrap
}

Invoke-Step -Number 2 -Do 'Note the current driver version before changing it.' -Command {
    Get-CimInstance Win32_VideoController | Format-List Name,DriverVersion,DriverDate,VideoProcessor
}

Invoke-Step -Number 3 -Do 'Uninstall the display driver completely and install the manufacturer''s current studio or WHQL build -- not a beta, and not the version Windows Update supplies.' -Why 'Windows Update carries a generic driver that is often several versions behind and occasionally mismatched to the card''s firmware. For graphics the vendor''s own package is the correct source.' -Manual

Invoke-Step -Number 4 -Do 'Stop Windows Update replacing it again on the next reboot while you test.' -Command {
    Get-WindowsDriver -Online | Where-Object ClassName -eq 'Display' | Format-Table Driver,ProviderName,Version
}

Invoke-Step -Number 5 -Do 'If the current version is the problem, install the previous one and test for a full day before concluding.' -Manual

Write-Rule "  Confirm it worked"
Show-Prose 'No 4101 events across a day of normal use.' 'White'
Write-Host ''
if (-not $DryRun) {
    Get-WinEvent -LogName System -MaxEvents 500 | Where-Object Id -eq 4101 | Measure-Object | Format-List Count
}

Write-Rule
if ($script:Failed -gt 0) {
    Write-Host ("  Finished with " + $script:Failed + " failed step(s).") -ForegroundColor Yellow
    Show-Prose 'Read the full write-up at https://jbtecwiz.com/support/win-hw-tdr' 'Yellow'
} else {
    Write-Host '  Finished.' -ForegroundColor Green
}
Write-Host ''
Show-Prose 'To undo: Reinstall the version noted in step two.' 'DarkGray'
Write-Rule