GUI/DismImage.ps1
|
#======================================================================= # Load Assemblies #======================================================================= # Assign current script directory to a global variable $Global:MyScriptDir = [System.IO.Path]::GetDirectoryName($myInvocation.MyCommand.Definition) # Load presentationframework and Dlls for the MahApps.Metro theme [System.Reflection.Assembly]::LoadWithPartialName("presentationframework") | Out-Null [System.Reflection.Assembly]::LoadFrom("$Global:MyScriptDir\assembly\MahApps.Metro.dll") | Out-Null [System.Reflection.Assembly]::LoadFrom("$Global:MyScriptDir\assembly\System.Windows.Interactivity.dll") | Out-Null #======================================================================= # Set PowerShell Window Title #======================================================================= $host.ui.RawUI.WindowTitle = "PowerShell Start-DismImage" #======================================================================= # Test-InWinPE #======================================================================= function Test-InWinPE { return Test-Path -Path Registry::HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlset\Control\MiniNT } #======================================================================= # LoadForm #======================================================================= function LoadForm { [CmdletBinding()] Param( [Parameter(Mandatory = $False, Position = 1)] [string]$XamlPath ) # Import the XAML code [xml]$Global:xmlWPF = Get-Content -Path $XamlPath # Add WPF and Windows Forms assemblies Try { Add-Type -AssemblyName PresentationCore, PresentationFramework, WindowsBase, system.windows.forms } Catch { Throw "Failed to load Windows Presentation Framework assemblies." } #Create the XAML reader using a new XML node reader $Global:xamGUI = [Windows.Markup.XamlReader]::Load((new-object System.Xml.XmlNodeReader $xmlWPF)) #Create hooks to each named object in the XAML $xmlWPF.SelectNodes("//*[@Name]") | foreach { Set-Variable -Name ($_.Name) -Value $xamGUI.FindName($_.Name) -Scope Global } } LoadForm -XamlPath (Join-Path $Global:MyScriptDir 'DismImage.xaml') #======================================================================= # GlobalVariables #======================================================================= $Global:DismImage = @{ BiosVersion = Get-MyBiosVersion AvailableDisks = Get-Disk.fixed | Where-Object {$_.IsBoot -eq $false} Manufacturer = Get-MyComputerManufacturer -Brief Model = Get-MyComputerModel -Brief RegCurrentVersion = Get-ItemProperty -Path 'HKLM:\SOFTWARE\Microsoft\Windows NT\CurrentVersion' SerialNumber = Get-MyBiosSerialNumber -Brief Title = 'DismImage' Version = Get-Module -Name OSDeploy | Sort-Object -Property Version | Select-Object -ExpandProperty Version -Last 1 } #======================================================================= # Subtitle #======================================================================= $Global:DismImage.ProductName = ($Global:DismImage.RegCurrentVersion).ProductName if ($Global:DismImage.RegCurrentVersion.DisplayVersion -gt 0) { $Global:DismImage.DisplayVersion = ($Global:DismImage.RegCurrentVersion).DisplayVersion } else { $Global:DismImage.DisplayVersion = ($Global:DismImage.RegCurrentVersion).ReleaseId } $Global:DismImage.BuildNumber = "$($Global:DismImage.RegCurrentVersion.CurrentBuild).$($Global:DismImage.RegCurrentVersion.UBR)" #======================================================================= # Content #======================================================================= $LabelTitle.Content = $Global:DismImage.Title $LabelVersion.Content = $Global:DismImage.Version $LabelManufacturer.Content = $Global:DismImage.Manufacturer $LabelModel.Content = $Global:DismImage.Model $LabelSerialNumber.Content = $Global:DismImage.SerialNumber $LabelSubtitle.Content = "$($Global:DismImage.ProductName) $($Global:DismImage.DisplayVersion) ($($Global:DismImage.BuildNumber))" $LabelBiosVersion.Content = "BIOS: $($Global:DismImage.BiosVersion)" $DismDescriptionTextbox.Text = "$($Global:DismImage.Manufacturer) $($Global:DismImage.Model) $($Global:DismImage.SerialNumber)" #======================================================================= # Right Side TPM #======================================================================= try { $TpmSpecVersion = (Get-CimInstance -Namespace "root\CIMV2\Security\MicrosoftTPM" -ClassName Win32_Tpm).SpecVersion } catch {} if ($TpmSpecVersion -match '2.0') { $LabelTpmVersion.Content = "TPM: 2.0" $LabelTpmVersion.Background = "Green" } elseif ($TpmSpecVersion -match '1.2') { $LabelTpmVersion.Content = "TPM: 1.2" $LabelTpmVersion.Background = "Red" } else { $LabelTpmVersion.Visibility = "Collapsed" } #======================================================================= # Hide #======================================================================= function Reset-DismSource { $DismSourceLabel.Content = "" $DismSourceLabel.Visibility = "Collapsed" $DismSourceCombobox.Items.Clear() $DismSourceCombobox.Visibility = "Collapsed" } function Reset-DismDestination { $DismDestinationLabel.Content = "" $DismDestinationLabel.Visibility = "Collapsed" $DismDestinationCombobox.Items.Clear() $DismDestinationCombobox.Visibility = "Collapsed" } function Reset-DismName { $DismNameLabel.Content = "" $DismNameLabel.Visibility = "Collapsed" $DismNameTextbox.Text = "" $DismNameTextbox.Visibility = "Collapsed" } function Reset-DismDescription { $DismDescriptionLabel.Content = "" $DismDescriptionLabel.Visibility = "Collapsed" $DismDescriptionTextbox.Text = "" $DismDescriptionTextbox.Visibility = "Collapsed" } function Reset-DismCompress { $DismCompressLabel.Content = "" $DismCompressLabel.Visibility = "Collapsed" $DismCompressCombobox.Items.Clear() $DismCompressCombobox.Visibility = "Collapsed" } #======================================================================= # Show #======================================================================= function Show-DismSource { $DismSourceLabel.Visibility = "Visible" $DismSourceCombobox.Visibility = "Visible" } function Show-DismDestination { $DismDestinationLabel.Visibility = "Visible" $DismDestinationCombobox.Visibility = "Visible" } function Show-DismName { $DismNameLabel.Visibility = "Visible" $DismNameTextbox.Visibility = "Visible" } function Show-DismDescription { $DismDescriptionLabel.Visibility = "Visible" $DismDescriptionTextbox.Visibility = "Visible" $DismDescriptionTextbox.Text = "$($Global:DismImage.Manufacturer) $($Global:DismImage.Model) $($Global:DismImage.SerialNumber)" } function Show-DismCompress { $DismCompressLabel.Visibility = "Visible" $DismCompressCombobox.Visibility = "Visible" } #======================================================================= # DismAction Defaults #======================================================================= function Set-DismAction { $DismActionLabel.Content = "Dism.exe" $DismActionCombobox.Items.Clear() $DismActionCombobox.Items.Add('Command Line Help') | Out-Null $DismActionCombobox.Items.Add('/Apply-FFU') | Out-Null $DismActionCombobox.Items.Add('/Capture-FFU') | Out-Null $DismActionCombobox.SelectedIndex = "0" $DismCommandTextbox.Text = "Dism.exe /?" $StartButton.Visibility = "Visible" } function Reset-DismAction { #Write-Verbose -Verbose 'Start Reset-DismAction' $DismCommandTextbox.Text = "Dism.exe /?" $StartButton.Visibility = "Visible" Reset-DismSource Reset-DismDestination Reset-DismName Reset-DismDescription Reset-DismCompress #Write-Verbose -Verbose 'End Reset-DismAction' } Set-DismAction #======================================================================= # DismAction add_SelectionChanged #======================================================================= $DismActionCombobox.add_SelectionChanged({ Reset-DismAction Start-DismAction }) #======================================================================= # Start-DismAction #======================================================================= function Start-DismAction { $DismCommandTextbox.Background = "#002846" $DismCommandTextbox.Foreground = "White" if ($DismActionCombobox.SelectedValue -eq 'Command Line Help') { } if ($DismActionCombobox.SelectedValue -eq '/Apply-FFU') { $DismCommandTextbox.Text = "Dism.exe /Apply-FFU /?" $StartButton.Visibility = "Collapsed" Start-SourceAction } if ($DismActionCombobox.SelectedValue -eq '/Capture-FFU') { $DismCommandTextbox.Text = "Dism.exe /Capture-FFU /?" $StartButton.Visibility = "Collapsed" $DismCompressLabel.Content = "/Compress:" $DismCompressCombobox.Items.Clear() $DismCompressCombobox.Items.Add('Default') | Out-Null $DismCompressCombobox.Items.Add('None') | Out-Null $DismCompressCombobox.SelectedIndex = "0" Start-SourceAction } } #======================================================================= # SourceAction #======================================================================= function Start-SourceAction { if ($DismActionCombobox.SelectedValue -eq '/Apply-FFU') { $DismSourceCombobox.Items.Clear() if (!($Global:DismImage.AvailableDisks)) { $DismCommandTextbox.Background = "Pink" $DismCommandTextbox.Foreground = "Red" $DismCommandTextbox.Text = "There are no disks present that can be used as the ApplyDrive. Make sure you are not booting to the disk that you want to Apply and try again" } else { $Global:ArrayOfDiskNumbers = @() $Global:DismImage.AvailableDisks | foreach { $DismSourceCombobox.Items.Add("\\.\PhysicalDrive$($_.DiskNumber) [$($_.BusType) $($_.MediaType) - $($_.FriendlyName)]") | Out-Null $Global:ArrayOfDiskNumbers += $_.Number } $DismSourceLabel.Content = "/ApplyDrive:" Show-DismSource } } if ($DismActionCombobox.SelectedValue -eq '/Capture-FFU') { $DismSourceCombobox.Items.Clear() if (!($Global:DismImage.AvailableDisks)) { $DismCommandTextbox.Background = "Pink" $DismCommandTextbox.Foreground = "Red" $DismCommandTextbox.Text = "There are no disks present that can be used as the CaptureDrive. Make sure you are not booting to the disk that you want to Capture and try again" } else { $Global:ArrayOfDiskNumbers = @() $Global:DismImage.AvailableDisks | foreach { $DismSourceCombobox.Items.Add("\\.\PhysicalDrive$($_.DiskNumber) [$($_.BusType) $($_.MediaType) - $($_.FriendlyName)]") | Out-Null $Global:ArrayOfDiskNumbers += $_.Number } $DismSourceLabel.Content = "/CaptureDrive:" Show-DismSource } } } #======================================================================= # DismSource add_SelectionChanged #======================================================================= $DismSourceCombobox.add_SelectionChanged({ Start-DestinationAction }) #======================================================================= # DismDestination Start-DestinationAction #======================================================================= function Start-DestinationAction { $DismCommandTextbox.Background = "#002846" $DismCommandTextbox.Foreground = "White" if ($DismActionCombobox.SelectedValue -eq '/Apply-FFU') { #Determine the Selected Disk information $Global:DismImage.SourceDiskNumber = (Get-Disk.fixed | Where-Object { $_.Number -eq $Global:ArrayOfDiskNumbers[$DismSourceCombobox.SelectedIndex] }).DiskNumber $Global:DismImage.PhysicalDrive = "\\.\PhysicalDrive$($Global:DismImage.SourceDiskNumber)" $DismCommandTextbox.Text = "Dism.exe /Apply-FFU /ApplyDrive:$($Global:DismImage.PhysicalDrive) /?" $DismDestinationCombobox.Items.Clear() $Global:DismImage.ApplyDrives = @() $Global:DismImage.ApplyDrives = Get-Disk.storage | Where-Object {$_.DiskNumber -ne $($Global:DismImage.SourceDiskNumber)} if (!($Global:DismImage.ApplyDrives)) { $DismCommandTextbox.Background = "Pink" $DismCommandTextbox.Foreground = "Red" $DismCommandTextbox.Text = "There are no drives present that can contain an ImageFile. Make sure you are not Applying on the drive that contains the ImageFile and try again" } else { $DismDestinationLabel.Content = "/ImageFile:" Show-DismDestination foreach ($ApplyDrive in $Global:DismImage.ApplyDrives) { if (Test-Path "$($ApplyDrive.DriveLetter):\Images") { $ApplyImageFiles += Get-ChildItem "$($ApplyDrive.DriveLetter):\Images" -Include *.ffu -File -Recurse -Force -ErrorAction Ignore | Select-Object -ExpandProperty FullName } } if ($ApplyImageFiles) { foreach ($Item in $ApplyImageFiles) { $DismDestinationCombobox.Items.Add($Item) | Out-Null } $DismDestinationCombobox.SelectedIndex = 0 } $DismDestinationCombobox.IsEditable = "True" Set-DismCommandApplyFFU $StartButton.Visibility = "Visible" } } if ($DismActionCombobox.SelectedValue -eq '/Capture-FFU') { #Determine the Selected Disk information $Global:DismImage.SourceDiskNumber = (Get-Disk.fixed | Where-Object { $_.Number -eq $Global:ArrayOfDiskNumbers[$DismSourceCombobox.SelectedIndex] }).DiskNumber $Global:DismImage.PhysicalDrive = "\\.\PhysicalDrive$($Global:DismImage.SourceDiskNumber)" $DismCommandTextbox.Text = "Dism.exe /Capture-FFU /CaptureDrive:$($Global:DismImage.PhysicalDrive) /?" $DismDestinationCombobox.Items.Clear() $Global:DismImage.ApplyDrives = @() $Global:DismImage.ApplyDrives = Get-Disk.storage | Where-Object {$_.DiskNumber -ne $Global:DismImage.SourceDiskNumber} if (!($Global:DismImage.ApplyDrives)) { $DismCommandTextbox.Background = "Pink" $DismCommandTextbox.Foreground = "Red" $DismCommandTextbox.Text = "There are no drives present that you can save an ImageFile on. Insert a FAST USB Drive or map a Network Drive and try again" } else { $DismDestinationLabel.Content = "/ImageFile:" Show-DismDestination #Dism Name $Global:DismImage.Name = "disk$($Global:DismImage.SourceDiskNumber)" $DismNameTextbox.Text = $Global:DismImage.Name foreach ($DestinationDrive in $Global:DismImage.ApplyDrives) { if ($DestinationDrive.DriveLetter -gt 0) { $Global:DismImage.ImageFile = "$($DestinationDrive.DriveLetter):\Images\$($Global:DismImage.Manufacturer)\$($Global:DismImage.Model)\$($Global:DismImage.SerialNumber)_$($Global:DismImage.Name).ffu" $DismDestinationCombobox.Items.Add($Global:DismImage.ImageFile) | Out-Null } } $DismDestinationCombobox.SelectedIndex = 0 $DismDestinationCombobox.IsEditable = "True" Show-DismName Show-DismDescription Show-DismCompress Set-DismCommandCaptureFFU $StartButton.Visibility = "Visible" } } } #======================================================================= # Set-DismCommand #======================================================================= function Set-DismCommandApplyFFU { $DismCommandTextbox.Text = "Dism.exe /Apply-FFU /ApplyDrive:$($Global:DismImage.PhysicalDrive) /ImageFile=`"$($DismDestinationCombobox.Text)`"" } function Set-DismCommandCaptureFFU { $DismCommandTextbox.Text = "Dism.exe /Capture-FFU /CaptureDrive:$($Global:DismImage.PhysicalDrive) /ImageFile=`"$($DismDestinationCombobox.Text)`" /Name:`"$($DismNameTextbox.Text)`" /Description:`"$($DismDescriptionTextbox.Text)`" /Compress:$($DismCompressCombobox.SelectedValue)" } #======================================================================= # add_SelectionChanged #======================================================================= $DismDestinationCombobox.add_SelectionChanged({ if ($DismActionCombobox.SelectedValue -eq '/Apply-FFU') { Set-DismCommandApplyFFU } if ($DismActionCombobox.SelectedValue -eq '/Capture-FFU') { Set-DismCommandCaptureFFU } }) $DismDestinationCombobox.add_DropDownClosed({ if ($DismActionCombobox.SelectedValue -eq '/Apply-FFU') { Set-DismCommandApplyFFU } if ($DismActionCombobox.SelectedValue -eq '/Capture-FFU') { Set-DismCommandCaptureFFU } }) $DismDestinationCombobox.add_KeyUp({ if ($DismActionCombobox.SelectedValue -eq '/Apply-FFU') { Set-DismCommandApplyFFU } if ($DismActionCombobox.SelectedValue -eq '/Capture-FFU') { Set-DismCommandCaptureFFU } }) $DismNameTextbox.add_KeyUp({ if ($DismActionCombobox.SelectedValue -eq '/Apply-FFU') { Set-DismCommandApplyFFU } if ($DismActionCombobox.SelectedValue -eq '/Capture-FFU') { Set-DismCommandCaptureFFU } }) $DismDescriptionTextbox.add_KeyUp({ if ($DismActionCombobox.SelectedValue -eq '/Apply-FFU') { Set-DismCommandApplyFFU } if ($DismActionCombobox.SelectedValue -eq '/Capture-FFU') { Set-DismCommandCaptureFFU } }) #======================================================================= # StartButton #======================================================================= $StartButton.add_Click({ #$xamGUI.Close() $Global:DismParams = @{} #======================================================================= # Command Line Help #======================================================================= if ($DismActionCombobox.SelectedValue -eq 'Command Line Help') { Start-Process PowerShell.exe -ArgumentList '-NoExit','-NoLogo','Dism.exe','/?' } #======================================================================= # /Apply-FFU #======================================================================= if ($DismActionCombobox.SelectedValue -eq '/Apply-FFU') { $ParentDirectory = Split-Path $DismDestinationCombobox.Text -Parent -ErrorAction Stop if (!(Test-Path $DismDestinationCombobox.Text)) { Write-Warning "ImageFile already exists. Rename the ImageFile and try again" } elseif (!(Test-InWinPE)) { Write-Warning "Dism.exe /Apply-FFU requires WinPE so nothing is going to happen here" } else { $xamGUI.Close() $Global:DismParams = @{ Dism = '/Apply-FFU' ApplyDrive = $Global:DismImage.PhysicalDrive ImageFile = $DismDestinationCombobox.Text } Get-OSDPower -Property High Start-Process PowerShell.exe -Wait -WorkingDirectory $ParentDirectory -ArgumentList '-NoExit','-NoLogo','Dism.exe','/Apply-FFU',"/ApplyDrive:'$($Global:DismParams.ApplyDrive)'","/ImageFile:'$($Global:DismParams.ImageFile)'" Get-OSDPower -Property Balanced } } #======================================================================= # /Capture-FFU #======================================================================= if ($DismActionCombobox.SelectedValue -eq '/Capture-FFU') { $ParentDirectory = Split-Path $DismDestinationCombobox.Text -Parent -ErrorAction Stop New-Item -Path $ParentDirectory -ItemType Directory -Force -ErrorAction Ignore | Out-Null if (Test-Path $DismDestinationCombobox.Text) { Write-Warning "ImageFile already exists. Rename the ImageFile and try again" } elseif (!(Test-Path "$ParentDirectory")) { Write-Warning "ImageFile appears to be in a Read Only directory or some junk path. Try another Path" } elseif (!(Test-InWinPE)) { Write-Warning "Dism.exe /Capture-FFU requires WinPE so nothing is going to happen here" } else { $xamGUI.Close() $Global:DismParams = @{ Dism = '/Capture-FFU' CaptureDrive = $Global:DismImage.PhysicalDrive ImageFile = $DismDestinationCombobox.Text Name = $DismNameTextbox.Text Description = $DismDescriptionTextbox.Text Compress = $DismCompressCombobox.SelectedValue } Get-OSDPower -Property High Start-Process PowerShell.exe -Wait -WorkingDirectory $ParentDirectory -ArgumentList '-NoExit','-NoLogo','Dism.exe','/Capture-FFU',"/CaptureDrive:'$($Global:DismParams.CaptureDrive)'","/ImageFile:'$($Global:DismParams.ImageFile)'","/Name:'$($Global:DismParams.Name)'","/Description:'$($Global:DismParams.Description)'","/Compress:$($Global:DismParams.Compress)" Get-OSDPower -Property Balanced if (Test-Path $DismDestinationCombobox.Text) { Get-WindowsImage -ImagePath $DismDestinationCombobox.Text -ErrorAction Ignore } } } }) #======================================================================= # ShowDialog #======================================================================= $xamGUI.ShowDialog() | Out-Null #======================================================================= |