Dart: Refactor Stubs

mojo/public/tools/bindings/generators/dart_templates/interface_definition.tmpl

 {%- for method in interface.methods %}
 const int _{{interface|name|lower_camel}}Method{{method|name|upper_camel}}Name = {{method.ordinal}};
 {%- endfor %}
 
 {#--- Interface Enums #}
 {%- from "enum_definition.tmpl" import enum_def -%}
 {%- for enum in interface.enums %}
   {{ enum_def(enum, typepkg, package) }}
 {%-  endfor %}
 
@@ skipping 44 matching lines @@
     [Function responseFactory = null]);
 {%- endif %}
 {%- endfor %}
 
 {#--- Interface Constants #}
 {%- for constant in interface.constants %}
   static const {{constant.kind|dart_type}} {{constant|name}} = {{constant.resolved_value}};
 {%-   endfor %}
 }
 
-
-class _{{interface|name}}ProxyControl extends bindings.ProxyMessageHandler
-                                      implements bindings.ProxyControl {
+class _{{interface|name}}ProxyControl
+    extends bindings.ProxyMessageHandler
+    implements bindings.ProxyControl {
   _{{interface|name}}ProxyControl.fromEndpoint(
       core.MojoMessagePipeEndpoint endpoint) : super.fromEndpoint(endpoint);
 
   _{{interface|name}}ProxyControl.fromHandle(
       core.MojoHandle handle) : super.fromHandle(handle);
 
   _{{interface|name}}ProxyControl.unbound() : super.unbound();
 
   {{descpkg}}ServiceDescription get serviceDescription =>
       new _{{interface|name}}ServiceDescription();
 
   String get serviceName => {{interface|name}}.serviceName;
 
-  @override
   void handleResponse(bindings.ServiceMessage message) {
     switch (message.header.type) {
 {%- for method in interface.methods %}
 {%- if method.response_parameters != None %}
 {%- set response_struct = method.response_param_struct %}
       case _{{interface|name|lower_camel}}Method{{method|name|upper_camel}}Name:
         var r = {{response_struct|name}}.deserialize(
             message.payload);
         if (!message.header.hasRequestId) {
           proxyError("Expected a message with a valid request Id.");
@@ skipping 21 matching lines @@
     }
   }
 
   @override
   String toString() {
     var superString = super.toString();
     return "_{{interface|name}}ProxyControl($superString)";
   }
 }
 
-
-class {{interface|name}}Proxy extends bindings.Proxy
-                              implements {{interface|name}} {
+class {{interface|name}}Proxy
+    extends bindings.Proxy
+    implements {{interface|name}} {
   {{interface|name}}Proxy.fromEndpoint(
       core.MojoMessagePipeEndpoint endpoint)
       : super(new _{{interface|name}}ProxyControl.fromEndpoint(endpoint));
 
   {{interface|name}}Proxy.fromHandle(core.MojoHandle handle)
       : super(new _{{interface|name}}ProxyControl.fromHandle(handle));
 
   {{interface|name}}Proxy.unbound()
       : super(new _{{interface|name}}ProxyControl.unbound());
 
@@ skipping 43 matching lines @@
     return ctrl.sendMessageWithRequestId(
         params,
         _{{interface|name|lower_camel}}Method{{method|name|upper_camel}}Name,
         -1,
         bindings.MessageHeader.kMessageExpectsResponse);
   }
 {%- endif %}
 {%- endfor %}
 }
 
-
-class {{interface|name}}Stub extends bindings.Stub {
+class _{{interface|name}}StubControl
+    extends bindings.StubMessageHandler
+    implements bindings.StubControl<{{interface|name}}> {
   {{interface|name}} _impl;
 
-  {{interface|name}}Stub.fromEndpoint(
+  _{{interface|name}}StubControl.fromEndpoint(
       core.MojoMessagePipeEndpoint endpoint, [{{interface|name}} impl])
       : super.fromEndpoint(endpoint, autoBegin: impl != null) {
     _impl = impl;
   }
 
-  {{interface|name}}Stub.fromHandle(
+  _{{interface|name}}StubControl.fromHandle(
       core.MojoHandle handle, [{{interface|name}} impl])
       : super.fromHandle(handle, autoBegin: impl != null) {
     _impl = impl;
   }
 
-  {{interface|name}}Stub.unbound([this._impl]) : super.unbound();
-
-  static {{interface|name}}Stub newFromEndpoint(
-      core.MojoMessagePipeEndpoint endpoint) {
-    assert(endpoint.setDescription("For {{interface|name}}Stub"));
-    return new {{interface|name}}Stub.fromEndpoint(endpoint);
-  }
+  _{{interface|name}}StubControl.unbound([this._impl]) : super.unbound();
 
 {% for method in interface.methods %}
 {%- if method.response_parameters != None %}
 {%- set response_struct = method.response_param_struct %}
   {{response_struct|name}} _{{response_struct|name|lower_camel}}Factory(
     {%- for param in method.response_parameters -%}
       {{param.kind|dart_type}} {{param|name}}{% if not loop.last %}, {% endif %}
     {%- endfor -%}
   ) {
     var result = new {{response_struct|name}}();
@@ skipping 74 matching lines @@
   }
 
   @override
   void bind(core.MojoMessagePipeEndpoint endpoint) {
     super.bind(endpoint);
     if (!isOpen && (_impl != null)) {
       beginHandlingEvents();
     }
   }
 
+  @override
   String toString() {
     var superString = super.toString();
-    return "{{interface|name}}Stub($superString)";
+    return "_{{interface|name}}StubControl($superString)";
   }
 
   int get version => {{interface.version}};
 
   static {{descpkg}}ServiceDescription _cachedServiceDescription;
   static {{descpkg}}ServiceDescription get serviceDescription {
     if (_cachedServiceDescription == null) {
       _cachedServiceDescription = new _{{interface|name}}ServiceDescription();
     }
     return _cachedServiceDescription;
   }
 }
+
+class {{interface|name}}Stub
+    extends bindings.Stub<{{interface|name}}>
+    implements {{interface|name}} {
+  {{interface|name}}Stub.fromEndpoint(
+      core.MojoMessagePipeEndpoint endpoint, [{{interface|name}} impl])
+      : super(new _{{interface|name}}StubControl.fromEndpoint(endpoint, impl));
+
+  {{interface|name}}Stub.fromHandle(
+      core.MojoHandle handle, [{{interface|name}} impl])
+      : super(new _{{interface|name}}StubControl.fromHandle(handle, impl));
+
+  {{interface|name}}Stub.unbound([{{interface|name}} impl])
+      : super(new _{{interface|name}}StubControl.unbound(impl));
+
+  static {{interface|name}}Stub newFromEndpoint(
+      core.MojoMessagePipeEndpoint endpoint) {
+    assert(endpoint.setDescription("For {{interface|name}}Stub"));
+    return new {{interface|name}}Stub.fromEndpoint(endpoint);
+  }
+
+  static {{descpkg}}ServiceDescription get serviceDescription =>
+      _{{interface|name}}StubControl.serviceDescription;
+
+{% for method in interface.methods %}
+{%- if method.response_parameters == None %}
+  void {{method|name}}(
+{%- for parameter in method.parameters -%}
+    {{parameter.kind|dart_type}} {{parameter|name}}{% if not loop.last %}, {% endif %}
+{%- endfor -%}
+  ) {
+    return impl.{{method|name}}(
+{%- for parameter in method.parameters -%}
+    {{parameter|name}}{% if not loop.last %}, {% endif %}
+{%- endfor -%}
+    );
+  }
+{%- else %}
+  dynamic {{method|name}}(
+{%- for parameter in method.parameters -%}
+    {{parameter.kind|dart_type}} {{parameter|name}},
+{%- endfor -%}
+    [Function responseFactory = null]) {
+    return impl.{{method|name}}(
+{%- for parameter in method.parameters -%}
+    {{parameter|name}},
+{%- endfor -%}
+    responseFactory);
+  }
+{%- endif %}
+{%- endfor %}
+}