96 lines
2.5 KiB
TypeScript
96 lines
2.5 KiB
TypeScript
import {
|
|
type QueryClient,
|
|
useQueryClient,
|
|
type UseQueryOptions,
|
|
} from "@tanstack/react-query";
|
|
import { useEffect } from "react";
|
|
|
|
import type { components, paths } from "./generated/schema";
|
|
|
|
import { useAuthenticatedQuery } from "./authenticated-query";
|
|
import { apiErrorCode, isBrokerageAccessLoss } from "./problem-details";
|
|
|
|
type InvestmentResourceMeta = components["schemas"]["InvestmentResourceMetaDto"];
|
|
|
|
type Resource = {
|
|
[key: string]: unknown;
|
|
data: unknown;
|
|
meta: InvestmentResourceMeta;
|
|
sync?: unknown;
|
|
};
|
|
|
|
type ResourcePath = {
|
|
[P in keyof paths]: paths[P]["get"] extends {
|
|
// eslint-disable-next-line @typescript-eslint/no-magic-numbers -- HTTP success response in the generated contract.
|
|
responses: { 200: { content: { "application/json": Resource } } };
|
|
} ? P : never;
|
|
}[keyof paths];
|
|
|
|
type ResourceQueryKey = readonly [
|
|
"get",
|
|
ResourcePath,
|
|
{ params?: { path?: object; query?: object } }?,
|
|
];
|
|
|
|
function invalidateResourceAccess(
|
|
client: QueryClient,
|
|
group: string,
|
|
error: unknown,
|
|
) {
|
|
if (!isBrokerageAccessLoss(apiErrorCode(error))) return;
|
|
|
|
const affected = {
|
|
predicate: (cached: { meta?: Record<string, unknown> }) =>
|
|
cached.meta?.accessGroup === group,
|
|
};
|
|
|
|
void client.cancelQueries(affected);
|
|
for (const cached of client.getQueryCache().findAll(affected))
|
|
cached.setState({
|
|
data: undefined,
|
|
dataUpdatedAt: 0,
|
|
error: error as Error,
|
|
fetchStatus: "idle",
|
|
status: "error",
|
|
});
|
|
}
|
|
|
|
function useResourceQuery<
|
|
T extends Resource,
|
|
E = unknown,
|
|
K extends readonly unknown[] = ResourceQueryKey,
|
|
>(
|
|
options: UseQueryOptions<T, E, T, K>,
|
|
) {
|
|
const client = useQueryClient();
|
|
|
|
const group
|
|
= typeof options.meta?.accessGroup === "string"
|
|
? options.meta.accessGroup
|
|
: undefined;
|
|
|
|
const query = useAuthenticatedQuery(options);
|
|
useEffect(() => {
|
|
if (group) invalidateResourceAccess(client, group, query.error);
|
|
}, [client, group, query.error]);
|
|
|
|
const data: T["data"] | undefined = query.data?.data;
|
|
const meta: T["meta"] | undefined = query.data?.meta;
|
|
|
|
// Project resource fields; refetch still returns the transport envelope.
|
|
// queryKey is kept for screen-sync cache bookkeeping.
|
|
return {
|
|
// eslint-disable-next-line @tanstack/query/no-rest-destructuring -- Single projection point; consumers already depend on the whole query state.
|
|
...query,
|
|
data: data ?? undefined,
|
|
meta,
|
|
queryKey: options.queryKey,
|
|
sync: query.data && "sync" in query.data ? query.data.sync : undefined,
|
|
};
|
|
}
|
|
|
|
export {
|
|
type InvestmentResourceMeta,
|
|
useResourceQuery,
|
|
};
|