Chris@0: module CodeRay
Chris@0: module Scanners
Chris@0: 
Chris@0:   load :html
Chris@0:   load :ruby
Chris@0: 
Chris@0:   # RHTML Scanner
Chris@0:   class RHTML < Scanner
Chris@0: 
Chris@0:     include Streamable
Chris@0:     register_for :rhtml
Chris@0:     title 'HTML ERB Template'
Chris@0:     
Chris@0:     KINDS_NOT_LOC = HTML::KINDS_NOT_LOC
Chris@0: 
Chris@0:     ERB_RUBY_BLOCK = /
Chris@0:       <%(?!%)[=-]?
Chris@0:       (?>
Chris@0:         [^\-%]*    # normal*
Chris@0:         (?>        # special
Chris@0:           (?: %(?!>) | -(?!%>) )
Chris@0:           [^\-%]*  # normal*
Chris@0:         )*
Chris@0:       )
Chris@0:       (?: -?%> )?
Chris@0:     /x
Chris@0: 
Chris@0:     START_OF_ERB = /
Chris@0:       <%(?!%)
Chris@0:     /x
Chris@0: 
Chris@0:   private
Chris@0: 
Chris@0:     def setup
Chris@0:       @ruby_scanner = CodeRay.scanner :ruby, :tokens => @tokens, :keep_tokens => true
Chris@0:       @html_scanner = CodeRay.scanner :html, :tokens => @tokens, :keep_tokens => true, :keep_state => true
Chris@0:     end
Chris@0: 
Chris@0:     def reset_instance
Chris@0:       super
Chris@0:       @html_scanner.reset
Chris@0:     end
Chris@0: 
Chris@0:     def scan_tokens tokens, options
Chris@0: 
Chris@0:       until eos?
Chris@0: 
Chris@0:         if (match = scan_until(/(?=#{START_OF_ERB})/o) || scan_until(/\z/)) and not match.empty?
Chris@0:           @html_scanner.tokenize match
Chris@0: 
Chris@0:         elsif match = scan(/#{ERB_RUBY_BLOCK}/o)
Chris@0:           start_tag = match[/\A<%[-=]?/]
Chris@0:           end_tag = match[/-?%?>?\z/]
Chris@0:           tokens << [:open, :inline]
Chris@0:           tokens << [start_tag, :inline_delimiter]
Chris@0:           code = match[start_tag.size .. -1 - end_tag.size]
Chris@0:           @ruby_scanner.tokenize code
Chris@0:           tokens << [end_tag, :inline_delimiter] unless end_tag.empty?
Chris@0:           tokens << [:close, :inline]
Chris@0: 
Chris@0:         else
Chris@0:           raise_inspect 'else-case reached!', tokens
Chris@0:         end
Chris@0: 
Chris@0:       end
Chris@0: 
Chris@0:       tokens
Chris@0: 
Chris@0:     end
Chris@0: 
Chris@0:   end
Chris@0: 
Chris@0: end
Chris@0: end