Replace stored with computed properties in input object structs

Fixes #185, fixes apollographql/apollo-ios#108.
apollographql · Jul 30, 2017 · dac1eb2 · dac1eb2
1 parent d8b4180
commit dac1eb2
Showing 1 changed file with 28 additions and 7 deletions.
diff --git a/src/swift/codeGeneration.js b/src/swift/codeGeneration.js
@@ -581,20 +581,41 @@ function structDeclarationForInputObjectType(generator, type) {
   });
 
   structDeclaration(generator, { structName, description, adoptedProtocols }, () => {
-    propertyDeclarations(generator, properties);
+    generator.printOnNewline(`public var graphQLMap: GraphQLMap`);
 
     generator.printNewlineIfNeeded();
+    generator.printOnNewline(`public init`);
+    generator.print('(');
+    generator.print(join(properties.map(({ propertyName, type, typeName, isOptional }) =>
+      join([
+        `${propertyName}: ${typeName}`,
+        isOptional && ' = nil'
+      ])
+    ), ', '));
+    generator.print(')');
 
-    initializerDeclarationForProperties(generator, properties);
-
-    generator.printNewlineIfNeeded();
-    generator.printOnNewline(`public var graphQLMap: GraphQLMap`);
     generator.withinBlock(() => {
       generator.printOnNewline(wrap(
-        `return [`,
+        `graphQLMap = [`,
         join(properties.map(({ name, propertyName }) => `"${name}": ${propertyName}`), ', ') || ':',
-        `]`,
+        `]`
       ));
     });
+
+    for (const { propertyName, typeName, description } of properties) {
+      generator.printNewlineIfNeeded();
+      comment(generator, description);
+      generator.printOnNewline(`public var ${propertyName}: ${typeName}`);
+      generator.withinBlock(() => {
+        generator.printOnNewline("get");
+        generator.withinBlock(() => {
+          generator.printOnNewline(`return graphQLMap["${propertyName}"] as! ${typeName}`);
+        });
+        generator.printOnNewline("set");
+        generator.withinBlock(() => {
+          generator.printOnNewline(`graphQLMap.updateValue(newValue, forKey: "${propertyName}")`);
+        });
+      });
+    }
   });
 }