CheckTSPrompt1.3.1.ps1

#Requires -version 2.0


# ***************************************************************************
# 
# File:      CheckTSPrompt.ps1
#
# Version:   1.3.1 (modified gui)
# 
# Author:    Jason Scheffelmaer 
# 
# 
# Purpose:   Checks selected task sequences to ensure that all packages they
#            reference are available on selected distribution points.
#
#            This requires PowerShell 2.0 and at least version 1.5
#            of SCCM.psm1.
#
# Usage:     Run this script.  No parameters are required.
#            You will be prompted for ConfigMgr Server to connect to,
#            Task Sequence to check, and Distribution Point to replicate
#            to. You will also be prompted to replicate or not and the end.
#
#            $dpExcludeList is a variable to hold all the distribution point
#            servers that you wish to exclude from the list or replication.       
#
# Version Change:
#
#            1.2: Workaround Issue for New-SCCMPackageDP now calling New-SCCMPackageDP_Custom
#            1.3: Now includes version 1.5 of SCCM.psm1 which has the correct fix in for 
#                 New-SCCMPackageDP to include sitecode.  Removed New-SCCMPackageDP_Custom.
#
#
# ------------- DISCLAIMER -------------------------------------------------
# This script code is provided as is with no guarantee or waranty concerning
# the usability or impact on systems and may be used, distributed, and
# modified in any way provided the parties agree and acknowledge the 
# Microsoft or Microsoft Partners have neither accountabilty or 
# responsibility for results produced by use of this script.
#
# Microsoft will not provide any support through any means.
# ------------- DISCLAIMER -------------------------------------------------
#
# ***************************************************************************



# DP Exclude List so they do not show up in the list
$dpExcludeList = "MYServer01", "MyServer02"

# ----------
# Functions
# ----------


# The CheckPackages function checks each task sequence, finding all the packages
# that it references.  It then checks each appropriate DP to make sure that the
# needed packages are available on on those DPs.


function CheckPackages
{
    process
    {
        # Get the list of packages referenced by this task sequence
        $tsID = $_.PackageID
        
        Write-Host "Getting a list of packages referenced by Task Sequence $tsID"
        $tsReferences = Get-SCCMObject SMS_TaskSequenceReferencesInfo "PackageID='$tsID' and ProgramName='*'"    
        
        # Check each package
        Write-Host "Checking all packages..."
        foreach ($tsPackage in $tsReferences)
        {
         
            # Get a list of DPs for this package
            $tsPackageID = $tsPackage.ReferencePackageID
            #write-host "Checking Package: $tspackageID"
            $tsReferenceDPs = Get-SCCMObject SMS_TaskSequenceReferenceDPs "TaskSequenceID='$tsID' and PackageID='$tsPackageID'"

            # Check each DP to see if the package is on it
            foreach ($dp in $global:SelecteddpList)
            {
                $dpSiteCode = $dp.SiteCode
                $found = $false
                foreach ($tsDP in $tsReferenceDPs)
                {
                    
                    if ($tsDP.ServerNALPath -eq $dp.NALPath)
                    {
                        #Write-Host "Found package"
                        $found = $true
                        break
                    }
                }
                if ($found)
                {
                    New-Object PSObject -Property @{SiteCode=$dpSiteCode; PackageID=($tsPackageID); ServerName=($dp.ServerName); ShareName=($dp.ShareName); NALPath=($dp.NALPath); Result="Found"}
                }
                else
                {
                    # Check to see if this DP is a PXE Service Point, and if so mark as Not Needed.
                    if ($tsPackage.ReferencePackageType -eq 258 -or (-not $dp.NALPath.Contains("SMSPXEIMAGES$")))
                    {
                        Write-Host " $tsPackageID - Package Not Found"
                        New-Object PSObject -Property @{SiteCode=$dpSiteCode; PackageID=($tsPackageID); ServerName=($dp.ServerName); ShareName=($dp.ShareName); NALPath=($dp.NALPath); Result="Not Found"}
                    }
                    else
                    {
                        New-Object PSObject -Property @{SiteCode=$dpSiteCode; PackageID=($tsPackageID); ServerName=($dp.ServerName); ShareName=($dp.ShareName); NALPath=($dp.NALPath); Result="Not Needed"}
                    }
                }
            }
        }
    }
}

# The InputTask function prompts for a task sequence PackageID, using an input box.
function InputPrompt
{
[CmdletBinding()]
    PARAM
    (
        [Parameter(Position=1)] $objectText,
        [Parameter(Position=2)] $TitleText,
        [Parameter(Position=3)] $PromptType
    )
    
[void] [System.Reflection.Assembly]::LoadWithPartialName("System.Windows.Forms")
[void] [System.Reflection.Assembly]::LoadWithPartialName("System.Drawing") 

# New Form
$objForm = New-Object System.Windows.Forms.Form 
$objForm.Text = $TitleText
$objForm.Size = New-Object System.Drawing.Size(500,405)
$objForm.StartPosition = "CenterScreen"

# Monitor KeyPress Events
$objForm.KeyPreview = $True
$objForm.Add_KeyDown({
    if ($_.KeyCode -eq "Enter") 
    {
        If ($checkbox.checked -eq $true)
        {$x='All';$objForm.Close()}
        Else
        {
            If ($PromptType -eq 'TS')
                {$x=$objListBox.SelectedItem;$objForm.Close()}
            ElseIf ($PromptType -eq 'DP')
                {$x=$objListBox.SelectedItem;$objForm.Close()}
            ElseIf ($PromptType -eq 'Summary')
                {$x='Yes';$objForm.Close()}
            Else
                {$x=$objTextBox.text;$objForm.Close()}
         }
    }})
$objForm.Add_KeyDown({if ($_.KeyCode -eq "Escape") 
    {$x='Cancel';$objForm.Close()}})

# Create OK Button
$OKButton = New-Object System.Windows.Forms.Button
$OKButton.Location = New-Object System.Drawing.Size(310,10)
$OKButton.Size = New-Object System.Drawing.Size(75,23)
$OKButton.Add_Click(
{
    If ($checkbox.checked -eq $true)
    {$x='All';$objForm.Close()}
    Else
    {
        If ($PromptType -eq 'TS')
            {$x=$objListBox.SelectedItem;$objForm.Close()}
        ElseIf ($PromptType -eq 'DP')
            {$x=$objListBox.SelectedItem;$objForm.Close()}
        ElseIf ($PromptType -eq 'Summary')
            {$x='Yes';$objForm.Close()}
        Else
            {$x=$objTextBox.text;$objForm.Close()}
     }
}
)
$objForm.Controls.Add($OKButton)

# Create Cancel Button
$CancelButton = New-Object System.Windows.Forms.Button
$CancelButton.Location = New-Object System.Drawing.Size(400,10)
$CancelButton.Size = New-Object System.Drawing.Size(75,23)
$CancelButton.Text = "Cancel"
$CancelButton.Add_Click({$x='Cancel';$objForm.Close()})
$objForm.Controls.Add($CancelButton)

# Create Label
$objLabel = New-Object System.Windows.Forms.Label
$objLabel.Location = New-Object System.Drawing.Size(10,14) 
$objLabel.Size = New-Object System.Drawing.Size(160,27) 
$objLabel.Text = $TitleText
$objForm.Controls.Add($objLabel)

# Create Texbox
$objTextBox = New-Object System.Windows.Forms.TextBox 
$objTextBox.Location = New-Object System.Drawing.Size(170,12) 
$objTextBox.Size = New-Object System.Drawing.Size(100,20) 

# Create ListBox
$objListBox = New-Object System.Windows.Forms.ListBox 
$objListBox.Location = New-Object System.Drawing.Size(10,41) 
$objListBox.Size = New-Object System.Drawing.Size(465,100) 
$objListBox.Height = 320

# Create Checbkox
$checkbox = New-Object System.Windows.Forms.CheckBox 
$checkbox.Location = New-Object System.Drawing.Size(170,1) 
$checkbox.Size = New-Object System.Drawing.Size(150,40) 
$checkbox.Add_Click({if ($checkbox.Checked -eq $true){$objListBox.enabled = $false}
else {$objListBox.enabled = $true}})

# Add DataGrid View
$dgDataGrid = new-object System.windows.forms.DataGrid
$dgDataGrid.AllowSorting = $True
$dgDataGrid.Location = new-object System.Drawing.Size(10,45)
$dgDataGrid.size = new-object System.Drawing.Size(470,300)


# Check to see if it is a TS or DP or ServerName Prompt
If ($PromptType -eq 'TS')
{
    $OKButton.Text = "OK"
    $checkbox.text = 'All Task Sequences'
    $objForm.Controls.Add($checkbox) 
    # Fill Listbox
    $objectText | ForEach-Object {[void] $objListBox.Items.Add($_.Name)}
    $objForm.Controls.Add($objListBox) 
}
ElseIf ($PromptType -eq 'DP')
{
    $OKButton.Text = "OK"
    $checkbox.text = 'All Distribution Points'
    $objForm.Controls.Add($checkbox) 
    # Fill Listbox
    $objectText | ForEach-Object {[void] $objListBox.Items.Add($_.Name)}
    $objForm.Controls.Add($objListBox) 
}
ElseIf ($PromptType -eq 'Summary')
{
    $OKButton.Text = "Yes"
    $objForm.Controls.Add($dgDataGrid)
    $dgDataGrid.DataSource = $Global:logTable
}
Else
{
    # Assume Textbox Prompt
    $objForm.Controls.Add($objTextBox) 
    $OKButton.Text = "Connect"
}

$objForm.Topmost = $True

$objForm.Add_Shown({$objForm.Activate()})
[void] $objForm.ShowDialog()

return $x
}

# ----------
# Main logic
# ----------

# Prompt for ConfigMgr Server Name
$ConfigMgr_Server = InputPrompt -TitleText "Enter ConfigMgr Server Name"
If ($ConfigMgr_Server -eq 'Cancel'){Exit}
$scriptStartDate = Get-Date
Write-Host "Script Started at: $scriptStartDate"

# Connect to ConfigMgr
Import-Module SCCM -force
Write-Host "Connecting to CongfigMgr Server $ConfigMgr_Server"
New-SCCMConnection -serverName $ConfigMgr_Server


# Get the list of all DPs to choose from via a menu
Write-Host "Getting a list of all Distribution Points to select from"
$global:dpList = Get-SCCMObject SMS_DistributionPointInfo | where {$dpExcludeList -notcontains $_.ServerName}

# Get the list of all task sequences to choose from via a menu
Write-Host "Getting a list of all Task Sequences to select from"
$TaskSequences = get-SCCMTaskSequencePackage | select-object Name, PackageID |  Sort-Object Name

# Prompt for Task Sequence to use when checking packages
$taskSequenceName = InputPrompt -objectText $TaskSequences -TitleText "Select Task Sequence" -PromptType 'TS'
If ($taskSequenceName -eq 'Cancel' -or $taskSequenceName -eq $null)
{

    Write-Host "No Task Sequence selected or user selected Cancel, exiting script..."
    Exit 
  
}
ElseIf ($taskSequenceName -eq 'All')
{
    $SelectedTS = $TaskSequences
}
Else
{
    $SelectedTS = $TaskSequences | ? {$_.Name -eq $taskSequenceName}
    $SelectedTSID = $SelectedTS.PackageID
}

# Prompt for Distribution Point to check the referenced packages against
$DPNameList = ($global:dpList| Select-Object @{Name="Name"; Expression={$_.ServerName}}, NalPath, SiteCode, SiteName) | Sort-Object Name
$global:DPName = InputPrompt -objectText ($DPNameList | Select-Object Name | Get-Unique -AsString) -TitleText "Select Distribution Point" -PromptType 'DP'
If ($global:DPName -eq 'Cancel' -or $global:DPName -eq $null)
{
    Write-Host "No Distribution Point selected or user selected Cancel, exiting script..."
    Exit 
    
}
ElseIf ($global:DPName -eq 'All')
{
    $global:SelecteddpList =  $DPNameList
}
Else
{
    $global:SelecteddpList =  $DPNameList | ? {$_.Name -eq $global:DPName}
}

# Get the package status for all packages referenced 
# by the task sequence selected.
Write-Host "Calling the CheckPackages function"
If ($taskSequenceName -eq 'All')
{
    $packageResults = Get-SCCMTaskSequencePackage | CheckPackages
}
Else
{
    $packageResults = Get-SCCMTaskSequencePackage -filter "PackageID='$SelectedTSID'"| CheckPackages
}

# Get a list of packages that need to be distributed to DP
$neededPackages = $packageResults | ? {$_.Result -eq 'Not Found'}
# $neededPackages | Format-Table

# If any packages are missing from DPs, add them to those DPs
if ($neededPackages -ne $null)
{
    # Fill Missing Packages Tables
    $Global:logTable = New-Object System.Data.DataTable
    $Global:logTable.TableName = "Missing Packages"
    $Global:logTable.Columns.Add("Package");
    $Global:logTable.Columns.Add("Status");
    $Global:logTable.Columns.Add("NALPath");
    $Global:logTable.Columns.Add("SiteCode");
    Foreach ($missingPackage in $neededPackages)
    {
        $Global:logTable.Rows.Add($missingPackage.PackageID,$missingPackage.Result,$missingPackage.NALPath, $missingPackage.SiteCode)
    }

    $replicate = InputPrompt -TitleText "Found Missing Packages, Replicate?" -PromptType 'Summary'
    If ($replicate -eq 'Yes')
    {
        Write-Host "Distribuing packages to distribution point"
        $newDPs = $neededPackages | New-SCCMPackageDP
        $newCount = ($newDPs | Measure-Object).Count
        Write-Host "$newCount package(s) distributed to distribution point."
    }
    Else
    {
        Write-Host "Cancel Selected, will not replicate packages"
    }
}
else
{
    Write-Host "No packages needed to be distributed."
}
$scriptEndDate = Get-Date
Write-Host "Script finished at: $scriptEndDate"