1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71
|
module RSpec
module Core
class Source
# @private
# Provides terminal syntax highlighting of code snippets
# when coderay is available.
class SyntaxHighlighter
def initialize(configuration)
@configuration = configuration
end
def highlight(lines)
implementation.highlight_syntax(lines)
end
private
if RSpec::Support::OS.windows?
# :nocov:
def implementation
WindowsImplementation
end
# :nocov:
else
def implementation
return color_enabled_implementation if @configuration.color_enabled?
NoSyntaxHighlightingImplementation
end
end
def color_enabled_implementation
@color_enabled_implementation ||= begin
require 'coderay'
CodeRayImplementation
rescue LoadError
NoSyntaxHighlightingImplementation
end
end
# @private
module CodeRayImplementation
RESET_CODE = "\e[0m"
def self.highlight_syntax(lines)
highlighted = begin
CodeRay.encode(lines.join("\n"), :ruby, :terminal)
rescue Support::AllExceptionsExceptOnesWeMustNotRescue
return lines
end
highlighted.split("\n").map do |line|
line.sub(/\S/) { |char| char.insert(0, RESET_CODE) }
end
end
end
# @private
module NoSyntaxHighlightingImplementation
def self.highlight_syntax(lines)
lines
end
end
# @private
# Not sure why, but our code above (and/or coderay itself) does not work
# on Windows, so we disable the feature on Windows.
WindowsImplementation = NoSyntaxHighlightingImplementation
end
end
end
end
|