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

    Fault : 0x80070643 installing KB5034441 -- WinRE partition too small
    Fix   : Shrink the OS partition and grow the recovery partition
    Source: https://jbtecwiz.com/support/win-kb5034441-winre

    Run as     : Elevated command prompt in Windows
    Expect     : 20-30 minutes
    Risk       : high
    Reversible : NO -- read the undo note below before running

    WHEN THIS IS THE RIGHT FIX
    You want the update installed. This repartitions the disk -- take a
    backup first, and do not run it on a machine without one.

    HOW TO UNDO IT
    There is no in-place rollback once the partition is deleted -- restore
    from the image taken in step 1.

    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 8"
    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 '  0x80070643 installing KB5034441 -- WinRE partition too small' -ForegroundColor White
Write-Host '  Shrink the OS partition and grow the recovery partition' -ForegroundColor Cyan
Write-Host ''
Write-Host '  Risk: high   NOT REVERSIBLE   20-30 minutes' -ForegroundColor Yellow
Write-Host ''
Show-Prose 'Undo: There is no in-place rollback once the partition is deleted -- restore from the image taken in step 1.' 'Yellow'
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 'Take a full backup or a disk image. This step is not optional; the rest of this procedure deletes a partition.' -Why 'diskpart delete override removes the recovery partition outright. If step 6 fails you need a way back.' -Manual

Invoke-Step -Number 2 -Do 'Disable WinRE and note the location it reports.' -Command {
    cmd.exe /c 'reagentc /disable & reagentc /info'
}

Invoke-Step -Number 3 -Do 'Start diskpart and select the disk and the Windows partition.' -Command {
    cmd.exe /c 'diskpart & list disk & sel disk 0 & list part'
}

Invoke-Step -Number 4 -Do 'Shrink the Windows partition by 250 MB. Substitute the partition number you saw for the large NTFS one.' -Command {
    cmd.exe /c 'sel part 3 & shrink desired=250 minimum=250'
}

Invoke-Step -Number 5 -Do 'Delete the old recovery partition. On a GPT disk it is the one with id de94bba4-06d1-4d40-a16a-bfd50179d6ac.' -Command {
    cmd.exe /c 'sel part 4 & delete partition override'
}

Invoke-Step -Number 6 -Do 'Create the replacement. GPT disks use the first block, MBR disks the second.' -Command {
    cmd.exe /c 'create partition primary id=de94bba4-06d1-4d40-a16a-bfd50179d6ac & gpt attributes=0x8000000000000001 & format quick fs=ntfs label="Windows RE tools" & exit'
}

Invoke-Step -Number 7 -Do 'Re-enable WinRE and confirm it registered.' -Command {
    cmd.exe /c 'reagentc /enable & reagentc /info'
}

Invoke-Step -Number 8 -Do 'Run Windows Update again.' -Manual

Write-Rule "  Confirm it worked"
Show-Prose 'reagentc /info reports Windows RE status: Enabled with a valid location, and the update installs.' 'White'
Write-Host ''
if (-not $DryRun) {
    cmd.exe /c 'reagentc /info'
}

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-kb5034441-winre' 'Yellow'
} else {
    Write-Host '  Finished.' -ForegroundColor Green
}
Write-Host ''
Show-Prose 'To undo: There is no in-place rollback once the partition is deleted -- restore from the image taken in step 1.' 'DarkGray'
Write-Rule