feat: allow UnionTypeComposer, ScalarTypeComposer, EnumTypeComposer to be as a source TC (#48)

* init

* chore: fix typescript definitions

* feat: allow ScalarTypeComposer & EnumTypeComposer to be as source TC

Co-authored-by: nodkz <[email protected]>

Author: theasmoth

src/pagination.ts

@@ -2,13 +2,23 @@ import { Resolver, inspect } from 'graphql-compose';
 import type {
   ObjectTypeComposer,
   InterfaceTypeComposer,
+  UnionTypeComposer,
+  ScalarTypeComposer,
+  EnumTypeComposer,
   ResolverResolveParams,
   ObjectTypeComposerArgumentConfigMap,
 } from 'graphql-compose';
 import { preparePaginationTC } from './types';
 
 export const DEFAULT_RESOLVER_NAME = 'pagination';
 export const DEFAULT_PER_PAGE = 20;
+const ALLOWED_TYPE_COMPOSERS = [
+  'ObjectTypeComposer',
+  'InterfaceTypeComposer',
+  'UnionTypeComposer',
+  'ScalarTypeComposer',
+  'EnumTypeComposer',
+];
 
 export type PaginationResolverOpts = {
   findManyResolver: Resolver;
@@ -40,12 +50,19 @@ export interface PaginationTArgs {
 }
 
 export function preparePaginationResolver<TSource, TContext>(
-  tc: ObjectTypeComposer<TSource, TContext> | InterfaceTypeComposer<TSource, TContext>,
+  tc:
+    | ObjectTypeComposer<TSource, TContext>
+    | InterfaceTypeComposer<TSource, TContext>
+    | UnionTypeComposer<TSource, TContext>
+    | ScalarTypeComposer<TContext>
+    | EnumTypeComposer<TContext>,
   opts: PaginationResolverOpts
 ): Resolver<TSource, TContext, PaginationTArgs> {
-  if (!tc || tc.constructor.name !== 'ObjectTypeComposer') {
+  if (!tc || !ALLOWED_TYPE_COMPOSERS.includes(tc.constructor.name)) {
     throw new Error(
-      'First arg for prepareConnectionResolver() should be instance of ObjectTypeComposer'
+      `First arg for preparePaginationResolver() should be instance of ${ALLOWED_TYPE_COMPOSERS.join(
+        ' or '
+      )}`
     );
   }
 

src/types.ts

@@ -3,6 +3,9 @@ import {
   ObjectTypeComposer,
   SchemaComposer,
   InterfaceTypeComposer,
+  UnionTypeComposer,
+  ScalarTypeComposer,
+  EnumTypeComposer,
 } from 'graphql-compose';
 
 // PaginationInfo should be global
@@ -41,7 +44,12 @@ export function preparePaginationInfoTC<TContext>(
 }
 
 export function preparePaginationTC<TSource, TContext>(
-  tc: ObjectTypeComposer<TSource, TContext> | InterfaceTypeComposer<TSource, TContext>,
+  tc:
+    | ObjectTypeComposer<TSource, TContext>
+    | InterfaceTypeComposer<TSource, TContext>
+    | UnionTypeComposer<TSource, TContext>
+    | ScalarTypeComposer<TContext>
+    | EnumTypeComposer<TContext>,
   resolverName?: string
 ): ObjectTypeComposer<TSource, TContext> {
   const schemaComposer = tc.schemaComposer;