Squashed 'apollo-ios-codegen/' changes from 7b4652c1..1b1661ba

1b1661ba fix: Local cache mutation generation when field merging disabled (#654)

git-subtree-dir: apollo-ios-codegen
git-subtree-split: 1b1661ba166f6cbb857c003d016e5891f4a6c554

Author: gh-action-runner

Sources/ApolloCodegenLib/ApolloCodegen.swift

@@ -299,34 +299,68 @@ public class ApolloCodegen {
     let mergeNamedFragmentFields = config.experimentalFeatures.fieldMerging.options
       .contains(.namedFragments)
 
+    /// A `ConfigurationContext` to use when generated local cache mutations.
+    ///
+    /// Local cache mutations require some codegen options to be overridden to generate valid objects. 
+    /// This context overrides only the necessary properties, copying all other values from the user-provided `context`.
+    lazy var cacheMutationContext: ConfigurationContext = {
+      ConfigurationContext(
+        config: ApolloCodegenConfiguration(
+          schemaNamespace: self.config.schemaNamespace,
+          input: self.config.input,
+          output: self.config.output,
+          options: self.config.options,
+          experimentalFeatures: ApolloCodegenConfiguration.ExperimentalFeatures(
+            fieldMerging: .all,
+            legacySafelistingCompatibleOperations: self.config.experimentalFeatures.legacySafelistingCompatibleOperations
+          ),
+          schemaDownload: self.config.schemaDownload,
+          operationManifest: self.config.operationManifest
+        ),
+        rootURL: self.config.rootURL
+      )
+    }()
+
     return try await nonFatalErrorCollectingTaskGroup() { group in
       for fragment in fragments {
+        let fragmentConfig = fragment.isLocalCacheMutation ? cacheMutationContext : self.config
+
         group.addTask {
           let irFragment = await ir.build(
             fragment: fragment,
-            mergingNamedFragmentFields: mergeNamedFragmentFields
+            mergingNamedFragmentFields: fragment.isLocalCacheMutation ? true : mergeNamedFragmentFields
           )
 
-          let errors = try await FragmentFileGenerator(irFragment: irFragment, config: self.config)
-            .generate(forConfig: self.config, fileManager: fileManager)
+          let errors = try await FragmentFileGenerator(
+            irFragment: irFragment,
+            config: fragmentConfig
+          ).generate(
+            forConfig: fragmentConfig,
+            fileManager: fileManager
+          )
           return (irFragment.name, errors)
         }
       }
 
       for operation in operations {
+        let operationConfig = operation.isLocalCacheMutation ? cacheMutationContext : self.config
+
         group.addTask {
           async let identifier = self.operationIdentifierFactory.identifier(for: operation)
 
           let irOperation = await ir.build(
             operation: operation,
-            mergingNamedFragmentFields: mergeNamedFragmentFields
+            mergingNamedFragmentFields: operation.isLocalCacheMutation ? true : mergeNamedFragmentFields
           )
 
           let errors = try await OperationFileGenerator(
             irOperation: irOperation,
             operationIdentifier: await identifier,
-            config: self.config
-          ).generate(forConfig: self.config, fileManager: fileManager)
+            config: operationConfig
+          ).generate(
+            forConfig: operationConfig,
+            fileManager: fileManager
+          )
           return (irOperation.name, errors)
         }
       }

Sources/ApolloCodegenLib/CodegenConfiguration/ApolloCodegenConfiguration.swift

@@ -1434,30 +1434,35 @@ extension ApolloCodegenConfiguration.OperationsFileOutput {
 extension ApolloCodegenConfiguration {
   /// Determine whether the operations files are output to the schema types module.
   func shouldGenerateSelectionSetInitializers(for operation: IR.Operation) -> Bool {
-    guard experimentalFeatures.fieldMerging == .all else { return false }
-
     if operation.definition.isLocalCacheMutation {
       return true
 
-    } else if options.selectionSetInitializers.contains(.operations) {
-      return true
-
     } else {
-      return options.selectionSetInitializers.contains(definitionNamed: operation.definition.name)
+      guard experimentalFeatures.fieldMerging == .all else { return false }
+
+      if options.selectionSetInitializers.contains(.operations) {
+        return true
+
+      } else {
+        return options.selectionSetInitializers.contains(definitionNamed: operation.definition.name)
+      }
     }
   }
 
   /// Determine whether the operations files are output to the schema types module.
   func shouldGenerateSelectionSetInitializers(for fragment: IR.NamedFragment) -> Bool {
-    guard experimentalFeatures.fieldMerging == .all else { return false }
-
-    if options.selectionSetInitializers.contains(.namedFragments) { return true }
-
     if fragment.definition.isLocalCacheMutation {
       return true
-    }
 
-    return options.selectionSetInitializers.contains(definitionNamed: fragment.definition.name)
+    } else {
+      guard experimentalFeatures.fieldMerging == .all else { return false }
+
+      if options.selectionSetInitializers.contains(.namedFragments) {
+        return true
+      } else {
+        return options.selectionSetInitializers.contains(definitionNamed: fragment.definition.name)
+      }
+    }
   }
 }