extend mutation resolver types

chriskalmar · Jan 30, 2021 · 2134bcd · 2134bcd
1 parent 299ff96
commit 2134bcd
Showing 1 changed file with 20 additions and 7 deletions.
diff --git a/src/graphqlProtocol/resolver.ts b/src/graphqlProtocol/resolver.ts
@@ -351,13 +351,26 @@ export const getNestedPayloadResolver = (
   };
 };
 
-export const getMutationResolver = (
-  entity: Entity | any,
-  entityMutation: Mutation,
-  typeName: string,
-  nested: boolean,
-  idResolver: Function,
-) => {
+export type GetMutationResolverParams = (params: {
+  entity: Entity;
+  entityMutation: Mutation;
+  typeName: string;
+  nested: boolean;
+  idResolver: ({ args }) => number | string;
+}) => (
+  source: any,
+  args: any,
+  context: Context,
+  info: GraphQLResolveInfo,
+) => Promise<Record<string, unknown>>;
+
+export const getMutationResolver: GetMutationResolverParams = ({
+  entity,
+  entityMutation,
+  typeName,
+  nested,
+  idResolver,
+}) => {
   const storageType = entity.storageType;
   const protocolConfiguration = ProtocolGraphQL.getProtocolConfiguration() as ProtocolGraphQLConfiguration;