523 lines
12 KiB
Ruby
523 lines
12 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
# ANSI color library
|
|
#
|
|
# Implementation per http://en.wikipedia.org/wiki/ANSI_escape_code
|
|
module Gitlab
|
|
module Ci
|
|
module Ansi2html
|
|
# keys represent the trailing digit in color changing command (30-37, 40-47, 90-97. 100-107)
|
|
COLOR = {
|
|
0 => 'black', # not that this is gray in the intense color table
|
|
1 => 'red',
|
|
2 => 'green',
|
|
3 => 'yellow',
|
|
4 => 'blue',
|
|
5 => 'magenta',
|
|
6 => 'cyan',
|
|
7 => 'white' # not that this is gray in the dark (aka default) color table
|
|
}.freeze
|
|
|
|
STYLE_SWITCHES = {
|
|
bold: 0x01,
|
|
italic: 0x02,
|
|
underline: 0x04,
|
|
conceal: 0x08,
|
|
cross: 0x10
|
|
}.freeze
|
|
|
|
def self.convert(ansi, state = nil)
|
|
Converter.new.convert(ansi, state)
|
|
end
|
|
|
|
class Converter
|
|
def on_0(_)
|
|
reset
|
|
end
|
|
|
|
def on_1(_)
|
|
enable(STYLE_SWITCHES[:bold])
|
|
end
|
|
|
|
def on_3(_)
|
|
enable(STYLE_SWITCHES[:italic])
|
|
end
|
|
|
|
def on_4(_)
|
|
enable(STYLE_SWITCHES[:underline])
|
|
end
|
|
|
|
def on_8(_)
|
|
enable(STYLE_SWITCHES[:conceal])
|
|
end
|
|
|
|
def on_9(_)
|
|
enable(STYLE_SWITCHES[:cross])
|
|
end
|
|
|
|
def on_21(_)
|
|
disable(STYLE_SWITCHES[:bold])
|
|
end
|
|
|
|
def on_22(_)
|
|
disable(STYLE_SWITCHES[:bold])
|
|
end
|
|
|
|
def on_23(_)
|
|
disable(STYLE_SWITCHES[:italic])
|
|
end
|
|
|
|
def on_24(_)
|
|
disable(STYLE_SWITCHES[:underline])
|
|
end
|
|
|
|
def on_28(_)
|
|
disable(STYLE_SWITCHES[:conceal])
|
|
end
|
|
|
|
def on_29(_)
|
|
disable(STYLE_SWITCHES[:cross])
|
|
end
|
|
|
|
def on_30(_)
|
|
set_fg_color(0)
|
|
end
|
|
|
|
def on_31(_)
|
|
set_fg_color(1)
|
|
end
|
|
|
|
def on_32(_)
|
|
set_fg_color(2)
|
|
end
|
|
|
|
def on_33(_)
|
|
set_fg_color(3)
|
|
end
|
|
|
|
def on_34(_)
|
|
set_fg_color(4)
|
|
end
|
|
|
|
def on_35(_)
|
|
set_fg_color(5)
|
|
end
|
|
|
|
def on_36(_)
|
|
set_fg_color(6)
|
|
end
|
|
|
|
def on_37(_)
|
|
set_fg_color(7)
|
|
end
|
|
|
|
def on_38(stack)
|
|
set_fg_color_256(stack)
|
|
end
|
|
|
|
def on_39(_)
|
|
set_fg_color(9)
|
|
end
|
|
|
|
def on_40(_)
|
|
set_bg_color(0)
|
|
end
|
|
|
|
def on_41(_)
|
|
set_bg_color(1)
|
|
end
|
|
|
|
def on_42(_)
|
|
set_bg_color(2)
|
|
end
|
|
|
|
def on_43(_)
|
|
set_bg_color(3)
|
|
end
|
|
|
|
def on_44(_)
|
|
set_bg_color(4)
|
|
end
|
|
|
|
def on_45(_)
|
|
set_bg_color(5)
|
|
end
|
|
|
|
def on_46(_)
|
|
set_bg_color(6)
|
|
end
|
|
|
|
def on_47(_)
|
|
set_bg_color(7)
|
|
end
|
|
|
|
def on_48(stack)
|
|
set_bg_color_256(stack)
|
|
end
|
|
|
|
def on_49(_)
|
|
set_bg_color(9)
|
|
end
|
|
|
|
def on_90(_)
|
|
set_fg_color(0, 'l')
|
|
end
|
|
|
|
def on_91(_)
|
|
set_fg_color(1, 'l')
|
|
end
|
|
|
|
def on_92(_)
|
|
set_fg_color(2, 'l')
|
|
end
|
|
|
|
def on_93(_)
|
|
set_fg_color(3, 'l')
|
|
end
|
|
|
|
def on_94(_)
|
|
set_fg_color(4, 'l')
|
|
end
|
|
|
|
def on_95(_)
|
|
set_fg_color(5, 'l')
|
|
end
|
|
|
|
def on_96(_)
|
|
set_fg_color(6, 'l')
|
|
end
|
|
|
|
def on_97(_)
|
|
set_fg_color(7, 'l')
|
|
end
|
|
|
|
def on_99(_)
|
|
set_fg_color(9, 'l')
|
|
end
|
|
|
|
def on_100(_)
|
|
set_bg_color(0, 'l')
|
|
end
|
|
|
|
def on_101(_)
|
|
set_bg_color(1, 'l')
|
|
end
|
|
|
|
def on_102(_)
|
|
set_bg_color(2, 'l')
|
|
end
|
|
|
|
def on_103(_)
|
|
set_bg_color(3, 'l')
|
|
end
|
|
|
|
def on_104(_)
|
|
set_bg_color(4, 'l')
|
|
end
|
|
|
|
def on_105(_)
|
|
set_bg_color(5, 'l')
|
|
end
|
|
|
|
def on_106(_)
|
|
set_bg_color(6, 'l')
|
|
end
|
|
|
|
def on_107(_)
|
|
set_bg_color(7, 'l')
|
|
end
|
|
|
|
def on_109(_)
|
|
set_bg_color(9, 'l')
|
|
end
|
|
|
|
attr_accessor :offset, :n_open_tags, :fg_color, :bg_color, :style_mask, :sections, :lineno_in_section
|
|
|
|
STATE_PARAMS = [:offset, :n_open_tags, :fg_color, :bg_color, :style_mask, :sections, :lineno_in_section].freeze
|
|
|
|
def convert(stream, new_state)
|
|
reset_state
|
|
restore_state(new_state, stream) if new_state.present?
|
|
|
|
append = false
|
|
truncated = false
|
|
|
|
cur_offset = stream.tell
|
|
if cur_offset > @offset
|
|
@offset = cur_offset
|
|
truncated = true
|
|
else
|
|
stream.seek(@offset)
|
|
append = @offset > 0
|
|
end
|
|
|
|
start_offset = @offset
|
|
|
|
stream.each_line do |line|
|
|
s = StringScanner.new(line)
|
|
|
|
until s.eos?
|
|
|
|
if s.scan(Gitlab::Regex.build_trace_section_regex)
|
|
handle_section(s)
|
|
elsif s.scan(/\e([@-_])(.*?)([@-~])/)
|
|
handle_sequence(s)
|
|
elsif s.scan(/\e(([@-_])(.*?)?)?$/)
|
|
break
|
|
elsif s.scan(/</)
|
|
write_in_tag '<'
|
|
elsif s.scan(/\r?\n/)
|
|
handle_new_line
|
|
else
|
|
write_in_tag s.scan(/./m)
|
|
end
|
|
|
|
@offset += s.matched_size
|
|
end
|
|
end
|
|
|
|
close_open_tags
|
|
|
|
# TODO: replace OpenStruct with a better type
|
|
# https://gitlab.com/gitlab-org/gitlab/issues/34305
|
|
OpenStruct.new(
|
|
html: @out.force_encoding(Encoding.default_external),
|
|
state: state,
|
|
append: append,
|
|
truncated: truncated,
|
|
offset: start_offset,
|
|
size: stream.tell - start_offset,
|
|
total: stream.size
|
|
)
|
|
end
|
|
|
|
def section_to_class_name(section)
|
|
section.to_s.downcase.gsub(/[^a-z0-9]/, '-')
|
|
end
|
|
|
|
def handle_new_line
|
|
write_in_tag %{<br/>}
|
|
|
|
close_open_tags if @sections.any? && @lineno_in_section == 0
|
|
@lineno_in_section += 1
|
|
end
|
|
|
|
def handle_section(scanner)
|
|
action = scanner[1]
|
|
timestamp = scanner[2]
|
|
section = scanner[3]
|
|
|
|
normalized_section = section_to_class_name(section)
|
|
|
|
if action == "start"
|
|
handle_section_start(normalized_section, timestamp)
|
|
elsif action == "end"
|
|
handle_section_end(normalized_section, timestamp)
|
|
end
|
|
end
|
|
|
|
def handle_section_start(section, timestamp)
|
|
return if @sections.include?(section)
|
|
|
|
@sections << section
|
|
write_raw %{<div class="section-start" data-timestamp="#{timestamp}" data-section="#{data_section_names}" role="button"></div>}
|
|
@lineno_in_section = 0
|
|
end
|
|
|
|
def handle_section_end(section, timestamp)
|
|
return unless @sections.include?(section)
|
|
|
|
# close all sections up to section
|
|
until @sections.empty?
|
|
write_raw %{<div class="section-end" data-section="#{data_section_names}"></div>}
|
|
|
|
last_section = @sections.pop
|
|
break if section == last_section
|
|
end
|
|
end
|
|
|
|
def data_section_names
|
|
@sections.join(" ")
|
|
end
|
|
|
|
def handle_sequence(scanner)
|
|
indicator = scanner[1]
|
|
commands = scanner[2].split ';'
|
|
terminator = scanner[3]
|
|
|
|
# We are only interested in color and text style changes - triggered by
|
|
# sequences starting with '\e[' and ending with 'm'. Any other control
|
|
# sequence gets stripped (including stuff like "delete last line")
|
|
return unless indicator == '[' && terminator == 'm'
|
|
|
|
close_open_tags
|
|
|
|
if commands.empty?
|
|
reset
|
|
return
|
|
end
|
|
|
|
evaluate_command_stack(commands)
|
|
end
|
|
|
|
def evaluate_command_stack(stack)
|
|
return unless command = stack.shift
|
|
|
|
if self.respond_to?("on_#{command}", true)
|
|
self.__send__("on_#{command}", stack) # rubocop:disable GitlabSecurity/PublicSend
|
|
end
|
|
|
|
evaluate_command_stack(stack)
|
|
end
|
|
|
|
def write_in_tag(data)
|
|
ensure_open_new_tag
|
|
@out << data
|
|
end
|
|
|
|
def write_raw(data)
|
|
close_open_tags
|
|
@out << data
|
|
end
|
|
|
|
def ensure_open_new_tag
|
|
open_new_tag if @n_open_tags == 0
|
|
end
|
|
|
|
def open_new_tag
|
|
css_classes = []
|
|
|
|
unless @fg_color.nil?
|
|
fg_color = @fg_color
|
|
# Most terminals show bold colored text in the light color variant
|
|
# Let's mimic that here
|
|
if @style_mask & STYLE_SWITCHES[:bold] != 0
|
|
fg_color.sub!(/fg-([a-z]{2,}+)/, 'fg-l-\1')
|
|
end
|
|
|
|
css_classes << fg_color
|
|
end
|
|
|
|
css_classes << @bg_color unless @bg_color.nil?
|
|
|
|
STYLE_SWITCHES.each do |css_class, flag|
|
|
css_classes << "term-#{css_class}" if @style_mask & flag != 0
|
|
end
|
|
|
|
if @sections.any?
|
|
css_classes << "section"
|
|
|
|
css_classes << if @lineno_in_section == 0
|
|
"section-header"
|
|
else
|
|
"line"
|
|
end
|
|
|
|
css_classes += sections.map { |section| "js-s-#{section}" }
|
|
end
|
|
|
|
close_open_tags
|
|
|
|
@out << if css_classes.any?
|
|
%{<span class="#{css_classes.join(' ')}">}
|
|
else
|
|
%{<span>}
|
|
end
|
|
|
|
@n_open_tags += 1
|
|
end
|
|
|
|
def close_open_tags
|
|
while @n_open_tags > 0
|
|
@out << %{</span>}
|
|
@n_open_tags -= 1
|
|
end
|
|
end
|
|
|
|
def reset_state
|
|
@offset = 0
|
|
@n_open_tags = 0
|
|
@out = +''
|
|
@sections = []
|
|
@lineno_in_section = 0
|
|
reset
|
|
end
|
|
|
|
def state
|
|
state = STATE_PARAMS.inject({}) do |h, param|
|
|
h[param] = send(param) # rubocop:disable GitlabSecurity/PublicSend
|
|
h
|
|
end
|
|
Base64.urlsafe_encode64(state.to_json)
|
|
end
|
|
|
|
def restore_state(new_state, stream)
|
|
state = Base64.urlsafe_decode64(new_state)
|
|
state = Gitlab::Json.parse(state, symbolize_names: true)
|
|
return if state[:offset].to_i > stream.size
|
|
|
|
STATE_PARAMS.each do |param|
|
|
send("#{param}=".to_sym, state[param]) # rubocop:disable GitlabSecurity/PublicSend
|
|
end
|
|
end
|
|
|
|
def reset
|
|
@fg_color = nil
|
|
@bg_color = nil
|
|
@style_mask = 0
|
|
end
|
|
|
|
def enable(flag)
|
|
@style_mask |= flag
|
|
end
|
|
|
|
def disable(flag)
|
|
@style_mask &= ~flag
|
|
end
|
|
|
|
def set_fg_color(color_index, prefix = nil)
|
|
@fg_color = get_term_color_class(color_index, ["fg", prefix])
|
|
end
|
|
|
|
def set_bg_color(color_index, prefix = nil)
|
|
@bg_color = get_term_color_class(color_index, ["bg", prefix])
|
|
end
|
|
|
|
def get_term_color_class(color_index, prefix)
|
|
color_name = COLOR[color_index]
|
|
return if color_name.nil?
|
|
|
|
get_color_class(["term", prefix, color_name])
|
|
end
|
|
|
|
def set_fg_color_256(command_stack)
|
|
css_class = get_xterm_color_class(command_stack, "fg")
|
|
@fg_color = css_class unless css_class.nil?
|
|
end
|
|
|
|
def set_bg_color_256(command_stack)
|
|
css_class = get_xterm_color_class(command_stack, "bg")
|
|
@bg_color = css_class unless css_class.nil?
|
|
end
|
|
|
|
def get_xterm_color_class(command_stack, prefix)
|
|
# the 38 and 48 commands have to be followed by "5" and the color index
|
|
return unless command_stack.length >= 2
|
|
return unless command_stack[0] == "5"
|
|
|
|
command_stack.shift # ignore the "5" command
|
|
color_index = command_stack.shift.to_i
|
|
|
|
return unless color_index >= 0
|
|
return unless color_index <= 255
|
|
|
|
get_color_class(["xterm", prefix, color_index])
|
|
end
|
|
|
|
def get_color_class(segments)
|
|
[segments].flatten.compact.join('-')
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|