Public/TokensAPI.ps1
|
Function Add-DatabricksApiToken { <# .SYNOPSIS Create and return a token. This call returns the error QUOTA_EXCEEDED if the caller exceeds the token quota, which is 600. .DESCRIPTION Create and return a token. This call returns the error QUOTA_EXCEEDED if the caller exceeds the token quota, which is 600. Official API Documentation: https://docs.databricks.com/api/latest/tokens.html#create .PARAMETER LifetimeSeconds The lifetime of the token, in seconds. If no lifetime is specified, the token remains valid indefinitely. .PARAMETER Comment Optional description to attach to the token. .EXAMPLE Add-DatabricksApiToken -LifetimeSeconds 360 -Comment "MyComment #> [CmdletBinding()] param ( [Parameter(Mandatory = $false, Position = 1)] [long] $LifetimeSeconds = -1, [Parameter(Mandatory = $false, Position = 2)] [string] $Comment ) begin { $requestMethod = "POST" $apiEndpoint = "/2.0/token/create" } process { Write-Verbose "Building Body/Parameters for final API call ..." #Set parameters $parameters = @{ } $parameters | Add-Property -Name "lifetime_seconds" -Value $LifetimeSeconds -NullValue -1 $parameters | Add-Property -Name "comment" -Value $Comment $result = Invoke-DatabricksApiRequest -Method $requestMethod -EndPoint $apiEndpoint -Body $parameters return $result } } Function Get-DatabricksApiToken { <# .SYNOPSIS List all the valid tokens for a user-workspace pair. .DESCRIPTION List all the valid tokens for a user-workspace pair. Official API Documentation: https://docs.databricks.com/api/latest/tokens.html#list .EXAMPLE Get-DatabricksApiToken #> [CmdletBinding()] param () begin { $requestMethod = "GET" $apiEndpoint = "/2.0/token/list" } process { Write-Verbose "Building Body/Parameters for final API call ..." #Set parameters $parameters = @{ } $result = Invoke-DatabricksApiRequest -Method $requestMethod -EndPoint $apiEndpoint -Body $parameters return $result.token_infos } } Function Remove-DatabricksApiToken { <# .SYNOPSIS Revoke an access token. This call returns the error RESOURCE_DOES_NOT_EXIST if a token with the specified ID is not valid. .DESCRIPTION Revoke an access token. This call returns the error RESOURCE_DOES_NOT_EXIST if a token with the specified ID is not valid. Official API Documentation: https://docs.databricks.com/api/latest/tokens.html#revoke .PARAMETER TokenID The ID of the token to be revoked. .EXAMPLE Remove-DatabricksApiToken -TokenID 1234 #> [CmdletBinding()] param ( #[Parameter(Mandatory = $true, Position = 1, ValueFromPipeline = $true, ValueFromPipelineByPropertyName = $true)] [Alias("token_id")] [string] $TokenID ) DynamicParam { #Create the RuntimeDefinedParameterDictionary $Dictionary = New-Object System.Management.Automation.RuntimeDefinedParameterDictionary $tokenIDValues = (Get-DynamicParamValues { Get-DatabricksApiToken }).token_id New-DynamicParam -Name TokenID -ValidateSet $tokenIDValues -Alias 'token_id' -Mandatory -ValueFromPipelineByPropertyName -DPDictionary $Dictionary #return RuntimeDefinedParameterDictionary return $Dictionary } begin { $requestMethod = "POST" $apiEndpoint = "/2.0/token/delete" } process { $TokenID = $PSBoundParameters.TokenID Write-Verbose "Building Body/Parameters for final API call ..." #Set parameters $parameters = @{ token_id = $TokenID } $result = Invoke-DatabricksApiRequest -Method $requestMethod -EndPoint $apiEndpoint -Body $parameters return $result } } |