Auth/Connect-ArmSubscription.ps1

Function Connect-ArmSubscription
{
    [CmdletBinding(DefaultParameterSetName='VisibleCredPrompt',SupportsShouldProcess=$true)]
    [OutputType([PsObject])]
    Param (
        [Parameter(Mandatory=$true,ParameterSetName='ConnectByCredObject')]
        [System.Management.Automation.PSCredential]$Credential,
        
        [Parameter(Mandatory=$False,ParameterSetName='VisibleCredPrompt')]
        [switch]$ForceShowUi,
        
        [Parameter(Mandatory=$True,ParameterSetName='ConnectByRefreshToken')]
        [String]$RefreshToken,

        [Parameter(Mandatory=$True,ParameterSetName='ConnectByRefreshToken')]
        [String]$LoginUrl,
        
        [String]$SubscriptionId,
        
        [String]$TenantId,

        [Switch]$BasicOutput
    )

    
    Write-Debug "Logging on using Parameter set $($PSCmdlet.ParameterSetName)"
    
    if ($PSCmdlet.ParameterSetName -eq "VisibleCredPrompt")
    {
        $Params = @{}
        if ($ForceShowUi -eq $true)
        {
            $Params.Add("PromptBehavior","Always")
        }
        Else
        {
            $Params.Add("PromptBehavior","Auto")
        }

        $Params.Add("LoginUrl",$Script:LoginUrl)
        $Params.Add("ResourceUrl",$Script:ResourceUrl)
        $Params.Add("ClientId",$Script:DefaultClientId)
        $Params.Add("RedirectUri",$Script:DefaultAuthRedirectUri)
        
        Try 
        {
            $authResult = Get-InternalAcquireToken @Params -ErrorAction Stop
        }
        Catch
        {
            #Error-handling here
        }
        
    }
    ElseIf($PSCmdlet.ParameterSetName -eq "ConnectByCredObject")
    {    
        $Params = @{}
        $Params.Add("LoginUrl",$Script:LoginUrl)
        $Params.Add("ResourceUrl",$Script:ResourceUrl)
        $Params.Add("ClientId",$Script:DefaultClientId)
        $Params.Add("Credential",$Credential)

        Try 
        {
            $authResult = Get-InternalAcquireToken @Params -ErrorAction Stop
        }
        Catch
        {
            #Error-handling here
        }
    }
    ElseIf($PScmdlet.ParameterSetName -eq "ConnectByRefreshToken")
    {
        $AuthResult = Get-InternalAcquireToken -RefreshToken $RefreshToken -ClientId $Script:DefaultClientId -LoginUrl $LoginUrl -resourceUrl $Script:ResourceUrl
        if ($AuthResult)
        {
            $script:AuthToken = $AuthResult.AccessToken
            $Script:RefreshToken = $AuthResult.RefreshToken
            $script:TokenExpirationUtc = $AuthResult.ExpiresOn

            #Remove the current subscription object from the array
            
            $CurrentSub = $Script:AllSubscriptions | where {$_.SubscriptionId -eq $script:CurrentSubscriptionId}
            $Script:AllSubscriptions = $Script:AllSubscriptions | where {$_.SubscriptionId -ne $script:CurrentSubscriptionId}
            

            $SubObj = "" | Select SubscriptionId,TenantId,AccessToken,RefreshToken, Expiry, SubscriptionObject, DisplayName, State, LoginUrl
            $subobj.SubscriptionId = $script:CurrentSubscriptionId
            $subobj.DisplayName = $CurrentSub.displayName
            $SubObj.State = $CurrentSub.state
            $subobj.TenantId = $CurrentSub.tenantId
            $subobj.AccessToken = $AuthResult.AccessToken
            $subobj.RefreshToken = $AuthResult.RefreshToken
            $subobj.Expiry = $AuthResult.ExpiresOn
            $subobj.SubscriptionObject = $CurrentSub.SubscriptionObject
            $subobj.LoginUrl = $CurrentSub.LoginUrl
            
            #Add it back to the array. At this point, subobj will contain the updated access/refresh tokens, and the updated exirys
            $Script:AllSubscriptions += $subobj
            
            #$ThisSubsciption.AccessToken = $script:AuthToken
            #$ThisSubsciption.RefreshToken = $Script:RefreshToken
            #$ThisSubsciption.Expiry = $script:TokenExpirationUtc
        }
        Else
        {
            #Error
        }

        Return
    }
    Else
    {
        Write-error "Could not understand how you wanted to log in"
    }
    
    if ($authResult)
    {
        $script:AuthToken = $AuthResult.AccessToken
        $Script:RefreshToken = $AuthResult.RefreshToken
        $script:TokenExpirationUtc = $AuthResult.ExpiresOn
        
        Write-Verbose -Message "Authenticated as $($AuthResult.UserInfo.DisplayableId)"
        
        $Result = Get-InternalRest -Uri "https://management.azure.com/tenants" -BearerToken $AuthResult.AccessToken -ApiVersion "2015-01-01"
    }
    Else
    {
        Write-error "Error Authenticating"
        Return
    }
    
    if (!$result)
    {
        Write-error "Error Authenticating"
        Return
    }
    
    if ($PScmdlet.ParameterSetName -ne "ConnectByRefreshToken")
    {
        #Create an array to hold the list of tenants, subscriptions and auth tokens
        $TenantAuthMap = @()
        $ThisSubscription = $null
        $Tenants = $Result.Value
    
        if ($TenantId)
        {
            $Tenants = $Tenants | where {$_.TenantId -eq $TenantId}
            if ($Tenants -eq $null)
            {
                Write-Error "The logged on user is not connected to tenant $tenantId"
                Return
            }
        }
    
    

    
        Foreach ($Tenant in $Tenants)
        {
        
            Write-verbose "Listing Subscriptions in tenant $($Tenant.tenantId)"
            $params["PromptBehavior"] = "Auto"
            $Params["LoginUrl"] = "https://login.windows.net/$($Tenant.tenantId)/oauth2/authorize/"
            $TenantauthResult = Get-InternalAcquireToken @Params
        
            Write-Debug "Using access key $($TenantauthResult.AccessToken)"
            $SubscriptionResult  = Get-InternalRest -Uri "https://management.azure.com/subscriptions" -BearerToken $TenantauthResult.AccessToken -Apiversion "2015-01-01"
            if ($SubscriptionResult.Value.count -gt 0)
            {
                foreach ($Subscription in $SubscriptionResult.Value)
                {
                    Write-verbose " Found subscription $($Subscription.subscriptionId)"
                    $SubObj = "" | Select SubscriptionId,TenantId,AccessToken,RefreshToken, Expiry, SubscriptionObject, DisplayName, State, LoginUrl
                    $subobj.SubscriptionId = $Subscription.subscriptionId
                    $subobj.DisplayName = $Subscription.displayName
                    $SubObj.State = $Subscription.state
                    $subobj.TenantId = $Tenant.tenantId
                    $subobj.AccessToken = $TenantAuthResult.AccessToken
                    $subobj.RefreshToken = $TenantauthResult.RefreshToken
                    $subobj.Expiry = $TenantauthResult.ExpiresOn
                    $subobj.SubscriptionObject = $Subscription
                    $subobj.LoginUrl = "https://login.windows.net/$($Tenant.tenantId)/oauth2/authorize/"
                    $TenantAuthMap += $SubObj
                }
            }
            Else
            {
                Write-verbose " Zero subscriptions found in tenant"
            }
        
        }
    
        #Add all subscriptions to global var
        $Script:AllSubscriptions = $TenantAuthMap
    
        #Figure out which subscription to choose
        if ($TenantAuthMap.count -eq 0)
        {
            #Error
        }
        ElseIf ($TenantAuthMap.count -gt 1 -and $SubscriptionId)
        {
            #Multiple returned, make surethe specified is in the list
            if (($TenantAuthMap | select -ExpandProperty SubscriptionId ) -notcontains $SubscriptionId)
            {
                Write-Error "specified subscriptionId $SubscriptionId was not found for tenant" -ErrorAction Stop
            }
        }
        ElseIf ($TenantAuthMap.count -eq 1)
        {
            #Only one returned, make sure its the right one
        
            #return the subscription
            $script:AuthToken = $TenantAuthMap[0].AccessToken
            $Script:RefreshToken = $TenantAuthMap[0].RefreshToken
            $script:TokenExpirationUtc = $TenantAuthMap[0].Expiry
            $script:CurrentSubscriptionId = $TenantAuthMap[0].SubscriptionId
            $ThisSubscription =  $TenantAuthMap[0]               
         
        }
        ElseIf ($TenantAuthMap.count -gt 1) {
            Write-Warning -Message "Multiple Subscriptions found and none specified. Please select the desired one"
            $i = 0
            $list = foreach ($T in $TenantAuthMap) {
                $i++
                '[{0}] - {1} - {2}' -f $i,$T.DisplayName,$T.SubscriptionId
            }
            do {
                $result = Read-Host -Prompt "Enter the index number of the desired subscription: `n$($List | Out-String)"
            } while ($result -gt $TenantAuthMap.count -or $result -eq 0)
            $script:AuthToken = $TenantAuthMap[$result -1].AccessToken
            $Script:RefreshToken = $TenantAuthMap[$result -1].RefreshToken
            $script:TokenExpirationUtc = $TenantAuthMap[$result -1].Expiry
            $ThisSubscription =  $TenantAuthMap[$result -1]
        }

        $script:CurrentSubscriptionId = $ThisSubscription.SubscriptionId
        $script:AuthToken = $ThisSubscription.AccessToken
        $script:RefreshToken = $ThisSubscription.AccessToken
        $script:TokenExpirationUtc = $ThisSubscription.Expiry
    
        #Grab the available locations for the subscriptions
        $LocationsResult = Get-InternalRest -Uri "https://management.azure.com/subscriptions/$script:CurrentSubscriptionId/locations" -ReturnType "Blue.AzureServiceLocation" -ReturnTypeSingular $false -ApiVersion "2015-01-01"
        $Script:AzureServiceLocations = $LocationsResult


        if ($BasicOutput)
        {
            return $ThisSubscription.SubscriptionId
        }
        Else
        {
            return $ThisSubscription.SubscriptionObject
        }
    
    }

    

    
}

Function Get-InternalAuthDetails
{
    Write-output $script:AuthToken
    Write-output $Script:RefreshToken
    Write-output $script:TokenExpirationUtc
}

Function Get-ArmSubscription
{
    Param (
    )
    
    Get-InternalAuthDetails
}