Sha256: 9eaa8ed459ba85c8ca8294d90515f8b359be6fe34aa6211d3f4025860a5373c5
Contents?: true
Size: 1.81 KB
Versions: 8
Compression:
Stored size: 1.81 KB
Contents
require "rouge"
require "htmlbeautifier"
require 'htmlentities'
module Lookbook
module CodeFormatter
class << self
def highlight(source, **opts)
coder = HTMLEntities.new
source = coder.decode source
language = opts[:language] || "ruby"
formatter = Formatter.new(**opts)
lexer = Rouge::Lexer.find(language.to_s) || Rouge::Lexer.find("plaintext")
formatter.format(lexer.lex(source)).html_safe
end
def beautify(source, **opts)
language = opts[:language] || "html"
source = source.strip
result = language.downcase == "html" ? HtmlBeautifier.beautify(source) : source
result.strip.html_safe
end
end
end
class Formatter < Rouge::Formatters::HTML
def initialize(**opts)
@opts = opts
@highlight_lines = opts[:highlight_lines].to_a || []
@start_line = opts[:start_line] || 1
@language = opts[:language]
end
def stream(tokens, &block)
lines = token_lines(tokens)
yield "<div class='wrapper'>"
if @opts[:line_numbers]
yield "<div class='line-numbers'>"
lines.each.with_index do |line, i|
yield "<div class='line #{"highlighted" if highlighted?(i)}'><span class='line-number'>#{line_number(i)}</span></div>"
end
yield "</div>"
end
yield "<pre class='code highlight' data-lang='#{@language}'><code>"
lines.each.with_index do |line_tokens, i|
yield "<div class='line#{" highlighted" if highlighted?(i)}'>"
line_tokens.each do |token, value|
yield span(token, value)
end
yield "</div>"
end
yield "</code></pre>"
yield "</div>"
end
def highlighted?(i)
@highlight_lines.include?(i + 1)
end
def line_number(i)
@start_line + i
end
end
end
Version data entries
8 entries across 8 versions & 1 rubygems