1 | import { __assign } from "tslib";
|
2 | import { useCallback, useEffect, useMemo, useRef } from 'react';
|
3 | import { mergeOptions } from "../../utilities/index.js";
|
4 | import { useInternalState } from "./useQuery.js";
|
5 | import { useApolloClient } from "./useApolloClient.js";
|
6 | var EAGER_METHODS = [
|
7 | 'refetch',
|
8 | 'reobserve',
|
9 | 'fetchMore',
|
10 | 'updateQuery',
|
11 | 'startPolling',
|
12 | 'subscribeToMore',
|
13 | ];
|
14 | export function useLazyQuery(query, options) {
|
15 | var _a;
|
16 | var abortControllersRef = useRef(new Set());
|
17 | var execOptionsRef = useRef();
|
18 | var merged = execOptionsRef.current ? mergeOptions(options, execOptionsRef.current) : options;
|
19 | var internalState = useInternalState(useApolloClient(options && options.client), (_a = merged === null || merged === void 0 ? void 0 : merged.query) !== null && _a !== void 0 ? _a : query);
|
20 | var useQueryResult = internalState.useQuery(__assign(__assign({}, merged), { skip: !execOptionsRef.current }));
|
21 | var initialFetchPolicy = useQueryResult.observable.options.initialFetchPolicy ||
|
22 | internalState.getDefaultFetchPolicy();
|
23 | var result = Object.assign(useQueryResult, {
|
24 | called: !!execOptionsRef.current,
|
25 | });
|
26 | var eagerMethods = useMemo(function () {
|
27 | var eagerMethods = {};
|
28 | var _loop_1 = function (key) {
|
29 | var method = result[key];
|
30 | eagerMethods[key] = function () {
|
31 | if (!execOptionsRef.current) {
|
32 | execOptionsRef.current = Object.create(null);
|
33 | internalState.forceUpdate();
|
34 | }
|
35 | return method.apply(this, arguments);
|
36 | };
|
37 | };
|
38 | for (var _i = 0, EAGER_METHODS_1 = EAGER_METHODS; _i < EAGER_METHODS_1.length; _i++) {
|
39 | var key = EAGER_METHODS_1[_i];
|
40 | _loop_1(key);
|
41 | }
|
42 | return eagerMethods;
|
43 | }, []);
|
44 | Object.assign(result, eagerMethods);
|
45 | useEffect(function () {
|
46 | return function () {
|
47 | abortControllersRef.current.forEach(function (controller) {
|
48 | controller.abort();
|
49 | });
|
50 | };
|
51 | }, []);
|
52 | var execute = useCallback(function (executeOptions) {
|
53 | var controller = new AbortController();
|
54 | abortControllersRef.current.add(controller);
|
55 | execOptionsRef.current = executeOptions ? __assign(__assign({}, executeOptions), { fetchPolicy: executeOptions.fetchPolicy || initialFetchPolicy }) : {
|
56 | fetchPolicy: initialFetchPolicy,
|
57 | };
|
58 | var promise = internalState
|
59 | .asyncUpdate(controller.signal)
|
60 | .then(function (queryResult) {
|
61 | abortControllersRef.current.delete(controller);
|
62 | return Object.assign(queryResult, eagerMethods);
|
63 | });
|
64 | promise.catch(function () {
|
65 | abortControllersRef.current.delete(controller);
|
66 | });
|
67 | return promise;
|
68 | }, []);
|
69 | return [execute, result];
|
70 | }
|
71 |
|
\ | No newline at end of file |