numerization.rb
133 lines
| 4.1 KiB
| text/x-ruby
|
RubyLexer
Jean-Philippe Lang
|
r638 | module CodeRay | ||
| module Encoders | ||||
| class HTML | ||||
| module Output | ||||
| def numerize *args | ||||
| clone.numerize!(*args) | ||||
| end | ||||
| =begin NUMERIZABLE_WRAPPINGS = { | ||||
| :table => [:div, :page, nil], | ||||
| :inline => :all, | ||||
| :list => [:div, :page, nil] | ||||
| } | ||||
| NUMERIZABLE_WRAPPINGS.default = :all | ||||
| =end | ||||
| def numerize! mode = :table, options = {} | ||||
| return self unless mode | ||||
| options = DEFAULT_OPTIONS.merge options | ||||
| start = options[:line_number_start] | ||||
| unless start.is_a? Integer | ||||
| raise ArgumentError, "Invalid value %p for :line_number_start; Integer expected." % start | ||||
| end | ||||
| #allowed_wrappings = NUMERIZABLE_WRAPPINGS[mode] | ||||
| #unless allowed_wrappings == :all or allowed_wrappings.include? options[:wrap] | ||||
| # raise ArgumentError, "Can't numerize, :wrap must be in %p, but is %p" % [NUMERIZABLE_WRAPPINGS, options[:wrap]] | ||||
| #end | ||||
| bold_every = options[:bold_every] | ||||
|
Jean-Philippe Lang
|
r3478 | highlight_lines = options[:highlight_lines] | ||
|
Jean-Philippe Lang
|
r638 | bolding = | ||
|
Jean-Philippe Lang
|
r3478 | if bold_every == false && highlight_lines == nil | ||
|
Jean-Philippe Lang
|
r638 | proc { |line| line.to_s } | ||
|
Jean-Philippe Lang
|
r3478 | elsif highlight_lines.is_a? Enumerable | ||
| highlight_lines = highlight_lines.to_set | ||||
| proc do |line| | ||||
| if highlight_lines.include? line | ||||
| "<strong class=\"highlighted\">#{line}</strong>" # highlighted line numbers in bold | ||||
| else | ||||
| line.to_s | ||||
| end | ||||
| end | ||||
|
Jean-Philippe Lang
|
r638 | elsif bold_every.is_a? Integer | ||
| raise ArgumentError, ":bolding can't be 0." if bold_every == 0 | ||||
| proc do |line| | ||||
| if line % bold_every == 0 | ||||
| "<strong>#{line}</strong>" # every bold_every-th number in bold | ||||
| else | ||||
| line.to_s | ||||
| end | ||||
| end | ||||
| else | ||||
| raise ArgumentError, 'Invalid value %p for :bolding; false or Integer expected.' % bold_every | ||||
| end | ||||
| case mode | ||||
| when :inline | ||||
| max_width = (start + line_count).to_s.size | ||||
|
Jean-Philippe Lang
|
r3478 | line_number = start | ||
|
Jean-Philippe Lang
|
r638 | gsub!(/^/) do | ||
|
Jean-Philippe Lang
|
r3478 | line_number_text = bolding.call line_number | ||
| indent = ' ' * (max_width - line_number.to_s.size) # TODO: Optimize (10^x) | ||||
| res = "<span class=\"no\">#{indent}#{line_number_text}</span> " | ||||
| line_number += 1 | ||||
|
Jean-Philippe Lang
|
r638 | res | ||
| end | ||||
| when :table | ||||
| # This is really ugly. | ||||
| # Because even monospace fonts seem to have different heights when bold, | ||||
| # I make the newline bold, both in the code and the line numbers. | ||||
| # FIXME Still not working perfect for Mr. Internet Exploder | ||||
| line_numbers = (start ... start + line_count).to_a.map(&bolding).join("\n") | ||||
| line_numbers << "\n" # also for Mr. MS Internet Exploder :-/ | ||||
| line_numbers.gsub!(/\n/) { "<tt>\n</tt>" } | ||||
| line_numbers_table_tpl = TABLE.apply('LINE_NUMBERS', line_numbers) | ||||
|
Jean-Philippe Lang
|
r4619 | gsub!("</div>\n", '</div>') | ||
| gsub!("\n", "<tt>\n</tt>") | ||||
|
Jean-Philippe Lang
|
r638 | wrap_in! line_numbers_table_tpl | ||
| @wrapped_in = :div | ||||
| when :list | ||||
| opened_tags = [] | ||||
| gsub!(/^.*$\n?/) do |line| | ||||
| line.chomp! | ||||
| open = opened_tags.join | ||||
| line.scan(%r!<(/)?span[^>]*>?!) do |close,| | ||||
| if close | ||||
| opened_tags.pop | ||||
| else | ||||
| opened_tags << $& | ||||
| end | ||||
| end | ||||
| close = '</span>' * opened_tags.size | ||||
|
Jean-Philippe Lang
|
r3478 | "<li>#{open}#{line}#{close}</li>\n" | ||
|
Jean-Philippe Lang
|
r638 | end | ||
|
Jean-Philippe Lang
|
r3478 | chomp!("\n") | ||
|
Jean-Philippe Lang
|
r638 | wrap_in! LIST | ||
| @wrapped_in = :div | ||||
| else | ||||
| raise ArgumentError, 'Unknown value %p for mode: expected one of %p' % | ||||
| [mode, [:table, :list, :inline]] | ||||
| end | ||||
| self | ||||
| end | ||||
| def line_count | ||||
| line_count = count("\n") | ||||
| position_of_last_newline = rindex(?\n) | ||||
| if position_of_last_newline | ||||
| after_last_newline = self[position_of_last_newline + 1 .. -1] | ||||
| ends_with_newline = after_last_newline[/\A(?:<\/span>)*\z/] | ||||
| line_count += 1 if not ends_with_newline | ||||
| end | ||||
| line_count | ||||
| end | ||||
| end | ||||
| end | ||||
| end | ||||
| end | ||||