Move server-side service stubs to .pbserver.dart

test/file_generator_test.dart

 #!/usr/bin/env dart
 // Copyright (c) 2013, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 library file_generator_test;
 
 import 'package:protoc_plugin/indenting_writer.dart';
 import 'package:protoc_plugin/src/descriptor.pb.dart';
 import 'package:protoc_plugin/src/plugin.pb.dart';
@@ skipping 279 matching lines @@
         new CodeGeneratorRequest(), new CodeGeneratorResponse());
 
     FileGenerator fg = new FileGenerator(fd);
     link(options, [fg]);
 
     var writer = new IndentingWriter();
     fg.writeMainHeader(writer);
     expect(writer.toString(), expected);
   });
 
-  test('FileGenerator outputs extra imports for a service', () {
-    String expected = r'''
+  test('FileGenerator outputs files for a service', () {
+    String expectedClient = r'''
 ///
 //  Generated code. Do not modify.
 ///
 library test;
 
 import 'dart:async';
 
 import 'package:protobuf/protobuf.dart';
 
+class Empty extends GeneratedMessage {
+  static final BuilderInfo _i = new BuilderInfo('Empty')
+    ..hasRequiredFields = false
+  ;
+
+  Empty() : super();
+  Empty.fromBuffer(List<int> i, [ExtensionRegistry r = ExtensionRegistry.EMPTY]) : super.fromBuffer(i, r);
+  Empty.fromJson(String i, [ExtensionRegistry r = ExtensionRegistry.EMPTY]) : super.fromJson(i, r);
+  Empty clone() => new Empty()..mergeFromMessage(this);
+  BuilderInfo get info_ => _i;
+  static Empty create() => new Empty();
+  static PbList<Empty> createRepeated() => new PbList<Empty>();
+  static Empty getDefault() {
+    if (_defaultInstance == null) _defaultInstance = new _ReadonlyEmpty();
+    return _defaultInstance;
+  }
+  static Empty _defaultInstance;
+  static void $checkItem(Empty v) {
+    if (v is !Empty) checkItemFailed(v, 'Empty');
+  }
+}
+
+class _ReadonlyEmpty extends Empty with ReadonlyMessageMixin {}
+
+class TestApi {

[Søren Gjesse, 2016/05/27 07:34:35]

Have you considered to also split out the client side API to a .pbclient.dart
file? For the use-cases where the client side API is not used. The
.pbclient.dart could re-export some of the other stuff so that only one import
is required when using the client side API.

[skybrian, 2016/05/27 18:27:50]

I considered it but I'm not sure it would accomplish that much? Normally the
requests and responses are in the same .proto file, so the client-side stub
doesn't require any additional imports.

+  RpcClient _client;
+  TestApi(this._client);
+
+  Future<Empty> ping(ClientContext ctx, Empty request) {
+    var emptyResponse = new Empty();
+    return _client.invoke(ctx, 'Test', 'Ping', request, emptyResponse);
+  }
+}
+
+''';
+
+    String expectedServer = r'''
+///
+//  Generated code. Do not modify.
+///
+library test_pbserver;
+
+import 'dart:async';
+
+import 'package:protobuf/protobuf.dart';
+
+import 'test.pb.dart';
+
 import 'test.pbjson.dart';
 
+abstract class TestServiceBase extends GeneratedService {
+  Future<Empty> ping(ServerContext ctx, Empty request);
+
+  GeneratedMessage createRequest(String method) {
+    switch (method) {
+      case 'Ping': return new Empty();
+      default: throw new ArgumentError('Unknown method: $method');
+    }
+  }
+
+  Future<GeneratedMessage> handleCall(ServerContext ctx, String method, GeneratedMessage request) {
+    switch (method) {
+      case 'Ping': return ping(ctx, request);
+      default: throw new ArgumentError('Unknown method: $method');
+    }
+  }
+
+  Map<String, dynamic> get $json => Test$json;
+  Map<String, dynamic> get $messageJson => Test$messageJson;
+}
+
 ''';
+
+    DescriptorProto empty = new DescriptorProto()..name = "Empty";
+
+    ServiceDescriptorProto sd = new ServiceDescriptorProto()
+      ..name = 'Test'
+      ..method.add(new MethodDescriptorProto()
+        ..name = 'Ping'
+        ..inputType = '.Empty'
+        ..outputType = '.Empty');
+
     FileDescriptorProto fd = new FileDescriptorProto()
       ..name = 'test'
-      ..service.add(new ServiceDescriptorProto());
+      ..messageType.add(empty)
+      ..service.add(sd);
 
     var options = parseGenerationOptions(
         new CodeGeneratorRequest(), new CodeGeneratorResponse());
 
     FileGenerator fg = new FileGenerator(fd);
     link(options, [fg]);
 
     var writer = new IndentingWriter();
     fg.writeMainHeader(writer);
-    expect(writer.toString(), expected);
+    expect(fg.generateMainFile(), expectedClient);
+    expect(fg.generateServerFile(), expectedServer);
   });
 
   test('FileGenerator handles field_name options', () {
     // NOTE: Below > 80 cols because it is matching generated code > 80 cols.
     String expected = r'''
 ///
 //  Generated code. Do not modify.
 ///
 library test;
 
@@ skipping 222 matching lines @@
     var request = new CodeGeneratorRequest();
     var response = new CodeGeneratorResponse();
     var options = parseGenerationOptions(request, response);
 
     FileGenerator fg = new FileGenerator(fd);
     link(options, [fg, new FileGenerator(fd1), new FileGenerator(fd2)]);
     expect(fg.generateMainFile(), expected);
     expect(fg.generateJsonFile(), expectedJson);
   });
 }