tentools

0.0.3

private/Write-ProgressHelper.ps1

function Write-ProgressHelper {
    # thanks adam!
    # https://www.adamtheautomator.com/building-progress-bar-powershell-scripts/
    param (
        [int]$StepNumber,
        [string]$Activity,
        [string]$Message,
        [int]$TotalSteps,
        [switch]$ExcludePercent
    )

    $caller = (Get-PSCallStack)[1].Command

    if ($caller -in "Invoke-SecureShellCommand","Set-TNCertificate","Backup-TNServer","Restore-TNServer","Send-TNAcasFile") {
        $TotalSteps = 20
    }

    if (-not $Activity) {
        $Activity = switch ($caller) {
            "Export-DbaInstance" {
                "Performing Instance Export for $instance"
            }
            default {
                "Executing $caller"
            }
        }
    }

    if ($ExcludePercent) {
        Write-Progress -Activity $Activity -Status $Message
    } else {
        if (-not $TotalSteps -and $caller -ne '<ScriptBlock>') {
            $TotalSteps = ([regex]::Matches((Get-Command -Module tentools -Name $caller).Definition, "Write-ProgressHelper")).Count
        }
        if (-not $TotalSteps) {
            $percentComplete = 0
        } else {
            $percentComplete = ($StepNumber / $TotalSteps) * 100
        }
        if ($percentComplete -gt 100) {
            $percentComplete = "100"
        }
        Write-Progress -Activity $Activity -Status $Message -PercentComplete $percentComplete
    }
}