Type safety in actionResult in shouldRevalidate?

[nikolailehbrink]

I wondered how I can get type safety for the actionResult parameter of the shouldRevalidate function. For example when using useActionData I can get the correct types by writing the following:

const actionData = useActionData<typeof action>();

Now how would I do that inside the shouldRevalidate function?

export const shouldRevalidate: ShouldRevalidateFunction = ({
  actionResult,
  defaultShouldRevalidate,
}) => {
  // This should be type safe but right now is any
  if (actionResult?.decryptedMessage) {
    return false; // Do not revalidate as the message has been handled
  }
  return defaultShouldRevalidate; // Fallback to the default revalidation behavior
};

[ankitjena]

I did something like this to infer the type

export const shouldRevalidate = ({
  defaultShouldRevalidate,
  actionResult,
}: Omit<ShouldRevalidateFunctionArgs, "actionResult"> & {
  actionResult: Awaited<ReturnType<Awaited<ReturnType<typeof action>>["json"]>>;
}) => {
  if (actionResult.data) {
    return true;
  }
  return defaultShouldRevalidate;
};

but there should be a better way to do it.