Add customization options to connection resolver

Author: michaelbeaumont

src/resolvers/connection.d.ts

@@ -1,4 +1,4 @@
-import { Resolver, ObjectTypeComposer } from 'graphql-compose';
+import { Resolver, ObjectTypeComposer, ObjectTypeComposerFieldConfigMap } from 'graphql-compose';
 // import { ConnectionSortMapOpts } from 'graphql-compose-connection';
 import { Model } from 'mongoose';
 import { MongoId } from '../types/mongoid';
@@ -7,12 +7,17 @@ import { FilterHelperArgs, SortHelperArgs } from './helpers';
 
 // @ts-todo The ConnectionSortMapOpts is not available yet since graphql-compose-connection doesn't have types for now,
 //          fallback to a simple object.
-export type ConnectionSortMapOpts = { [opt: string]: any };
+export type ConnectionOpts<TContext> = { [opt: string]: any } & {
+  connectionResolverName?: string,
+  findResolverName?: string;
+  countResolverName?: string;
+  edgeFields?: ObjectTypeComposerFieldConfigMap<any, TContext>;
+};
 
 export default function connection(
   model: Model<any>,
   tc: ObjectTypeComposer<any>,
-  opts?: ConnectionSortMapOpts,
+  opts?: ConnectionOpts<any>,
 ): Resolver<any, any> | undefined;
 
 export function prepareCursorQuery(

src/resolvers/connection.js

@@ -3,19 +3,28 @@
 
 import type { MongooseDocument } from 'mongoose';
 import type { ConnectionSortMapOpts as _ConnectionSortMapOpts } from 'graphql-compose-connection';
-import type { Resolver, ObjectTypeComposer } from 'graphql-compose';
+import type {
+  Resolver,
+  ObjectTypeComposer,
+  ObjectTypeComposerFieldConfigMap,
+} from 'graphql-compose';
 import {
   getUniqueIndexes,
   extendByReversedIndexes,
   type IndexT,
 } from '../utils/getIndexesFromModel';
 
-export type ConnectionSortMapOpts = _ConnectionSortMapOpts;
+export type ConnectionOpts<TContext> = _ConnectionSortMapOpts & {
+  edgeFields?: ObjectTypeComposerFieldConfigMap<any, TContext>,
+  connectionResolverName?: string,
+  findResolverName?: string,
+  countResolverName?: string,
+};
 
 export default function connection<TSource: MongooseDocument, TContext>(
   model: Class<TSource>, // === MongooseModel
   tc: ObjectTypeComposer<TSource, TContext>,
-  opts?: ConnectionSortMapOpts
+  opts: ConnectionOpts<TContext> = {}
 ): ?Resolver<TSource, TContext> {
   try {
     require.resolve('graphql-compose-connection');
@@ -56,14 +65,22 @@ export default function connection<TSource: MongooseDocument, TContext>(
       },
     };
   });
-
+  const {
+    connectionResolverName = 'connection',
+    findResolverName = 'findMany',
+    countResolverName = 'count',
+    edgeFields,
+    ...sortOptions
+  } = opts
   return prepareConnectionResolver(tc, {
-    findResolverName: 'findMany',
-    countResolverName: 'count',
+    connectionResolverName,
+    findResolverName,
+    countResolverName,
     sort: {
       ...sortConfigs,
-      ...opts,
+      ...sortOptions,
     },
+    edgeFields,
   });
 }