Dartium 45 roll

tools/dom/scripts/database.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.
 
 """Module to manage IDL files."""
 
 import copy
 import pickle
 import logging
@@ skipping 26 matching lines @@
       be created.
     """
     self._root_dir = root_dir
     if not os.path.exists(root_dir):
       _logger.debug('creating root directory %s' % root_dir)
       os.makedirs(root_dir)
     self._all_interfaces = {}
     self._interfaces_to_delete = []
     self._enums = {}
     self._all_dictionaries = {}
+    # TODO(terry): Hack to remember all typedef unions.
+    self._all_type_defs = {}
 
   def Clone(self):
     new_database = Database(self._root_dir)
     new_database._all_interfaces = copy.deepcopy(self._all_interfaces)
     new_database._interfaces_to_delete = copy.deepcopy(
         self._interfaces_to_delete)
     new_database._enums = copy.deepcopy(self._enums)
     new_database._all_dictionaries = copy.deepcopy(self._all_dictionaries)
+    new_database._all_type_defs = copy.deepcopy(self._all_type_defs)
 
     return new_database
 
   def Delete(self):
     """Deletes the database by deleting its directory"""
     if os.path.exists(self._root_dir):
       shutil.rmtree(self._root_dir)
     # reset in-memory constructs
     self._all_interfaces = {}
 
@@ skipping 210 matching lines @@
       raise RuntimeError('Dictionary %s already exists' % dictionary_name)
     self._all_dictionaries[dictionary_name] = dictionary
 
   def GetDictionaries(self):
     """Returns a list of all loaded dictionaries."""
     res = []
     for _, dictionary in sorted(self._all_dictionaries.items()):
       res.append(dictionary)
     return res
 
+  def HasTypeDef(self, type_def_name):
+    """Returns True if the typedef is in memory"""
+    return type_def_name in self._all_type_defs
+
+  def GetTypeDef(self, type_def_name):
+    """Returns an IDLTypeDef corresponding to the type_def_name
+    from memory.
+
+    Args:
+      type_def_name -- the name of the typedef.
+    """
+    if type_def_name not in self._all_type_defs:
+      raise RuntimeError('Typedef %s is not loaded' % type_def_name)
+    return self._all_type_defs[type_def_name]
+
+  def AddTypeDef(self, type_def):
+    """Add only a typedef that a unions they map to any (no type)."""
+    type_def_name = type_def.id
+    if type_def_name in self._all_type_defs:
+      raise RuntimeError('Typedef %s already exists' % type_def_name)
+    self._all_type_defs[type_def_name] = type_def
+    print '  Added typedef %s' % type_def_name
+
   def TransitiveSecondaryParents(self, interface, propagate_event_target):
     """Returns a list of all non-primary parents.
 
     The list contains the interface objects for interfaces defined in the
     database, and the name for undefined interfaces.
     """
     def walk(parents):
       for parent in parents:
         parent_name = parent.type.id
         if IsDartCollectionType(parent_name):
           result.append(parent_name)
           continue
         if self.HasInterface(parent_name):
           parent_interface = self.GetInterface(parent_name)
           result.append(parent_interface)
           walk(parent_interface.parents)
 
     result = []
     if interface.parents:
       parent = interface.parents[0]
       if (IsPureInterface(parent.type.id) or
           (propagate_event_target and parent.type.id == 'EventTarget')):
         walk(interface.parents)
       else:
         walk(interface.parents[1:])
     return result