WakeUpAzureWebApp.psm1

. "$PSScriptRoot/Private/GetUrls.ps1";
. "$PSScriptRoot/Private/GetInstanceGuidsForWebApp.ps1";
. "$PSScriptRoot/Private/MapUrlsWithHostName.ps1";

function Start-WakeUpAzureWebApp{

    <#
        .SYNOPSIS
            Wake up Azure WebApp for the current subsribtion with name $TargetWebAppName
        .DESCRIPTION
            Use a list of $WarmupUrls to warm up the specified Azure WebApp for the current subsribtion with name $TargetWebAppName
            Targeting all instances one by one to make sure every instance is warmed up
        .PARAMETER TargetWebAppName
            The actual name of the Azure WebApp
        .PARAMETER WarmupUrls
            A list of string containing relative urls
        .OUTPUTS
            Some details about the progress and the response code of the urls
        .NOTES
            This module is designed for use in Azure context, so Start-WakeUpAzureWebApp needs to be called before executing the sript.
            This is designed with Octopus deploy in mind.
        .LINK
            https://gitlab.com/Valtech-Amsterdam/PowerShell/WakeUpAzureWebApp
        .EXAMPLE
            Import-Module WakeUpAzureWebApp;
 
            $targetWebAppName = #"#{Environment.WebApp.Name.CD1}";
            $warmupUrls = @(
                "/nl-nl"
                "/nl-nl/some url/"
                "/nl-nl/another url/"
            );
 
            Start-WakeUpAzureWebApp `
                -targetWebAppName $targetWebAppName `
                -warmupUrls $warmupUrls;
    #>


    [CmdletBinding()] Param(
        [Parameter(Mandatory=$True)]
        [string]$TargetWebAppName,
        [Parameter(Mandatory=$True)]
        [string[]]$WarmupUrls
    )

    BEGIN {

        try{
            $webApp = Get-AzureRmWebApp -Name $TargetWebAppName;

            $defaultHostName = $webApp.DefaultHostName;
            $instanceGuids = GetInstanceGuidsForWebApp -webApp $webApp;
            $domainUrls = MapUrlsWithHostName -defaultHostName $defaultHostName -urls $WarmupUrls;
        } catch {
            Write-Host "Something went wrong contacting Azure, are you sure you are logged in? (Login-AzureRMAccount)"
            exit 1;
        }
        $instanceCount = @($instanceGuids).length;

    }

    PROCESS {

        Write-Host "Getting urls for $instanceCount instances"
        for($i = 0; $i -lt $instanceCount; $i++){
            $instanceGuid = @($instanceGuids)[$i];

            $index = $i + 1;
            Write-Host "Warming up instance $instanceGuid ($index / $instanceCount)";
            Write-Progress -Id 91 -Activity "Getting urls for $instanceCount instances" `
                -status "Warming up instance $instanceGuid $index / $instanceCount)" `
                -percentComplete (($i / $instanceCount) * 100);

            GetUrls -defaultHostName $defaultHostName -urls $domainUrls -instanceGuid $instanceGuid;
        }

    }

    End {

        Write-Host "Completed warming up $instanceCount instances";
        Write-Progress -Id 91 -Activity "Getting urls for $instanceCount instances" `
            -status "Completed warming up $instanceCount instances" `
            -percentComplete 100 `
            -Completed;

    }
}

Export-ModuleMember -Function Start-WakeUpAzureWebApp