pure-pwsh

0.1.0

pure-pwsh.psm1

filter color {$_.Split('*')[0]} # the part from '*' is only for `$pure` display

function global:prompt {
  $isError = !$?

  $curPath = $ExecutionContext.SessionState.Path.CurrentLocation.Path
  if ($curPath.ToLower().StartsWith($Home.ToLower())) {
    $curPath = "~" + $curPath.SubString($Home.Length)
  }

  $prompt = "`n$($pure.pwdColor | color)$curPath "

  $gitStatus = if ($null -ne (Get-Module posh-git)) {get-gitstatus} else {$null}
  if ($gitStatus) {
    $prompt += "$($pure.branchColor | color)$($gitStatus.branch)"
    if ($gitStatus.HasWorking -or $gitStatus.HasIndex) {
      $prompt += "*"
    }
    $prompt += " "
    if ($gitStatus.BehindBy) {
      $prompt += "$($pure.remoteColor | color)$($pure.downChar)"
    }

    if ($gitStatus.AheadBy) {
      $prompt += "$($pure.remoteColor | color)$($pure.upChar)"
    }
  }

  if ($lastCmd = Get-History -Count 1) {
    $diff = $lastCmd.EndExecutionTime - $lastCmd.StartExecutionTime
    if ($diff.TotalSeconds -gt 2) {
      $prompt += "$($pure.errorColor | color) ($("{0:f2}" -f $diff.TotalSeconds)s)"
    }
  }

  $promptColor = if ($isError) {$pure.errorColor} else {$pure.PromptColor}
  $prompt += "`n$($promptColor | color)$($pure.PromptChar) "
  $prompt
}

function Set-PureOption() {
  [CmdletBinding()]
  param (
    [ValidateSet(
      'PwdColor',
      'BranchColor',
      'RemoteColor',
      'ErrorColor',
      'PromptColor',
      'PromptChar',
      'UpChar',
      'DownChar')]
    $Option,

    [String]
    $Value
  )

  if ($Option -like '*Color') {
    $val = if ($Value -match '^\d*m$') {$null} else {$Value}
    $Value = valueOrDefault $val $Value
  }
  $Global:pure.$option = $value
}

function valueOrDefault($value, $default) {
  "$(if ($value) {$value} else {"$([char]0x1b)[$default"})" +
  "*$([char]0x1b)[0m"
}

function init() {
  $psrOptions = Get-PSReadlineOption
  if ($psrOptions) {
    Set-PSReadLineOption -PromptText "❯ "
    Set-PSReadLineOption -ContinuationPrompt "❯❯ "
    Set-PSReadLineOption -Colors @{ ContinuationPrompt = $psrOptions.EmphasisColor }
  }

  $Global:pure = [ordered]@{
    'PwdColor'    = valueOrDefault $psrOptions.CommentColor "32m"
    'BranchColor' = valueOrDefault $psrOptions.StringColor "36m"
    'RemoteColor' = valueOrDefault $psrOptions.TypeColor "37m"
    'ErrorColor'  = valueOrDefault $psrOptions.ErrorColor "91m"
    'PromptColor' = valueOrDefault $psrOptions.EmphasisColor "96m"
    'PromptChar'  = '❯'
    'UpChar'      = '⇡'
    'DownChar'    = '⇣'
  }
}

init