feat: add FragmentRefs import

Author: renanmav

example/ts/__relay_artifacts__/TodoListFooter_viewer.graphql.ts

@@ -1,6 +1,7 @@
 /* tslint:disable */
 
 import { ReaderFragment } from "relay-runtime";
+import { FragmentRefs } from "relay-runtime";
 export type TodoListFooter_viewer = {
     readonly id: string;
     readonly completedCount: number | null;

example/ts/__relay_artifacts__/Todo_todo.graphql.ts

@@ -1,6 +1,7 @@
 /* tslint:disable */
 
 import { ReaderFragment } from "relay-runtime";
+import { FragmentRefs } from "relay-runtime";
 export type Todo_todo = {
     readonly complete: boolean | null;
     readonly id: string;

example/ts/__relay_artifacts__/Todo_viewer.graphql.ts

@@ -1,47 +1,50 @@
 /* tslint:disable */
 
-import { ReaderFragment } from "relay-runtime"
+import { ReaderFragment } from "relay-runtime";
+import { FragmentRefs } from "relay-runtime";
 export type Todo_viewer = {
-  readonly id: string
-  readonly totalCount: number | null
-  readonly completedCount: number | null
-  readonly " $refType": "Todo_viewer"
-}
-export type Todo_viewer$data = Todo_viewer
+    readonly id: string;
+    readonly totalCount: number | null;
+    readonly completedCount: number | null;
+    readonly " $refType": "Todo_viewer";
+};
+export type Todo_viewer$data = Todo_viewer;
 export type Todo_viewer$key = {
-  readonly " $data"?: Todo_viewer$data
-  readonly " $fragmentRefs": FragmentRefs<"Todo_viewer">
-}
+    readonly " $data"?: Todo_viewer$data;
+    readonly " $fragmentRefs": FragmentRefs<"Todo_viewer">;
+};
+
+
 
 const node: ReaderFragment = {
-  kind: "Fragment",
-  name: "Todo_viewer",
-  type: "User",
-  metadata: null,
-  argumentDefinitions: [],
-  selections: [
+  "kind": "Fragment",
+  "name": "Todo_viewer",
+  "type": "User",
+  "metadata": null,
+  "argumentDefinitions": [],
+  "selections": [
     {
-      kind: "ScalarField",
-      alias: null,
-      name: "id",
-      args: null,
-      storageKey: null,
+      "kind": "ScalarField",
+      "alias": null,
+      "name": "id",
+      "args": null,
+      "storageKey": null
     },
     {
-      kind: "ScalarField",
-      alias: null,
-      name: "totalCount",
-      args: null,
-      storageKey: null,
+      "kind": "ScalarField",
+      "alias": null,
+      "name": "totalCount",
+      "args": null,
+      "storageKey": null
     },
     {
-      kind: "ScalarField",
-      alias: null,
-      name: "completedCount",
-      args: null,
-      storageKey: null,
-    },
-  ],
-}
-;(node as any).hash = "1e2b17bb7b92d4521c4e72309d996339"
-export default node
+      "kind": "ScalarField",
+      "alias": null,
+      "name": "completedCount",
+      "args": null,
+      "storageKey": null
+    }
+  ]
+};
+(node as any).hash = '1e2b17bb7b92d4521c4e72309d996339';
+export default node;

src/TypeScriptGenerator.ts

@@ -334,7 +334,8 @@ function createVisitor(options: TypeGeneratorOptions): IRVisitor.NodeVisitor {
     useHaste: options.useHaste,
     useSingleArtifactDirectory: options.useSingleArtifactDirectory,
     noFutureProofEnums: options.noFutureProofEnums,
-    matchFields: new Map()
+    matchFields: new Map(),
+    runtimeImports: new Set()
   };
 
   return {
@@ -466,10 +467,11 @@ function createVisitor(options: TypeGeneratorOptions): IRVisitor.NodeVisitor {
               baseType
             ])
           : baseType;
+        state.runtimeImports.add("FragmentRefs");
 
         return [
-          ...getFragmentRefsTypeImport(state),
           ...getEnumDefinitions(state),
+          importTypes(Array.from(state.runtimeImports).sort(), "relay-runtime"),
           exportType(node.name, type),
           exportType(dataTypeName, dataType),
           exportType(

src/TypeScriptTypeTransformers.ts

@@ -25,6 +25,7 @@ export type State = {
   };
   generatedFragments: Set<string>;
   matchFields: Map<string, ts.TypeNode>;
+  runtimeImports: Set<string>;
 } & TypeGeneratorOptions;
 
 function getInputObjectTypeIdentifier(type: GraphQLInputObjectType): string {