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 {
   // This should match the extension in dart_options.proto.
   static const int implementMapByDefaultOption = 95333044;
 
   // Returns true if option implement_map_by_default is on for this file.
   static bool _shouldImplementMapByDefault(FileDescriptorProto desc) {
     if (!desc.hasOptions()) return false;
 
     var val = desc.options.unknownFields.getField(implementMapByDefaultOption);
     if (val == null || val.length != 1) return false;
 
     return val.values[0] == 1;
   }
 
   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);
 
     bool implementMap = _shouldImplementMapByDefault(_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, implementMap));
     }
     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';"
     );
 
     if (needsMapMixinImport) {
       out.println("import 'dart:collection' show MapMixin;");
     }
 
     for (String import in _fileDescriptor.dependency) {
       Uri importPath = new Uri.file(import);
@@ skipping 38 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);
+    }
   }
 
   bool get needsMapMixinImport {
     for (var m in messageGenerators) {
       if (m.needsMapMixinImport) {
         return true;
       }
     }
     return false;
   }
@@ skipping 13 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];
 }