22 lines
653 B
TypeScript
22 lines
653 B
TypeScript
"use client";
|
|
|
|
import { type DehydratedState, Hydrate } from "@tanstack/react-query";
|
|
import { useMemo } from "react";
|
|
|
|
import type { DataTransformer } from "@trpc/server";
|
|
|
|
export function createHydrateClient(opts: { transformer?: DataTransformer }) {
|
|
return function HydrateClient(props: { children: React.ReactNode; state: DehydratedState }) {
|
|
const { state, children } = props;
|
|
|
|
const transformedState: DehydratedState = useMemo(() => {
|
|
if (opts.transformer) {
|
|
return opts.transformer.deserialize(state);
|
|
}
|
|
return state;
|
|
}, [state]);
|
|
|
|
return <Hydrate state={transformedState}>{children}</Hydrate>;
|
|
};
|
|
}
|