common.ps1

function Resolve-Exception {
    [CmdletBinding()]
    Param (
      [parameter(Mandatory = $true, Position = 0, HelpMessage = 'The exception object to handle')]
      [ValidateNotNull()]
      [Object] $ExceptionObject,
      [parameter(Mandatory = $false, HelpMessage = 'Toogle stacktrace output')]
      [switch] $HideStackTrace = $false,
      [parameter(Mandatory = $false, HelpMessage = 'The error action to use')]
      [String] $CustomErrorAction = 'Stop'
    )

    Begin {
    }

    Process
    {
        $sst = ''
        $st = ''
        $e = $ExceptionObject.Exception
        if ($null -ne (Get-Member -InputObject $ExceptionObject -Name 'ScriptStackTrace')) {
            $sst = $ExceptionObject.ScriptStackTrace
        }
        if ($null -ne (Get-Member -InputObject $ExceptionObject -Name 'StackTrace')) {
            $st = $ExceptionObject.StackTrace
        }

        $msg = $e.Message
        while ($e.InnerException) {
            $e = $e.InnerException

            $msg += "`n" + $e.Message
            if ($null -ne (Get-Member -InputObject $e -Name 'ScriptStackTrace')) {
                $sst += "`n" + $e.ScriptStackTrace + "`n---"
            }

            if ($null -ne (Get-Member -InputObject $e -Name 'StackTrace')) {
                $st += "`n" + $e.StackTrace + "`n---"
            }
        }

        if (-not ($HideStackTrace)) {
            $msg += "`n---[ScriptStackTrace]---`n" + $sst + "`n---[StackTrace]---`n" + $st
        }

        Write-Error -Message $msg -ErrorAction $CustomErrorAction
    }

    End {
    }
}

function Add-FileToAppDomain {
    [CmdletBinding()]
    param (
        [parameter(Mandatory = $true, HelpMessage = 'The base path to load files from.')]
        [ValidateNotNull()]
        [string] $BasePath,
        [parameter(Mandatory = $true, HelpMessage = 'The file to load into the AppDomain.')]
        [ValidateNotNull()]
        [string] $File
    )

    if (-not (Test-Path "$BasePath" -PathType Container))
    {
        Throw "[!] Can't find or access folder ${BasePath}."
    }

    $FileToLoad = Join-Path "${BasePath}" "$File"

    if (-not (Test-Path "$FileToLoad" -PathType Leaf))
    {
        Throw "[!] Can't find or access file ${FileToLoad}."
    }

    if ( -Not ([appdomain]::currentdomain.getassemblies() |Where-Object Location -Like ${FileToLoad})) {
        try {
            [System.Reflection.Assembly]::LoadFrom($FileToLoad) | Out-Null
            $clientVersion = [System.Diagnostics.FileVersionInfo]::GetVersionInfo($FileToLoad).ProductVersion
            Write-Debug "[+] File ${File} loaded with version ${clientVersion} from ${BasePath}."
        } catch {
            Resolve-Exception -ExceptionObject $PSitem
        }
    }
}

function Add-NuGetDependencies {
    [CmdletBinding()]
    param (
        [parameter(Mandatory = $true, HelpMessage = 'The Nuget packages to load.')]
        [Hashtable] $NugetPackages
    )

    $CurrentFileName = Split-Path $PSScriptRoot -Leaf
    $memstream = [IO.MemoryStream]::new([byte[]][char[]]$CurrentFileName)
    $CurrentFileNameHash = (Get-FileHash -InputStream $memstream -Algorithm SHA1).Hash

    # Get a unique temporary directory to store all the NuGet packages we need later
    $TempWorkDir = Join-Path "$($env:TEMP)" "$CurrentFileNameHash"

    if (-not (Test-Path "$TempWorkDir" -PathType Container))
    {
        New-Item -Path "$($env:TEMP)" -Name "$CurrentFileNameHash" -ItemType Directory
    }

    foreach ($dep in $NugetPackages.Keys) {
        $version = $NugetPackages[$dep]
        $destinationPath = Join-Path "$TempWorkDir" "${dep}.${version}"
        if (-not (Test-Path "$destinationPath" -PathType Container))
        {
            Install-Package -Name $dep -RequiredVersion $version -Destination "$TempWorkDir" -SkipDependencies -ProviderName NuGet -Source nuget.org -Force
            Write-Information "[+] Install package ${dep} with version ${version} into folder ${TempWorkDir}"
        }

        # Prioritise version 4.8 over 4.5
        $BasePath = Join-Path (Join-Path "$destinationPath" "lib") "net48"
        if (-not (Test-Path "$BasePath" -PathType Container)) {
            $BasePath = Join-Path (Join-Path "$destinationPath" "lib") "net45"
        }

        Add-FileToAppDomain -BasePath $BasePath -File "${dep}.dll"
    }
}

function Get-AutomationButton {
    [CmdletBinding()]
    param (
        [parameter(Mandatory = $true, Position = 0, HelpMessage = 'The base automation element')]
        [ValidateNotNull()]
        [FlaUI.Core.AutomationElements.AutomationElement] $BaseAutomationElement,
        [parameter(Mandatory = $false, HelpMessage = 'The name of the button to get')]
        [string] $Name,
        [parameter(Mandatory = $false, HelpMessage = 'The automation id of the button to get')]
        [string] $Id,
        [parameter(Mandatory = $false, HelpMessage = 'Delay in milliseconds to wait for the control')]
        [int] $WaitDelay = 1200
    )

    return Get-AutomationElement -BaseAutomationElement $BaseAutomationElement -Name $Name -Id $Id -ControlType 'Button' -WaitDelay $WaitDelay
}

function Get-AutomationTextBox {
    [CmdletBinding()]
    param (
        [parameter(Mandatory = $true, Position = 0, HelpMessage = 'The base automation element')]
        [ValidateNotNull()]
        [FlaUI.Core.AutomationElements.AutomationElement] $BaseAutomationElement,
        [parameter(Mandatory = $false, HelpMessage = 'The name of the Textbox to get')]
        [string] $Name,
        [parameter(Mandatory = $false, HelpMessage = 'The automation id of the Textbox to get')]
        [string] $Id,
        [parameter(Mandatory = $false, HelpMessage = 'Delay in milliseconds to wait for the control')]
        [int] $WaitDelay = 50
    )

    return Get-AutomationElement -BaseAutomationElement $BaseAutomationElement -Name $Name -Id $Id -ControlType 'Text' -WaitDelay $WaitDelay
}

function Get-AutomationElement {
    [CmdletBinding()]
    param (
        [parameter(Mandatory = $true, Position = 0, HelpMessage = 'The base automation element')]
        [ValidateNotNull()]
        [FlaUI.Core.AutomationElements.AutomationElement] $BaseAutomationElement,
        [parameter(Mandatory = $true, Position = 1, HelpMessage = 'The control type of the element to get')]
        [ValidateNotNull()]
        [string] $ControlType,
        [parameter(Mandatory = $false, HelpMessage = 'The name of the element to get')]
        [string] $Name,
        [parameter(Mandatory = $false, HelpMessage = 'The automation id of the element to get')]
        [string] $Id,
        [parameter(Mandatory = $false, HelpMessage = 'Delay in milliseconds to wait for the control')]
        [int] $WaitDelay = 250
    )

    $ElementCondition = $BaseAutomationElement.ConditionFactory.ByControlType($ControlType)
    [FlaUI.Core.AutomationElements.AutomationElement] $FoundElement = $null
    $MaxRetries = 2500

    $CallStack = Get-PSCallStack

    while ($true) {
        if (($PSBoundParameters.ContainsKey('Id')) -and (-not ([string]::IsNullOrWhiteSpace("$Id")))) {
            $FoundElement = $BaseAutomationElement.FindFirstDescendant($ElementCondition.And($BaseAutomationElement.ConditionFactory.ByAutomationId($Id)))
        } else {
            $FoundElement = $BaseAutomationElement.FindFirstDescendant($ElementCondition.And($BaseAutomationElement.ConditionFactory.ByName($Name)))
        }

        if ($null -ne $FoundElement) {
            if ($FoundElement.IsAvailable -and $FoundElement.IsEnabled) {
                break
            } else {
                Write-Debug "$($CallStack[2].ScriptName): $($CallStack[2].ScriptLineNumber):`n`tFound element has state isAvailable: $($FoundElement.IsAvailable) and isEnabled: $($FoundElement.IsEnabled). Retries to go: $($MaxRetries) - $FoundElement"
            }
        } elseif (0 -eq ($MaxRetries % 250)) {
            Write-Debug "$($CallStack[2].ScriptName): $($CallStack[2].ScriptLineNumber):`n`tStill searching for element of type '$ControlType' with Id '$Id' or Name '$Name'."
        }

        Start-Sleep -Milliseconds $WaitDelay
        if (0 -eq $MaxRetries) {
            throw "$($CallStack[2].ScriptName): $($CallStack[2].ScriptLineNumber):`n`tCouldn't find element of type '$ControlType' with Id '$Id' or Name '$Name'."
        }

        $MaxRetries--
    }

    try {
        switch ($ControlType) {
            'List' { $ControlType = 'ListBox' }
            'ListItem' { $ControlType = 'ListBoxItem' }
            'Edit' { $ControlType = 'TextBox' }
            'Pane' { $ControlType = 'TextBox' }
            'Text' { $ControlType = 'TextBox' }
            'Hyperlink' { $ControlType = 'AutomationElement' }
            'ScrollBar' { $ControlType = 'AutomationElement' }
            'MenuBar' { $ControlType = 'AutomationElement' }
            'MenuItem' { $ControlType = 'AutomationElement' }
            'ToolBar' { $ControlType = 'AutomationElement' }
            'DataItem' { $ControlType = 'AutomationElement' }
            'TabItem' { $ControlType = 'AutomationElement' }
            'Window' { $ControlType = 'AutomationElement' }
        }

        Set-Variable -Name 'RetVal' -Scope 'Local' -Value ($FoundElement.FrameworkAutomationElement -As ("FlaUI.Core.AutomationElements.$ControlType" -As [Type]))
    } catch {
        Resolve-Exception -ExceptionObject $PSitem
    }

    return $RetVal
}