feat: update context

parameter in relay functions

Author: productdevbook

.gitignore

@@ -3,4 +3,6 @@
 node_modules
 dist
 package-lock.json
-.eslintcache
+.eslintcache
+
+*.tgz

src/relay.ts

@@ -53,6 +53,7 @@ export async function resolveOffsetConnection<
     offset: number
     limit: number
   }) => U & (MaybePromise<readonly T[] | null> | null),
+  context?: (params: ResolveCursorConnectionArgs) => I,
 ): Promise<Connection<T, I>> {
   const { limit, offset, expectedSize, hasPreviousPage, hasNextPage } = offsetForArgs(options)
 
@@ -77,6 +78,10 @@ export async function resolveOffsetConnection<
       hasPreviousPage,
       hasNextPage: hasNextPage(nodes.length),
     },
+    context: {
+      ...context?.({ before: undefined, after: undefined, limit, inverted: false }) ?? {},
+      ...(nodes as any).context ?? {},
+    },
   }
 }
 
@@ -94,6 +99,7 @@ export function offsetToCursor(offset: number): string {
 export function resolveArrayConnection<T, I>(
   options: ResolveArrayConnectionOptions,
   array: readonly T[],
+  context?: (params: ResolveCursorConnectionArgs) => I,
 ): Connection<T, I> {
   const { limit, offset, expectedSize, hasPreviousPage, hasNextPage } = offsetForArgs(options)
 
@@ -117,6 +123,10 @@ export function resolveArrayConnection<T, I>(
       hasPreviousPage,
       hasNextPage: hasNextPage(nodes.length),
     },
+    context: {
+      ...context?.({ before: undefined, after: undefined, limit, inverted: false }) ?? {},
+      ...(nodes as any).context ?? {},
+    },
   }
 }
 
@@ -154,6 +164,7 @@ export async function resolveCursorConnection<
 >(
   options: ResolveCursorConnectionOptions<NodeType<U>>,
   resolve: (params: ResolveCursorConnectionArgs) => U,
+  context?: (params: ResolveCursorConnectionArgs) => I,
 ): Promise<Connection<NodeType<U>, I>> {
   const { before, after, limit, inverted, expectedSize, hasPreviousPage, hasNextPage }
     = parseCurserArgs(options)
@@ -192,5 +203,9 @@ export async function resolveCursorConnection<
       hasPreviousPage: hasPreviousPage(nodes.length),
       hasNextPage: hasNextPage(nodes.length),
     },
+    context: {
+      ...context?.({ before, after, limit, inverted }) ?? {},
+      ...(nodes as any).context ?? {},
+    },
   }
 }

src/types.ts

@@ -68,7 +68,9 @@ export interface ResolveArrayConnectionOptions {
 export interface Connection<T, I> {
   edges: Array<Edge<T>>
   pageInfo: PageInfo
-  context?: I
+  context?: {
+    [key: string]: I
+  }
 }
 
 /**