Add getGraphModels to exports (#407)

Author: eskrm

src/graphql/graphql.ts

@@ -845,6 +845,16 @@ function isScalar(type: any): boolean {
   );
 }
 
+export function getGraphModels(graphSchema: gql.GraphQLSchema): string[] {
+  const models: string[] = [];
+  for (const [name, type] of Object.entries(graphSchema.getTypeMap())) {
+    if (isV2ModelType(type)) {
+      models.push(name);
+    }
+  }
+  return models;
+}
+
 interface IncrementalQueryConfig {
   readonly type: gql.GraphQLObjectType;
   readonly resolvedPrimaryKeys?: Dictionary<string>;

src/index.ts

@@ -61,6 +61,7 @@ export {
   crossMerge,
   flattenIterable,
   flattenV2,
+  getGraphModels,
   paginatedQueryV2,
   pathToModelV2,
   readerFromQuery,

test/graphql.test.ts

@@ -476,4 +476,12 @@ describe('graphql', () => {
       incremental: false,
     });
   });
+
+  test('get graph models', () => {
+    expect(sut.getGraphModels(graphSchemaV2)).toIncludeAllMembers([
+      'cicd_Artifact',
+      'cicd_Build',
+      'cicd_Pipeline',
+    ]);
+  });
 });