infisical/frontend/src/context/SubscriptionContext/SubscriptionContext.tsx

import { useSuspenseQuery } from "@tanstack/react-query";
import { useRouteContext } from "@tanstack/react-router";

import { fetchOrgSubscription, subscriptionQueryKeys } from "@app/hooks/api/subscriptions/queries";

export const useSubscription = (refreshCache?: boolean) => {
  const organizationId = useRouteContext({
    from: "/_authenticate/_inject-org-details",
    select: (el) => el.organizationId
  });

  const { data: subscription } = useSuspenseQuery({
    queryKey: subscriptionQueryKeys.getOrgSubsription(organizationId),
    queryFn: () => fetchOrgSubscription(organizationId, refreshCache),
    staleTime: Infinity
  });

  return { subscription };
};