Public/Protect-RubrikTag.ps1

#Requires -Version 3

function Protect-RubrikTag
{
  <#
      .SYNOPSIS
      Connects to Rubrik and assigns an SLA Domain based on a vSphere category and tag value
 
      .DESCRIPTION
      The Protect-RubrikTag cmdlet will update a virtual machine's SLA Domain assignment within the Rubrik cluster.
      The SLA Domain contains all policy-driven values needed to protect workloads.
      Make sure you have PowerCLI installed and connect to the required vCenter Server.
 
      .NOTES
      Written by Jason Burrell for community usage
      Twitter: @jasonburrell2
 
      .LINK
      https://github.com/rubrikinc/PowerShell-Module
 
      .EXAMPLE
      Protect-RubrikTag -Tag 'Gold' -Category 'Rubrik'
      This will assign the Gold SLA Domain to any VM tagged with Gold in the Rubrik category
 
      .EXAMPLE
      Protect-RubrikTag -Tag 'Gold' -Category 'Rubrik' -SLA 'Titanium'
      This will assign the Titanium SLA Domain to any VM tagged with Gold in the Rubrik category
 
      .EXAMPLE
      Protect-RubrikTag -Tag 'Gold' -Category 'Rubrik' -DoNotProtect
      This will remove protection from any VM tagged with Gold in the Rubrik category
 
      .EXAMPLE
      Protect-RubrikTag -Tag 'Gold' -Category 'Rubrik' -Inherit
      This will flag any VM tagged with Gold in the Rubrik category to inherit the SLA Domain of its parent object
  #>


  [CmdletBinding(SupportsShouldProcess = $true,ConfirmImpact = 'High')]
  Param(
    # vSphere Tag
    [Parameter(Mandatory = $true)]
    [String]$Tag,
    # vSphere Tag Category
    [Parameter(Mandatory = $true)]
    [String]$Category,
    # The SLA Domain in Rubrik
    [Parameter(ParameterSetName = 'SLA_Explicit')]
    [String]$SLA,
    # Removes the SLA Domain assignment
    [Parameter(ParameterSetName = 'SLA_Unprotected')]
    [Switch]$DoNotProtect,
    # Inherits the SLA Domain assignment from a parent object
    [Parameter(ParameterSetName = 'SLA_Inherit')]
    [Switch]$Inherit,
    # Rubrik server IP or FQDN
    [String]$Server = $global:RubrikConnection.server,
    # API version
    [String]$api = $global:RubrikConnection.api
  )

  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
    
    # Check to ensure that a session to the Rubrik cluster exists and load the needed header data for authentication
    Test-RubrikConnection
    
    # Check to ensure that a session to the vSphere Center Server exists and load the needed header data for authentication
    Test-VMwareConnection

    # API data references the name of the function
    # For convenience, that name is saved here to $function
    $function = $MyInvocation.MyCommand.Name
        
    # Retrieve all of the URI, method, body, query, result, filter, and success details for the API endpoint
    Write-Verbose -Message "Gather API Data for $function"
    $resources = (Get-RubrikAPIData -endpoint $function).$api
    Write-Verbose -Message "Load API data for $($resources.Function)"
    Write-Verbose -Message "Description: $($resources.Description)"

  }

  Process {

    #region One-off
    Write-Verbose -Message 'Gathering the SLA Domain id'
    if (!$SLA -and !$Inherit -and !$DoNotProtect) 
    {
      $SLA = $Tag
    }
    $SLAID = Test-RubrikSLA -SLA $SLA -Inherit $Inherit -DoNotProtect $DoNotProtect    
    
    Write-Verbose -Message "Gathering a list of VMs associated with Category $Category and Tag $Tag"
    try 
    {
      $vmlist = Get-VM -Tag (Get-Tag -Name $Tag -Category $Category) | Get-View
      # This will pull out the vCenter UUID assigned to the parent vCenter Server by Rubrik
      # Reset switches to prevent Get-RubrikVM from picking them up (must be a better way?)
      $DoNotProtect = $false    
      $Inherit = $false
      $vcuuid = ((Get-RubrikVM -VM ($vmlist[0].Name)).vCenterId) -replace 'vCenter:::', ''
    }
    catch
    {
      throw $_
    }

    Write-Verbose -Message 'Building an array of Rubrik Managed IDs'
    [array]$vmbulk = @()
    foreach ($_ in $vmlist)
    {
      $vmbulk += 'VirtualMachine:::' + $vcuuid + '-' + $($_.moref.value)
      Write-Verbose -Message "Found $($vmbulk.count) records"
    }
    #endregion
       
    $uri = New-URIString -server $Server -endpoint ($resources.URI) -id $SLAID
    $uri = Test-QueryParam -querykeys ($resources.Query.Keys) -parameters ((Get-Command $function).Parameters.Values) -uri $uri

    #region One-off
    Write-Verbose -Message 'Creating the array to mass assign the list of IDs'
    $body = @{
      managedIds = $vmbulk
    }
    $body = ConvertTo-Json -InputObject $body
    #endregion

    $result = Submit-Request -uri $uri -header $Header -method $($resources.Method) -body $body
    $result = Test-ReturnFormat -api $api -result $result -location $resources.Result
    $result = Test-FilterObject -filter ($resources.Filter) -result $result

    return $result

  } # End of process
} # End of function