oh-my-posh.psm1
|
#requires -Version 2 -Modules posh-git . "$PSScriptRoot\Themes\Tools.ps1" . "$PSScriptRoot\defaults.ps1" $global:ThemeSettings = New-Object -TypeName PSObject -Property @{ CurrentThemeLocation = "$PSScriptRoot\Themes\Agnoster.psm1" MyThemesLocation = '~\Documents\WindowsPowerShell\PoshThemes' ErrorCount = 0 GitSymbols = @{ BranchSymbol = [char]::ConvertFromUtf32(0xE0A0) BeforeStashSymbol = '{' AfterStashSymbol = '}' DelimSymbol = '|' LocalWorkingStatusSymbol = '!' LocalStagedStatusSymbol = '~' LocalDefaultStatusSymbol = '' BranchUntrackedSymbol = [char]::ConvertFromUtf32(0x2262) BranchIdenticalStatusToSymbol = [char]::ConvertFromUtf32(0x2263) BranchAheadStatusSymbol = [char]::ConvertFromUtf32(0x2191) BranchBehindStatusSymbol = [char]::ConvertFromUtf32(0x2193) } PromptSymbols = @{ StartSymbol = ' ' TruncatedFolderSymbol = '..' PromptIndicator = [char]::ConvertFromUtf32(0x25B6) FailedCommandSymbol = [char]::ConvertFromUtf32(0x2A2F) ElevatedSymbol = [char]::ConvertFromUtf32(0x26A1) SegmentForwardSymbol = [char]::ConvertFromUtf32(0xE0B0) SegmentBackwardSymbol = [char]::ConvertFromUtf32(0x26A1) SegmentSeparatorForwardSymbol = [char]::ConvertFromUtf32(0x26A1) SegmentSeparatorBackwardSymbol = [char]::ConvertFromUtf32(0x26A1) PathSeparator = '\' } Colors = @{ GitDefaultColor = [ConsoleColor]::DarkGreen GitLocalChangesColor = [ConsoleColor]::DarkYellow GitNoLocalChangesAndAheadColor = [ConsoleColor]::DarkMagenta PromptForegroundColor = [ConsoleColor]::White PromptHighlightColor = [ConsoleColor]::DarkBlue DriveForegroundColor = [ConsoleColor]::DarkBlue PromptBackgroundColor = [ConsoleColor]::DarkBlue PromptSymbolColor = [ConsoleColor]::White SessionInfoBackgroundColor = [ConsoleColor]::Magenta SessionInfoForegroundColor = [ConsoleColor]::White CommandFailedIconForegroundColor = [ConsoleColor]::DarkRed AdminIconForegroundColor = [ConsoleColor]::DarkYellow WithBackgroundColor = [ConsoleColor]::DarkRed WithForegroundColor = [ConsoleColor]::White GitForegroundColor = [ConsoleColor]::Black } } <# .SYNOPSIS Method called at each launch of Powershell .DESCRIPTION Sets up things needed in each console session, aside from prompt #> function Start-Up { if(Test-Path -Path ~\.last) { (Get-Content -Path ~\.last) | Set-Location Remove-Item -Path ~\.last } # Makes git diff work $env:TERM = 'msys' Set-Prompt } <# .SYNOPSIS Generates the prompt before each line in the console #> function Set-Prompt { Import-Module $sl.CurrentThemeLocation function global:prompt { $lastCommandFailed = $global:error.Count -gt $sl.ErrorCount $sl.ErrorCount = $global:error.Count #Start the vanilla posh-git when in a vanilla window, else: go nuts if(Test-IsVanillaWindow) { Write-Host -Object ($pwd.ProviderPath) -NoNewline Write-VcsStatus return '> ' } Write-Theme -lastCommandFailed $lastCommandFailed return ' ' } } function global:Write-WithPrompt() { param( [string] $command ) $lastCommandFailed = $global:error.Count -gt $sl.ErrorCount $sl.ErrorCount = $global:error.Count if(Test-IsVanillaWindow) { Write-ClassicPrompt -command $command return } Write-Theme -lastCommandFailed $lastCommandFailed -with $command Write-Host ' ' -NoNewline } function Show-ThemeColors { Write-Host -Object '' Write-ColorPreview -text 'GitDefaultColor ' -color $sl.Colors.GitDefaultColor Write-ColorPreview -text 'GitLocalChangesColor ' -color $sl.Colors.GitLocalChangesColor Write-ColorPreview -text 'GitNoLocalChangesAndAheadColor ' -color $sl.Colors.GitNoLocalChangesAndAheadColor Write-ColorPreview -text 'GitForegroundColor ' -color $sl.Colors.GitForegroundColor Write-ColorPreview -text 'PromptForegroundColor ' -color $sl.Colors.PromptForegroundColor Write-ColorPreview -text 'PromptBackgroundColor ' -color $sl.Colors.PromptBackgroundColor Write-ColorPreview -text 'PromptSymbolColor ' -color $sl.Colors.PromptSymbolColor Write-ColorPreview -text 'PromptHighlightColor ' -color $sl.Colors.PromptHighlightColor Write-ColorPreview -text 'SessionInfoBackgroundColor ' -color $sl.Colors.SessionInfoBackgroundColor Write-ColorPreview -text 'SessionInfoForegroundColor ' -color $sl.Colors.SessionInfoForegroundColor Write-ColorPreview -text 'CommandFailedIconForegroundColor ' -color $sl.Colors.CommandFailedIconForegroundColor Write-ColorPreview -text 'AdminIconForegroundColor ' -color $sl.Colors.AdminIconForegroundColor Write-ColorPreview -text 'WithBackgroundColor ' -color $sl.Colors.WithBackgroundColor Write-ColorPreview -text 'WithForegroundColor ' -color $sl.Colors.WithForegroundColor Write-Host -Object '' } function Write-ColorPreview { param ( [string] $text, [ConsoleColor] $color ) Write-Host -Object $text -NoNewline Write-Host -Object ' ' -BackgroundColor $color } function Show-Colors { for($i = 0; $i -lt 16; $i++) { $color = [ConsoleColor]$i Write-Host -Object $color -BackgroundColor $i } } function Set-Theme { param( [Parameter(Mandatory=$true)] [string] $name ) if (Test-Path "$($sl.MyThemesLocation)\$($name).psm1") { $sl.CurrentThemeLocation = "$($sl.MyThemesLocation)\$($name).psm1" } elseif (Test-Path "$PSScriptRoot\Themes\$($name).psm1") { $sl.CurrentThemeLocation = "$PSScriptRoot\Themes\$($name).psm1" } else { Write-Host '' Write-Host "Theme $name not found. Available themes are:" Show-Themes } Set-Prompt } # Helper function to create argument completion results function New-CompletionResult { param( [Parameter(Mandatory)] [string]$CompletionText, [string]$ListItemText = $CompletionText, [System.Management.Automation.CompletionResultType]$CompletionResultType = [System.Management.Automation.CompletionResultType]::ParameterValue, [string]$ToolTip = $CompletionText ) New-Object System.Management.Automation.CompletionResult $CompletionText, $ListItemText, $CompletionResultType, $ToolTip } function ThemeCompletion { param( $commandName, $parameterName, $wordToComplete, $commandAst, $fakeBoundParameter ) $themes = @() Get-ChildItem -Path "$($ThemeSettings.MyThemesLocation)\*" -Include '*.psm1' -Exclude Tools.ps1 | ForEach-Object -Process { $themes += $_.BaseName } Get-ChildItem -Path "$PSScriptRoot\Themes\*" -Include '*.psm1' -Exclude Tools.ps1 | Sort-Object Name | ForEach-Object -Process { $themes += $_.BaseName } $themes | Where-Object {$_.ToLower().StartsWith($wordToComplete)} | ForEach-Object { New-CompletionResult -CompletionText $_ } } Register-ArgumentCompleter ` -CommandName Set-Theme ` -ParameterName name ` -ScriptBlock $function:ThemeCompletion $sl = $global:ThemeSettings #local settings $sl.ErrorCount = $global:error.Count Start-Up # Executes the Start-Up function, better encapsulation |