class Rouge::Formatters::HTMLPygments

def initialize(inner, css_class='codehilite')
  @inner = HTML.assert_html_formatter!(inner)
  @css_class = css_class
end

def stream(tokens, &b)
  yield %(<div class="highlight"><pre class="#{@css_class}"><code>)
  @inner.stream(tokens, &b)
  yield "</code></pre></div>"
end