Make interfaced objects override resolveType

comigor · Jun 11, 2019 · a954a22 · a954a22
1 parent 5d4e7d4
commit a954a22
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/lib/helpers.dart b/lib/helpers.dart
@@ -73,7 +73,11 @@ void _generateClassProperty(StringBuffer buffer, GraphQLSchema schema,
   buffer.writeln('  $dartTypeStr ${field.name};');
 }
 
-void _generateResolveTypeProperty(StringBuffer buffer) {
+void _generateResolveTypeProperty(StringBuffer buffer,
+    {bool override = false}) {
+  if (override) {
+    buffer.writeln('  @override');
+  }
   buffer.writeln('''  @JsonKey(name: '__resolveType')
   String resolveType;''');
 }
@@ -165,8 +169,10 @@ void _generateClass(StringBuffer buffer, GraphQLSchema schema, GraphQLType type,
       buffer.writeln('@JsonSerializable()');
       buffer.writeln('class $className $mixins {');
 
-      if (type.kind == GraphQLTypeKind.INTERFACE) {
-        _generateResolveTypeProperty(buffer);
+      if (type.kind == GraphQLTypeKind.INTERFACE ||
+          (type.kind == GraphQLTypeKind.OBJECT && type.interfaces.isNotEmpty)) {
+        _generateResolveTypeProperty(buffer,
+            override: type.kind == GraphQLTypeKind.OBJECT);
       }
 
       final interfaceFields = type.interfaces