1 files changed, 88 insertions, 86 deletions
diff --git a/lib/coderay/token_kinds.rb b/lib/coderay/token_kinds.rb
index 756f75e..1a8f85f 100755
--- a/lib/coderay/token_kinds.rb
+++ b/lib/coderay/token_kinds.rb
@@ -1,89 +1,91 @@
 module CodeRay
-  class Tokens
-    AbbreviationForKind = Hash.new do |h, k|  # :nodoc:
-      if $CODERAY_DEBUG
-        raise 'Undefined Token kind: %p' % [k]  # :nodoc:
-      else
-        false
-      end
-    end
-    AbbreviationForKind.update with = {  # :nodoc:
-      :annotation => 'at',
-      :attribute_name => 'an',
-      :attribute_value => 'av',
-      :bin => 'bi',
-      :char => 'ch',
-      :class => 'cl',
-      :class_variable => 'cv',
-      :color => 'cr',
-      :comment => 'c',
-      :complex => 'cm',
-      :constant => 'co',
-      :content => 'k',
-      :decorator => 'de',
-      :definition => 'df',
-      :delimiter => 'dl',
-      :directive => 'di',
-      :doc => 'do',
-      :doctype => 'dt',
-      :doc_string => 'ds',
-      :entity => 'en',
-      :error => 'er',
-      :escape => 'e',
-      :exception => 'ex',
-      :filename => 'filename',
-      :float => 'fl',
-      :function => 'fu',
-      :global_variable => 'gv',
-      :hex => 'hx',
-      :imaginary => 'cm',
-      :important => 'im',
-      :include => 'ic',
-      :inline => 'il',
-      :inline_delimiter => 'idl',
-      :instance_variable => 'iv',
-      :integer => 'i',
-      :interpreted => 'in',
-      :key => 'ke',
-      :keyword => 'kw',
-      :label => 'la',
-      :local_variable => 'lv',
-      :modifier => 'mod',
-      :namespace => 'ns',
-      :oct => 'oc',
-      :predefined => 'pd',
-      :preprocessor => 'pp',
-      :pre_constant => 'pc',
-      :pre_type => 'pt',
-      :pseudo_class => 'ps',
-      :regexp => 'rx',
-      :reserved => 'r',
-      :shell => 'sh',
-      :string => 's',
-      :symbol => 'sy',
-      :tag => 'ta',
-      :tag_special => 'ts',
-      :type => 'ty',
-      :value => 'vl',
-      :variable => 'v',
-      
-      :insert => 'ins',
-      :delete => 'del',
-      :change => 'chg',
-      :head => 'head',
-      
-      :eyecatcher => 'eye',
-      
-      :ident => false, # 'id'
-      :operator => false,  # 'op'
-      
-      :space => false,  # 'sp'
-      :plain => false,
-    }
-    AbbreviationForKind[:method] = AbbreviationForKind[:function]
-    AbbreviationForKind[:nesting_delimiter] = AbbreviationForKind[:delimiter]
-    AbbreviationForKind[:escape] = AbbreviationForKind[:delimiter]
-    AbbreviationForKind[:docstring] = AbbreviationForKind[:comment]
-    #AbbreviationForKind.default = AbbreviationForKind[:error] or raise 'no class found for :error!'
+  
+  # A Hash of all known token kinds and their associated CSS classes.
+  TokenKinds = Hash.new do |h, k|
+    warn 'Undefined Token kind: %p' % [k] if $CODERAY_DEBUG
+    false
   end
+  
+  # speedup
+  TokenKinds.compare_by_identity if TokenKinds.respond_to? :compare_by_identity
+  
+  TokenKinds.update with = {  # :nodoc:
+    :annotation => 'at',
+    :attribute_name => 'an',
+    :attribute_value => 'av',
+    :bin => 'bi',
+    :char => 'ch',
+    :class => 'cl',
+    :class_variable => 'cv',
+    :color => 'cr',
+    :comment => 'c',
+    :complex => 'cm',
+    :constant => 'co',
+    :content => 'k',
+    :decorator => 'de',
+    :definition => 'df',
+    :delimiter => 'dl',
+    :directive => 'di',
+    :doc => 'do',
+    :doctype => 'dt',
+    :doc_string => 'ds',
+    :entity => 'en',
+    :error => 'er',
+    :escape => 'e',
+    :exception => 'ex',
+    :filename => 'filename',
+    :float => 'fl',
+    :function => 'fu',
+    :global_variable => 'gv',
+    :hex => 'hx',
+    :imaginary => 'cm',
+    :important => 'im',
+    :include => 'ic',
+    :inline => 'il',
+    :inline_delimiter => 'idl',
+    :instance_variable => 'iv',
+    :integer => 'i',
+    :interpreted => 'in',
+    :key => 'ke',
+    :keyword => 'kw',
+    :label => 'la',
+    :local_variable => 'lv',
+    :modifier => 'mod',
+    :namespace => 'ns',
+    :oct => 'oc',
+    :predefined => 'pd',
+    :preprocessor => 'pp',
+    :pre_constant => 'pc',  # FIXME: rename to :predefined_constant
+    :pre_type => 'pt',  # FIXME: rename to :predefined_type
+    :pseudo_class => 'ps',
+    :regexp => 'rx',
+    :reserved => 'r',
+    :shell => 'sh',
+    :string => 's',
+    :symbol => 'sy',
+    :tag => 'ta',
+    :tag_special => 'ts',
+    :type => 'ty',
+    :value => 'vl',
+    :variable => 'v',
+    
+    :insert => 'ins',
+    :delete => 'del',
+    :change => 'chg',
+    :head => 'head',
+    
+    :eyecatcher => 'eye',
+    
+    :ident => false, # 'id'
+    :operator => false,  # 'op'
+    
+    :space => false,  # 'sp'
+    :plain => false,
+  }
+  
+  TokenKinds[:method]    = TokenKinds[:function]
+  TokenKinds[:escape]    = TokenKinds[:delimiter]
+  TokenKinds[:docstring] = TokenKinds[:comment]
+  
+  TokenKinds.freeze
 end