Extending Windows Computer class from a CSV file in SCOM

  <TypeDefinitions>
    <EntityTypes>
      <ClassTypes>
        <ClassType ID="DemoCSV.Windows.Computer.Extended.Class" Accessibility="Public" Abstract="false" Base="Windows!Microsoft.Windows.Computer" Hosted="false" Singleton="false" Extension="false">
          <Property ID="TIER" Type="string" AutoIncrement="false" Key="false" CaseSensitive="false" MaxLength="256" MinLength="0" Required="false" Scale="0" />
          <Property ID="GROUPID" Type="string" AutoIncrement="false" Key="false" CaseSensitive="false" MaxLength="256" MinLength="0" Required="false" Scale="0" />
          <Property ID="OWNER" Type="string" AutoIncrement="false" Key="false" CaseSensitive="false" MaxLength="256" MinLength="0" Required="false" Scale="0" />
        </ClassType>
      </ClassTypes>
    </EntityTypes>
  </TypeDefinitions>

#=================================================================================
# Extend Windows Computer class from CSV
#=================================================================================

Param($SourceId,$ManagedEntityId)

# For testing discovery manually in PowerShell:
# $SourceId = '{00000000-0000-0000-0000-000000000000}'
# $ManagedEntityId = '{00000000-0000-0000-0000-000000000000}'

#=================================================================================
# Constants section - modify stuff here:
$CSVPath = "\servershareserverlist.csv"

# Assign script name variable for use in event logging
$ScriptName = "DemoCSV.Windows.Computer.Extended.Class.Discovery.Script.ps1"
#=================================================================================


#=================================================================================
# function Is-ClassMember
# Purpose:  To ensure we only return discovery data for computers that
#           already exist in SCOM, otherwise it will be rejected    
# Arguments: 
#  -$InstanceDisplayName - The name of the object instance like 'servername.domain.com'
#==================================================================================
function Is-ClassMember
  {
  param($InstanceDisplayName)
  If ($InstanceDisplayName -in $ComputerNames)
    {
    $value = "True"
    }
  Else
    {
    $value = "False"
    }
  Return $value
  }
# End of function Is-ClassMember




# Gather script start time
$StartTime = Get-Date

$MServer = $env:COMPUTERNAME

# Gather who the script is running as
$WhoAmI = whoami

# Load MOMScript API
$momapi = New-Object -comObject MOM.ScriptAPI

# Load SCOM Discovery module
$DiscoveryData = $momapi.CreateDiscoveryData(0, $SourceId, $ManagedEntityId)

# Log an event for the script starting              
$momapi.LogScriptEvent($ScriptName,7777,0, "Script is starting. Running, as $WhoAmI.")

# Clear any previous errors
if($Error)
  {
  $Error.Clear()
  }

# Import the OperationsManager module and connect to the management group
Try
  {
  $SCOMPowerShellKey = "HKLM:SOFTWAREMicrosoftSystem Center Operations Manager12SetupPowershellV2"
  $SCOMModulePath = Join-Path (Get-ItemProperty $SCOMPowerShellKey).InstallDirectory "OperationsManager"
  Import-module $SCOMModulePath
  }
Catch
  {
  $momapi.LogScriptEvent($ScriptName,7778,2, "Unable to load the OperationsManager module, Error is: $error")
  }
Try
  {
  New-DefaultManagementGroupConnection $MServer
  }
Catch
  {
  $momapi.LogScriptEvent($ScriptName,7778,2, "Unable to connect to the management server: $MServer.  Error when calling New-DefaultManagementGroupConnection.  Error is: $error")
  }


# Get all instances of a existing Windows Computer class
# We need this to check and make sure each computer in the CSV exists in SCOM or discvoery data will be rejected
  $WindowsComputers = Get-SCOMClass -DisplayName "Windows Computer" | Get-SCOMClassInstance
  $ComputerNames = $WindowsComputers.DisplayName
  $ComputerCount = $ComputerNames.count
# Log an event for command ending
  $momapi.LogScriptEvent($ScriptName,7777,0, "Get all Windows Computers has completed.  Returned $ComputerCount Windows Computers.")

# Clear any previous errors
if($Error)
  {
  $Error.Clear()
  }

#Test the CSV path and make sure we can read it:
  If (Test-Path $CSVPath)
    {
    # Log an event for CSV path good
    $momapi.LogScriptEvent($ScriptName,7777,0, "CSV file was found at $CSVPath")
    }
  Else
    {
    # Log an event for CSV path bad
    $momapi.LogScriptEvent($ScriptName,7778,2, "CSV file was NOT found at $CSVPath  This is a fatal script error, ending script.  Error is $Error")
    exit
    }

# Query the CSV file to get the servers and properties
  $CSVContents = Import-Csv $CSVPath

# Loop through the CSV and add discovery data for existing SCOM computers
  $i=0;
  foreach ($row in $CSVContents)
    {
    # Get the FQDN and assign it to a variable
    $FQDN = $row.FQDN

    #Check and see if the $FQDN value contains a computer that exists as a Windows Computer in SCOM
    $IsSCOMComputer = Is-ClassMember $FQDN
    If ($IsSCOMComputer -eq "True")
        {
        $i=$i+1
        # Get each property in your CSV and assign it to a variable
        $TIER = $row.TIER
        $GROUPID = $row.GROUPID
        $OWNER = $row.OWNER

        # Create discovery data for each computer that exists in both the CSV and SCOM
        $Inst = $DiscoveryData.CreateClassInstance("$MPElement[Name='DemoCSV.Windows.Computer.Extended.Class']$")
        $Inst.AddProperty("$MPElement[Name='Windows!Microsoft.Windows.Computer']/PrincipalName$", $FQDN)
        $Inst.AddProperty("$MPElement[Name='DemoCSV.Windows.Computer.Extended.Class']/TIER$", $TIER)
        $Inst.AddProperty("$MPElement[Name='DemoCSV.Windows.Computer.Extended.Class']/GROUPID$", $GROUPID)
        $Inst.AddProperty("$MPElement[Name='DemoCSV.Windows.Computer.Extended.Class']/OWNER$", $OWNER)
        $DiscoveryData.AddInstance($Inst)
        } #End If
    } #End foreach


# Return Discovery Items
  $DiscoveryData

# Return Discovery Bag to the command line for testing (does not work from ISE):
# $momapi.Return($DiscoveryData)

$CSVMatchComputerCount = $i
$CSVRowCount = $CSVContents.Count

# End script and record total runtime
$EndTime = Get-Date
$ScriptTime = ($EndTime - $StartTime).TotalSeconds
# Log an event for script ending and total execution time.
$momapi.LogScriptEvent($ScriptName,7777,0, "Script has completed.  CSV returned $CSVRowCount computers.  SCOM returned $ComputerCount Computers.  Discovery returned $CSVMatchComputerCount matching computers from the CSV and SCOM.  Runtime was $ScriptTime seconds")