PowerStub

2.0.2

private/functions/Invoke-PowerStubUpdate.ps1

<#
.SYNOPSIS
    Handles the 'pstb update' virtual verb for updating git repositories.

.DESCRIPTION
    Updates git repositories for registered stubs. Supports updating all stubs
    or a specific stub, with an optional --check flag for status-only mode.

.PARAMETER Command
    The command argument (may contain a stub name).

.PARAMETER RemainingArgs
    Additional arguments (may contain --check flag or stub name).

.OUTPUTS
    None. Writes status information to the host.
#>

function Invoke-PowerStubUpdate {
    [CmdletBinding()]
    param(
        [string]$Command,
        [object[]]$RemainingArgs
    )

    if (-not $Script:GitEnabled) {
        throw "Git integration is disabled. Set GitEnabled to true in the file returned by Get-PowerStubConfiguration -Key ConfigFile, then reload PowerStub."
    }
    if (-not $Script:GitAvailable) {
        throw "Git is not available on this system."
    }

    # Parse arguments for --check flag and stub name
    $checkOnly = $false
    $targetStub = $null
    $allArgs = @()
    if ($Command) { $allArgs += $Command }
    if ($RemainingArgs) { $allArgs += $RemainingArgs }

    foreach ($arg in $allArgs) {
        if ($arg -eq '--check' -or $arg -eq '-Check') {
            $checkOnly = $true
        }
        elseif (-not $arg.StartsWith('-')) {
            $targetStub = $arg
        }
    }

    $stubs = Get-PowerStubConfigurationKey 'Stubs'
    $processedRepos = @{}
    $showStatus = {
        param($StubName, $GitInfo)
        if ($GitInfo.Status -ne 'Ready') {
            Write-Host "Stub '$StubName': $($GitInfo.StatusMessage)." -ForegroundColor Yellow
        }
        elseif ($GitInfo.BehindCount -gt 0 -and $GitInfo.AheadCount -gt 0) {
            Write-Host "Stub '$StubName' has diverged: $($GitInfo.BehindCount) commit(s) behind and $($GitInfo.AheadCount) commit(s) ahead." -ForegroundColor Yellow
        }
        elseif ($GitInfo.BehindCount -gt 0) {
            Write-Host "Stub '$StubName' is $($GitInfo.BehindCount) commit(s) behind." -ForegroundColor Yellow
        }
        elseif ($GitInfo.AheadCount -gt 0) {
            Write-Host "Stub '$StubName' is $($GitInfo.AheadCount) commit(s) ahead." -ForegroundColor Cyan
        }
        else {
            Write-Host "Stub '$StubName' is up to date." -ForegroundColor Green
        }
    }

    if ($targetStub) {
        # Process specific stub
        if (-not ($stubs.Keys -contains $targetStub)) {
            throw "Stub '$targetStub' not found."
        }
        $stubConfig = $stubs[$targetStub]
        $stubPath = Get-PowerStubPath -StubConfig $stubConfig
        $gitInfo = Get-PowerStubGitInfo -Path $stubPath -Fetch:$checkOnly
        if (-not $gitInfo.IsRepo) {
            throw "Stub '$targetStub' is not in a Git repository."
        }

        if ($checkOnly) {
            & $showStatus $targetStub $gitInfo
        }
        else {
            Write-Host "Updating stub '$targetStub'..." -ForegroundColor Cyan
            $result = Update-PowerStubGitRepo -Path $stubPath
            if ($result.Success) {
                Write-Host " $($result.Message)" -ForegroundColor Green
            }
            else {
                Write-Host " $($result.Message)" -ForegroundColor Red
            }
        }
    }
    else {
        # Process all stubs
        $behindCount = 0
        $aheadCount = 0
        $uncheckedCount = 0
        $updatedCount = 0
        $failedCount = 0
        foreach ($stubName in $stubs.Keys) {
            $stubConfig = $stubs[$stubName]
            $stubPath = Get-PowerStubPath -StubConfig $stubConfig
            $gitInfo = Get-PowerStubGitInfo -Path $stubPath -Fetch:$checkOnly
            if ($gitInfo.IsRepo -and $gitInfo.RepoRoot -and -not $processedRepos.ContainsKey($gitInfo.RepoRoot)) {
                if ($checkOnly) {
                    & $showStatus $stubName $gitInfo
                    if ($gitInfo.Status -ne 'Ready') {
                        $uncheckedCount++
                    }
                    else {
                        if ($gitInfo.BehindCount -gt 0) { $behindCount++ }
                        if ($gitInfo.AheadCount -gt 0) { $aheadCount++ }
                    }
                }
                else {
                    Write-Host "Updating stub '$stubName' ($($gitInfo.RepoRoot))..." -ForegroundColor Cyan
                    $result = Update-PowerStubGitRepo -Path $stubPath
                    if ($result.Success) {
                        $updatedCount++
                        Write-Host " $($result.Message)" -ForegroundColor Green
                    }
                    else {
                        $failedCount++
                        Write-Host " $($result.Message)" -ForegroundColor Red
                    }
                }
                $processedRepos[$gitInfo.RepoRoot] = $true
            }
        }
        if ($processedRepos.Count -eq 0) {
            Write-Host "No stubs with Git repositories found." -ForegroundColor Yellow
        }
        elseif ($checkOnly) {
            if ($behindCount -gt 0) {
                Write-Host "`n$behindCount stub(s) have updates available. Run 'pstb update' to pull changes." -ForegroundColor Yellow
            }
            if ($aheadCount -gt 0) {
                Write-Host "`n$aheadCount repository(ies) have local commits ahead of their tracking branch." -ForegroundColor Cyan
            }
            if ($uncheckedCount -gt 0) {
                Write-Host "`n$uncheckedCount repository(ies) could not be checked." -ForegroundColor Yellow
            }
            if ($behindCount -eq 0 -and $aheadCount -eq 0 -and $uncheckedCount -eq 0) {
                Write-Host "`nAll $($processedRepos.Count) stub(s) are up to date." -ForegroundColor Green
            }
        }
        else {
            Write-Host "`nUpdated $updatedCount repository(ies)." -ForegroundColor Cyan
            if ($failedCount -gt 0) {
                Write-Host "$failedCount repository(ies) failed to update." -ForegroundColor Red
            }
        }
    }
}