Private/Invoke-VagrantCloudLocalToken.ps1
|
<#
.SYNOPSIS Grabs an API token from the local machine used for Vagrant Cloud. .DESCRIPTION This function will check the environment variables for the location of the Vagrant Cloud Token. Vagrant Cloud looks for the variable "VARGRANT_CLOUD_TOKEN". This token should be present. If the token is not present, this function will attempt to create a token. .NOTES This function only works on Windows at this time. .EXAMPLE Query the token and return a hash table, with the name and token values. Invoke-VagrantCloudLocalToken .EXAMPLE Query the token and copy the token value to clipboard. (Invoke-VagrantCloudLocalToken).token | clip .INPUTS none .OUTPUTS System.Management.Automation.PSCustomObject .NOTES Author: Steven Jennings <steven@automatingops.com> Github: https://github.com/deyjcode/PSVagrantCloud #> Function Invoke-VagrantCloudLocalToken { [CmdletBinding()] param ( ) $TokenName = "VAGRANT_CLOUD_TOKEN" try { $GrabToken = Get-Item "Env:\$TokenName" -ErrorAction Stop Write-Verbose "API Token already exists." $TokenObject = [PSCustomObject]@{ Name = $GrabToken.Name Token = $GrabToken.Value } } catch { Write-Warning "The environment variable $TokenName must be present!" $AskTokenQuestion = Read-Host -Prompt "Would you like to create the token now? [Y/n]" switch ($AskTokenQuestion) { 'Y' { $TokenDescription = New-VagrantCloudAPIToken -VagrantTokenDescription (Read-Host -Prompt "Enter a description for the token") Write-Verbose "Writing Token Values..." try { # https://docs.microsoft.com/en-us/dotnet/api/system.environment.setenvironmentvariable?view=netframework-4.7.2 [Environment]::SetEnvironmentVariable("$TokenName", "$($TokenDescription.token)", "User") } catch { Write-Error "An unknown error occurred while setting the environment variable." } Write-Verbose "The environment variable $TokenName has been created." Write-Warning "It is recommended to reboot the system to ensure system variables are set!" $AskRestartQuestion = Read-Host -Prompt "Would you like to reboot? [Y/n]" switch ($AskRestartQuestion) { 'Y' { Restart-Computer -Confirm } 'N' { return } } } 'N' { Write-Warning "Unable to continue without a token. Exiting to prompt..." return } } } return $TokenObject } Invoke-VagrantCloudLocalToken |