Create Invoke-SecondaryDiskFormat.ps1

This commit is contained in:
freedbygrace
2023-10-17 10:32:34 -04:00
committed by GitHub
parent 287c8bff1c
commit 752da30a47
+148
View File
@@ -0,0 +1,148 @@
Try
{
#Initialize the task sequence environment
Try {[System.__ComObject]$TSEnvironment = New-Object -ComObject "Microsoft.SMS.TSEnvironment"} Catch {}
#Define variable(s)
[Regex]$BusTypeExclusions = "USB"
[String[]]$RequiredModules = @("Storage")
$IsUEFI = [Boolean]::Parse($TSEnvironment.Value('_SMSTSBootUEFI'))
$LogMessage = "IsUEFI = $($IsUEFI.ToString())"
Write-Verbose -Message "$($LogMessage)" -Verbose
[Int]$OperatingSystemDiskNumber = $TSEnvironment.Value('OSDiskNumber')
$LogMessage = "Operating System Disk Number = $($OperatingSystemDiskNumber)"
Write-Verbose -Message "$($LogMessage)" -Verbose
$LogMessage = "Operating System disk WILL NOT be wiped because it has already been formatted by the operating system deployment solution."
Write-Verbose -Message "$($LogMessage)" -Verbose
#Import required module(s)
ForEach ($RequiredModule In $RequiredModules)
{
If (!(Get-Module -Name $RequiredModule))
{
$Null = Import-Module -Name ($RequiredModule) -DisableNameChecking -Force -ErrorAction Stop
}
}
#Get the physical disks attached to the device
$PhysicalDisks = Get-PhysicalDisk | Where-Object {($_.BusType -inotmatch $BusTypeExclusions.ToString())} | Sort-Object -Property @('Size')
#Measure the amount of physical disks detected within the device (Ensure that storage interface(s) are not disabled within the BIOS settings)
$PhysicalDiskCount = $PhysicalDisks | Measure-Object | Select-Object -ExpandProperty Count
#Determine the disk that will store the operating system
$OperatingSystemDisk = $PhysicalDisks | Where-Object {($_.DeviceID -ieq $OperatingSystemDiskNumber)}
#Determine the disk(s) that will not store the operating system
$DataDisks = $PhysicalDisks | Where-Object {($_.DeviceID -ine $OperatingSystemDisk.DeviceID)}
$DataDiskCount = $DataDisks | Measure-Object | Select-Object -ExpandProperty Count
#Get the partitions/volume(s) on the disk
If ($DataDiskCount -gt 0)
{
ForEach ($DataDisk In $DataDisks)
{
[Int]$DataDiskNumber = $DataDisk.DeviceID
[String]$DataDiskVolumeName = "DATA00$($DataDiskNumber)"
$TaskSequenceVariableName = "OSDFormatDataDisk00$($DataDiskNumber)"
$TSEnvironment.Value($TaskSequenceVariableName) = "False"
[ScriptBlock]$GetDataDiskInfo = {Get-Disk -Number ($DataDiskNumber)}
[ScriptBlock]$FormatAndPartitionDisk = {
#Clean the disk and format it according to the specification(s)
Switch ($IsUEFI)
{
{($_ -eq $True)}
{
$PartitionStyle = [Microsoft.PowerShell.Cmdletization.GeneratedTypes.Disk.PartitionStyle]::GPT
}
{($_ -eq $False)}
{
$PartitionStyle = [Microsoft.PowerShell.Cmdletization.GeneratedTypes.Disk.PartitionStyle]::MBR
}
}
#Perform disk partitioning and formatting
[String]$ActivityMessage = "Attempting to format and partition data disk $($DataDiskNumber). Please Wait..."
[String]$StatusMessage = "$($ActivityMessage)"
[Int]$RandomNumber = Get-Random -Minimum 100 -Maximum 999
Write-Progress -ID ($RandomNumber) -Activity ($ActivityMessage) -Status ($StatusMessage)
Switch ($GetDataDiskInfo.Invoke())
{
{($_.PartitionStyle -imatch "^RAW$")}
{
$Null = Initialize-Disk -InputObject ($GetDataDiskInfo.Invoke()) -PartitionStyle ($PartitionStyle) -PassThru -Confirm:$False -Verbose
}
{($_.PartitionStyle -inotmatch "^RAW$")}
{
$Null = Clear-Disk -InputObject ($GetDataDiskInfo.Invoke()) -RemoveData -Confirm:$False -PassThru -Verbose
$Null = Initialize-Disk -InputObject ($GetDataDiskInfo.Invoke()) -PartitionStyle ($PartitionStyle) -PassThru -Confirm:$False -Verbose
}
}
$Null = New-Partition -InputObject ($GetDataDiskInfo.Invoke()) -UseMaximumSize -AssignDriveLetter -Verbose | Format-Volume -FileSystem NTFS -NewFileSystemLabel ($DataDiskVolumeName) -Confirm:$False -Verbose
Write-Progress -ID ($RandomNumber) -Activity ($ActivityMessage) -Completed
}
If (($GetDataDiskInfo.Invoke()).NumberOfPartitions -gt 0)
{
$DataDiskVolumes = $GetDataDiskInfo.Invoke() | Get-Partition | Get-Volume | Where-Object {($_.DriveLetter -imatch "^[a-zA-Z]$")}
$DataDiskVolumeCount = $DataDiskVolumes | Measure-Object | Select-Object -ExpandProperty Count
If ($DataDiskVolumeCount -gt 0)
{
ForEach ($DataDiskVolume In $DataDiskVolumes)
{
$FoldersInVolumeRoot = Get-ChildItem -Path "$($DataDiskVolume.DriveLetter):\" -Force | Where-Object {($_ -is [System.IO.DirectoryInfo])}
If ($FoldersInVolumeRoot.Name -imatch "^Windows|Windows\.Old$")
{
#Set the value of the 'OSDFormatDataDisk' variable which will tell the task sequence that this disk is ok to wipe
$TSEnvironment.Value($TaskSequenceVariableName) = "True"
$LogMessage = "Disk $($DataDiskNumber) will be wiped and formatted because a previous Windows installation was detected and could cause errors with the installation of the new operating system."
Write-Verbose -Message "$($LogMessage)" -Verbose
#Clean the disk and format it according to the specification(s)
$FormatAndPartitionDisk.Invoke()
#Exit the for loop because at least one volume has already met the requirements for the disk to be formatted
Break
}
}
}
}
If (($GetDataDiskInfo.Invoke().NumberOfPartitions -eq 0) -or ($GetDataDiskInfo.Invoke().PartitionStyle -imatch '^RAW$'))
{
$LogMessage = "Disk $($DataDiskNumber) will be wiped and formatted because no partition(s) or volume(s) could be found. The disk has not been provisioned."
Write-Verbose -Message "$($LogMessage)" -Verbose
#Set the value of the 'OSDFormatDataDisk' variable which will tell the task sequence that this disk is ok to wipe
$TSEnvironment.Value($TaskSequenceVariableName) = "True"
#Clean the disk and format it according to the specification(s)
$FormatAndPartitionDisk.Invoke()
}
}
}
}
Catch
{
Write-Error -ErrorRecord ($_)
}