Initial support for generating client and server stubs for Dart.

lib/file_generator.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.
 
 part of protoc;
 
 class FileGenerator extends ProtobufContainer {
 
   /// Returns the the mixin to use by default in this file,
   /// or null for no mixin by default.
@@ skipping 10 matching lines @@
     return mixin;
   }
 
   final FileDescriptorProto _fileDescriptor;
   final ProtobufContainer _parent;
   final GenerationContext _context;
 
   final List<EnumGenerator> enumGenerators = <EnumGenerator>[];
   final List<MessageGenerator> messageGenerators = <MessageGenerator>[];
   final List<ExtensionGenerator> extensionGenerators = <ExtensionGenerator>[];
+  final List<ClientApiGenerator> clientApiGenerators = <ClientApiGenerator>[];
+  final List<ServiceGenerator> serviceGenerators = <ServiceGenerator>[];
 
   FileGenerator(this._fileDescriptor, this._parent, this._context) {
     _context.register(this);
 
     var defaultMixin = _getDefaultMixin(_fileDescriptor);
 
     // Load and register all enum and message types.
     for (EnumDescriptorProto enumType in _fileDescriptor.enumType) {
       enumGenerators.add(new EnumGenerator(enumType, this, _context));
     }
     for (DescriptorProto messageType in _fileDescriptor.messageType) {
       messageGenerators.add(
           new MessageGenerator(messageType, this, _context, defaultMixin));
     }
     for (FieldDescriptorProto extension in _fileDescriptor.extension) {
       extensionGenerators.add(
           new ExtensionGenerator(extension, this, _context));
     }
+    for (ServiceDescriptorProto service in _fileDescriptor.service) {
+      serviceGenerators.add(new ServiceGenerator(service, this, _context));
+      clientApiGenerators.add(new ClientApiGenerator(service, this, _context));
+    }
   }
 
   String get package => _fileDescriptor.package;
   String get classname => '';
   String get fqname => '.${_fileDescriptor.package}';
 
   // Extract the filename from a URI and remove the extension.
   String _fileNameWithoutExtension(Uri filePath) {
     String fileName = filePath.pathSegments.last;
     int index = fileName.lastIndexOf(".");
@@ skipping 24 matching lines @@
 
   void generate(IndentingWriter out) {
     Uri filePath = new Uri.file(_fileDescriptor.name);
     if (filePath.isAbsolute) {
         // protoc should never generate a file descriptor with an absolute path.
         throw("FAILURE: File with an absolute path is not supported");
     }
 
     String libraryName = _generateLibraryName(filePath);
 
+    // Print header and imports. We only add the dart:async import if there
+    // are services in the FileDescriptorProto.
     out.println(
       '///\n'
       '//  Generated code. Do not modify.\n'
       '///\n'
-      'library $libraryName;\n'
-      '\n'
+      'library $libraryName;\n');
+    if (_fileDescriptor.service.isNotEmpty) {
+      out.println("import 'dart:async';\n");
+    }
+    out.println(
       "import 'package:fixnum/fixnum.dart';\n"
       "import 'package:protobuf/protobuf.dart';"
     );
 
     var mixinImports = findMixinsToImport();
     var importNames = mixinImports.keys.toList();
     importNames.sort();
     for (var imp in importNames) {
       var symbols = mixinImports[imp];
       out.println("import '${imp}' show ${symbols.join(', ')};");
@@ skipping 42 matching lines @@
           x.generate(out);
         }
         out.println('static void registerAllExtensions(ExtensionRegistry '
             'registry) {');
         for (ExtensionGenerator x in extensionGenerators) {
           out.println('  registry.add(${x.name});');
         }
         out.println('}');
       });
     }
+
+    for (ClientApiGenerator c in clientApiGenerators) {
+      c.generate(out);
+    }
+    for (ServiceGenerator s in serviceGenerators) {
+      s.generate(out);
+    }
   }
 
   /// Returns a map from import names to the Dart symbols to be imported.
   Map<String, List<String>> findMixinsToImport() {
     var mixins = new Set<PbMixin>();
     for (MessageGenerator m in messageGenerators) {
       m.addMixinsTo(mixins);
     }
 
     var imports = {};
@@ skipping 29 matching lines @@
     _registry[container.fqname] = container;
     if (container is FileGenerator) {
       _files[container._fileDescriptor.name] = container;
     }
   }
 
   ProtobufContainer operator [](String fqname) => _registry[fqname];
 
   FileGenerator lookupFile(String name) => _files[name];
 }