Public/Tags.ps1
|
function Get-TMTag { <# .SYNOPSIS Gets a Tag from TransitionManager .DESCRIPTION This function will retrieve one or more Tags from TransitionManager by name or Id .PARAMETER TMSession The name of the TM Session to use when retrieving a Tag .PARAMETER Server The URI of the TransitionManager instance .PARAMETER ResetIDs Switch indicating that the Tag(s) should be returned without Id's .PARAMETER Name The name of the Tag to be retrieved .PARAMETER Id The Id of the Tag to be retrieved .EXAMPLE Get-TMTag -TMSession 'TMDDEV' -Name 'TestTag' .EXAMPLE $AllTags = Get-TMTag -TMSession 'TMDDEV' .EXAMPLE Get-TMTag -TMSession 'TMDDEV' -Id @(123, 546, 23) .OUTPUTS TMTag object representing the Tag in TransitionManager #> [CmdletBinding(DefaultParameterSetName = 'ByName')] param ( [Parameter(Mandatory = $false)] [PSObject]$TMSession = "Default", [Parameter(Mandatory = $false)] [Switch]$ResetIDs, [Parameter(Mandatory = $false, Position = 0, ValueFromPipeline = $true, ParameterSetName = 'ByName')] [String[]]$Name, [Parameter(Mandatory = $false, ValueFromPipelineByPropertyName = $true, ParameterSetName = 'ById')] [Int[]]$Id ) begin { Write-Verbose "ParameterSet = $($PSCmdlet.ParameterSetName)" ## Get Session Configuration $TMSession = Get-TMSession $TMSession if (-not $TMSession) { Write-Host 'TMSession: [' -NoNewline Write-Host $TMSession -ForegroundColor Cyan Write-Host '] was not Found. Please use the New-TMSession command.' Throw "TM Session Not Found. Use New-TMSession command before using features." } # Format the request parameters $Instance = $TMSession.TMServer.Replace('/tdstm', '').Replace('https://', '').Replace('http://', '') $Uri = "https://$Instance/tdstm/ws/tag" Set-TMHeaderAccept -TMSession $TMSession -Accept 'JSON' $WebRequestSplat = @{ Method = 'GET' Uri = $Uri WebSession = $TMSession.TMWebSession SkipCertificateCheck = $TMSession.AllowInsecureSSL } # Make the request try { Write-Verbose "Web Request Parameters:" Write-Verbose ($WebRequestSplat | ConvertTo-Json -Depth 10) Write-Verbose "Invoking web request" $Response = Invoke-WebRequest @WebRequestSplat Write-Verbose "Response status code: $($Response.StatusCode)" Write-Verbose "Response Content: $($Response.Content)" } catch { throw $_ } if ($Response.StatusCode -eq 200) { $Result = ($Response.Content | ConvertFrom-Json -Depth 10).data } else { throw "Couldn't get tag(s)" } } process { ## Sort the tags $Result = $Result | Sort-Object -Property 'name' # Filter the results by the Name(s) passed in if ($Name) { $Result = $Result | Where-Object name -in $Name } # Filter the results by the ID(s) passed in if ($Id) { $Result = $Result | Where-Object id -in $Id } # Remove thew IDs if the ResetIDs switch was passed if ($ResetIDs) { for ($i = 0; $i -lt $Result.Count; $i++) { $Result[$i].id = $null } } $Result | ForEach-Object { [TMTag]::new($_.id, $_.name, $_.description, $_.color) } } } function New-TMTag { <# .SYNOPSIS Creates a new Tag in TransitionManager .DESCRIPTION This function will create a new Tag in TransitionManager .PARAMETER TMSession The name of the TM Session to use when creating a Tag .PARAMETER Name The name of the Tag to be created .PARAMETER Description The new Tag's description .PARAMETER Color The color of the Tag to be created .PARAMETER InputObject A TMTag object representing the Tag to be created .PARAMETER Passthru Switch indicating that the new Tag should be returned after creation .EXAMPLE $NewTagSplat = @{ Name = 'VM' Description = "This asset is a virtual machine" Color = 'Blue' TMSession = 'TMDDEV' Passthru = $true } New-TMTag @NewTagSplat .EXAMPLE Get-TMTag -TMSession TMDDEV2 | New-TMTag -TMSession TMDDEV -Passthru .OUTPUTS If Passthru switch is used, a TMTag object representing the created Tag. Otherwise, none #> [CmdletBinding(DefaultParameterSetName = 'ByProperty')] param ( [Parameter(Mandatory = $false)] [PSObject]$TMSession = "Default", [Parameter(Mandatory = $false, Position = 0, ValueFromPipelineByPropertyName = $true, ParameterSetName = 'ByProperty')] [String]$Name, [Parameter(Mandatory = $false, ValueFromPipelineByPropertyName = $true, ParameterSetName = 'ByProperty')] [String]$Description = "", [Parameter(Mandatory = $false, ValueFromPipelineByPropertyName = $true, ParameterSetName = 'ByProperty')] [ArgumentCompleter({ [TMTag]::ValidColors })] [ValidateScript( { $_ -in [TMTag]::ValidColors } )] [String]$Color = "Grey", [Parameter(Mandatory = $true, ParameterSetName = 'ByObject')] [TMTag]$InputObject, [Parameter(Mandatory = $false)] [Switch]$Passthru ) begin { Write-Verbose "ParameterSet = $($PSCmdlet.ParameterSetName)" ## Get Session Configuration $TMSession = Get-TMSession $TMSession if (-not $TMSession) { Write-Host 'TMSession: [' -NoNewline Write-Host $TMSession -ForegroundColor Cyan Write-Host '] was not Found. Please use the New-TMSession command.' Throw "TM Session Not Found. Use New-TMSession command before using features." } # Format the request parameters $Instance = $TMSession.TMServer.Replace('/tdstm', '').Replace('https://', '').Replace('http://', '') $Uri = "https://$Instance/tdstm/ws/tag" Set-TMHeaderAccept -TMSession $TMSession -Accept 'Any' Set-TMHeaderContentType -TMSession $TMSession -ContentType 'JSON' $WebRequestSplat = @{ Method = 'POST' Uri = $Uri WebSession = $TMSession.TMWebSession SkipCertificateCheck = $TMSession.AllowInsecureSSL Body = "" } } process { $TagsToAdd = [System.Collections.ArrayList]::new() # Format the object to be sent to TM switch ($PSCmdlet.ParameterSetName) { 'ByObject' { foreach ($Object in $InputObject) { if ([String]::IsNullOrWhiteSpace($Object.Name)) { Write-Error "InputObject does not have a 'Name' property" return } [void]$TagsToAdd.Add([TMTag]::new($Object.Name, $Object.Description, $Object.Color)) } } 'ByProperty' { [void]$TagsToAdd.Add([TMTag]::new($Name, $Description, $Color)) } } # Process each of the tags to add foreach ($Tag in $TagsToAdd) { # Make sure the tag doesn't already exist $TagCheck = Get-TMTag -TMSession $TMSession -Name $Tag.Name if ($TagCheck) { if ($Passthru) { $TagCheck } return } # Make sure the color is valid if ($Tag.Color -notin [TMTag]::ValidColors) { Write-Verbose "Tag '$($Tag.Name)' did not have a valid color, so it has been set to Grey" $Tag.Color = 'Grey' } # Format the body of the request $WebRequestSplat.Body = ($Tag | ConvertTo-Json -Compress) # Make the request try { Write-Verbose "Web Request Parameters:" Write-Verbose ($WebRequestSplat | ConvertTo-Json -Depth 10) Write-Verbose "Invoking web request" $Response = Invoke-WebRequest @WebRequestSplat Write-Verbose "Response status code: $($Response.StatusCode)" Write-Verbose "Response Content: $($Response.Content)" } catch { throw $_ } if ($Response.StatusCode -eq 200) { if ($Passthru) { try { New-Object TMTag -Property ($Response.Content | ConvertFrom-Json).data } catch { Get-TMTag -TMSession $TMSession -Name $Name } } } elseif ($Response.StatusCode -eq 204) { if ($Passthru) { Get-TMTag -TMSession $TMSession -Name $Name } } else { Write-Error "Tag '$Name' could not be created" } } } } function Remove-TMTag { <# .SYNOPSIS Removes a Tag from TransitionManager .DESCRIPTION This function will remove a Tag in TransitionManager by Name or Id .PARAMETER TMSession The TransitionManager session to be used when deleting a Tag .PARAMETER Name The name of the Tag to be removed .PARAMETER Id The Id of the Tag to be removed .EXAMPLE Get-TMTag -TMSession TMDDEV | Remove-TMTag -TMSession TMDDEV .EXAMPLE Remove-TMTag -Name 'Tag1', 'Tag2' .EXAMPLE Remove-TMTag -Id 489 .OUTPUTS None #> [CmdletBinding(DefaultParameterSetName = 'ByName')] param ( [Parameter(Mandatory = $false)] [PSObject]$TMSession = "Default", [Parameter(Mandatory = $false, Position = 0, ParameterSetName = 'ByName')] [String[]]$Name, [Parameter(Mandatory = $false, ValueFromPipelineByPropertyName = $true, ParameterSetName = 'ById')] [Int[]]$Id ) begin { Write-Verbose "ParameterSet = $($PSCmdlet.ParameterSetName)" ## Get Session Configuration $TMSession = Get-TMSession $TMSession if (-not $TMSession) { Write-Host 'TMSession: [' -NoNewline Write-Host $TMSession -ForegroundColor Cyan Write-Host '] was not Found. Please use the New-TMSession command.' Throw "TM Session Not Found. Use New-TMSession command before using features." } # Format the request parameters $Instance = $TMSession.TMServer.Replace('/tdstm', '').Replace('https://', '').Replace('http://', '') Set-TMHeaderAccept -TMSession $TMSession -Accept 'Any' } process { $IDsToRemove = [System.Collections.ArrayList]::new() switch ($PSCmdlet.ParameterSetName) { 'ByName' { (Get-TMTag -TMSession $TMSession -Name $Name).id | ForEach-Object { [void]$IDsToRemove.Add($_) } } 'ById' { $Id | ForEach-Object { [void]$IDsToRemove.Add($_) } } } foreach ($TagId in $IDsToRemove) { $WebRequestSplat = @{ Method = 'DELETE' Uri = "https://$Instance/tdstm/ws/tag/$TagId" WebSession = $TMSession.TMWebSession SkipCertificateCheck = $TMSession.AllowInsecureSSL } # Make the request try { Write-Verbose "Web Request Parameters:" Write-Verbose ($WebRequestSplat | ConvertTo-Json -Depth 10) Write-Verbose "Invoking web request" $Response = Invoke-WebRequest @WebRequestSplat Write-Verbose "Response status code: $($Response.StatusCode)" Write-Verbose "Response Content: $($Response.Content)" } catch { throw $_ } if ($Response.StatusCode -notin 200, 204) { Write-Error "Tag with Id $TagId could not be removed" } } } } |