posh-winget

0.0.5

posh-winget.ps1

function Invoke-WingetList {
    [CmdletBinding()]
    param (
        [Parameter()]
        [string]
        $Query,
        [Parameter()]
        [string]
        $Id,
        [Parameter()]
        [string]
        $Name,
        [Parameter()]
        [string]
        $Moniker,
        [Parameter()]
        [string]
        $Source,
        [Parameter()]
        [string]
        $Tag,
        [Parameter()]
        [string]
        $Command,
        [Parameter()]
        [int]
        $Count,
        [Parameter()]
        [switch]
        $Exact
    )
    $arguments = @()
    $arguments += $Query
    if ($Id) {
        $arguments += "--id"
        $arguments += $Id
    }
    if ($Name) {
        $arguments += "--name"
        $arguments += $Name
    }
    if ($Moniker) {
        $arguments += "--moniker"
        $arguments += $Moniker
    }
    if ($Source) {
        $arguments += "--source"
        $arguments += $Source
    }
    if ($Tag) {
        $arguments += "--tag"
        $arguments += $Tag
    }
    if ($Command) {
        $arguments += "--command"
        $arguments += $Command
    }
    if ($Count) {
        $arguments += "--count"
        $arguments += $Count.ToString()
    }
    if ($Exact) {
        $arguments += "--exact"
    }

    $result = (Invoke-Winget list $arguments)
    if ($result.IsSuccess) {
        Read-Package ($result.StandardOutput -split "`r`n")
    }
    else {
        @()
    }
}

function Invoke-WingetInstall {
    [CmdletBinding()]
    param (
        [Parameter(Mandatory)]
        [string]
        $Id
    )

    winget install --id $Id
}

function Invoke-WingetImport {
    [CmdletBinding()]
    param (
        [Parameter(Mandatory)]
        [string]
        $Path
    )

    $config = Get-Content -Path $Path | ConvertFrom-Yaml
    $config | ForEach-Object {
        $package = $_
        $list = Invoke-WingetList -Id $package.id -Exact
        if($list.Length -eq 0) {
            $arguments = @()
            $arguments += 'install'
            if($package.silent) {
                $arguments += '--silent'
            }
            $arguments += '--id'
            $arguments += $package.id

            if($null -ne $package.packageParameters) {
                $arguments += '--override'
                $arguments += "`"$($package.packageParameters)`""
            }

            $argumentList = [string]::Join(' ', $arguments)
            Start-Process winget -NoNewWindow -Wait -ArgumentList $argumentList
        } else {
            Write-Host "$($package.id) is installed."
        }
    }
}

function Read-Package {
    [CmdletBinding()]
    param (
        [Parameter()]
        [string[]]
        $Packages
    )
    if($Packages.Length -eq 1) {
        @()
    }

    $header = $Packages[0]
    $header = $header.Substring($header.LastIndexOf("`r") + 1).Trim()
    $indexs = @()
    $header.Split(" ") | Where-Object { $_.Length -ne 0} | ForEach-Object {
        $column = $_
        $indexs += Get-Width $header.Substring(0, $header.IndexOf($column))
    }

    [array]::Reverse( $indexs )

    $result = @()
    $Packages | Select-Object -Skip 2 | ForEach-Object {
        $line = $_
        $columns = @()
        $indexs | ForEach-Object {
            $index = Get-Index $line $_
            if( $index -ge 0) {
                $column = $line.Substring($index).Trim()
                if ($column.EndsWith("…")) {
                    $column = $column.Substring(0, $column.Length - 1)
                }
                if ($column.Length -eq 0) {
                    $column = $null
                }
                $columns += $column
                $line = $line.Substring(0, $index)
            }
            else {
                $columns += $null
            }
        }
        if ($columns.Length -eq 5) {
            $result += [PSCustomObject]@{
                NamePrefix = $columns[4]
                IdPrefix = $columns[3]
                Version = $columns[2]
                Available = $columns[1]
                Source = $columns[0]
            }
        }
        elseif ($columns.Length -eq 4) {
            $result += [PSCustomObject]@{
                NamePrefix = $columns[3]
                IdPrefix = $columns[3]
                Version = $columns[1]
                Available = $null
                Source = $columns[0]
            }
        }
        else {
            $result += [PSCustomObject]@{
                NamePrefix = $columns[2]
                IdPrefix = $columns[1]
                Version = $columns[0]
                Available = $null
                Source = $null
            }
        }
    }
    $result
}

function Get-Width
{
    [CmdletBinding()]
    param (
        [Parameter()]
        [string]
        $str
    )
    $encoding = [system.Text.Encoding]::UTF8
    $charArray = $str.ToCharArray()
    $width = 0
    for ($i = 0; $i -lt $charArray.Count; $i++) {
        $char = $charArray[$i]
        if($char -eq "…") {
            $width += 1;
        }
        elseif ($encoding.GetBytes($char).Length -eq 1) {
            $width += 1;
        }
        else {
            $width += 2;
        }
    }
    $width
}

function Get-Index
{
    [CmdletBinding()]
    param (
        [Parameter()]
        [string]
        $Word,
        [Parameter()]
        [int]
        $halfCharIndex
    )
    $encoding = [system.Text.Encoding]::UTF8

    if ($halfCharIndex -eq 0) {
        0
        return
    }

    $wordChars = $Word.ToCharArray()
    $currentIndex = 0
    for ($i = 0; $i -lt $wordChars.Count; $i++) {
        $current = $Word[$i]
        if($current -eq "…") {
            $currentIndex += 1;
        }
        elseif ($encoding.GetBytes($current).Length -eq 1) {
            $currentIndex += 1;
        }
        else {
            $currentIndex += 2;
        }
        if($currentIndex -eq $halfCharIndex) {
            ($i + 1)
            break
        }
    }
}

function Invoke-Winget {
    [CmdletBinding()]
    param (
        [Parameter(Mandatory)]
        [string]
        $Command,
        [Parameter()]
        [array]
        $Arguments
    )
    $argument = [string]::Join(' ', ($Arguments | Where-Object { $_.Length -ne 0 } | ForEach-Object { 
        $value = $_
        if ($value.StartsWith('--')) {
            $value
        }
        else {
            "`"${value}`"" 
        }
    }))

    $psi = New-Object Diagnostics.ProcessStartInfo
    $psi.FileName = $env:LOCALAPPDATA + "\Microsoft\WindowsApps\winget.exe"
    $psi.Arguments = "$Command $argument"
    $psi.UseShellExecute = $false
    $psi.StandardOutputEncoding = [Text.Encoding]::UTF8
    $psi.RedirectStandardOutput = $true
    Using-Object ($p = [Diagnostics.Process]::Start($psi)) {
        $s = $p.StandardOutput.ReadToEnd()
        $p.WaitForExit()
        if ($p.ExitCode -eq 0) {
            [PSCustomObject]@{
                IsSuccess = $true
                StandardOutput = $s
            }
        }
        else {
            [PSCustomObject]@{
                IsSuccess = $false
                StandardOutput = $s
            }
        }
    }
}

function Using-Object
{
    [CmdletBinding()]
    param (
        [Parameter(Mandatory = $true)]
        [AllowEmptyString()]
        [AllowEmptyCollection()]
        [AllowNull()]
        [Object]
        $InputObject,

        [Parameter(Mandatory = $true)]
        [scriptblock]
        $ScriptBlock
    )

    try
    {
        . $ScriptBlock
    }
    finally
    {
        if ($null -ne $InputObject -and $InputObject -is [System.IDisposable])
        {
            $InputObject.Dispose()
        }
    }
}