display_code_with_line_numbers.rb

# frozen_string_literal: true

module SyntaxSuggest
  # Outputs code with highlighted lines
  #
  # Whatever is passed to this class will be rendered
  # even if it is "marked invisible" any filtering of
  # output should be done before calling this class.
  #
  #   DisplayCodeWithLineNumbers.new(
  #     lines: lines,
  #     highlight_lines: [lines[2], lines[3]]
  #   ).call
  #   # =>
  #       1
  #       2  def cat
  #     > 3    Dir.chdir
  #     > 4    end
  #       5  end
  #       6
  class DisplayCodeWithLineNumbers
    TERMINAL_HIGHLIGHT = "\e[1;3m" # Bold, italics
    TERMINAL_END = "\e[0m"

def initialize(lines:, highlight_lines: [], terminal: false)
      @lines = Array(lines).sort
      @terminal = terminal
      @highlight_line_hash = Array(highlight_lines).each_with_object({}) { |line, h| h[line] = true }
      @digit_count = @lines.last&.line_number.to_s.length
    end

def call
      @lines.map do |line|
        format_line(line)
      end.join
    end

private def format_line(code_line)
      # Handle trailing slash lines
      code_line.original.lines.map.with_index do |contents, i|
        format(
          empty: code_line.empty?,
          number: (code_line.number + i).to_s,
          contents: contents,
          highlight: @highlight_line_hash[code_line]
        )
      end.join
    end

private def format(contents:, number:, empty:, highlight: false)
      string = +""
      string << if highlight
        "> "
      else
        "  "
      end

string << number.rjust(@digit_count).to_s
      if empty
        string << contents
      else
        string << "  "
        string << TERMINAL_HIGHLIGHT if @terminal && highlight
        string << contents
        string << TERMINAL_END if @terminal
      end
      string
    end
  end
end

docroot: /home/induvedam/public_html

curdir: /opt/alt/ruby34/share/ruby/syntax_suggest

	.	4096
	..	4096
X	api.rb	6219
X	around_block_scan.rb	7069
X	block_expand.rb	4992
	capture	73
X	capture_code_context.rb	6826
X	clean_document.rb	8943
X	cli.rb	3190
X	code_block.rb	2183
X	code_frontier.rb	5748
X	code_line.rb	6739
X	code_search.rb	3905
X	core_ext.rb	2709
X	display_code_with_line_numbers.rb	1812
X	display_invalid_blocks.rb	1797
X	explain_syntax.rb	2756
X	left_right_lex_count.rb	4156
X	lex_all.rb	1704
X	lex_value.rb	1533
X	mini_stringio.rb	774
X	parse_blocks_from_indent_line.rb	1537
X	pathname_from_message.rb	1456
X	priority_engulf_queue.rb	1295
X	priority_queue.rb	2046
X	ripper_errors.rb	960
X	scan_history.rb	2989
X	unvisited_lines.rb	705
X	version.rb	76