|
1 | 1 | import { useCallback } from 'react'; |
2 | 2 | import { useRouter } from 'next/router'; |
3 | 3 | import { get, post, put, del, getItem } from 'lib/web'; |
4 | | -import { AUTH_TOKEN } from 'lib/constants'; |
| 4 | +import { AUTH_TOKEN, SHARE_TOKEN_HEADER } from 'lib/constants'; |
| 5 | +import useStore from 'store/app'; |
5 | 6 |
|
6 | | -function includeAuthToken(headers = {}) { |
7 | | - const authToken = getItem(AUTH_TOKEN); |
| 7 | +const selector = state => state.shareToken; |
8 | 8 |
|
| 9 | +function parseHeaders(headers = {}, { authToken, shareToken }) { |
9 | 10 | if (authToken) { |
10 | | - headers.Authorization = `Bearer ${authToken}`; |
| 11 | + headers.authorization = `Bearer ${authToken}`; |
| 12 | + } |
| 13 | + |
| 14 | + if (shareToken) { |
| 15 | + headers[SHARE_TOKEN_HEADER] = shareToken.token; |
11 | 16 | } |
12 | 17 |
|
13 | 18 | return headers; |
14 | 19 | } |
15 | 20 |
|
16 | 21 | export default function useApi() { |
17 | 22 | const { basePath } = useRouter(); |
| 23 | + const authToken = getItem(AUTH_TOKEN); |
| 24 | + const shareToken = useStore(selector); |
18 | 25 |
|
19 | 26 | return { |
20 | 27 | get: useCallback( |
21 | 28 | async (url, params, headers) => { |
22 | | - return get(`${basePath}/api${url}`, params, includeAuthToken(headers)); |
| 29 | + return get( |
| 30 | + `${basePath}/api${url}`, |
| 31 | + params, |
| 32 | + parseHeaders(headers, { authToken, shareToken }), |
| 33 | + ); |
23 | 34 | }, |
24 | 35 | [get], |
25 | 36 | ), |
26 | 37 |
|
27 | 38 | post: useCallback( |
28 | 39 | async (url, params, headers) => { |
29 | | - return post(`${basePath}/api${url}`, params, includeAuthToken(headers)); |
| 40 | + return post( |
| 41 | + `${basePath}/api${url}`, |
| 42 | + params, |
| 43 | + parseHeaders(headers, { authToken, shareToken }), |
| 44 | + ); |
30 | 45 | }, |
31 | 46 | [post], |
32 | 47 | ), |
33 | 48 |
|
34 | 49 | put: useCallback( |
35 | 50 | async (url, params, headers) => { |
36 | | - return put(`${basePath}/api${url}`, params, includeAuthToken(headers)); |
| 51 | + return put( |
| 52 | + `${basePath}/api${url}`, |
| 53 | + params, |
| 54 | + parseHeaders(headers, { authToken, shareToken }), |
| 55 | + ); |
37 | 56 | }, |
38 | 57 | [put], |
39 | 58 | ), |
40 | 59 |
|
41 | 60 | del: useCallback( |
42 | 61 | async (url, params, headers) => { |
43 | | - return del(`${basePath}/api${url}`, params, includeAuthToken(headers)); |
| 62 | + return del( |
| 63 | + `${basePath}/api${url}`, |
| 64 | + params, |
| 65 | + parseHeaders(headers, { authToken, shareToken }), |
| 66 | + ); |
44 | 67 | }, |
45 | 68 | [del], |
46 | 69 | ), |
|
0 commit comments