Generate Mojom Types in Go

mojo/public/tools/bindings/generators/go_templates/source.tmpl

 // 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.
 
 // This file is autogenerated by:
 //     mojo/public/tools/bindings/mojom_bindings_generator.py
 // For:
 //     {{module.path}}
 //
 
 package {{package}}
 
 import (
 {% for i in imports %}
         {{i}}
 {% endfor %}
 )
 
 {% import "enum.tmpl" as enum_macros %}
 {% import "interface.tmpl" as interface_macros %}
 {% import "struct.tmpl" as struct_macros %}
 {% import "union.tmpl" as union_macros %}
+{% import "mojom_reference_macros.tmpl" as mojom_reference_macros %}
+{% import "mojom_util_macros.tmpl" as util %}
 
-{#- Enum definitions #}
+{#- This init function initializes a mapping between identifiers and their
+    corresponding user-defined type. This could have been a MojomDescriptor,
+    but since it allows Types to be unresolved, it is more complex than is
+    necessary. It's simpler to resolve everything in codegen.
+    It is also difficult to insert changes to the description during the
+    process of generating each type, so this mapping is computed in Python.
+    Note: While we may register items into the map multiple times, this is okay
+    because the keys will match.
+-#}
+
 {% for enum in enums %}
-{{enum_macros.define(enum)}}
+var ID_{{util.enumName(enum, typepkg, package)}} string = "{{util.enumName(enum, typepkg, package)}}"
+{% endfor %}
+{% for struct in structs %}
+var ID_{{util.structName(struct, typepkg, package)}} string = "{{util.structName(struct, typepkg, package)}}"
+{% endfor %}
+{% for union in unions %}
+var ID_{{util.unionName(union, typepkg, package)}} string = "{{util.unionName(union, typepkg, package)}}"
+{% endfor %}
+{% for interface in interfaces %}
+var ID_{{util.interfaceName(interface, typepkg, package)}} string = "{{util.interfaceName(interface, typepkg, package)}}"
+{% endfor %}
+
+{%- if need_mojom_desc -%}
+{% set mapping = package|qualified(None, true) ~ 'Desc__' %}
+var {{mapping}} = make(map[string]{{descpkg}}UserDefinedType)
+func init() {
+  {% for enum in enums %}
+  {{mojom_reference_macros.registerType(mapping, descpkg, typepkg, package, enum)}}
+  {%- endfor %}
+  {% for struct in structs %}
+  {{mojom_reference_macros.registerType(mapping, descpkg, typepkg, package, struct)}}
+  {%- endfor %}
+  {% for union in unions %}
+  {{mojom_reference_macros.registerType(mapping, descpkg, typepkg, package, union)}}
+  {%- endfor %}
+  {% for interface in interfaces %}
+  {{mojom_reference_macros.registerType(mapping, descpkg, typepkg, package, interface)}}
+  {%- endfor %}
+
+  {% for mi in mojom_imports.values() %}
+  for s, udt := range {{mi}}.Descriptor() {
+    {{mapping}}[s] = udt
+  }
+  {% endfor %}
+}
+func Descriptor() map[string]{{descpkg}}UserDefinedType {
+  return {{mapping}}
+}
+{% endif %}
+
+{# Enum definitions #}
+{%- for enum in enums %}
+{{enum_macros.define(enum, typepkg, package)}}
 {%- endfor %}
 
 {#- Interface definitions #}
 {% for interface in interfaces %}
-{{interface_macros.define(interface)}}
+{{interface_macros.define(interface, typepkg, package)}}
 {%- endfor %}
 
 {#- Struct definitions #}
 {% for struct in structs %}
-{{struct_macros.define(struct)}}
+{{struct_macros.define(struct, typepkg, package)}}
 {%- endfor %}
 
 {#- Union definitions #}
 {% for union in unions %}
-{{union_macros.define(union)}}
+{{union_macros.define(union, typepkg, package)}}
 {%- endfor %}