Merge pull request #250 from nikodevv/bugfix-add-node-missing-imports

bugfix: Added fetch and websockets imports for node environments.

Author: aexol

src/TreeToTS/index.ts

@@ -10,6 +10,7 @@ import {
   graphqlErrorTypeScript,
   typescriptFunctions,
 } from './templates/typescript';
+import { commonImports, envSpecificImports } from './templates/typescript/indexImports';
 
 export interface OperationName {
   name: string;
@@ -124,7 +125,7 @@ export class TreeToTS {
   }) {
     const operations = bodyTypeScript(env, TreeToTS.resolveOperations(tree));
     return {
-      indexImports: `import { AllTypesProps, ReturnTypes } from './const${esModule ? '.js' : ''}';`,
+      indexImports: commonImports(esModule).concat(envSpecificImports(env)),
       const: TreeToTS.resolveBasisCode(tree),
       index: TreeToTS.resolveBasisTypes(tree)
         .concat(graphqlErrorTypeScript.concat('\n').concat(constantTypesTypescript).concat('\n\n'))

src/TreeToTS/templates/typescript/browser/fetchImport.ts

@@ -0,0 +1 @@
+export default ``;

src/TreeToTS/templates/typescript/browser/websocketsImport.ts

@@ -0,0 +1 @@
+export default ``;

src/TreeToTS/templates/typescript/indexImports.ts

@@ -0,0 +1,7 @@
+export const commonImports = (esModule?: boolean): string =>
+  `import { AllTypesProps, ReturnTypes } from './const${esModule ? '.js' : ''}';`;
+
+export const envSpecificImports = (env: string) => `
+${require(`./${env}/fetchImport`).default}
+${require(`./${env}/websocketsImport`).default}
+`;

src/TreeToTS/templates/typescript/node/fetchImport.ts

@@ -0,0 +1 @@
+export default `import fetch from 'node-fetch';`;

src/TreeToTS/templates/typescript/node/websocketsImport.ts

@@ -0,0 +1 @@
+export default `import WebSocket from 'ws';`;