|
1 | | -import { useCallback, useState, useMemo } from 'react' |
2 | | -import useUncontrolledTokenPagination from './uncontrolled' |
| 1 | +import { useCallback, useEffect, useState, useMemo } from 'react' |
3 | 2 | import { assertNumber } from './utils' |
4 | 3 |
|
5 | | -const DEFAULTS = { |
6 | | - defaultPageNumber: 1, |
7 | | - resetPageNumberOnPageSizeChange: true, |
8 | | -} |
9 | | - |
10 | | -export default function useControlledTokenPagination(options) { |
11 | | - options = { ...DEFAULTS, ...options } |
12 | | - |
13 | | - assertNumber('defaultPageNumber', options.defaultPageNumber) |
14 | | - assertNumber('defaultPageSize', options.defaultPageSize) |
15 | | - |
16 | | - const [{ pageNumber, pageSize }, setPagination] = useState({ |
17 | | - pageNumber: options.defaultPageNumber, |
18 | | - pageSize: options.defaultPageSize, |
19 | | - }) |
| 4 | +export default function useControlledTokenPagination(pageNumber) { |
| 5 | + assertNumber('pageNumber', pageNumber) |
20 | 6 |
|
21 | | - const change = useCallback( |
22 | | - (property, changer) => { |
23 | | - const pageNumberReset = options.resetPageNumberOnPageSizeChange |
24 | | - ? { pageNumber: options.defaultPageNumber } |
25 | | - : {} |
| 7 | + const [mapping, setMapping] = useState({}) |
26 | 8 |
|
27 | | - switch (typeof changer) { |
28 | | - case 'function': |
29 | | - return setPagination(p => ({ |
30 | | - ...p, |
31 | | - ...pageNumberReset, |
32 | | - [property]: changer(p[property]), |
33 | | - })) |
34 | | - case 'number': |
35 | | - return setPagination(p => ({ |
36 | | - ...p, |
37 | | - ...pageNumberReset, |
38 | | - [property]: changer, |
39 | | - })) |
40 | | - default: |
41 | | - throw new Error( |
42 | | - `Unsupported value ${changer} of type ${typeof changer} for ${property}` |
43 | | - ) |
44 | | - } |
| 9 | + const updateToken = useCallback( |
| 10 | + nextToken => { |
| 11 | + setMapping(m => ({ |
| 12 | + ...m, |
| 13 | + [pageNumber + 1]: nextToken, |
| 14 | + })) |
45 | 15 | }, |
46 | | - [options.defaultPageNumber, options.resetPageNumberOnPageSizeChange] |
| 16 | + [pageNumber] |
47 | 17 | ) |
48 | 18 |
|
49 | | - const changePageNumber = useCallback(c => change('pageNumber', c), [change]) |
50 | | - const changePageSize = useCallback(c => change('pageSize', c), [change]) |
51 | | - |
52 | | - const uncontrolled = useUncontrolledTokenPagination(pageNumber) |
| 19 | + const useUpdateToken = useCallback( |
| 20 | + function useUpdateToken(nextToken) { |
| 21 | + useEffect(() => updateToken(nextToken), [nextToken]) |
| 22 | + }, |
| 23 | + [updateToken] |
| 24 | + ) |
53 | 25 |
|
54 | 26 | return useMemo( |
55 | 27 | () => ({ |
56 | | - ...uncontrolled, |
57 | | - pageNumber, |
58 | | - pageSize, |
59 | | - changePageNumber, |
60 | | - changePageSize, |
| 28 | + currentToken: mapping[pageNumber], |
| 29 | + useUpdateToken, |
| 30 | + updateToken, |
61 | 31 | }), |
62 | | - [changePageNumber, changePageSize, pageNumber, pageSize, uncontrolled] |
| 32 | + [mapping, pageNumber, updateToken, useUpdateToken] |
63 | 33 | ) |
64 | 34 | } |
0 commit comments