CorkScrew

1.0.66

Public/Get-GithubRepo.ps1

function Get-GithubRepo {
    [CmdletBinding()]
    Param(
        [Parameter(Mandatory = $true, Position = 0)]
        [string]$Owner,

        [Parameter(Mandatory = $true, Position = 1)]
        [string]$Repository,

        [Parameter(Mandatory = $false)]
        [string]$Branch = 'master',

        [Parameter(Mandatory = $false)]
        [string]$TargetPath = (Get-Location).Path,

        [Parameter(ParameterSetName = "Credential", Mandatory = $false)]
        [System.Management.Automation.PSCredential]
        [System.Management.Automation.Credential()]
        $Credential = $global:GithubCredential,

        [Parameter(Mandatory = $false)]
        [switch]$Release
    )

    Begin {
        $VerbosePrefix = "Get-GithubRepo:"
        try {
            $ResolvedTargetPath = Resolve-Path -Path $TargetPath -ErrorAction Stop
            $LocalFile = Join-Path -Path $ResolvedTargetPath -ChildPath "$Repository.zip"
            $ExtractDirectory = Join-Path -Path $ResolvedTargetPath -ChildPath $Repository
        } catch [System.Management.Automation.SessionStateException] {
            $PSCmdlet.ThrowTerminatingError(
                [System.Management.Automation.ErrorRecord]::new(
                    ([System.ArgumentException]"TargetPath not found."),
                    'TargetPathNotFound',
                    [System.Management.Automation.ErrorCategory]::CloseError,
                    $TargetPath
                )
            )
        }

        if ($Credential) {
            $NetworkCredential = $Credential.GetNetworkCredential()
            $Headers = @{
                Authorization = 'Basic ' + [System.Convert]::ToBase64String(
                    [System.Text.Encoding]::ASCII.GetBytes(
                        "$($NetworkCredential.UserName):$($NetworkCredential.Password)"
                    )
                )
            }
        } else {
            $Headers = @{ }
        }

        # Legacy PowerShell config
        # Enable TLS1.2 for Invoke-WebRequest
        # Enable -UseBasicParsing for Invoke-WebRequest
        if ($global:PSVersionTable.PSEdition -ne 'Core') {
            [Net.ServicePointManager]::SecurityProtocol = [Net.SecurityProtocolType]::Tls12
            $InvokeWebRequestParams = @{ 'UseBasicParsing' = $true }
        } else {
            $InvokeWebRequestParams = @{ }
        }

        $InitialUrl = 'https://api.github.com/repos/' + $Owner + "/" + $Repository
        try {
            $RepoInfo = Invoke-WebRequest -Uri $InitialUrl -Headers $Headers @InvokeWebRequestParams
        } catch {
            $ErrorMessage = ($_.ErrorDetails.Message | ConvertFrom-Json).message
            switch -Regex ($ErrorMessage) {
                'two-factor' {
                    $PSCmdlet.ThrowTerminatingError(
                        [System.Management.Automation.ErrorRecord]::new(
                            ([System.ArgumentException]"MFA is no longer supported by this API. Use a personal access token instead. https://help.github.com/en/github/authenticating-to-github/creating-a-personal-access-token-for-the-command-line"),
                            'RepoUrlNotFound',
                            [System.Management.Automation.ErrorCategory]::CloseError,
                            $InitialUrl
                        )
                    )

                }
                default {
                    $PSCmdlet.ThrowTerminatingError(
                        [System.Management.Automation.ErrorRecord]::new(
                            ([System.ArgumentException]"URL not found. If this is a private repo, specify -Credential,"),
                            'RepoUrlNotFound',
                            [System.Management.Automation.ErrorCategory]::CloseError,
                            $InitialUrl
                        )
                    )
                }
            }

        }

        if ($Release) {
            $Url = $RepoInfo.releases_url -replace '{/id}', '/latest'
            try {
                # Get latest release zipfile url
                $LatestRelease = Invoke-WebRequest -Uri $Url -Headers $Headers @InvokeWebRequestParams
                $Content = ConvertFrom-Json $LatestRelease.Content
                $DownloadUrl = $Content.zipball_url
            } catch {
                $PSCmdlet.ThrowTerminatingError(
                    [System.Management.Automation.ErrorRecord]::new(
                        ([System.ArgumentException]"URL not found. If this is a private repo, specify -Credential,"),
                        'RepoUrlNotFound',
                        [System.Management.Automation.ErrorCategory]::CloseError,
                        $InitialUrl
                    )
                )
            }
        } else {
            $DownloadUrl = "https://github.com/" + $Owner + "/" + $Repository + "/archive/" + $Branch + ".zip"
        }

    }

    Process {
        # Download zip file
        try {
            $DownloadFile = Invoke-WebRequest -Uri $DownloadUrl -OutFile $LocalFile -Headers $Headers @InvokeWebRequestParams
        } catch {
            $PSCmdlet.ThrowTerminatingError(
                [System.Management.Automation.ErrorRecord]::new(
                    ([System.ArgumentException]"URL not found. If this is a private repo, specify -Credential,"),
                    'RepoUrlNotFound',
                    [System.Management.Automation.ErrorCategory]::CloseError,
                    $InitialUrl
                )
            )
        }

        # Expand File
        Write-Verbose "Extracting Zip: $LocalFile -> $ExtractDirectory"
        Expand-Archive -Path $LocalFile -DestinationPath $ExtractDirectory


        # Move Files to root of targer directory
        $ExtraDirectory = $Repository + '-*'
        $ExtractedFolder = (Get-ChildItem -Path $ExtractDirectory -Filter $ExtraDirectory).FullName
        Write-Verbose "ExtractedFolder: $ExtractedFolder"
        $ExtractedDirectories = Get-ChildItem -Path $ExtractedFolder -Recurse -Directory
        foreach ($directory in $ExtractedDirectories) {
            $thisSource = $directory.FullName
            Write-Verbose "$VerbosePrefix checking for directory: $thisSource"
            $thisDestination = ($directory.FullName).Replace($ExtractedFolder, $ExtractDirectory)
            if (!(Test-Path -Path $thisDestination)) {
                New-Item -Path $thisDestination -ItemType Directory | Out-Null
            }
        }
        $ExtractedFiles = Get-ChildItem -Path $ExtractedFolder -Recurse -File
        foreach ($file in $ExtractedFiles) {
            $thisSource = $file.FullName
            $thisDestination = ($file.FullName).Replace($ExtractedFolder, $ExtractDirectory)
            $Move = Move-Item -Path $thisSource -Destination $thisDestination -Force
        }
        $RemoveExtraFolder = Remove-Item -Path $ExtractedFolder -Recurse -Force

        # Delete zip file
        Remove-Item -Path $LocalFile | Out-Null

    }
}

New-Alias -Name ggh -Value Get-GithubRepo