Update mojo sdk to rev 8af2ccff2eee4bfca1043015abee30482a030b30

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

Index: third_party/mojo/src/mojo/public/tools/bindings/generators/mojom_go_generator.py
diff --git a/third_party/mojo/src/mojo/public/tools/bindings/generators/mojom_go_generator.py b/third_party/mojo/src/mojo/public/tools/bindings/generators/mojom_go_generator.py
index 3f7c938baea6f8b4fd6946449167d8d4a872adfe..11962767e4b816a7119a1648e5fed38cd42b16f7 100644
--- a/third_party/mojo/src/mojo/public/tools/bindings/generators/mojom_go_generator.py
+++ b/third_party/mojo/src/mojo/public/tools/bindings/generators/mojom_go_generator.py
@@ -4,12 +4,148 @@
 
 '''Generates Go source files from a mojom.Module.'''
 
+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
+
+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),
+  mojom.UINT8:                 KindInfo('uint8', 'Uint8', 'Uint8', 8),
+  mojom.INT16:                 KindInfo('int16', 'Int16', 'Int16', 16),
+  mojom.UINT16:                KindInfo('uint16', 'Uint16', 'Uint16', 16),
+  mojom.INT32:                 KindInfo('int32', 'Int32', 'Int32', 32),
+  mojom.UINT32:                KindInfo('uint32', 'Uint32', 'Uint32', 32),
+  mojom.FLOAT:                 KindInfo('float32', 'Float32', 'Float32', 32),
+  mojom.HANDLE:                KindInfo(
+      'system.Handle', 'Handle', 'Handle', 32),
+  mojom.DCPIPE:                KindInfo(
+      'system.ConsumerHandle', 'Handle', 'ConsumerHandle', 32),
+  mojom.DPPIPE:                KindInfo(
+      'system.ProducerHandle', 'Handle', 'ProducerHandle', 32),
+  mojom.MSGPIPE:               KindInfo(
+      'system.MessagePipeHandle', 'Handle', 'MessagePipeHandle', 32),
+  mojom.SHAREDBUFFER:          KindInfo(
+      'system.SharedBufferHandle', 'Handle', 'SharedBufferHandle', 32),
+  mojom.NULLABLE_HANDLE:       KindInfo(
+      'system.Handle', 'Handle', 'Handle', 32),
+  mojom.NULLABLE_DCPIPE:       KindInfo(
+      'system.ConsumerHandle', 'Handle', 'ConsumerHandle', 32),
+  mojom.NULLABLE_DPPIPE:       KindInfo(
+      'system.ProducerHandle', 'Handle', 'ProducerHandle', 32),
+  mojom.NULLABLE_MSGPIPE:      KindInfo(
+      'system.MessagePipeHandle', 'Handle', 'MessagePipeHandle', 32),
+  mojom.NULLABLE_SHAREDBUFFER: KindInfo(
+      'system.SharedBufferHandle', 'Handle', 'SharedBufferHandle', 32),
+  mojom.INT64:                 KindInfo('int64', 'Int64', 'Int64', 64),
+  mojom.UINT64:                KindInfo('uint64', 'Uint64', 'Uint64', 64),
+  mojom.DOUBLE:                KindInfo('float64', 'Float64', 'Float64', 64),
+  mojom.STRING:                KindInfo('string', 'String', 'String', 64),
+  mojom.NULLABLE_STRING:       KindInfo('string', 'String', 'String', 64),
+}
+
+def GetBitSize(kind):
+  if isinstance(kind, (mojom.Array, mojom.Map, mojom.Struct)):
+    return 64
+  if isinstance(kind, (mojom.InterfaceRequest, mojom.Interface)):
+    kind = mojom.MSGPIPE
+  if isinstance(kind, mojom.Enum):
+    kind = mojom.INT32
+  return _kind_infos[kind].bit_size
+
+def GetGoType(kind, nullable = True):
+  if nullable and mojom.IsNullableKind(kind):
+    return '*%s' % GetNonNullableGoType(kind)
+  return GetNonNullableGoType(kind)
+
+def GetNonNullableGoType(kind):
+  if mojom.IsStructKind(kind):
+    return '%s' % FormatName(kind.name)
+  if mojom.IsArrayKind(kind):
+    if kind.length:
+      return '[%s]%s' % (kind.length, GetGoType(kind.kind))
+    return '[]%s' % GetGoType(kind.kind)
+  if mojom.IsMapKind(kind):
+    return 'map[%s]%s' % (GetGoType(kind.key_kind), GetGoType(kind.value_kind))
+  if mojom.IsInterfaceKind(kind) or mojom.IsInterfaceRequestKind(kind):
+    return GetGoType(mojom.MSGPIPE)
+  if mojom.IsEnumKind(kind):
+    return GetNameForNestedElement(kind)
+  return _kind_infos[kind].go_type
+
+def NameToComponent(name):
+  # insert '_' between anything and a Title name (e.g, HTTPEntry2FooBar ->
+  # HTTP_Entry2_FooBar)
+  name = re.sub('([^_])([A-Z][^A-Z_]+)', r'\1_\2', name)
+  # insert '_' between non upper and start of upper blocks (e.g.,
+  # HTTP_Entry2_FooBar -> HTTP_Entry2_Foo_Bar)
+  name = re.sub('([^A-Z_])([A-Z])', r'\1_\2', name)
+  return [x.lower() for x in name.split('_')]
+
+def UpperCamelCase(name):
+  return ''.join([x.capitalize() for x in NameToComponent(name)])
+
+def FormatName(name, exported=True):
+  if exported:
+    return UpperCamelCase(name)
+  # Leave '_' symbols for unexported names.
+  return name[0].lower() + name[1:]
+
+def GetNameForNestedElement(element):
+  if element.parent_kind:
+    return "%s_%s" % (GetNameForElement(element.parent_kind),
+        FormatName(element.name))
+  return FormatName(element.name)
+
+def GetNameForElement(element, exported=True):
+  if (mojom.IsInterfaceKind(element) or mojom.IsStructKind(element) or
+      isinstance(element, (mojom.EnumField,
+                           mojom.Field,
+                           mojom.Method,
+                           mojom.Parameter))):
+    return FormatName(element.name, exported)
+  if isinstance(element, (mojom.Enum,
+                          mojom.Constant,
+                          mojom.ConstantValue)):
+    return GetNameForNestedElement(element)
+  raise Exception('Unexpected element: %s' % element)
+
+def ExpressionToText(token):
+  if isinstance(token, mojom.EnumValue):
+    return "%s_%s" % (GetNameForNestedElement(token.enum),
+        FormatName(token.name, True))
+  if isinstance(token, mojom.ConstantValue):
+    return GetNameForNestedElement(token)
+  if isinstance(token, mojom.Constant):
+    return ExpressionToText(token.value)
+  return token
+
+def DecodeSuffix(kind):
+  if mojom.IsEnumKind(kind):
+    return DecodeSuffix(mojom.INT32)
+  if mojom.IsInterfaceKind(kind) or mojom.IsInterfaceRequestKind(kind):
+    return DecodeSuffix(mojom.MSGPIPE)
+  return _kind_infos[kind].decode_suffix
+
+def EncodeSuffix(kind):
+  if mojom.IsEnumKind(kind):
+    return EncodeSuffix(mojom.INT32)
+  if mojom.IsInterfaceKind(kind) or mojom.IsInterfaceRequestKind(kind):
+    return EncodeSuffix(mojom.MSGPIPE)
+  return _kind_infos[kind].encode_suffix
 
 def GetPackage(module):
   if module.namespace:
@@ -22,12 +158,63 @@ def GetPackagePath(module):
     path = os.path.join(path, i)
   return path
 
+def GetStructFromMethod(method):
+  params_class = "%s_%s_Params" % (GetNameForElement(method.interface),
+      GetNameForElement(method))
+  struct = mojom.Struct(params_class, module=method.interface.module)
+  for param in method.parameters:
+    struct.AddField("in%s" % GetNameForElement(param),
+        param.kind, param.ordinal)
+  struct.packed = pack.PackedStruct(struct)
+  struct.bytes = pack.GetByteLayout(struct.packed)
+  struct.versions = pack.GetVersionInfo(struct.packed)
+  return struct
+
+def GetResponseStructFromMethod(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)
+  struct.packed = pack.PackedStruct(struct)
+  struct.bytes = pack.GetByteLayout(struct.packed)
+  struct.versions = pack.GetVersionInfo(struct.packed)
+  return struct
+
 class Generator(generator.Generator):
-  go_filters = {}
+  go_filters = {
+    'array': lambda kind: mojom.Array(kind),
+    'bit_size': GetBitSize,
+    'decode_suffix': DecodeSuffix,
+    'encode_suffix': EncodeSuffix,
+    'go_type': GetGoType,
+    'expression_to_text': ExpressionToText,
+    'is_array': mojom.IsArrayKind,
+    'is_enum': mojom.IsEnumKind,
+    'is_handle': mojom.IsAnyHandleKind,
+    'is_map': mojom.IsMapKind,
+    'is_none_or_empty': lambda array: array == None or len(array) == 0,
+    'is_nullable': mojom.IsNullableKind,
+    'is_pointer': mojom.IsObjectKind,
+    'is_struct': mojom.IsStructKind,
+    'name': GetNameForElement,
+    'response_struct_from_method': GetResponseStructFromMethod,
+    'struct_from_method': GetStructFromMethod,
+    'tab_indent': lambda s, size = 1: ('\n' + '\t' * size).join(s.splitlines())
+  }
+
+  def GetAllEnums(self):
+    data = [self.module] + self.GetStructs() + self.module.interfaces
+    enums = [x.enums for x in data]
+    return [i for i in chain.from_iterable(enums)]
 
   def GetParameters(self):
     return {
+      'enums': self.GetAllEnums(),
+      'interfaces': self.module.interfaces,
       'package': GetPackage(self.module),
+      'structs': self.GetStructs(),
     }
 
   @UseJinja('go_templates/source.tmpl', filters=go_filters)