class Lookbook::Formatter

def highlighted?(i)
  @highlight_lines.include?(i + 1)
end

def initialize(**opts)
  @opts = opts
  @highlight_lines = opts[:highlight_lines].to_a || []
  @start_line = opts[:start_line] || 1
  @language = opts[:language]
end

def line_number(i)
  @start_line + i
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