create aggregateRuntimeImports function

Author: renanmav

package.json

@@ -89,21 +89,21 @@
     "jest": "^24.8.0",
     "jest-cli": "^24.8.0",
     "lint-staged": "^9.2.1",
-    "prettier": "^1.18.2",
+    "prettier": "^1.19.1",
     "relay-compiler": "^7.0.0",
     "relay-runtime": "^7.0.0",
     "relay-test-utils-internal": "^7.0.0",
     "ts-jest": "^22.0.1",
     "tslint": "^5.18.0",
     "tslint-config-prettier": "^1.18.0",
-    "typescript": "3.6.4"
+    "typescript": "^3.7.2"
   },
   "peerDependencies": {
     "@types/react-relay": ">=7.0.0",
     "@types/relay-runtime": ">=6.0.7",
     "relay-compiler": ">=7.0.0",
     "relay-runtime": ">=7.0.0",
-    "typescript": ">=3.6.4"
+    "typescript": ">=3.7.2"
   },
   "publishConfig": {
     "registry": "https://registry.npmjs.org/"

src/TypeScriptGenerator.ts

@@ -48,11 +48,13 @@ const MODULE_IMPORT_FIELD = "MODULE_IMPORT_FIELD";
 const DIRECTIVE_NAME = "raw_response_type";
 
 export const generate: TypeGenerator["generate"] = (schema, node, options) => {
-  const ast: ts.Statement[] = IRVisitor.visit(
+  let ast: ts.Statement[] = IRVisitor.visit(
     node,
     createVisitor(schema, options)
   );
 
+  ast = aggregateRuntimeImports(ast);
+
   const printer = ts.createPrinter({ newLine: ts.NewLineKind.LineFeed });
 
   const resultFile = ts.createSourceFile(
@@ -68,6 +70,57 @@ export const generate: TypeGenerator["generate"] = (schema, node, options) => {
   return printer.printNode(ts.EmitHint.SourceFile, fullProgramAst, resultFile);
 };
 
+function aggregateRuntimeImports(ast: ts.Statement[]) {
+  const importNodes = ast.filter(declaration =>
+    ts.isImportDeclaration(declaration)
+  ) as ts.ImportDeclaration[];
+
+  const runtimeImports = importNodes.filter(
+    importDeclaration =>
+      // @ts-ignore
+      importDeclaration.moduleSpecifier.text === "relay-runtime"
+  );
+
+  if (runtimeImports.length > 1) {
+    const namedImports: string[] = [];
+    runtimeImports.map(node => {
+      // @ts-ignore
+      node.importClause?.namedBindings.elements.map(element => {
+        namedImports.push(element.name.escapedText);
+      });
+    });
+
+    const importSpecifiers: ts.ImportSpecifier[] = [];
+    namedImports.map(namedImport => {
+      const specifier = ts.createImportSpecifier(
+        undefined,
+        ts.createIdentifier(namedImport)
+      );
+      importSpecifiers.push(specifier);
+    });
+
+    const namedBindings = ts.createNamedImports(importSpecifiers);
+    const aggregatedRuntimeImportDeclaration = ts.createImportDeclaration(
+      undefined,
+      undefined,
+      ts.createImportClause(undefined, namedBindings),
+      ts.createStringLiteral("relay-runtime")
+    );
+
+    const aggregatedRuntimeImportAST = ast.reduce<ts.Statement[]>(
+      (prev, curr) => {
+        if (!ts.isImportDeclaration(curr)) prev.push(curr);
+        return prev;
+      },
+      [aggregatedRuntimeImportDeclaration]
+    );
+
+    return aggregatedRuntimeImportAST;
+  } else {
+    return ast;
+  }
+}
+
 function nullthrows<T>(obj: T | null | undefined): T {
   if (obj == null) {
     throw new Error("Obj is null");

test/__snapshots__/TypeScriptGenerator-test.ts.snap

@@ -1535,8 +1535,7 @@ export type RefetchableFragment$key = {
 
 
 // RefetchableFragmentQuery.graphql
-import { FragmentReference } from "relay-runtime";
-import { FragmentRefs } from "relay-runtime";
+import { FragmentReference, FragmentRefs } from "relay-runtime";
 export type RefetchableFragmentQueryVariables = {
     id: string;
 };

yarn.lock

@@ -4854,10 +4854,10 @@ preserve@^0.2.0:
   resolved "https://registry.yarnpkg.com/preserve/-/preserve-0.2.0.tgz#815ed1f6ebc65926f865b310c0713bcb3315ce4b"
   integrity sha1-gV7R9uvGWSb4ZbMQwHE7yzMVzks=
 
-prettier@^1.18.2:
-  version "1.18.2"
-  resolved "https://registry.yarnpkg.com/prettier/-/prettier-1.18.2.tgz#6823e7c5900017b4bd3acf46fe9ac4b4d7bda9ea"
-  integrity sha512-OeHeMc0JhFE9idD4ZdtNibzY0+TPHSpSSb9h8FqtP+YnoZZ1sl8Vc9b1sasjfymH3SonAF4QcA2+mzHPhMvIiw==
+prettier@^1.19.1:
+  version "1.19.1"
+  resolved "https://registry.yarnpkg.com/prettier/-/prettier-1.19.1.tgz#f7d7f5ff8a9cd872a7be4ca142095956a60797cb"
+  integrity sha512-s7PoyDv/II1ObgQunCbB9PdLmUcBZcnWOcxDh7O0N/UwDEsHyqkW+Qh28jW+mVuCdx7gLB0BotYI1Y6uI9iyew==
 
 pretty-format@^22.4.0, pretty-format@^22.4.3:
   version "22.4.3"
@@ -5943,11 +5943,16 @@ type-fest@^0.6.0:
   resolved "https://registry.yarnpkg.com/type-fest/-/type-fest-0.6.0.tgz#8d2a2370d3df886eb5c90ada1c5bf6188acf838b"
   integrity sha512-q+MB8nYR1KDLrgr4G5yemftpMC7/QLqVndBmEEdqzmNj5dcFOO4Oo8qlwZE3ULT3+Zim1F8Kq4cBnikNhlCMlg==
 
-typescript@3.6.4, typescript@^3.0.0:
+typescript@^3.0.0:
   version "3.6.4"
   resolved "https://registry.yarnpkg.com/typescript/-/typescript-3.6.4.tgz#b18752bb3792bc1a0281335f7f6ebf1bbfc5b91d"
   integrity sha512-unoCll1+l+YK4i4F8f22TaNVPRHcD9PA3yCuZ8g5e0qGqlVlJ/8FSateOLLSagn+Yg5+ZwuPkL8LFUc0Jcvksg==
 
+typescript@^3.7.2:
+  version "3.7.2"
+  resolved "https://registry.yarnpkg.com/typescript/-/typescript-3.7.2.tgz#27e489b95fa5909445e9fef5ee48d81697ad18fb"
+  integrity sha512-ml7V7JfiN2Xwvcer+XAf2csGO1bPBdRbFCkYBczNZggrBZ9c7G3riSUeJmqEU5uOtXNPMhE3n+R4FA/3YOAWOQ==
+
 ua-parser-js@^0.7.18:
   version "0.7.20"
   resolved "https://registry.yarnpkg.com/ua-parser-js/-/ua-parser-js-0.7.20.tgz#7527178b82f6a62a0f243d1f94fd30e3e3c21098"