# ANSI color library # # Implementation per http://en.wikipedia.org/wiki/ANSI_escape_code 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(s) reset() end def on_1(s) enable(STYLE_SWITCHES[:bold]) end def on_3(s) enable(STYLE_SWITCHES[:italic]) end def on_4(s) enable(STYLE_SWITCHES[:underline]) end def on_8(s) enable(STYLE_SWITCHES[:conceal]) end def on_9(s) enable(STYLE_SWITCHES[:cross]) end def on_21(s) disable(STYLE_SWITCHES[:bold]) end def on_22(s) disable(STYLE_SWITCHES[:bold]) end def on_23(s) disable(STYLE_SWITCHES[:italic]) end def on_24(s) disable(STYLE_SWITCHES[:underline]) end def on_28(s) disable(STYLE_SWITCHES[:conceal]) end def on_29(s) disable(STYLE_SWITCHES[:cross]) end def on_30(s) set_fg_color(0) end def on_31(s) set_fg_color(1) end def on_32(s) set_fg_color(2) end def on_33(s) set_fg_color(3) end def on_34(s) set_fg_color(4) end def on_35(s) set_fg_color(5) end def on_36(s) set_fg_color(6) end def on_37(s) set_fg_color(7) end def on_38(s) set_fg_color_256(s) end def on_39(s) set_fg_color(9) end def on_40(s) set_bg_color(0) end def on_41(s) set_bg_color(1) end def on_42(s) set_bg_color(2) end def on_43(s) set_bg_color(3) end def on_44(s) set_bg_color(4) end def on_45(s) set_bg_color(5) end def on_46(s) set_bg_color(6) end def on_47(s) set_bg_color(7) end def on_48(s) set_bg_color_256(s) end def on_49(s) set_bg_color(9) end def on_90(s) set_fg_color(0, 'l') end def on_91(s) set_fg_color(1, 'l') end def on_92(s) set_fg_color(2, 'l') end def on_93(s) set_fg_color(3, 'l') end def on_94(s) set_fg_color(4, 'l') end def on_95(s) set_fg_color(5, 'l') end def on_96(s) set_fg_color(6, 'l') end def on_97(s) set_fg_color(7, 'l') end def on_99(s) set_fg_color(9, 'l') end def on_100(s) set_bg_color(0, 'l') end def on_101(s) set_bg_color(1, 'l') end def on_102(s) set_bg_color(2, 'l') end def on_103(s) set_bg_color(3, 'l') end def on_104(s) set_bg_color(4, 'l') end def on_105(s) set_bg_color(5, 'l') end def on_106(s) set_bg_color(6, 'l') end def on_107(s) set_bg_color(7, 'l') end def on_109(s) set_bg_color(9, 'l') end attr_accessor :offset, :n_open_tags, :fg_color, :bg_color, :style_mask STATE_PARAMS = [:offset, :n_open_tags, :fg_color, :bg_color, :style_mask].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 open_new_tag stream.each_line do |line| s = StringScanner.new(line) until s.eos? if s.scan(/\e([@-_])(.*?)([@-~])/) handle_sequence(s) elsif s.scan(/\e(([@-_])(.*?)?)?$/) break elsif s.scan(/</) @out << '<' elsif s.scan(/\r?\n/) @out << '<br>' else @out << s.scan(/./m) end @offset += s.matched_size end end close_open_tags() 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 handle_sequence(s) indicator = s[1] commands = s[2].split ';' terminator = s[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) open_new_tag end def evaluate_command_stack(stack) return unless command = stack.shift() if self.respond_to?("on_#{command}", true) self.send("on_#{command}", stack) end evaluate_command_stack(stack) 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-(\w{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 return if css_classes.empty? @out << %{<span class="#{css_classes.join(' ')}">} @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 = '' reset end def state state = STATE_PARAMS.inject({}) do |h, param| h[param] = send(param) h end Base64.urlsafe_encode64(state.to_json) end def restore_state(new_state, stream) state = Base64.urlsafe_decode64(new_state) state = 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]) 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 nil 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