104 lines
2.5 KiB
PowerShell
104 lines
2.5 KiB
PowerShell
$script:bg = [Console]::BackgroundColor;
|
|
$script:fg = [System.ConsoleColor]::White;
|
|
$script:first = $true;
|
|
$script:last = 0;
|
|
|
|
$global:PromptReplacementDirs = @{
|
|
$env:USERPROFILE = '~';
|
|
}
|
|
|
|
function Write-PromptFancyEnd {
|
|
Write-Host -NoNewline -ForegroundColor $script:fg
|
|
|
|
$script:bg = [System.ConsoleColor]::Black
|
|
$script:fg = [System.ConsoleColor]::White
|
|
}
|
|
|
|
function Write-PromptSegment {
|
|
param(
|
|
[Parameter(
|
|
Position=0,
|
|
Mandatory=$true,
|
|
ValueFromPipeline=$true,
|
|
ValueFromPipelineByPropertyName=$true
|
|
)][string]$Text,
|
|
|
|
[Parameter(Position=1)][System.ConsoleColor] $Foreground = [System.ConsoleColor]::White,
|
|
[Parameter(Position=2)][System.ConsoleColor] $Background = [Console]::BackgroundColor,
|
|
[switch] $NoSegment
|
|
)
|
|
|
|
Write-Host $text -NoNewline -BackgroundColor $Background -ForegroundColor $Foreground
|
|
|
|
if (-not $NoSegment) {
|
|
Write-Host -NoNewline -BackgroundColor $Background -ForegroundColor $Foreground
|
|
}
|
|
|
|
$script:bg = $Background;
|
|
$script:fg = $Foreground;
|
|
}
|
|
|
|
function Get-FancyDir {
|
|
$location = $(Get-Location).ToString();
|
|
foreach($replacement in $global:PromptReplacementDirs.GetEnumerator()) {
|
|
$location = $location.Replace($replacement.Name, $replacement.Value);
|
|
}
|
|
|
|
return $location.Replace('\', ' ');
|
|
}
|
|
|
|
function Get-GitBranch {
|
|
$HEAD = Get-Content $(Join-Path $(Get-GitDirectory) HEAD)
|
|
if($HEAD -like 'ref: refs/heads/*') {
|
|
return $HEAD -replace 'ref: refs/heads/(.*?)', "$1";
|
|
} else {
|
|
return $HEAD.Substring(0, 8);
|
|
}
|
|
}
|
|
|
|
function Write-PromptStatus {
|
|
if($script:last) {
|
|
Write-PromptSegment ' ✔ ' Green
|
|
} else {
|
|
Write-PromptSegment " ✖ $lastexitcode " Red
|
|
}
|
|
}
|
|
|
|
function Write-PromptUser {
|
|
if($global:admin) {
|
|
Write-PromptSegment ' ADMIN ' Red
|
|
}
|
|
}
|
|
|
|
function Write-PromptVirtualEnv {
|
|
if($env:VIRTUAL_ENV) {
|
|
Write-PromptSegment " $(split-path $env:VIRTUAL_ENV -leaf) " Cyan
|
|
}
|
|
}
|
|
|
|
function Write-PromptDir {
|
|
Write-PromptSegment " $(Get-FancyDir) " DarkGray -NoSegment
|
|
}
|
|
|
|
# Depends on posh-git
|
|
function Write-PromptGit {
|
|
if(Get-GitDirectory) {
|
|
Write-PromptSegment " $(Get-GitBranch) " Blue
|
|
}
|
|
}
|
|
|
|
function prompt {
|
|
$script:last = $?;
|
|
$script:first = $true;
|
|
|
|
Write-PromptStatus
|
|
Write-PromptUser
|
|
Write-PromptVirtualEnv
|
|
Write-PromptGit
|
|
Write-PromptDir
|
|
|
|
Write-PromptFancyEnd
|
|
|
|
return ' '
|
|
}
|