Postgres Initializer

pgcli.ps1

<#
.LINK
https://www.postgresql.org/docs/current/app-pg-ctl.html
#>
param(
    [string]$DataDir,
    [switch]$InitDataDir,

    [string]$ServiceName,
    [switch]$Register,
    [switch]$Unregister,

    [switch]$Start,
    [switch]$Stop,
    [switch]$AdhocInstance,

    [switch]$CreateDb,
    [string]$DbName
)

trap { throw $_ }
$ctl = (Get-Command -ErrorAction Stop "pg_ctl.exe").Path

$env:LC_ALL = "en_US.UTF-8"
& $ctl --version

if ($DataDir -eq '') {
    $DataDir = Join-Path $PSScriptRoot "data"
}

$Username = "postgres"

if ($ServiceName -eq '') {
    $ServiceName = "postgres"
}

if ($DbName -eq '') {
    $DbName = "main"
}

$script:currentUser = [Security.Principal.WindowsPrincipal][Security.Principal.WindowsIdentity]::GetCurrent()

function Test-IsAdmin {
    param(
        [switch]$NoThrow
    )
    $isAdmin = $script:currentUser.IsInRole([Security.Principal.WindowsBuiltInRole]::Administrator)
    if (-not $isAdmin -and -not $NoThrow) {
        Write-Error -ErrorAction Stop "Not run as Admin"
        throw
    }
}

if ($Stop) {
    if ($AdhocInstance) {
        Write-Host "pg_ctl stop -D $DataDir"
        & $ctl stop -D $DataDir
        if ($LASTEXITCODE -ne 0) {
            Write-Host "pg_ctl stop exited with $LASTEXITCODE"
            exit 1
        }
    }
    else {
        Test-IsAdmin
        Stop-Service -ErrorAction Stop -Name $ServiceName
        Get-Service -ErrorAction Stop -Name $ServiceName
    }
}

if ($Unregister) {
    Test-IsAdmin
    Write-Host "pg_ctl unregister -N $ServiceName"
    & $ctl unregister -N $ServiceName
    if ($LASTEXITCODE -ne 0) {
        Write-Host "pg_ctl unregister exited with $LASTEXITCODE"
        exit 1
    }
}

if ($InitDataDir) {
    Write-Host "pg_ctl initdb -D $DataDir"
    & $ctl initdb -D $DataDir -o "--username=$Username --locale=en_US.UTF-8 --encoding=UTF8 --set config_file=.\postgresql.conf"
    if ($LASTEXITCODE -ne 0) {
        Write-Host "pg_ctl initdb exited with $LASTEXITCODE"
        exit 1
    }
}

if ($Register) {
    Test-IsAdmin
    Write-Host "pg_ctl register -N $ServiceName"
    # -S - start type
    # -w - wait for server to start
    & $ctl register -N $ServiceName -D $DataDir -l postgres.log -U LocalSystem -S auto --wait
    if ($LASTEXITCODE -ne 0) {
        Write-Host "pg_ctl register exited with $LASTEXITCODE"
        exit 1
    }
}

if ($Start) {
    if ($AdhocInstance) {
        Write-Host "pg_ctl start -D $DataDir"
        & $ctl start -D $DataDir
        if ($LASTEXITCODE -ne 0) {
            Write-Host "pg_ctl start exited with $LASTEXITCODE"
            exit 1
        }
    }
    else {
        Test-IsAdmin
        Start-Service -ErrorAction Stop -Name $ServiceName
        Get-Service -ErrorAction Stop -Name $ServiceName
    }
}

if ($CreateDb) {
    & psql -U $Username -c "CREATE DATABASE $DbName WITH OWNER $Username ENCODING 'utf-8';"
    if ($LASTEXITCODE -ne 0) {
        Write-Host "psql exited with $LASTEXITCODE"
        exit 1
    }
}