Add support for devOnlyAssignments

src/formatGeneratedModule.ts

@@ -5,14 +5,13 @@ import addAnyTypeCast from "./addAnyTypeCast";
 export const formatterFactory = (
   compilerOptions: ts.CompilerOptions = {}
 ): FormatModule => ({
-  moduleName,
   documentType,
   docText,
   concreteText,
   typeText,
-  hash,
   relayRuntimeModule = "relay-runtime",
-  sourceHash
+  sourceHash,
+  devOnlyAssignments
 }) => {
   const documentTypeImport = documentType
     ? `import { ${documentType} } from "${relayRuntimeModule}";`
@@ -23,13 +22,23 @@ export const formatterFactory = (
   if (compilerOptions.noImplicitAny) {
     nodeStatement = addAnyTypeCast(nodeStatement).trim();
   }
+  const devOnlyAssignmentsText =
+    devOnlyAssignments != null && devOnlyAssignments.length > 0
+      ? `\nif (process.env.NODE_ENV !== 'production') {\n  ${devOnlyAssignments.replace(
+          // `devOnlyAssignments` uses a flow comment, we can simply replace it with a
+          // ts cast.
+          "(node/*: any*/)",
+          "(node as any)"
+        )}\n}`
+      : "";
   return `/* tslint:disable */
 
 ${documentTypeImport}
 ${typeText || ""}
 
 ${docTextComment}
 ${nodeStatement}
+${devOnlyAssignmentsText}
 (node as any).hash = '${sourceHash}';
 export default node;
 `;

test/fixtures/generated-module/complete-example-no-cast.ts

@@ -5,5 +5,6 @@ export type CompleteExample = { readonly id: string }
 
 
 const node: ConcreteFragment = {"the":{"fragment":{"data":42}}};
+
 (node as any).hash = 'edcba';
 export default node;

test/fixtures/generated-module/complete-example.ts

@@ -5,5 +5,6 @@ export type CompleteExample = { readonly id: string }
 
 
 const node: ConcreteFragment = ({ "the": { "fragment": { "data": 42 } } } as any);
+
 (node as any).hash = 'edcba';
 export default node;

test/fixtures/generated-module/dev-only-assignments.ts

@@ -0,0 +1,13 @@
+/* tslint:disable */
+
+import { ConcreteFragment } from "relay-runtime";
+export type CompleteExample = { readonly id: string }
+
+
+const node: ConcreteFragment = {"the":{"fragment":{"data":42}}};
+
+if (process.env.NODE_ENV !== 'production') {
+  (node as any).params.text = "query CompleteExampleQuery { id }";
+}
+(node as any).hash = 'edcba';
+export default node;

test/formatGeneratedModule-test.ts

@@ -75,4 +75,18 @@ describe('formatGeneratedModule', () => {
       sourceHash: 'edcba',
     })).toMatchFile(join(__dirname, 'fixtures/generated-module/complete-example-no-cast.ts'))
   })
+
+  it('works with devOnlyAssignments', () => {
+    const formatGeneratedModule = formatterFactory();
+    expect(formatGeneratedModule({
+      moduleName: 'complete-example',
+      documentType: 'ConcreteFragment',
+      docText: null,
+      concreteText: JSON.stringify({ the: { fragment: { data: 42 } }}),
+      typeText: 'export type CompleteExample = { readonly id: string }',
+      hash: 'abcde',
+      sourceHash: 'edcba',
+      devOnlyAssignments: '(node/*: any*/).params.text = "query CompleteExampleQuery { id }";',
+    })).toMatchFile(join(__dirname, 'fixtures/generated-module/dev-only-assignments.ts'))
+  })
 })