fix: added support for graphql schema

Author: abhinav-from-contentstack

src/commands/tsgen.ts

@@ -43,6 +43,11 @@ export default class TypeScriptCodeGeneratorCommand extends Command {
       default: true,
       allowNo: true,
     }),
+
+    graphql: flags.boolean({
+      description: 'support graphql schema',
+      default: false,
+    }),
   };
 
   async run() {
@@ -51,6 +56,7 @@ export default class TypeScriptCodeGeneratorCommand extends Command {
 
       const token = this.getToken(flags['token-alias'])
       const prefix = flags.prefix
+      const graphql = flags.graphql
       const includeDocumentation = flags.doc
       const outputPath = flags.output
 
@@ -65,14 +71,14 @@ export default class TypeScriptCodeGeneratorCommand extends Command {
       const config: StackConnectionConfig = {
         apiKey: token.apiKey,
         token: token.token,
-        region: (this.region.name === 'eu') ? 'eu': undefined,
+        region: (this.region.name === 'EU') ? 'EU' : null,
         environment: token.environment || '',
       }
 
       const client = await stackConnect(this.deliveryAPIClient.Stack, config)
 
       if (client.types) {
-        const result = await tsgenRunner(outputPath, client.types, prefix, includeDocumentation)
+        const result = await tsgenRunner(outputPath, client.types, prefix, includeDocumentation, graphql)
         this.log(`Wrote ${result.definitions} Content Types to '${result.outputPath}'.`)
       } else {
         this.log('No Content Types exist in the Stack.')

src/lib/tsgen/runner.ts

@@ -7,6 +7,7 @@ import {DocumentationGenerator} from './docgen/doc'
 import JSDocumentationGenerator from './docgen/jsdoc'
 import NullDocumentationGenerator from './docgen/nulldoc'
 import tsgenFactory from './factory'
+import {supportGraphql} from '../util'
 
 async function format(definition: string) {
   const prettierConfig = await prettier.resolveConfig(process.cwd())
@@ -26,7 +27,7 @@ function createOutputPath(outputFile: string) {
   return outputPath
 }
 
-export default async function tsgenRunner(outputFile: string, contentTypes: any[], prefix = '', includeDocumentation = true) {
+export default async function tsgenRunner(outputFile: string, contentTypes: any[], prefix = '', includeDocumentation = true, graphql: boolean) {
   const docgen: DocumentationGenerator = includeDocumentation ? new JSDocumentationGenerator() : new NullDocumentationGenerator()
 
   const outputPath = createOutputPath(outputFile)
@@ -40,7 +41,10 @@ export default async function tsgenRunner(outputFile: string, contentTypes: any[
     },
   })
 
-  for (const contentType of contentTypes) {
+  for (let contentType of contentTypes) {
+    if (graphql) {
+      contentType = supportGraphql(contentType)
+    }
     const tsgenResult = tsgen(contentType)
 
     definitions.push(tsgenResult.definition)
@@ -52,6 +56,7 @@ export default async function tsgenRunner(outputFile: string, contentTypes: any[
 
   const output = await format(
     [
+
       defaultInterfaces(prefix).join('\n\n'),
       [...globalFields].join('\n\n'),
       definitions.join('\n\n'),

src/lib/util/index.ts

@@ -0,0 +1,9 @@
+export function supportGraphql(contentType: any) {
+	contentType.schema = contentType.schema.map((element: any) => {
+		if (element.data_type === 'reference') {
+			element.uid += 'Connection'
+		}
+		return element
+	})
+	return contentType
+}