| 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 11962767e4b816a7119a1648e5fed38cd42b16f7..3f7c938baea6f8b4fd6946449167d8d4a872adfe 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,148 +4,12 @@
|
|
|
| '''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:
|
| @@ -158,63 +22,12 @@
|
| 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 = {
|
| - '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)]
|
| + go_filters = {}
|
|
|
| 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)
|
|
|