class RDoc::Markup::ToAnsi

# File lib/rdoc/markup/to_ansi.rb, line 10
def initialize(markup = nil)
  super

  @headings.clear
  @headings[1] = ["\e[1;32m", "\e[m"] # bold
  @headings[2] = ["\e[4;32m", "\e[m"] # underline
  @headings[3] = ["\e[32m",   "\e[m"] # just green
end

# File lib/rdoc/markup/to_ansi.rb, line 69
def accept_list_item_end(list_item)
  width = case @list_type.last
          when :BULLET then
            2
          when :NOTE, :LABEL then
            if @prefix then
              @res << @prefix.strip
              @prefix = nil
            end

            @res << "\n" unless res.length == 1
            2
          else
            bullet = @list_index.last.to_s
            @list_index[-1] = @list_index.last.succ
            bullet.length + 2
          end

  @indent -= width
end

# File lib/rdoc/markup/to_ansi.rb, line 93
def accept_list_item_start(list_item)
  bullet = case @list_type.last
           when :BULLET then
             '*'
           when :NOTE, :LABEL then
             labels = Array(list_item.label).map do |label|
               attributes(label).strip
             end.join "\n"

             labels << ":\n" unless labels.empty?

             labels
           else
             @list_index.last.to_s + '.'
           end

  case @list_type.last
  when :NOTE, :LABEL then
    @indent += 2
    @prefix = bullet + (' ' * @indent)
  else
    @prefix = (' ' * @indent) + bullet.ljust(bullet.length + 1)

    width = bullet.gsub(/\e\[[\d;]*m/, '').length + 1

    @indent += width
  end
end

# File lib/rdoc/markup/to_ansi.rb, line 50
def add_text(text)
  attrs = @attributes.keys
  if @applied_attributes != attrs
    apply_attributes(attrs)
  end
  emit_inline(text)
end

# File lib/rdoc/markup/to_ansi.rb, line 34
def apply_attributes(attributes)
  before, after = [@applied_attributes, attributes].map do |attrs|
    attrs.map { |attr| ANSI_STYLE_CODES[attr] }.compact.sort
  end
  return if before == after

  if after.empty?
    emit_inline("\e[m")
  elsif (before - after).empty?
    emit_inline("\e[#{(after - before).join(';')}m")
  else
    emit_inline("\e[#{[0, after].join(';')}m")
  end
  @applied_attributes = attributes
end

# File lib/rdoc/markup/to_ansi.rb, line 122
def calculate_text_width(text)
  text.gsub(/\e\[[\d;]*m/, '').size
end

# File lib/rdoc/markup/to_ansi.rb, line 58
def handle_inline(text)
  @applied_attributes = []
  res = super
  res << "\e[m" unless @applied_attributes.empty?
  @applied_attributes = []
  res
end

# File lib/rdoc/markup/to_ansi.rb, line 129
def start_accepting
  super

  @res = ["\e[0m"]
end