Map enums to strings.

tools/dom/scripts/idlnode.py

 #!/usr/bin/python
 # Copyright (c) 2011, the Dart project authors.  Please see the AUTHORS file
 # for details. All rights reserved. Use of this source code is governed by a
 # BSD-style license that can be found in the LICENSE file.
 
 import os
 import sys
 
 
 class IDLNode(object):
@@ skipping 241 matching lines @@
   """IDLFile is the top-level node in each IDL file. It may contain interfaces."""
 
   def __init__(self, ast, filename=None):
     IDLNode.__init__(self, ast)
     self.filename = filename
     self.interfaces = self._convert_all(ast, 'Interface', IDLInterface)
     modules = self._convert_all(ast, 'Module', IDLModule)
     self.implementsStatements = self._convert_all(ast, 'ImplStmt',
       IDLImplementsStatement)
     self.typeDefs = self._convert_all(ast, 'TypeDef', IDLTypeDef)
+    self.enums = self._convert_all(ast, 'Enum', IDLEnum)
     for module in modules:
       self.interfaces.extend(module.interfaces)
       self.implementsStatements.extend(module.implementsStatements)
       self.typeDefs.extend(module.typeDefs)
 
 
 class IDLModule(IDLNode):
   """IDLModule has an id, and may contain interfaces, type defs and
   implements statements."""
   def __init__(self, ast):
     IDLNode.__init__(self, ast)
     self._convert_ext_attrs(ast)
     self._convert_annotations(ast)
     self.interfaces = self._convert_all(ast, 'Interface', IDLInterface)
     self.typeDefs = self._convert_all(ast, 'TypeDef', IDLTypeDef)
+    self.enums = self._convert_all(ast, 'Enum', IDLNode)
     self.implementsStatements = self._convert_all(ast, 'ImplStmt',
       IDLImplementsStatement)
 
 
 class IDLExtAttrs(IDLDictNode):
   """IDLExtAttrs is an IDLDictNode that stores IDL Extended Attributes.
   Modules, interfaces, members and arguments can all own IDLExtAttrs."""
   def __init__(self, ast=None):
     IDLDictNode.__init__(self, None)
     if not ast:
@@ skipping 84 matching lines @@
     elif label.endswith('Type'):
       # Omit 'Type' suffix and lowercase the rest.
       label = '%s%s' % (label[0].lower(), label[1:-4])
 
     # Add unsigned qualifier.
     if self._has(ast, 'Unsigned'):
       label = 'unsigned %s' % label
     return label
 
 
+class IDLEnum(IDLNode):
+  """IDLNode for 'enum [id] { [string]+ }'"""
+  def __init__(self, ast):
+    IDLNode.__init__(self, ast)
+    self._convert_annotations(ast)
+    # TODO(antonm): save enum values.
+
+
 class IDLTypeDef(IDLNode):
   """IDLNode for 'typedef [type] [id]' declarations."""
   def __init__(self, ast):
     IDLNode.__init__(self, ast)
     self._convert_annotations(ast)
     self.type = self._convert_first(ast, 'Type', IDLType)
 
 
 class IDLInterface(IDLNode):
   """IDLInterface node contains operations, attributes, constants,
@@ skipping 136 matching lines @@
   """IDLDictNode specialization for one annotation."""
   def __init__(self, ast=None):
     IDLDictNode.__init__(self, ast)
     self.id = None
     if not ast:
       return
     for arg in self._find_all(ast, 'AnnotationArg'):
       name = self._find_first(arg, 'Id')
       value = self._find_first(arg, 'AnnotationArgValue')
       self[name] = value