Allow application of external mixins to generated dart protos.

lib/message_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 MessageGenerator extends ProtobufContainer {
-  // Returns the mixin for this message, or null if none.
-  static PbMixin _getMixin(DescriptorProto desc, PbMixin defaultValue) {
-    if (!desc.hasOptions()) return defaultValue;
-    if (!desc.options.hasExtension(Dart_options.mixin)) return defaultValue;
+  /// Returns the mixin for this message, or null if none.
+  ///
+  /// First searches [declaredMixins], then internal mixins declared by
+  /// [findMixin].
+  static PbMixin _getMixin(DescriptorProto desc,
+      Map<String, PbMixin> declaredMixins, String defaultMixin) {
+    PbMixin getMixinByName(String name) {
+      if (name.isEmpty) return null; // don't use a mixin (override any default)
+      return declaredMixins[name] ?? findMixin(name);
+    }
+
+    if (!desc.hasOptions() || !desc.options.hasExtension(Dart_options.mixin)) {
+      return getMixinByName(defaultMixin);
+    }
 
     String name = desc.options.getExtension(Dart_options.mixin);
-    if (name.isEmpty) return null; // don't use a mixin (override any default)
-    var mixin = findMixin(name);
-    if (mixin == null) {
-      throw ("unknown mixin class: ${name}");
-    }
-    return mixin;
+    return getMixinByName(name);
   }
 
   final String classname;
   final String fqname;
   final PbMixin mixin;
 
   final ProtobufContainer _parent;
   final DescriptorProto _descriptor;
   final List<EnumGenerator> _enumGenerators = <EnumGenerator>[];
   final List<MessageGenerator> _messageGenerators = <MessageGenerator>[];
   final List<ExtensionGenerator> _extensionGenerators = <ExtensionGenerator>[];
 
   // populated by resolve()
   List<ProtobufField> _fieldList;
 
   MessageGenerator(DescriptorProto descriptor, ProtobufContainer parent,
-      PbMixin defaultMixin)
+      Map<String, PbMixin> declaredMixins, String defaultMixin)

[skybrian, 2016/06/23 19:01:42]

Maybe change defaultMixin back to having type PbMixin so we don't need to look
it up again.

[frederikmutzel, 2016/06/27 08:44:31]

Done.

       : _descriptor = descriptor,
         _parent = parent,
         classname = (parent.classname == '')
             ? descriptor.name
             : '${parent.classname}_${descriptor.name}',
         fqname = (parent == null || parent.fqname == null)
             ? descriptor.name
             : (parent.fqname == '.'
                 ? '.${descriptor.name}'
                 : '${parent.fqname}.${descriptor.name}'),
-        mixin = _getMixin(descriptor, defaultMixin) {
+        mixin = _getMixin(descriptor, declaredMixins, defaultMixin) {
     for (EnumDescriptorProto e in _descriptor.enumType) {
       _enumGenerators.add(new EnumGenerator(e, this));
     }
 
     for (DescriptorProto n in _descriptor.nestedType) {
-      _messageGenerators.add(new MessageGenerator(n, this, defaultMixin));
+      _messageGenerators
+          .add(new MessageGenerator(n, this, declaredMixins, defaultMixin));
     }
 
     for (FieldDescriptorProto x in _descriptor.extension) {
       _extensionGenerators.add(new ExtensionGenerator(x, this));
     }
   }
 
   String get package => _parent.package;
 
   /// The generator of the .pb.dart file that will declare this type.
@@ skipping 310 matching lines @@
 
     for (var m in _messageGenerators) {
       m.generateConstants(out);
     }
 
     for (var e in _enumGenerators) {
       e.generateConstants(out);
     }
   }
 }