@connectrpc/connect-query
Version:
TypeScript-first expansion pack for TanStack Query that gives you Protobuf superpowers.
65 lines • 2.63 kB
JavaScript
;
// Copyright 2021-2023 The Connect Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
Object.defineProperty(exports, "__esModule", { value: true });
exports.createUseInfiniteQueryOptions = void 0;
const call_unary_method_js_1 = require("./call-unary-method.js");
const connect_query_key_js_1 = require("./connect-query-key.js");
const utils_js_1 = require("./utils.js");
function createUnaryInfiniteQueryFn(methodType, input, { callOptions, transport, pageParamKey, }) {
return async (context) => {
(0, utils_js_1.assert)(input !== utils_js_1.disableQuery, "Disabled query cannot be fetched");
(0, utils_js_1.assert)("pageParam" in context, "pageParam must be part of context");
const inputCombinedWithPageParam = {
...input,
[pageParamKey]: context.pageParam,
};
return (0, call_unary_method_js_1.callUnaryMethod)(methodType, inputCombinedWithPageParam, {
callOptions: {
...callOptions,
signal: callOptions?.signal ?? context.signal,
},
transport,
});
};
}
/**
* Query the method provided. Maps to useInfiniteQuery on tanstack/react-query
*
* @param methodSig
* @returns
*/
function createUseInfiniteQueryOptions(methodSig, input, { transport, getNextPageParam, pageParamKey, callOptions, }) {
const queryKey = (0, connect_query_key_js_1.createConnectInfiniteQueryKey)(methodSig, input === utils_js_1.disableQuery
? undefined
: {
...input,
[pageParamKey]: undefined,
});
return {
getNextPageParam,
initialPageParam: input === utils_js_1.disableQuery
? undefined
: input[pageParamKey],
queryKey,
queryFn: createUnaryInfiniteQueryFn(methodSig, input, {
transport,
callOptions,
pageParamKey,
}),
enabled: input !== utils_js_1.disableQuery,
};
}
exports.createUseInfiniteQueryOptions = createUseInfiniteQueryOptions;
//# sourceMappingURL=create-use-infinite-query-options.js.map