common.ps1

<#
Library of common functions shared by the Install-foo scripts.
#>

$proEdition = $null

$script:shell = 'PowerShell'
if ($PSVersionTable.PSVersion.Major -lt 6) { $script:shell = 'WindowsPowerShell' }


function Chocolatize
{
    param($name)
    if (UnChocolatized $name)
    {
        HighTitle $name
        choco install -y $name
    }
    else
    {
        WriteOK "$name already installed by chocolatey"
    }
}

function Chocolatized
{
    param($name)
    ((choco list -l $name | Select-string "$name ").count -gt 0)
}

function UnChocolatized
{
    param($name)
    ((choco list -l $name | Select-string "$name ").count -eq 0)
}

function DownloadBootstrap
{
    # source=filename, target=folder
    param($source, $target)
    $zip = Join-Path $target $source

    if ($false) #$env:GITHUB_TOKEN)
    {
        curl -s -H "Authorization: token $($env:GITHUB_TOKEN)" `
            -H 'Accept: application/vnd.github.v3.raw' `
            -o $zip -L "https://api.github.com/repos/stevencohn/bootstraps/contents/$source`?ref=main"
    }
    else
    {
        curl -s "https://raw.githubusercontent.com/stevencohn/bootstraps/main/$source" -o $zip
    }

    Expand-Archive $zip -DestinationPath $target -Force | Out-Null
    Remove-Item $zip -Force -Confirm:$false
}

function EnsureHKCRDrive
{
    if (!(Test-Path 'HKCR:'))
    {
        New-PSDrive -Name HKCR -PSProvider Registry -Root HKEY_CLASSES_ROOT -Scope global | Out-Null
    }
}

function GetCommandList
{
    Get-ChildItem function:\ | `
        where HelpUri -match 'cmd' | `
        sort-object -property @{expression={ $_.Name } } | `
        foreach { [PsCustomObject]@{Name=$_.Name; Description=$_.scriptblock.attributes[0].description } }
}

function Highlight
{
    param($text = '', $color = 'Yellow')
    $text | Write-Host -ForegroundColor Black -BackgroundColor $color
}

function HighTitle
{
    param($title, $action = 'Installing')
    Highlight '', "---- $action $title ---------------------------"
}

function InstallChocolatey
{
    # Modules/Scripts contains a better version but this is a stand-alone copy for the
    # top-level Install scripts so they can remain independent of the Module scripts
    if ((Get-Command choco -ErrorAction:SilentlyContinue) -eq $null)
    {
        HighTitle 'Chocolatey'
        Set-ExecutionPolicy Bypass -Scope Process -Force
        Invoke-Expression ((New-Object System.Net.WebClient).DownloadString('https://chocolatey.org/install.ps1'))
    }
}

function InstallCurl
{
    if ((Get-Alias curl -ErrorAction:SilentlyContinue) -ne $null) {
        Remove-Item alias:curl -ErrorAction:SilentlyContinue
    }

    $cmd = Get-Command curl -ErrorAction:SilentlyContinue
    if ($cmd -ne $null)
    {
        if ($cmd.Source.Contains('curl.exe')) { return }
    }

    if ((Get-Command choco -ErrorAction:SilentlyContinue) -eq $null)
    {
        HighTitle 'Installing Chocolatey'
        Set-ExecutionPolicy Bypass -Scope Process -Force
        Invoke-Expression ((New-Object System.Net.WebClient).DownloadString('https://chocolatey.org/install.ps1'))
    }

    if ((choco list -l 'curl' | Select-string 'curl ').count -gt 0) { return }

    HighTitle 'Curl'
    choco install -y curl
}

function InstallGit
{
    if ((Get-Command git -ErrorAction:SilentlyContinue) -eq $null)
    {
        HighTitle 'Git'
        choco install -y git
        # Git adds its path to the Machine PATH but not the Process PATH; copy it so we don't need to restart the shell
        $gitpath = [Environment]::GetEnvironmentVariable("PATH", [EnvironmentVariableTarget]::Machine) -split ';' | ? { $_ -match 'Git\\cmd' }
        $env:Path = "${env:Path};$gitpath"

        Write-Host "git installed; consider running the command" -ForegroundColor Yellow
        Write-Host '>> git config --global --add safe.directory *' -ForegroundColor DarkYellow
    }

    if ((Get-Command gpg -ErrorAction:SilentlyContinue) -eq $null)
    {
        HighTitle 'GnuPG'
        choco install -y gnupg
        # Git adds its path to the Machine PATH but not the Process PATH; copy it so we don't need to restart the shell
        $gpgpath = "$([Environment]::GetEnvironmentVariable("ProgramFiles(x86)"))\gnupg"
        $env:Path = "${env:Path};$gpgpath"
    }
}

function InvokeCommand
{
    param($command)
    $fn = Get-ChildItem function:\ | where Name -eq $command
    if ($fn -and ($fn.HelpUri -match 'cmd'))
    {
        Highlight "... invoking command $($fn.Name)"
        Invoke-Expression $fn.Name
    }
    else
    {
        Write-Host "$command is not a recognized command" -ForegroundColor Yellow
        Write-Host 'Use -List argument to see all commands' -ForegroundColor DarkYellow
    }
}
function IsElevated
{
    # Modules/Scripts contains a better/alt version but this is a stand-alone copy for the
    # top-level Install scripts so they can remain independent of the Module scripts
    if (([Security.Principal.WindowsPrincipal][Security.Principal.WindowsIdentity]::GetCurrent()`
        ).IsInRole([Security.Principal.WindowsBuiltInRole] "Administrator"))
    {
        return $true
    }

    Write-Host
    WriteWarn '... This script must be run from an elevated console'
    WriteWarn '... Open an administrative PowerShell window and run again'
    return $false
}

function IsWindows11
{
    $0 = 'HKLM:\SOFTWARE\Microsoft\Windows NT\CurrentVersion'
    ([int](Get-ItemPropertyValue -path $0 -name CurrentBuild) -ge 22000)
}

function IsWindowsHomeEdition
{
    return (-not (IsWindowsProEdition))
}

function IsWindowsProEdition
{
    if ($null -eq $proEdition)
    {
        $script:proEdition = (Get-WindowsEdition -online).Edition -eq 'Professional'
    }

    $proEdition
}

function RebootWithContinuation
{
    param([string] $cargs)

    # prep a logon continuation task
    $trigger = New-ScheduledTaskTrigger -AtLogOn;
    # note here that the -Command arg string must be wrapped with double-quotes
    $action = New-ScheduledTaskAction -Execute 'powershell.exe' -Argument "-Command ""$($MyInvocation.ScriptName) -Continuation $cargs"""
    $principal = New-ScheduledTaskPrincipal -GroupId "BUILTIN\Administrators" -RunLevel Highest
    $name = "$([System.IO.Path]::GetFileNameWithoutExtension(($MyInvocation.ScriptName | split-path -leaf)))-continuation"
    Register-ScheduledTask -Action $action -Trigger $trigger -TaskName $name -Principal $principal | Out-Null

    Write-Host
	Write-Host '... Press Enter for required reboot ' -BackgroundColor DarkRed -ForegroundColor Black -NoNewline
    Read-Host

    Restart-Computer -Force
}

function CleanupContinuation
{
    $name = "$([System.IO.Path]::GetFileNameWithoutExtension(($MyInvocation.ScriptName | split-path -leaf)))-continuation"
    if (Get-ScheduledTask -TaskName $name -ErrorAction:silentlycontinue) {
		Unregister-ScheduledTask -TaskName $name -Confirm:$false
	}
}

function WriteOK
{
    param($text)
    $text | Write-Host -ForegroundColor Green
}
function WriteWarn
{
    param($text)
    $text | Write-Host -ForegroundColor Yellow
}