Public/Set-ArmorAccountContext.ps1

Function Set-ArmorAccountContext
{
    <#
        .SYNOPSIS
        Sets the Armor account context, so that all future requests reference the specified account.

        .DESCRIPTION
        { required: more detailed description of the function's purpose }

        .NOTES
        Troy Lindsay
        Twitter: @troylindsay42
        GitHub: tlindsay42

        .PARAMETER ID
        The Armor account ID to use for all subsequent requests.

        .INPUTS
        System.UInt16

        .OUTPUTS
        None

        .LINK
        https://github.com/tlindsay42/ArmorPowerShell

        .LINK
        https://docs.armor.com/display/KBSS/Armor+API+Guide

        .LINK
        https://developer.armor.com/

        .EXAMPLE
        {required: show one or more examples using the function}
    #>


    [CmdletBinding()]
    Param 
    (
        [Parameter( Position = 0, ValueFromPipeline = $true )]
        [ValidateScript( { $_ -in ( Get-ArmorAccount ).ID } )]
        [UInt16] $ID = $null
    )

    Begin
    {
        # The Begin section is used to perform one-time loads of data necessary to carry out the function's purpose
        # If a command needs to be run with each iteration or pipeline input, place it in the Process section

        # API data references the name of the function
        # For convenience, that name is saved here to $function
        $function = $MyInvocation.MyCommand.Name

        Write-Verbose  -Message ( 'Beginning {0}.' -f $function )

        # Check to ensure that a session to the Armor cluster exists and load the needed header data for authentication
        Test-ArmorConnection
    } # End of Begin

    Process
    {
        $Global:ArmorConnection.AccountContextID = $ID

        $Global:ArmorConnection.Headers.'X-Account-Context' = $ID
    } # End of Process

    End
    {
        Write-Verbose -Message ( 'Ending {0}.' -f $function )
    } # End of End
} # End of Function