aboutsummaryrefslogtreecommitdiffstats
path: root/tests/utils/graphql/connections.ts
blob: f38fa59721e249c7de1768ffe832e623b24950b1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
import type { EdgesResponse, GraphQLEdges, Maybe } from '../../../src/types';
import { CONFIG } from '../../../src/utils/config';

/**
 * Retrieve the edges.
 *
 * @param {T[]} data - An array of objects.
 * @param {number} offset - The offset.
 * @returns {Array<Edge<T>>} The edges.
 */
export const getEdges = <T>(data: T[], offset: number): GraphQLEdges<T>[] =>
  data.map((singleData, index) => {
    const currentItemNumber = index + 1;

    return {
      cursor: `cursor${currentItemNumber + offset}`,
      node: singleData,
    };
  });

type GetConnectionProps<T> = {
  data: Maybe<T[]>;
  first: Maybe<number>;
  after: Maybe<string>;
};

/**
 * Retrieve a GraphQL connection.
 *
 * @param props - An object.
 * @param props.after - The number of items before.
 * @param props.data - An array of items.
 * @param props.first - The number of items per page.
 * @returns {Connection<T>} The connection.
 */
export const getConnection = <T>({
  after,
  data = [],
  first = CONFIG.postsPerPage,
}: GetConnectionProps<T>): EdgesResponse<T> => {
  const afterInt = after ? Number(after.replace('cursor', '')) : 0;
  const edges = getEdges(data.slice(afterInt, afterInt + first), afterInt);
  const endCursor =
    edges.length > 0 ? edges[edges.length - 1].cursor : 'cursor1';

  return {
    edges,
    pageInfo: {
      endCursor,
      hasNextPage: data.length - afterInt > first,
      total: data.length,
    },
  };
};