# Assign script name variable for use in event logging
$ScriptName = "SCOM.Management.OMSWorkspaceAdd.WA.ps1"
#=================================================================================
# Gather who the script is running as
$whoami = whoami
#Load the MOMScript API and discovery propertybag
$momapi = New-Object -comObject "Mom.ScriptAPI"
#Log script event that we are starting task
$momapi.LogScriptEvent($ScriptName,1347,0, "`n Starting script. `n Running as ($whoami)")
# Begin Main Script
#=================================================================================
Write-Host "Task Starting. Running as $whoami"
#Load agent scripting object
Write-Host "Loading agent scripting objects."
$AgentCfg = New-Object -ComObject AgentConfigManager.MgmtSvcCfg
#Check to see if this agent supports OMS
$AgentSupportsOMS = $AgentCfg | Get-Member -Name 'GetCloudWorkspaces'
IF (!$AgentSupportsOMS)
{
Write-Host "FATAL ERROR: This agent version does not support Cloud Workspaces. You must upgrade this agent. Terminating."
EXIT
}
#Add OMS Workspace
Write-Host "Adding OMS Workspace: ($WorkspaceID)"
$Error.Clear()
$AgentCfg.AddCloudWorkspace($WorkspaceID,$WorkspaceKey)
IF ($Error)
{
Write-Host "There was a critical error adding OMS Workspace. Error is: $Error"
EXIT
}
Write-Host "OMS Workspace Added."
#Add Proxy URL
IF ($ProxyURL)
{
Write-Host "A proxy URL was specified: ($ProxyURL). Adding Proxy configuration to agent."
$AgentCfg.SetProxyUrl($ProxyURL)
Write-Host "Proxy Added."
}
#Restart Agent
Write-Host "Restarting Agent now."
#=================================================================================
# End Main Script
# Restart Agent
#=================================================================================
#We need a reliable way to restart the SCOM Agent out of band so that tasks can complete with success
$Command = "Start-Sleep -s 5;Restart-Service HealthService"
$Process = ([wmiclass]"root\cimv2:Win32_ProcessStartup").CreateInstance()
$Process.ShowWindow = 0
$Process.CreateFlags = 16777216
([wmiclass]"root\cimv2:Win32_Process").Create("powershell.exe $Command")|Out-Null
#================================================================================= </Script></ScriptBody>
<Parameters>
<Parameter>
<Name>WorkspaceID</Name>
<Value>$Config/WorkspaceID$</Value>
</Parameter>
<Parameter>
<Name>WorkspaceKey</Name>
<Value>$Config/WorkspaceKey$</Value>
</Parameter>
<Parameter>
<Name>ProxyURL</Name>
<Value>$Config/ProxyURL$</Value>
</Parameter>
</Parameters>
<TimeoutSeconds>$Config/TimeoutSeconds$</TimeoutSeconds>
</WriteAction>
</MemberModules>
<Composition>
<Node ID="PSWA"/>
</Composition>
</Composite>
</ModuleImplementation>
<OutputType>System!System.BaseData</OutputType>
<InputType>System!System.BaseData</InputType>
</WriteActionModuleType>