Generate Mojom Types for Dart

mojo/public/tools/bindings/generators/mojom_go_generator.py

 # Copyright 2015 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 '''Generates Go source files from a mojom.Module.'''
 
+from collections import defaultdict
 from itertools import chain
 import os
 import re
 
 from mojom.generate.template_expander import UseJinja
 
 import mojom.generate.generator as generator
 import mojom.generate.module as mojom
 import mojom.generate.pack as pack
 
+GENERATOR_PREFIX = 'go'
+
 class KindInfo(object):
   def __init__(self, go_type, encode_suffix, decode_suffix, bit_size):
     self.go_type = go_type
     self.encode_suffix = encode_suffix
     self.decode_suffix = decode_suffix
     self.bit_size = bit_size
 
 _kind_infos = {
   mojom.BOOL:                  KindInfo('bool', 'Bool', 'Bool', 1),
   mojom.INT8:                  KindInfo('int8', 'Int8', 'Int8', 8),
@@ skipping 262 matching lines @@
     return
   path = GetPackagePath(imported['module'])
   if path in _imports:
     return
   name = GetPackageName(imported['module'])
   while name in _imports.values(): # This avoids repeated names.
     name += '_'
   _imports[path] = name
   _mojom_imports[path] = name
 
-# The identifier cache is used by the Type generator to determine if a type has
-# already been generated or not. This prevents over-generation of the same type
-# when it is referred to in multiple ways.
-identifier_cache = {}
+# The identifier_store tracks identifiers that have been used so that code
+# generation can avoid over-generating code for the same identifier.
+identifier_store = defaultdict(bool)
 def GetIdentifier(kind):
-  # Use the kind's module to determine the package name.
+  # Use the kind's module to determine the package and name.
+  # Note: InterfaceRequestKind's should use the Interface inside them.
   if hasattr(kind, 'module'):
     package = GetPackageName(kind.module)
+    name = kind.name
   elif mojom.IsInterfaceRequestKind(kind):
     package = GetPackageName(kind.kind.module)
+    name = kind.kind.name
   else:
+    # This is for a type that should not be stored. Return an empty identifier.
     return ''
 
-  # Most kinds have a name, but those that don't should rely on their spec.
-  # Since spec can have : and ? characters, these must be replaced. Since ? is
-  # replaced with '', the caller must keep track of optionality on its own.
-  name_or_spec = (kind.name if hasattr(kind, 'name') else kind.spec)
-  package_unique = name_or_spec.replace(':', '_').replace('?', '')
-  return '%s_%s' % (package, package_unique)
+  return '%s_%s' % (package, name)
 
-def StoreIdentifier(identifier, cache_name):
-  if not cache_name in identifier_cache:
-    identifier_cache[cache_name] = {}
-  identifier_cache[cache_name][identifier] = True
+def StoreIdentifier(identifier):
+  identifier_store[identifier] = True
   return ''
 
-def CheckIdentifier(identifier, cache_name):
-  if not cache_name in identifier_cache:
-    identifier_cache[cache_name] = {}
-  return identifier in identifier_cache[cache_name]
+def CheckIdentifier(identifier):
+  return identifier in identifier_store
 
 # Get the mojom type's identifier suffix.
 def GetMojomTypeIdentifier(kind):
   # Since this should be unique, it is based on the type's identifier.
   return "%s__" % GetIdentifier(kind)
 
 class Generator(generator.Generator):
   go_filters = {
     'array': lambda kind: mojom.Array(kind),
     'bit_size': GetBitSize,
@@ skipping 19 matching lines @@
     'is_union': mojom.IsUnionKind,
     'qualified': GetQualifiedName,
     'mojom_type': GetMojomTypeValue,
     'mojom_type_identifier': GetMojomTypeIdentifier,
     'name': GetNameForElement,
     'unqualified_name': GetUnqualifiedNameForElement,
     'package': GetPackageNameForElement,
     'tab_indent': lambda s, size = 1: ('\n' + '\t' * size).join(s.splitlines())
   }
 
-  # TODO: This value should be settable via arguments. If False, then mojom type
-  # information will not be generated.
-  should_gen_mojom_types = True
+  # If set to True, then mojom type information will be generated.
+  should_gen_mojom_types = False
 
   def GetParameters(self):
     package = GetPackageName(self.module)
     return {
       'enums': GetAllEnums(self.module),
       'imports': self.GetImports()[0],
       'interfaces': self.GetInterfaces(),
       'mojom_imports': self.GetMojomImports(),
       'package': package,
       'structs': self.GetStructs(),
       'descpkg': '%s.' % _service_describer_pkg_short \
         if package != _service_describer_pkg_short else '',
       'typepkg': '%s.' % _mojom_types_pkg_short \
         if package != _mojom_types_pkg_short else '',
       'unions': self.GetUnions()
     }
 
   @UseJinja('go_templates/source.tmpl', filters=go_filters)
   def GenerateSource(self):
     return self.GetParameters()
 
   def GenerateFiles(self, args):
+    if "--go_gen_types" in args:
+      self.should_gen_mojom_types = True
+
     self.Write(self.GenerateSource(), os.path.join("go", "src",
         GetPackagePath(self.module), "%s.go" % self.module.name))
 
   def GetJinjaParameters(self):
     return {
       'lstrip_blocks': True,
       'trim_blocks': True,
     }
 
   def GetGlobals(self):
@@ skipping 91 matching lines @@
   # Overrides the implementation from the base class in order to customize the
   # struct and field names.
   def _GetResponseStructFromMethod(self, method):
     params_class = "%s_%s_ResponseParams" % (
         GetNameForElement(method.interface), GetNameForElement(method))
     struct = mojom.Struct(params_class, module=method.interface.module)
     for param in method.response_parameters:
       struct.AddField("out%s" % GetNameForElement(param),
           param.kind, param.ordinal, attributes=param.attributes)
     return self._AddStructComputedData(False, struct)