public/Update-VPASPSMSettingsByPlatformID.ps1

<#
.Synopsis
   UPDATE PSM SETTINGS BY PLATFORMID
   CREATED BY: Vadim Melamed, EMAIL: vmelamed5@gmail.com
.DESCRIPTION
   USE THIS FUNCTION TO UPDATE PSM SETTINGS LIKE CONNECTION COMPONENTS AND PSMSERVERID FOR A SPECIFIC PLATFORM
.PARAMETER NoSSL
   If the environment is not set up for SSL, API calls will be made via HTTP not HTTPS (Not Recommended!)
.PARAMETER token
   HashTable of data containing various pieces of login information (PVWA, LoginToken, HeaderType, etc).
   If -token is not passed, function will use last known hashtable generated by New-VPASToken
.PARAMETER PlatformID
   Unique PlatformID that will be updated
.PARAMETER ConnectionComponentID
   Unique ConnectionComponentID that will be added or removed
.PARAMETER Action
   Which action will be taken on the updated fields
   Possible values: ADD, REMOVE
.PARAMETER PSMServerID
   Unique target PSMServerID that will be added or removed
.EXAMPLE
   $UpdatePSMSettingsStatus = Update-VPASPSMSettingsByPlatformID -PlatformID {PLATFORMID VALUE} -ConnectionComponentID {CONNECTION COMPONENT ID VALUE}
.EXAMPLE
   $UpdatePSMSettingsStatus = Update-VPASPSMSettingsByPlatformID -PlatformID {PLATFORMID VALUE} -PSMServerID {PSM SERVER ID VALUE}
.OUTPUTS
   $true if successful
   $false if failed
#>

function Update-VPASPSMSettingsByPlatformID{
    [OutputType([bool])]
    [CmdletBinding()]
    Param(

        [Parameter(Mandatory=$true,ValueFromPipelineByPropertyName=$true,HelpMessage="Enter target platformID to update (for example: WinServerLocal)",Position=0)]
        [String]$PlatformID,

        [Parameter(Mandatory=$false,ValueFromPipelineByPropertyName=$true,Position=1)]
        [String]$ConnectionComponentID,

        [Parameter(Mandatory=$false,ValueFromPipelineByPropertyName=$true,Position=2)]
        [ValidateSet('ADD','REMOVE')]
        [String]$Action,

        [Parameter(Mandatory=$false,ValueFromPipelineByPropertyName=$true,Position=3)]
        [String]$PSMServerID,

        [Parameter(Mandatory=$false,ValueFromPipelineByPropertyName=$true,Position=4)]
        [hashtable]$token,

        [Parameter(Mandatory=$false,ValueFromPipelineByPropertyName=$true,Position=5)]
        [Switch]$NoSSL

    )

    Begin{
        $tokenval,$sessionval,$PVWA,$Header,$ISPSS,$IdentityURL = Get-VPASSession -token $token
    }
    Process{

        Write-Verbose "SUCCESSFULLY PARSED PVWA VALUE"
        Write-Verbose "SUCCESSFULLY PARSED TOKEN VALUE"
        Write-Verbose "SUCCESSFULLY PARSED PLATFORMID VALUE: $PlatformID"
        Write-Verbose "SUCCESSFULLY PARSED SSL VALUE"

        try{

            Write-Verbose "INVOKING PLATFORMID HELPER FUNCTION"
            if($NoSSL){
                Write-Verbose "NO SSL ENABLED, USING HTTP INSTEAD OF HTTPS"
                $platID = Get-VPASPlatformIDHelper -token $token -platformID $PlatformID -NoSSL
            }
            else{
                Write-Verbose "SSL ENABLED BY DEFAULT, USING HTTPS"
                $platID = Get-VPASPlatformIDHelper -token $token -platformID $PlatformID
            }

            if($platID -eq -1){
                Write-Verbose "COULD NOT FIND TARGET PLATFORMID: $PlatformID"
                Write-VPASOutput -str "COULD NOT FIND TARGET PLATFORMID: $PlatformID" -type E
                return $false
            }
            else{
                Write-Verbose "GETTING CURRENT PLATFORM PSM SETTINGS"
                $curvals = Get-VPASPSMSettingsByPlatformID -token $token -PlatformID $PlatformID
                $curPSMServerID = $curvals.PSMServerId
                $curPSMConnectors = @()

                $res = $curvals.PSMConnectors
                foreach($rec in $res){
                    $minirec = @{
                        PSMConnectorID = $rec.PSMConnectorID
                        Enabled = $rec.Enabled
                    }
                    $curPSMConnectors += $minirec
                }

                if(![String]::IsNullOrEmpty($ConnectionComponentID) -and [String]::IsNullOrEmpty($Action)){
                    Write-VPASOutput -str "$ConnectionComponentID PASSED WITH NO ACTION, ADD OR REMOVE $ConnectionComponentID : " -type Y
                    $Action = Read-Host
                }

                $params = @{}
                Write-Verbose "INITIALIZING API PARAMS"

                if([String]::IsNullOrEmpty($ConnectionComponentID)){
                    $params += @{
                        PSMConnectors = $curPSMConnectors
                    }
                }
                else{
                    Write-Verbose "HANDLING $ConnectionComponentID AND ACTION INTO API PARAMS"
                    $arrNew = @()

                    $foundConnector = $false
                    foreach($providedRec in $curPSMConnectors){
                        $Enabled = $providedRec.Enabled
                        $PSMConnector = $providedRec.PSMConnectorID

                        if($PSMConnector -eq $ConnectionComponentID){
                            $foundConnector = $true

                            if($Action -eq "ADD"){
                                Write-VPASOutput -str "CONNECTION COMPONENT ALREADY EXISTS ON $PlatformID" -type M
                                Write-Verbose "CONNECTION COMPONENT ALREADY EXISTS IN $PlatformID"
                                Write-Verbose "RETURNING FALSE"
                                return $false
                            }
                            elseif($Action -eq "REMOVE"){
                                Write-Verbose "$ConnectionComponentID WILL BE IGNORED FROM API PARAMETERS"
                            }
                        }
                        else{
                            $arrADD = @{
                                PSMConnectorID = $PSMConnector
                                Enabled = $Enabled
                            }
                            $arrNew += $arrADD
                        }
                    }
                    if(!$foundConnector -and $Action -eq "ADD"){
                        $arrADD = @{
                            PSMConnectorID = $ConnectionComponentID
                            Enabled = $true
                        }
                        $arrNew += $arrADD
                    }
                    $params += @{
                        PSMConnectors = $arrNew
                    }
                }

                if([String]::IsNullOrEmpty($PSMServerID)){
                    $params += @{
                        PSMServerId = $curPSMServerID
                    }
                }
                else{
                    Write-Verbose "ADDING $PSMServerID TO API PARAMS"
                    $params += @{
                        PSMServerId = $PSMServerID
                    }
                }

                $params = $params | ConvertTo-Json

                Write-Verbose "MAKING API CALL TO CYBERARK"
                if($NoSSL){
                    Write-Verbose "NO SSL ENABLED, USING HTTP INSTEAD OF HTTPS"
                    $uri = "http://$PVWA/passwordvault/api/Platforms/Targets/$platID/PrivilegedSessionManagement/"
                }
                else{
                    Write-Verbose "SSL ENABLED BY DEFAULT, USING HTTPS"
                    $uri = "https://$PVWA/passwordvault/api/Platforms/Targets/$platID/PrivilegedSessionManagement/"
                }

                if($sessionval){
                    $response = Invoke-RestMethod -Headers @{"Authorization"=$Header} -Uri $uri -Method PUT -Body $params -ContentType "application/json" -WebSession $sessionval
                }
                else{
                    $response = Invoke-RestMethod -Headers @{"Authorization"=$Header} -Uri $uri -Method PUT -Body $params -ContentType "application/json"
                }
                Write-Verbose "SUCCESSFULLY UPDATED PSM SETTINGS FOR PLATFORM: $PlatformID"
                Write-Verbose "RETURNING TRUE"
                return $true

            }
        }catch{
            Write-Verbose "UNABLE TO UPDATE PSM SETTINGS FOR PLATFORM: $PlatformID"
            Write-VPASOutput -str $_ -type E
            return $false
        }
    }
    End{

    }
}