PwshSyntaxHighlight/public/PwshSyntaxHighlight.ps1
|
$script:Themes = @{ Github = @{ Function = @{ R = 255; G = 123; B = 114 } Generic = @{ R = 199; G = 159; B = 252 } String = @{ R = 143; G = 185; B = 221 } Variable = @{ R = 255; G = 255; B = 255 } Identifier = @{ R = 110; G = 174; B = 231 } Number = @{ R = 255; G = 255; B = 255 } Keyword = @{ R = 255; G = 123; B = 114 } Default = @{ R = 200; G = 200; B = 200 } ForegroundRgb = @{ R = 102; G = 102; B = 102 } BackgroundRgb = @{ R = 35; G = 35; B = 35 } HighlightRgb = @{ R = 231; G = 72; B = 86 } } Matrix = @{ Function = @{ R = 255; G = 255; B = 255 } Generic = @{ R = 113; G = 255; B = 96 } String = @{ R = 202; G = 255; B = 194 } Variable = @{ R = 200; G = 255; B = 200 } Identifier = @{ R = 131; G = 193; B = 26 } Number = @{ R = 255; G = 255; B = 255 } Keyword = @{ R = 40; G = 220; B = 20 } Default = @{ R = 0; G = 120; B = 0 } ForegroundRgb = @{ R = 102; G = 190; B = 102 } BackgroundRgb = @{ R = 15; G = 45; B = 15 } HighlightRgb = @{ R = 255; G = 221; B = 0 } } } function Write-Codeblock { <# .SYNOPSIS Writes a code block to the host. Intended for internal use only when you want to show a code block with some nicer formatting. .DESCRIPTION The Write-Codeblock function outputs a code block to the host console with optional line numbers, syntax highlighting, and line or extent highlighting. The function also supports custom foreground and background colors. .NOTES Author: Shaun Lawrie This was originally going to be using a screenbuffer but I wanted to support really long functions that may scroll the terminal so this streams the lines out from top to bottom which isn't the fastest way to render but it was the most reliable way I found to avoid mangling the code as it was being written out. #> param ( # The text containing the code to write to the host [Parameter(ValueFromPipeline=$true, Mandatory=$true)] [string] $Text, # Show a gutter with line numbers [switch] $ShowLineNumbers, # Syntax highlight the code block [switch] $SyntaxHighlight, # Extents to highlight in the code block [array] $HighlightExtents, # Lines to highlight in the code block [array] $HighlightLines, # The theme to use to render the code [ValidateSet("Github", "Matrix")] [string] $Theme = "Github" ) $ForegroundRgb = $script:Themes[$Theme].ForegroundRgb $BackgroundRgb = $script:Themes[$Theme].BackgroundRgb # Work out the width of the console minus the line-number gutter $gutterSize = 0 if($ShowLineNumbers) { $gutterSize = $Text.Split("`n").Count.ToString().Length + 1 } $codeWidth = $Host.UI.RawUI.WindowSize.Width - $gutterSize try { Set-CursorVisible $false $functionLineNumber = 1 $resetEscapeCode = "$([Char]27)[0m" $foregroundColorEscapeCode = "$([Char]27)[38;2;{0};{1};{2}m" -f $ForegroundRgb.R, $ForegroundRgb.G, $ForegroundRgb.B $backgroundColorEscapeCode = "$([Char]27)[48;2;{0};{1};{2}m" -f $BackgroundRgb.R, $BackgroundRgb.G, $BackgroundRgb.B # Get all code tokens $tokens = @() [System.Management.Automation.Language.Parser]::ParseInput($Text, [ref]$tokens, [ref]$null) | Out-Null $lineTokens = Expand-Tokens -Tokens $tokens | Where-Object { -not [string]::IsNullOrWhiteSpace($_.Text) } | Group-Object { $_.Extent.StartLineNumber } $lineExtents = $HighlightExtents | Group-Object { $_.StartLineNumber } $functionLinesToRender = $Text.Split("`n") foreach($line in $functionLinesToRender) { $gutterText = "" if($ShowLineNumbers) { $gutterText = $functionLineNumber.ToString().PadLeft($gutterSize - 1) + " " } # Disable syntax highlighting for specifically highlighted lines $lineSyntax = $SyntaxHighlight $lineHighlight = $false if($HighlightLines -contains $functionLineNumber) { $lineSyntax = $false $lineHighlight = $true } # Work out the lines that will be wrapped in the terminal because they're too long and draw the background $lineBackground = $foregroundColorEscapeCode + $gutterText + $backgroundColorEscapeCode + (" " * $codeWidth) + $resetEscapeCode if($line.Length -gt $codeWidth) { # How many times can this line be wrapped in the code editor width available $wrappedLineSegments = ($line | Select-String -Pattern ".{1,$codeWidth}" -AllMatches).Matches.Value # Render the background line plus additional background lines without the gutter line number for each wrapped line $wrappedLinesBackground = (" " * $gutterSize) + $backgroundColorEscapeCode + (" " * $codeWidth) + $resetEscapeCode [Console]::WriteLine($lineBackground + ($wrappedLinesBackground * ($wrappedLineSegments.Count - 1))) # Correct terminal line position if the window scrolled $terminalLine = $Host.UI.RawUI.CursorPosition.Y - $wrappedLineSegments.Count } else { # Render the background [Console]::WriteLine($lineBackground) $terminalLine = $Host.UI.RawUI.CursorPosition.Y - 1 } # Render the tokens that are on this line ($lineTokens | Where-Object { $_.Name -eq $functionLineNumber }).Group | ForEach-Object { if($null -ne $_) { Write-Token -Token $_ -TerminalLine $terminalLine -BackgroundRgb $BackgroundRgb -GutterSize $gutterSize -Highlight:$lineHighlight -Theme $Theme -SyntaxHighlight:$lineSyntax } } # Highlight all extents on this line that have been requested to be emphasized ($lineExtents | Where-Object { $_.Name -eq $functionLineNumber }).Group | Foreach-Object { if($null -ne $_) { Write-Token -Extent $_ -TerminalLine $terminalLine -BackgroundRgb $BackgroundRgb -GutterSize $gutterSize -Highlight -Theme $Theme } } $functionLineNumber++ } } catch { throw $_ } finally { Set-CursorVisible } } |