Chris@210: module CodeRay
Chris@210: module Scanners
Chris@210: 
Chris@210:   class Java < Scanner
Chris@210: 
Chris@210:     include Streamable
Chris@210:     register_for :java
Chris@210:     helper :builtin_types
Chris@210:     
Chris@210:     # http://java.sun.com/docs/books/tutorial/java/nutsandbolts/_keywords.html
Chris@210:     KEYWORDS = %w[
Chris@210:       assert break case catch continue default do else
Chris@210:       finally for if instanceof import new package
Chris@210:       return switch throw try typeof while
Chris@210:       debugger export
Chris@210:     ]
Chris@210:     RESERVED = %w[ const goto ]
Chris@210:     CONSTANTS = %w[ false null true ]
Chris@210:     MAGIC_VARIABLES = %w[ this super ]
Chris@210:     TYPES = %w[
Chris@210:       boolean byte char class double enum float int interface long
Chris@210:       short void
Chris@210:     ] << '[]'  # because int[] should be highlighted as a type
Chris@210:     DIRECTIVES = %w[
Chris@210:       abstract extends final implements native private protected public
Chris@210:       static strictfp synchronized throws transient volatile
Chris@210:     ]
Chris@210:     
Chris@210:     IDENT_KIND = WordList.new(:ident).
Chris@210:       add(KEYWORDS, :keyword).
Chris@210:       add(RESERVED, :reserved).
Chris@210:       add(CONSTANTS, :pre_constant).
Chris@210:       add(MAGIC_VARIABLES, :local_variable).
Chris@210:       add(TYPES, :type).
Chris@210:       add(BuiltinTypes::List, :pre_type).
Chris@210:       add(BuiltinTypes::List.select { |builtin| builtin[/(Error|Exception)$/] }, :exception).
Chris@210:       add(DIRECTIVES, :directive)
Chris@210: 
Chris@210:     ESCAPE = / [bfnrtv\n\\'"] | x[a-fA-F0-9]{1,2} | [0-7]{1,3} /x
Chris@210:     UNICODE_ESCAPE =  / u[a-fA-F0-9]{4} | U[a-fA-F0-9]{8} /x
Chris@210:     STRING_CONTENT_PATTERN = {
Chris@210:       "'" => /[^\\']+/,
Chris@210:       '"' => /[^\\"]+/,
Chris@210:       '/' => /[^\\\/]+/,
Chris@210:     }
Chris@210:     IDENT = /[a-zA-Z_][A-Za-z_0-9]*/
Chris@210:     
Chris@210:     def scan_tokens tokens, options
Chris@210: 
Chris@210:       state = :initial
Chris@210:       string_delimiter = nil
Chris@210:       import_clause = class_name_follows = last_token_dot = false
Chris@210: 
Chris@210:       until eos?
Chris@210: 
Chris@210:         kind = nil
Chris@210:         match = nil
Chris@210:         
Chris@210:         case state
Chris@210: 
Chris@210:         when :initial
Chris@210: 
Chris@210:           if match = scan(/ \s+ | \\\n /x)
Chris@210:             tokens << [match, :space]
Chris@210:             next
Chris@210:           
Chris@210:           elsif match = scan(%r! // [^\n\\]* (?: \\. [^\n\\]* )* | /\* (?: .*? \*/ | .* ) !mx)
Chris@210:             tokens << [match, :comment]
Chris@210:             next
Chris@210:           
Chris@210:           elsif import_clause && scan(/ #{IDENT} (?: \. #{IDENT} )* /ox)
Chris@210:             kind = :include
Chris@210:           
Chris@210:           elsif match = scan(/ #{IDENT} | \[\] /ox)
Chris@210:             kind = IDENT_KIND[match]
Chris@210:             if last_token_dot
Chris@210:               kind = :ident
Chris@210:             elsif class_name_follows
Chris@210:               kind = :class
Chris@210:               class_name_follows = false
Chris@210:             else
Chris@210:               import_clause = true if match == 'import'
Chris@210:               class_name_follows = true if match == 'class' || match == 'interface'
Chris@210:             end
Chris@210:           
Chris@210:           elsif scan(/ \.(?!\d) | [,?:()\[\]}] | -- | \+\+ | && | \|\| | \*\*=? | [-+*\/%^~&|<>=!]=? | <<<?=? | >>>?=? /x)
Chris@210:             kind = :operator
Chris@210:           
Chris@210:           elsif scan(/;/)
Chris@210:             import_clause = false
Chris@210:             kind = :operator
Chris@210:           
Chris@210:           elsif scan(/\{/)
Chris@210:             class_name_follows = false
Chris@210:             kind = :operator
Chris@210:           
Chris@210:           elsif check(/[\d.]/)
Chris@210:             if scan(/0[xX][0-9A-Fa-f]+/)
Chris@210:               kind = :hex
Chris@210:             elsif scan(/(?>0[0-7]+)(?![89.eEfF])/)
Chris@210:               kind = :oct
Chris@210:             elsif scan(/\d+[fFdD]|\d*\.\d+(?:[eE][+-]?\d+)?[fFdD]?|\d+[eE][+-]?\d+[fFdD]?/)
Chris@210:               kind = :float
Chris@210:             elsif scan(/\d+[lL]?/)
Chris@210:               kind = :integer
Chris@210:             end
Chris@210: 
Chris@210:           elsif match = scan(/["']/)
Chris@210:             tokens << [:open, :string]
Chris@210:             state = :string
Chris@210:             string_delimiter = match
Chris@210:             kind = :delimiter
Chris@210: 
Chris@210:           elsif scan(/ @ #{IDENT} /ox)
Chris@210:             kind = :annotation
Chris@210: 
Chris@210:           else
Chris@210:             getch
Chris@210:             kind = :error
Chris@210: 
Chris@210:           end
Chris@210: 
Chris@210:         when :string
Chris@210:           if scan(STRING_CONTENT_PATTERN[string_delimiter])
Chris@210:             kind = :content
Chris@210:           elsif match = scan(/["'\/]/)
Chris@210:             tokens << [match, :delimiter]
Chris@210:             tokens << [:close, state]
Chris@210:             string_delimiter = nil
Chris@210:             state = :initial
Chris@210:             next
Chris@210:           elsif state == :string && (match = scan(/ \\ (?: #{ESCAPE} | #{UNICODE_ESCAPE} ) /mox))
Chris@210:             if string_delimiter == "'" && !(match == "\\\\" || match == "\\'")
Chris@210:               kind = :content
Chris@210:             else
Chris@210:               kind = :char
Chris@210:             end
Chris@210:           elsif scan(/\\./m)
Chris@210:             kind = :content
Chris@210:           elsif scan(/ \\ | $ /x)
Chris@210:             tokens << [:close, state]
Chris@210:             kind = :error
Chris@210:             state = :initial
Chris@210:           else
Chris@210:             raise_inspect "else case \" reached; %p not handled." % peek(1), tokens
Chris@210:           end
Chris@210: 
Chris@210:         else
Chris@210:           raise_inspect 'Unknown state', tokens
Chris@210: 
Chris@210:         end
Chris@210: 
Chris@210:         match ||= matched
Chris@210:         if $CODERAY_DEBUG and not kind
Chris@210:           raise_inspect 'Error token %p in line %d' %
Chris@210:             [[match, kind], line], tokens
Chris@210:         end
Chris@210:         raise_inspect 'Empty token', tokens unless match
Chris@210:         
Chris@210:         last_token_dot = match == '.'
Chris@210:         
Chris@210:         tokens << [match, kind]
Chris@210: 
Chris@210:       end
Chris@210: 
Chris@210:       if state == :string
Chris@210:         tokens << [:close, state]
Chris@210:       end
Chris@210: 
Chris@210:       tokens
Chris@210:     end
Chris@210: 
Chris@210:   end
Chris@210: 
Chris@210: end
Chris@210: end