fix: Don't remove when. (#361)

Author: budde377

packages/graphql_codegen/CHANGELOG.md

@@ -1,3 +1,7 @@
+# 1.0.1
+
+- Don't remove `when` and `maybeWhen` when `disableCopyWithGeneration = true`.
+
 # 1.0.0
 
 - Supports `@oneOf` directive on input.

packages/graphql_codegen/example/pubspec.lock

@@ -342,7 +342,7 @@ packages:
       path: ".."
       relative: true
     source: path
-    version: "1.0.0"
+    version: "1.0.1"
   graphql_flutter:
     dependency: "direct main"
     description:

packages/graphql_codegen/lib/src/printer/base/document.dart

@@ -151,10 +151,6 @@ Class printContext(PrintContext c) {
 List<Spec> printContextExtension(PrintContext c) {
   final context = c.context;
 
-  if (context.config.disableCopyWithGeneration) {
-    return [];
-  }
-
   final properties = c.context.properties;
 
   final whenMethod = _printWhen(
@@ -168,26 +164,29 @@ List<Spec> printContextExtension(PrintContext c) {
     context.typenameProperty,
     context.possibleTypes,
   );
-
+  final methods = [
+    if (!context.config.disableCopyWithGeneration)
+      _printCopyWithMethod(
+        c.namePrinter.printClassName(context.path),
+        c,
+      ),
+    if (whenMethod != null) whenMethod,
+    if (maybeWhenMethod != null) maybeWhenMethod,
+  ];
   return [
-    Extension(
-      (b) => b
-        ..name = c.namePrinter.printClassExtensionName(context.path)
-        ..on = refer(c.namePrinter.printClassName(context.path))
-        ..methods = ListBuilder([
-          _printCopyWithMethod(
-            c.namePrinter.printClassName(context.path),
-            c,
-          ),
-          if (whenMethod != null) whenMethod,
-          if (maybeWhenMethod != null) maybeWhenMethod,
-        ]),
-    ),
-    ...printCopyWithClasses(
-      c,
-      c.namePrinter.printClassName(context.path),
-      properties,
-    ),
+    if (methods.isNotEmpty)
+      Extension(
+        (b) => b
+          ..name = c.namePrinter.printClassExtensionName(context.path)
+          ..on = refer(c.namePrinter.printClassName(context.path))
+          ..methods = ListBuilder(methods),
+      ),
+    if (!context.config.disableCopyWithGeneration)
+      ...printCopyWithClasses(
+        c,
+        c.namePrinter.printClassName(context.path),
+        properties,
+      ),
   ];
 }
 

packages/graphql_codegen/pubspec.yaml

@@ -3,7 +3,7 @@ description: |
   Simple, opinionated, codegen library for GraphQL. It allows you to
   generate serializers and client helpers to easily call and parse your data.
 
-version: 1.0.0
+version: 1.0.1
 homepage: https://github.com/heftapp/graphql_codegen/tree/main/packages/graphql_codegen
 repository: https://github.com/heftapp/graphql_codegen/tree/main/packages/graphql_codegen
 

packages/graphql_codegen/test/assets/disable_copy_with/document.graphql

@@ -1,13 +1,39 @@
 scalar DateTime
 
-type Query {
+input DataQuery {
   time: DateTime
 }
 
-query FetchScalars {
-  time
+type Query {
+  data(input: DataQuery): I
 }
 
-input CreateScalar {
-  time: DateTime
+
+interface I {
+  value: String
+}
+
+type A implements I {
+  value: String
+}
+
+type B implements I {
+  value: String
 }
+
+
+fragment FA on A {
+  value
+}
+
+fragment FB on B {
+  value
+}
+
+query FetchScalars {
+  data {
+    ...FA
+    ...FB
+  }
+}
+