#requires -Version 5.1
<#
.SYNOPSIS
NN Rescue Copy - Nerdy Neighbor GUI rescue-backup tool.
Run: irm copy.nerdyneighbor.net | iex
.DESCRIPTION
Copies user data from a slaved customer drive to a backup drive, including
OneDrive/Dropbox cloud-reparse files that robocopy/xcopy/Explorer cannot open
(ERROR 1920) when cldflt.sys is not running. Read-only on the source drive.
Full Windows only (WPF). See github.com/nerd-industries/rescue-copy.
#>
$NNIsWindows = [Environment]::OSVersion.Platform -eq 'Win32NT'
# Captured at top level so the STA relauncher can persist the running script to disk.
$NNScriptPath = $PSCommandPath
#region Constants
$NNOsExcludeRoot = @('Windows','Program Files','Program Files (x86)','ProgramData','PerfLogs','Recovery',
'System Volume Information','$Recycle.Bin','$WinREAgent','Users','OneDriveTemp','Config.Msi',
'Intel','AMD','NVIDIA','Drivers','MSOCache','inetpub','Documents and Settings')
$NNProfileExclude = @('Default','Default User','Public','All Users','defaultuser0','WDAGUtilityAccount')
$NNVisibleFolders = @('Desktop','Documents','Pictures','Downloads','Videos','Music','Favorites')
# Each def: Label shown in GUI, Folder = destination folder name under AppData-Rescue, Segs = path under the profile dir
$NNAppDataDefs = @(
@{ Label = 'Chrome (bookmarks, passwords, profiles)'; Folder = 'Chrome'; Segs = @('AppData','Local','Google','Chrome','User Data') }
@{ Label = 'Edge (bookmarks, passwords, profiles)'; Folder = 'Edge'; Segs = @('AppData','Local','Microsoft','Edge','User Data') }
@{ Label = 'Firefox profiles'; Folder = 'Firefox'; Segs = @('AppData','Roaming','Mozilla','Firefox','Profiles') }
@{ Label = 'Outlook data files (PST/OST)'; Folder = 'Outlook'; Segs = @('AppData','Local','Microsoft','Outlook') }
@{ Label = 'Outlook Files (Documents)'; Folder = 'Outlook Files'; Segs = @('Documents','Outlook Files') }
@{ Label = 'Windows Mail'; Folder = 'WindowsMail'; Segs = @('AppData','Local','Comms') }
)
#endregion
#region Path utilities
function Join-NNParts {
param([string[]]$Parts)
$p = $Parts[0]
for ($i = 1; $i -lt $Parts.Count; $i++) { $p = Join-Path $p $Parts[$i] }
return $p
}
function Test-NNPathOverlap {
param([string]$PathA, [string]$PathB)
$a = $PathA.TrimEnd('\', '/') + [IO.Path]::DirectorySeparatorChar
$b = $PathB.TrimEnd('\', '/') + [IO.Path]::DirectorySeparatorChar
if ($a.StartsWith($b, [StringComparison]::OrdinalIgnoreCase)) { return $true }
if ($b.StartsWith($a, [StringComparison]::OrdinalIgnoreCase)) { return $true }
return $false
}
function ConvertTo-NNLongPath {
param([string]$Path)
if ($Path.StartsWith('\\?\')) { return $Path }
if ($Path.StartsWith('\\')) { return '\\?\UNC\' + $Path.Substring(2) }
return '\\?\' + $Path
}
function Get-NNSafeName {
param([string]$Name)
# Fixed Windows-invalid set (NOT [IO.Path]::GetInvalidFileNameChars(): on the
# Linux test box that set lacks ':' '?' etc., but job folders land on NTFS/exFAT).
$invalid = [char[]]@('<', '>', ':', '"', '/', '\', '|', '?', '*')
$sb = New-Object System.Text.StringBuilder
foreach ($ch in $Name.ToCharArray()) {
if ([int]$ch -ge 32 -and $invalid -notcontains $ch) { $null = $sb.Append($ch) }
}
return (($sb.ToString() -replace '\s+', ' ').Trim())
}
function Get-NNJobRoot {
param([string]$BackupRoot, [string]$JobName)
return Join-NNParts @($BackupRoot, 'NN-Rescue', (Get-NNSafeName $JobName))
}
function Format-NNBytes {
param([long]$Bytes)
if ($Bytes -ge 1TB) { return ('{0:N1} TB' -f ($Bytes / 1TB)) }
if ($Bytes -ge 1GB) { return ('{0:N1} GB' -f ($Bytes / 1GB)) }
if ($Bytes -ge 1MB) { return ('{0:N1} MB' -f ($Bytes / 1MB)) }
if ($Bytes -ge 1KB) { return ('{0:N1} KB' -f ($Bytes / 1KB)) }
return ('{0} B' -f $Bytes)
}
function New-NNTarget {
param([string]$Category, [string]$User, [string]$Label, [string]$SourcePath, [string]$DestRel, [bool]$Selected)
[pscustomobject]@{
Category = $Category
User = $User
Label = $Label
SourcePath = $SourcePath
DestRel = $DestRel
SizeBytes = $null
Selected = $Selected
}
}
#endregion
#region Scanner
function Get-NNUserProfileDirs {
param([string]$SourceRoot)
$users = Join-Path $SourceRoot 'Users'
if (-not (Test-Path -LiteralPath $users)) { return @() }
@(Get-ChildItem -LiteralPath $users -Directory -Force -ErrorAction SilentlyContinue |
Where-Object { $NNProfileExclude -notcontains $_.Name })
}
function Get-NNProfileTargets {
param([IO.DirectoryInfo]$ProfileDir)
$u = $ProfileDir.Name
$out = @()
foreach ($n in $NNVisibleFolders) {
$p = Join-Path $ProfileDir.FullName $n
if (Test-Path -LiteralPath $p) {
$out += New-NNTarget 'Profile' $u $n $p (Join-NNParts @('Users', $u, $n)) $true
}
}
foreach ($od in @(Get-ChildItem -LiteralPath $ProfileDir.FullName -Directory -Force -Filter 'OneDrive*' -ErrorAction SilentlyContinue)) {
$out += New-NNTarget 'Profile' $u $od.Name $od.FullName (Join-NNParts @('Users', $u, $od.Name)) $true
}
$out
}
function Get-NNAppDataTargets {
param([IO.DirectoryInfo]$ProfileDir)
$u = $ProfileDir.Name
$out = @()
foreach ($d in $NNAppDataDefs) {
$p = Join-NNParts (@($ProfileDir.FullName) + $d.Segs)
if (Test-Path -LiteralPath $p) {
$out += New-NNTarget 'AppData' $u $d.Label $p (Join-NNParts @('Users', $u, 'AppData-Rescue', $d.Folder)) $true
}
}
$pk = Join-NNParts @($ProfileDir.FullName, 'AppData', 'Local', 'Packages')
if (Test-Path -LiteralPath $pk) {
foreach ($sn in @(Get-ChildItem -LiteralPath $pk -Directory -Filter 'Microsoft.MicrosoftStickyNotes_*' -ErrorAction SilentlyContinue)) {
$ls = Join-Path $sn.FullName 'LocalState'
if (Test-Path -LiteralPath $ls) {
$out += New-NNTarget 'AppData' $u 'Sticky Notes' $ls (Join-NNParts @('Users', $u, 'AppData-Rescue', 'StickyNotes')) $true
}
}
}
$out
}
function Get-NNExtraTargets {
param([string]$SourceRoot)
$out = @()
foreach ($d in @(Get-ChildItem -LiteralPath $SourceRoot -Directory -Force -ErrorAction SilentlyContinue)) {
if ($NNOsExcludeRoot -notcontains $d.Name) {
$out += New-NNTarget 'Extra' $null $d.Name $d.FullName (Join-NNParts @('Extras', $d.Name)) $false
}
}
$pub = Join-NNParts @($SourceRoot, 'Users', 'Public')
foreach ($n in @('Documents', 'Desktop', 'Pictures', 'Videos', 'Music', 'Downloads')) {
$p = Join-Path $pub $n
if (Test-Path -LiteralPath $p) {
$sel = ($n -eq 'Documents') -or ($n -eq 'Desktop')
$out += New-NNTarget 'Public' 'Public' $n $p (Join-NNParts @('Public', $n)) $sel
}
}
foreach ($k in @(@('ProgramData', 'Intuit'), @('ProgramData', 'Sage'))) {
$p = Join-NNParts (@($SourceRoot) + $k)
if (Test-Path -LiteralPath $p) {
$out += New-NNTarget 'Extra' $null ($k -join ' ') $p (Join-NNParts @('Extras', ($k -join '-'))) $false
}
}
$out
}
function Get-NNFolderSize {
param([string]$Path)
$sum = [long]0
foreach ($f in @(Get-ChildItem -LiteralPath $Path -Recurse -Force -File -ErrorAction SilentlyContinue)) {
$sum += $f.Length
}
return $sum
}
#endregion
#region Hostname
function ConvertFrom-NNRegQuery {
param([string[]]$Lines, [string]$ValueName)
foreach ($l in $Lines) {
if ($l -match ('^\s*' + [regex]::Escape($ValueName) + '\s+REG_\w+\s+(.+?)\s*$')) {
return $Matches[1]
}
}
return $null
}
function Get-NNSourceHostname {
param([string]$SourceRoot)
if (-not $NNIsWindows) { return $null }
$hive = Join-NNParts @($SourceRoot, 'Windows', 'System32', 'config', 'SYSTEM')
if (-not (Test-Path -LiteralPath $hive)) { return $null }
$mount = 'HKLM\NN-RESCUE-SYS'
& reg.exe load $mount $hive > $null 2>&1
if ($LASTEXITCODE -ne 0) { return $null }
try {
$sel = & reg.exe query "$mount\Select" /v Current 2>$null
$cur = ConvertFrom-NNRegQuery $sel 'Current'
if (-not $cur) { return $null }
$n = [Convert]::ToInt32($cur, 16)
$key = ('{0}\ControlSet{1:D3}\Control\ComputerName\ComputerName' -f $mount, $n)
$q = & reg.exe query $key /v ComputerName 2>$null
return (ConvertFrom-NNRegQuery $q 'ComputerName')
} catch {
return $null
} finally {
& reg.exe unload $mount > $null 2>&1
}
}
#endregion
#region Raw copy engine
if (-not ('RawFileNative' -as [type])) {
Add-Type -TypeDefinition @'
using System;
using System.Runtime.InteropServices;
using Microsoft.Win32.SafeHandles;
public static class RawFileNative {
[DllImport("kernel32.dll", SetLastError=true, CharSet=CharSet.Unicode)]
public static extern SafeFileHandle CreateFileW(string name, uint access, uint share,
IntPtr sec, uint disposition, uint flags, IntPtr template);
}
'@
}
# Decimal constants: PS5.1 parses large hex literals as negative int32
$NNGenericRead = [uint32]2147483648 # 0x80000000
$NNShareAll = [uint32]7 # READ | WRITE | DELETE
$NNOpenExisting = [uint32]3
$NNRawFlags = [uint32]35651584 # FILE_FLAG_BACKUP_SEMANTICS | FILE_FLAG_OPEN_REPARSE_POINT
$NNCloudOnlyAttrs = 4198400 # FILE_ATTRIBUTE_OFFLINE (0x1000) | RECALL_ON_DATA_ACCESS (0x400000)
function Open-NNSourceStream {
param([string]$Path)
if (-not $NNIsWindows) { return [IO.File]::OpenRead($Path) } # test seam: prod is Windows-only
$h = [RawFileNative]::CreateFileW((ConvertTo-NNLongPath $Path), $NNGenericRead, $NNShareAll,
[IntPtr]::Zero, $NNOpenExisting, $NNRawFlags, [IntPtr]::Zero)
if ($h.IsInvalid) {
throw (New-Object ComponentModel.Win32Exception ([Runtime.InteropServices.Marshal]::GetLastWin32Error()))
}
return (New-Object IO.FileStream ($h, [IO.FileAccess]::Read))
}
function Test-NNCloudOnly {
param([IO.FileInfo]$File)
return ((([int]$File.Attributes) -band $NNCloudOnlyAttrs) -ne 0)
}
function Copy-NNFile {
param([IO.FileInfo]$Src, [string]$DestPath, [bool]$Force, [byte[]]$Buffer)
if (Test-NNCloudOnly $Src) { return 'CLOUD-ONLY' }
# Long-path-safe destination (>260 chars); no-op off Windows (test seam)
if ($NNIsWindows) { $DestPath = ConvertTo-NNLongPath $DestPath }
$dstInfo = New-Object IO.FileInfo ($DestPath)
if (-not $Force -and $dstInfo.Exists -and $dstInfo.Length -eq $Src.Length) { return 'SKIP-EXISTS' }
$in = $null
try { $in = Open-NNSourceStream $Src.FullName }
catch {
$code = 'err=' + ($_.Exception.HResult -band 0xFFFF)
if ($_.Exception -is [ComponentModel.Win32Exception]) { $code = 'err=' + $_.Exception.NativeErrorCode }
if ($_.Exception -is [IO.FileNotFoundException] -or $_.Exception -is [IO.DirectoryNotFoundException]) { $code = 'err=2' }
if ($_.Exception.InnerException -is [IO.FileNotFoundException] -or $_.Exception.InnerException -is [IO.DirectoryNotFoundException]) { $code = 'err=2' }
return "OPEN-FAIL($code)"
}
$out = $null
try {
$out = [IO.File]::Create($DestPath)
while (($n = $in.Read($Buffer, 0, $Buffer.Length)) -gt 0) { $out.Write($Buffer, 0, $n) }
} catch {
return "READ-FAIL($($_.Exception.Message))"
} finally {
if ($in) { $in.Dispose() }
if ($out) { $out.Dispose() }
}
$dstInfo.Refresh()
if ($dstInfo.Length -ne $Src.Length) { return "SIZE-MISMATCH($($dstInfo.Length)/$($Src.Length))" }
try {
$dstInfo.CreationTimeUtc = $Src.CreationTimeUtc
$dstInfo.LastWriteTimeUtc = $Src.LastWriteTimeUtc
} catch { }
return 'OK'
}
function Invoke-NNCopyJob {
param($Targets, [string]$JobRoot, $Control, $Queue, [bool]$Force)
# Pass 1: enumerate work
$files = New-Object System.Collections.Generic.List[object]
$problems = New-Object System.Collections.Generic.List[string]
$totalBytes = [long]0
$enumErrs = @()
foreach ($t in $Targets) {
$src = Get-Item -Force -LiteralPath $t.SourcePath -ErrorAction SilentlyContinue
if (-not $src) { continue }
if ($src.PSIsContainer) {
foreach ($f in @(Get-ChildItem -LiteralPath $src.FullName -Recurse -Force -File -ErrorAction SilentlyContinue -ErrorVariable +enumErrs)) {
$rel = $f.FullName.Substring($src.FullName.Length).TrimStart('\', '/')
$files.Add(@{ File = $f; Dest = (Join-Path (Join-Path $JobRoot $t.DestRel) $rel) })
$totalBytes += $f.Length
}
} else {
$files.Add(@{ File = $src; Dest = (Join-Path $JobRoot $t.DestRel) })
$totalBytes += $src.Length
}
}
if (@($enumErrs).Count -gt 0) {
$cap = [Math]::Min(50, @($enumErrs).Count)
for ($ei = 0; $ei -lt $cap; $ei++) {
$problems.Add('ENUM-FAIL(' + $enumErrs[$ei].Exception.Message + ')')
}
}
$Queue.Enqueue([pscustomobject]@{ Type = 'plan'; Count = $files.Count; Bytes = $totalBytes })
$buffer = New-Object byte[] 1048576
$summary = @{}
$pairs = New-Object System.Collections.Generic.List[object]
$done = [long]0
$i = 0
$lastDir = ''
try {
$null = [IO.Directory]::CreateDirectory($JobRoot)
$csvPath = Join-Path $JobRoot '_RescueLog.csv'
$newCsv = -not (Test-Path -LiteralPath $csvPath)
$csv = New-Object IO.StreamWriter ($csvPath, $true, [Text.Encoding]::UTF8)
if ($newCsv) { $csv.WriteLine('Time,Result,Bytes,Source,Destination') }
try {
foreach ($item in $files) {
while ($Control.Pause -and -not $Control.Cancel) { Start-Sleep -Milliseconds 200 }
if ($Control.Cancel) { break }
$destDir = Split-Path $item.Dest
if ($destDir -ne $lastDir) {
$mkDir = $destDir
if ($NNIsWindows) { $mkDir = ConvertTo-NNLongPath $mkDir }
$null = [IO.Directory]::CreateDirectory($mkDir)
$lastDir = $destDir
}
$r = Copy-NNFile -Src $item.File -DestPath $item.Dest -Force $Force -Buffer $buffer
if ($summary.ContainsKey($r)) { $summary[$r]++ } else { $summary[$r] = 1 }
if ($r -eq 'OK') { $pairs.Add(@{ Src = $item.File.FullName; Dst = $item.Dest }) }
if ($r -ne 'OK' -and $r -ne 'SKIP-EXISTS') { $problems.Add("$r $($item.File.FullName)") }
$done += $item.File.Length
$i++
$csv.WriteLine(('{0:o},"{1}",{2},"{3}","{4}"' -f [DateTime]::UtcNow, $r.Replace('"', '""'), $item.File.Length,
$item.File.FullName.Replace('"', '""'), $item.Dest.Replace('"', '""')))
$Queue.Enqueue([pscustomobject]@{ Type = 'file'; Result = $r; Path = $item.File.FullName; BytesDone = $done; Index = $i })
if ($r -like 'READ-FAIL*') {
$free = -1
try { $free = (New-Object IO.DriveInfo([IO.Path]::GetPathRoot($JobRoot))).AvailableFreeSpace } catch { }
if ($free -ge 0 -and $free -lt 1048576) {
$problems.Add('FATAL(Destination drive is full - copy halted. Free space and re-run to resume.)')
break
}
}
}
} finally {
$csv.Dispose()
}
} catch {
$problems.Add('FATAL(' + $_.Exception.Message + ')')
}
$Queue.Enqueue([pscustomobject]@{ Type = 'done'; Summary = $summary; Problems = [object[]]$problems; Pairs = [object[]]$pairs;
Cancelled = [bool]$Control.Cancel; BytesDone = $done })
}
#endregion
#region Verify
function Get-NNFileHash {
param([string]$Path)
$sha = $null
$s = $null
try {
$sha = [Security.Cryptography.SHA256]::Create()
$s = Open-NNSourceStream $Path
$buf = New-Object byte[] 1048576
while (($n = $s.Read($buf, 0, $buf.Length)) -gt 0) {
$null = $sha.TransformBlock($buf, 0, $n, $null, 0)
}
$null = $sha.TransformFinalBlock($buf, 0, 0)
return ([BitConverter]::ToString($sha.Hash) -replace '-', '')
} finally {
if ($s) { $s.Dispose() }
if ($sha) { $sha.Dispose() }
}
}
function Invoke-NNVerifyPass {
param($Pairs, $Control, $Queue)
$mismatch = New-Object System.Collections.Generic.List[string]
$i = 0
try {
foreach ($p in $Pairs) {
if ($Control.Cancel) { break }
try {
$a = Get-NNFileHash $p.Src
$b = Get-NNFileHash $p.Dst
if ($a -ne $b) { $mismatch.Add("HASH-MISMATCH $($p.Src)") }
} catch {
$mismatch.Add("VERIFY-FAIL($($_.Exception.Message)) $($p.Src)")
}
$i++
$Queue.Enqueue([pscustomobject]@{ Type = 'verify'; Index = $i; Total = @($Pairs).Count })
}
} catch {
$mismatch.Add('VERIFY-FATAL(' + $_.Exception.Message + ')')
}
$Queue.Enqueue([pscustomobject]@{ Type = 'verifydone'; Mismatches = $mismatch })
return $mismatch
}
#endregion
#region Report
function New-NNHtmlReport {
param([hashtable]$Summary, [string[]]$Problems, [string]$JobName, [long]$Bytes, [string]$OutPath)
function Encode-NNHtml { param([string]$s)
return $s.Replace('&', '&').Replace('<', '<').Replace('>', '>').Replace('"', '"')
}
$rows = ''
foreach ($k in ($Summary.Keys | Sort-Object)) {
$color = '#e2e8f0'
if ($k -eq 'OK') { $color = '#4ade80' }
elseif ($k -eq 'SKIP-EXISTS') { $color = '#94a3b8' }
elseif ($k -eq 'CLOUD-ONLY') { $color = '#fbbf24' }
else { $color = '#f87171' }
$rows += ('{1} {2} {0}Files needing attention ' -f $items)
}
$cloudHtml = ''
$cloudCount = 0
foreach ($k in $Summary.Keys) { if ($k -eq 'CLOUD-ONLY') { $cloudCount = $Summary[$k] } }
if ($cloudCount -gt 0) {
$cloudHtml = ('{0} cloud-only file(s) were skipped. ' +
'These files live only in the customer''s cloud account (OneDrive/Dropbox) - there is no local data on the drive. ' +
'To recover them, sign into the account on a working machine and let them download.
') -f $cloudCount
}
$html = @"
NN Rescue Report - $(Encode-NNHtml $JobName)
NN Rescue Copy
Job: $(Encode-NNHtml $JobName) · $(Format-NNBytes $Bytes) processed · $([DateTime]::Now.ToString('yyyy-MM-dd HH:mm'))
Results
$cloudHtml
$probHtml
Nerdy Neighbor · copy.nerdyneighbor.net
"@
[IO.File]::WriteAllText($OutPath, $html, [Text.Encoding]::UTF8)
return $OutPath
}
#endregion
#region Selection model
function Build-NNSelectionModel {
param([string]$SourceRoot, [bool]$IncludeAppData)
$groups = @()
foreach ($pd in @(Get-NNUserProfileDirs $SourceRoot)) {
$items = @(Get-NNProfileTargets $pd)
if ($IncludeAppData) { $items += @(Get-NNAppDataTargets $pd) }
if ($items.Count -gt 0) { $groups += @{ Header = "User: $($pd.Name)"; Items = $items } }
}
$extras = @(Get-NNExtraTargets $SourceRoot)
if ($extras.Count -gt 0) { $groups += @{ Header = 'Extras found on drive'; Items = $extras } }
return ,$groups
}
function Get-NNSelectedBytes {
param($Model)
$s = [long]0
foreach ($g in $Model) {
foreach ($i in $g.Items) {
if ($i.Selected -and $null -ne $i.SizeBytes) { $s += [long]$i.SizeBytes }
}
}
return $s
}
#endregion
#region XAML
$NNXaml = @'
'@
#endregion
#region Background
function Start-NNBackground {
param([string]$FunctionName, [object[]]$Arguments)
$iss = [System.Management.Automation.Runspaces.InitialSessionState]::CreateDefault()
foreach ($fn in @(Get-ChildItem function: | Where-Object { $_.Name -like '*-NN*' })) {
$iss.Commands.Add((New-Object System.Management.Automation.Runspaces.SessionStateFunctionEntry ($fn.Name, $fn.Definition)))
}
foreach ($vn in @('NNIsWindows','NNOsExcludeRoot','NNProfileExclude','NNVisibleFolders','NNAppDataDefs',
'NNGenericRead','NNShareAll','NNOpenExisting','NNRawFlags','NNCloudOnlyAttrs')) {
$v = Get-Variable -Name $vn -ErrorAction SilentlyContinue
if ($v) {
$iss.Variables.Add((New-Object System.Management.Automation.Runspaces.SessionStateVariableEntry ($vn, $v.Value, $null)))
}
}
$ps = [powershell]::Create($iss)
$null = $ps.AddCommand($FunctionName)
foreach ($a in $Arguments) { $null = $ps.AddArgument($a) }
return @{ PS = $ps; Handle = $ps.BeginInvoke() }
}
function Get-NNVolumes {
$out = @()
foreach ($d in [IO.DriveInfo]::GetDrives()) {
if (-not $d.IsReady) { continue }
$root = $d.RootDirectory.FullName
$badge = @()
if (Test-Path -LiteralPath (Join-Path $root 'Windows')) { $badge += 'Windows' }
if (Test-Path -LiteralPath (Join-Path $root 'Users')) { $badge += 'Users' }
$out += [pscustomobject]@{
Drive = $d.Name.TrimEnd('\', '/')
Label = $d.VolumeLabel
Size = Format-NNBytes $d.TotalSize
Free = Format-NNBytes $d.AvailableFreeSpace
FreeBytes = [long]$d.AvailableFreeSpace
Contents = ($badge -join ', ')
Root = $root
}
}
return $out
}
#endregion
#region GUI
# All GUI helpers are top-level functions operating on $script:NNCtx - WPF event
# handler scriptblocks execute against the session scope chain and would NOT see
# functions nested inside Start-NNRescueGui.
function Show-NNStep {
param([int]$n)
$script:NNCtx.Step = $n
$ui = $script:NNCtx.UI
for ($i = 1; $i -le 5; $i++) {
$ui["PanelStep$i"].Visibility = 'Collapsed'
$ui["RailStep$i"].Foreground = '#94A3B8'
$ui["RailStep$i"].FontWeight = 'Normal'
}
$ui["PanelStep$n"].Visibility = 'Visible'
$ui["RailStep$n"].Foreground = '#38BDF8'
$ui["RailStep$n"].FontWeight = 'Bold'
$ui.BtnBack.Visibility = 'Hidden'
if ($n -eq 2 -or $n -eq 3) { $ui.BtnBack.Visibility = 'Visible' }
$ui.BtnNext.Visibility = 'Visible'
if ($n -eq 1 -or $n -eq 2) { $ui.BtnNext.Content = 'Next' }
if ($n -eq 3) { $ui.BtnNext.Content = 'Start Copy' }
if ($n -ge 4) { $ui.BtnNext.Visibility = 'Hidden' }
}
function Update-NNVolumeLists {
$vols = @(Get-NNVolumes)
$script:NNCtx.UI.LvSource.ItemsSource = $vols
$script:NNCtx.UI.LvDest.ItemsSource = $vols
}
function Update-NNTotals {
$c = $script:NNCtx
if (-not $c.Model) { return }
$sel = Get-NNSelectedBytes $c.Model
$c.UI.TxtTotals.Text = ('Selected: {0} Destination free: {1}' -f (Format-NNBytes $sel), (Format-NNBytes $c.DestFreeBytes))
if ($sel -gt $c.DestFreeBytes) { $c.UI.TxtTotals.Foreground = '#F87171' }
else { $c.UI.TxtTotals.Foreground = '#4ADE80' }
}
function Add-NNTreeGroup {
param($Group)
$c = $script:NNCtx
$node = New-Object System.Windows.Controls.TreeViewItem
$node.IsExpanded = $true
$head = New-Object System.Windows.Controls.CheckBox
$head.Content = $Group.Header
$head.FontWeight = 'Bold'
$head.Foreground = '#38BDF8'
$node.Header = $head
$childBoxes = New-Object System.Collections.Generic.List[object]
foreach ($item in $Group.Items) {
$cb = New-Object System.Windows.Controls.CheckBox
$sz = ''
if ($null -ne $item.SizeBytes) { $sz = ' (' + (Format-NNBytes $item.SizeBytes) + ')' }
$cb.Content = $item.Label + $sz
$cb.IsChecked = $item.Selected
$cb.Tag = $item
$cb.Add_Checked({ param($s, $e) $s.Tag.Selected = $true; Update-NNTotals })
$cb.Add_Unchecked({ param($s, $e) $s.Tag.Selected = $false; Update-NNTotals })
$leaf = New-Object System.Windows.Controls.TreeViewItem
$leaf.Header = $cb
$null = $node.Items.Add($leaf)
$childBoxes.Add($cb)
}
$head.Tag = $childBoxes
$allOn = $true
foreach ($b in $childBoxes) { if (-not $b.IsChecked) { $allOn = $false } }
$head.IsChecked = $allOn
$head.Add_Checked({ param($s, $e) foreach ($b in $s.Tag) { $b.IsChecked = $true } })
$head.Add_Unchecked({ param($s, $e) foreach ($b in $s.Tag) { $b.IsChecked = $false } })
$null = $c.UI.TreeSel.Items.Add($node)
}
function Start-NNScan {
$c = $script:NNCtx
if ($c.ScanJob) { try { $c.ScanJob.PS.Stop(); $c.ScanJob.PS.Dispose() } catch { } }
$c.UI.TreeSel.Items.Clear()
$c.Model = New-Object System.Collections.Generic.List[object]
$c.UI.TxtScanStatus.Text = 'Scanning...'
$c.ScanGen++
$c.ScanJob = Start-NNBackground 'Invoke-NNScanJob' @($c.SourceRoot, [bool]$c.UI.ChkAppData.IsChecked, $c.Queue, $c.ScanGen)
}
function Read-NNQueue {
$c = $script:NNCtx
$m = $null
while ($c.Queue.TryDequeue([ref]$m)) {
if ($m.Type -eq 'scangroup') {
if ($m.Gen -eq $c.ScanGen) {
$c.Model.Add($m.Group)
Add-NNTreeGroup $m.Group
Update-NNTotals
}
}
elseif ($m.Type -eq 'scandone') {
if ($m.Gen -eq $c.ScanGen) {
if ($m.Error) { $c.UI.TxtScanStatus.Text = 'Scan failed: ' + $m.Error }
else { $c.UI.TxtScanStatus.Text = 'Scan complete.' }
}
}
elseif ($m.Type -eq 'plan') {
$c.PlanBytes = [long]$m.Bytes; $c.PlanCount = [int]$m.Count
$c.UI.PbOverall.Maximum = [double]([math]::Max(1, $m.Bytes / 1MB))
$c.UI.TxtPhase.Text = ('Copying {0} files ({1})...' -f $m.Count, (Format-NNBytes $m.Bytes))
}
elseif ($m.Type -eq 'file') {
$c.BytesDone = [long]$m.BytesDone
$c.UI.PbOverall.Value = [double]($m.BytesDone / 1MB)
$c.UI.TxtCurrentFile.Text = $m.Path
if ($c.Tally.ContainsKey($m.Result)) { $c.Tally[$m.Result]++ } else { $c.Tally[$m.Result] = 1 }
$parts = @()
foreach ($k in ($c.Tally.Keys | Sort-Object)) { $parts += ('{0}: {1}' -f $k, $c.Tally[$k]) }
$c.UI.TxtCounters.Text = ($parts -join ' ')
if ($m.Result -ne 'OK' -and $m.Result -ne 'SKIP-EXISTS') {
$null = $c.UI.LbProblems.Items.Add(('{0} {1}' -f $m.Result, $m.Path))
}
if ($c.CopyWatch -and $c.PlanBytes -gt 0 -and $m.BytesDone -gt 0) {
$elapsed = $c.CopyWatch.Elapsed
$etaSec = ($elapsed.TotalSeconds / $m.BytesDone) * ($c.PlanBytes - $m.BytesDone)
if ($etaSec -lt 0) { $etaSec = 0 }
$c.UI.TxtPhase.Text = ('Copying... {0:hh\:mm\:ss} elapsed, ~{1:hh\:mm\:ss} left' -f $elapsed, [TimeSpan]::FromSeconds($etaSec))
}
}
elseif ($m.Type -eq 'done') {
$c.Summary = $m.Summary; $c.Pairs = $m.Pairs
foreach ($p in $m.Problems) { $c.Problems.Add($p) }
if ($c.UI.ChkVerify.IsChecked -and @($m.Pairs).Count -gt 0 -and -not $m.Cancelled) {
$c.UI.TxtPhase.Text = ('Verifying {0} copied files (SHA-256)...' -f @($m.Pairs).Count)
$c.UI.PbOverall.Value = 0
$c.UI.PbOverall.Maximum = [double]@($m.Pairs).Count
$c.VerifyJob = Start-NNBackground 'Invoke-NNVerifyPass' @($m.Pairs, $c.Control, $c.Queue)
} else {
Complete-NNJob
}
}
elseif ($m.Type -eq 'verify') {
$c.UI.PbOverall.Value = [double]$m.Index
$c.UI.TxtCurrentFile.Text = ('Verified {0} / {1}' -f $m.Index, $m.Total)
}
elseif ($m.Type -eq 'verifydone') {
foreach ($mm in $m.Mismatches) { $c.Problems.Add($mm) }
$hashCount = 0
$failCount = 0
foreach ($mm in $m.Mismatches) {
if ($mm -like 'HASH-MISMATCH*') { $hashCount++ } else { $failCount++ }
}
if ($hashCount -gt 0) { $c.Summary['HASH-MISMATCH'] = $hashCount }
if ($failCount -gt 0) { $c.Summary['VERIFY-FAIL'] = $failCount }
Complete-NNJob
}
}
}
function Complete-NNJob {
$c = $script:NNCtx
try {
$reportPath = Join-Path $c.JobRoot '_RescueReport.html'
$null = New-NNHtmlReport -Summary $c.Summary -Problems $c.Problems -JobName $c.JobName -Bytes $c.BytesDone -OutPath $reportPath
$c.ReportPath = $reportPath
} catch {
$c.ReportPath = $null
}
$lines = @()
foreach ($k in ($c.Summary.Keys | Sort-Object)) { $lines += ('{0,-22} {1}' -f $k, $c.Summary[$k]) }
$c.UI.TxtSummary.Text = ($lines -join [Environment]::NewLine)
$cloud = 0
if ($c.Summary.ContainsKey('CLOUD-ONLY')) { $cloud = $c.Summary['CLOUD-ONLY'] }
if ($cloud -gt 0) {
$c.UI.TxtCloudNote.Text = ('{0} file(s) live only in the customer''s cloud account (OneDrive/Dropbox). ' +
'Sign into the account on a working machine to download them.') -f $cloud
} else {
$c.UI.TxtCloudNote.Text = ''
}
$c.UI.LbFinalProblems.Items.Clear()
foreach ($p in $c.Problems) { $null = $c.UI.LbFinalProblems.Items.Add($p) }
Show-NNStep 5
}
function Start-NNRescueGui {
$win = [Windows.Markup.XamlReader]::Parse($NNXaml)
$ui = @{}
foreach ($n in @('RailStep1','RailStep2','RailStep3','RailStep4','RailStep5',
'PanelStep1','PanelStep2','PanelStep3','PanelStep4','PanelStep5',
'LvSource','LvDest','BtnBrowseSource','BtnRescan','TxtSrcPick',
'TxtJobName','TxtDestPreview','TxtJobHint',
'ChkAppData','ChkVerify','ChkForce','TreeSel','TxtTotals','TxtScanStatus',
'PbOverall','TxtPhase','TxtCurrentFile','TxtCounters','LbProblems','BtnPause','BtnCancelCopy',
'TxtSummary','TxtCloudNote','LbFinalProblems','BtnOpenDest','BtnOpenReport',
'BtnBack','BtnNext')) {
$ui[$n] = $win.FindName($n)
}
$script:NNCtx = @{
Win = $win; UI = $ui; Step = 1
SourceRoot = $null; BackupRoot = $null; DestFreeBytes = [long]0
JobRoot = $null; JobName = ''
Model = $null; ScanJob = $null; CopyJob = $null; VerifyJob = $null
Control = [hashtable]::Synchronized(@{ Cancel = $false; Pause = $false })
Queue = (New-Object 'System.Collections.Concurrent.ConcurrentQueue[object]')
Tally = @{}; Problems = (New-Object System.Collections.Generic.List[string])
Summary = $null; Pairs = $null; PlanBytes = [long]0; PlanCount = 0
ReportPath = $null; BytesDone = [long]0; ScanGen = 0; CopyWatch = $null
}
# --- events ---
$ui.BtnRescan.Add_Click({ Update-NNVolumeLists })
$ui.BtnBrowseSource.Add_Click({
$dlg = New-Object System.Windows.Forms.FolderBrowserDialog
$dlg.Description = 'Pick the customer data root (e.g. a mounted image)'
if ($dlg.ShowDialog() -eq 'OK') {
$script:NNCtx.SourceRoot = $dlg.SelectedPath
$script:NNCtx.UI.TxtSrcPick.Text = 'Source: ' + $dlg.SelectedPath
$script:NNCtx.UI.LvSource.SelectedItem = $null
}
})
$ui.LvSource.Add_SelectionChanged({
if ($script:NNCtx.UI.LvSource.SelectedItem) {
$script:NNCtx.SourceRoot = $script:NNCtx.UI.LvSource.SelectedItem.Root
$script:NNCtx.UI.TxtSrcPick.Text = 'Source: ' + $script:NNCtx.SourceRoot
}
})
$ui.TxtJobName.Add_TextChanged({
$c = $script:NNCtx
if ($c.BackupRoot) {
$c.UI.TxtDestPreview.Text = Get-NNJobRoot $c.BackupRoot $c.UI.TxtJobName.Text
}
})
$ui.ChkAppData.Add_Click({ if ($script:NNCtx.Step -eq 3) { Start-NNScan } })
$ui.BtnBack.Add_Click({
$c = $script:NNCtx
if ($c.Step -eq 3 -and $c.ScanJob) { try { $c.ScanJob.PS.Stop() } catch { } }
if ($c.Step -gt 1) { Show-NNStep ($c.Step - 1) }
})
$ui.BtnNext.Add_Click({
$c = $script:NNCtx
if ($c.Step -eq 1) {
if (-not $c.SourceRoot) {
[System.Windows.MessageBox]::Show('Pick the customer (source) drive first.', 'NN Rescue Copy') | Out-Null; return
}
if (-not $c.UI.LvDest.SelectedItem) {
[System.Windows.MessageBox]::Show('Pick the backup (destination) drive.', 'NN Rescue Copy') | Out-Null; return
}
$c.BackupRoot = $c.UI.LvDest.SelectedItem.Root
$c.DestFreeBytes = [long]$c.UI.LvDest.SelectedItem.FreeBytes
if (Test-NNPathOverlap $c.BackupRoot $c.SourceRoot) {
[System.Windows.MessageBox]::Show('Destination overlaps the source drive. Pick a different backup drive.', 'NN Rescue Copy') | Out-Null; return
}
if (-not $c.UI.TxtJobName.Text) {
$hn = Get-NNSourceHostname $c.SourceRoot
if ($hn) { $c.UI.TxtJobName.Text = $hn }
else { $c.UI.TxtJobHint.Text = 'Could not read the drive''s registry - enter a name manually.' }
}
Show-NNStep 2
}
elseif ($c.Step -eq 2) {
$name = Get-NNSafeName $c.UI.TxtJobName.Text
if (-not $name) {
[System.Windows.MessageBox]::Show('Enter a customer/job name.', 'NN Rescue Copy') | Out-Null; return
}
$c.JobName = $name
$c.JobRoot = Get-NNJobRoot $c.BackupRoot $name
Show-NNStep 3
Start-NNScan
}
elseif ($c.Step -eq 3) {
$targets = @()
foreach ($g in $c.Model) { foreach ($i in $g.Items) { if ($i.Selected) { $targets += $i } } }
if ($targets.Count -eq 0) {
[System.Windows.MessageBox]::Show('Nothing selected.', 'NN Rescue Copy') | Out-Null; return
}
$sel = Get-NNSelectedBytes $c.Model
if ($sel -gt $c.DestFreeBytes) {
$r = [System.Windows.MessageBox]::Show(
('Selection ({0}) exceeds destination free space ({1}). Continue anyway?' -f (Format-NNBytes $sel), (Format-NNBytes $c.DestFreeBytes)),
'NN Rescue Copy', 'YesNo', 'Warning')
if ($r -ne 'Yes') { return }
}
$c.Control.Cancel = $false; $c.Control.Pause = $false
$c.Tally = @{}; $c.Problems.Clear(); $c.UI.LbProblems.Items.Clear()
Show-NNStep 4
$c.CopyWatch = [Diagnostics.Stopwatch]::StartNew()
$c.CopyJob = Start-NNBackground 'Invoke-NNCopyJob' @($targets, $c.JobRoot, $c.Control, $c.Queue, [bool]$c.UI.ChkForce.IsChecked)
}
})
$ui.BtnPause.Add_Click({
$c = $script:NNCtx
$c.Control.Pause = -not $c.Control.Pause
if ($c.Control.Pause) { $c.UI.BtnPause.Content = 'Resume'; $c.UI.TxtPhase.Text = 'Paused.' }
else { $c.UI.BtnPause.Content = 'Pause'; $c.UI.TxtPhase.Text = 'Copying...' }
})
$ui.BtnCancelCopy.Add_Click({ $script:NNCtx.Control.Cancel = $true })
$ui.BtnOpenDest.Add_Click({ Start-Process explorer.exe $script:NNCtx.JobRoot })
$ui.BtnOpenReport.Add_Click({ if ($script:NNCtx.ReportPath) { Start-Process $script:NNCtx.ReportPath } })
# UI pump
$timer = New-Object System.Windows.Threading.DispatcherTimer
$timer.Interval = [TimeSpan]::FromMilliseconds(250)
$timer.Add_Tick({ Read-NNQueue })
$timer.Start()
Update-NNVolumeLists
Show-NNStep 1
$null = $win.ShowDialog()
$timer.Stop()
foreach ($j in @($script:NNCtx.ScanJob, $script:NNCtx.CopyJob, $script:NNCtx.VerifyJob)) {
if ($j) { try { $j.PS.Stop(); $j.PS.Dispose() } catch { } }
}
}
function Invoke-NNScanJob {
param([string]$SourceRoot, [bool]$IncludeAppData, $Queue, [int]$Gen)
try {
$model = Build-NNSelectionModel $SourceRoot $IncludeAppData
foreach ($g in $model) {
foreach ($i in $g.Items) { $i.SizeBytes = Get-NNFolderSize $i.SourcePath }
$Queue.Enqueue([pscustomobject]@{ Type = 'scangroup'; Group = $g; Gen = $Gen })
}
$Queue.Enqueue([pscustomobject]@{ Type = 'scandone'; Error = $null; Gen = $Gen })
} catch {
$Queue.Enqueue([pscustomobject]@{ Type = 'scandone'; Error = $_.Exception.Message; Gen = $Gen })
}
}
#endregion
#region Entry
function Start-NNRescue {
if (-not $NNIsWindows) {
Write-Host 'NN Rescue Copy needs full Windows (it is a WPF app). On Linux, mount with ntfs-3g and copy normally.' -ForegroundColor Red
return
}
try {
Add-Type -AssemblyName PresentationFramework, PresentationCore, WindowsBase, System.Windows.Forms -ErrorAction Stop
} catch {
Write-Host 'This Windows has no GUI stack (WinPE / Server Core). Slave the drive into a full Windows machine or VM.' -ForegroundColor Red
return
}
if ([Threading.Thread]::CurrentThread.GetApartmentState() -ne [Threading.ApartmentState]::STA) {
$tmp = Join-Path $env:TEMP 'NN-RescueCopy.ps1'
$saved = $false
if ($NNScriptPath -and (Test-Path -LiteralPath $NNScriptPath)) {
Copy-Item -Force -LiteralPath $NNScriptPath -Destination $tmp
$saved = $true
} else {
try { Invoke-RestMethod 'https://copy.nerdyneighbor.net' -OutFile $tmp; $saved = $true } catch { }
}
if ($saved) {
$exe = 'powershell.exe'
if ($PSVersionTable.PSEdition -eq 'Core') { $exe = 'pwsh.exe' }
Write-Host 'Relaunching in STA mode for the GUI...' -ForegroundColor Cyan
Start-Process $exe -ArgumentList '-Sta', '-ExecutionPolicy', 'Bypass', '-File', $tmp
} else {
Write-Host 'Not in STA mode and could not self-relaunch. Download the script and run: powershell -Sta -File NN-RescueCopy.ps1' -ForegroundColor Red
}
return
}
Start-NNRescueGui
}
#endregion
if ($env:NN_RESCUE_NOGUI -ne '1') { Start-NNRescue }