117 lines
2.9 KiB
TypeScript
117 lines
2.9 KiB
TypeScript
import { useEffect, useRef, useState } from "react";
|
|
import { useQuery, useMutation, useSession } from "blitz";
|
|
|
|
import type { Subscription } from "db";
|
|
import getSubscription from "../queries/get-subscription";
|
|
import getPayments from "../queries/get-payments";
|
|
import usePaddle from "./use-paddle";
|
|
import useCurrentUser from "../../core/hooks/use-current-user";
|
|
import updateSubscription from "../mutations/update-subscription";
|
|
|
|
type Params = {
|
|
initialData?: Subscription;
|
|
};
|
|
|
|
export default function useSubscription({ initialData }: Params = {}) {
|
|
const session = useSession();
|
|
const { user } = useCurrentUser();
|
|
const [isWaitingForSubChange, setIsWaitingForSubChange] = useState(false);
|
|
const [subscription] = useQuery(getSubscription, null, {
|
|
initialData,
|
|
refetchInterval: isWaitingForSubChange ? 1500 : false,
|
|
});
|
|
const [payments] = useQuery(getPayments, null);
|
|
const [updateSubscriptionMutation] = useMutation(updateSubscription);
|
|
|
|
const resolve = useRef<() => void>();
|
|
const promise = useRef<Promise<void>>();
|
|
|
|
const { Paddle } = usePaddle({
|
|
eventCallback(data) {
|
|
if (["Checkout.Close", "Checkout.Complete"].includes(data.event)) {
|
|
resolve.current!();
|
|
promise.current = new Promise((r) => (resolve.current = r));
|
|
}
|
|
},
|
|
});
|
|
|
|
// cancel subscription polling when we get a new subscription
|
|
useEffect(() => setIsWaitingForSubChange(false), [subscription?.paddleSubscriptionId]);
|
|
|
|
useEffect(() => {
|
|
promise.current = new Promise((r) => (resolve.current = r));
|
|
}, []);
|
|
|
|
type BuyParams = {
|
|
planId: number;
|
|
coupon?: string;
|
|
};
|
|
|
|
async function subscribe(params: BuyParams) {
|
|
if (!user || !session.orgId) {
|
|
return;
|
|
}
|
|
|
|
const { planId, coupon } = params;
|
|
const checkoutOpenParams = {
|
|
email: user.email,
|
|
product: planId,
|
|
allowQuantity: false,
|
|
passthrough: JSON.stringify({ organizationId: session.orgId }),
|
|
coupon: "",
|
|
};
|
|
|
|
if (coupon) {
|
|
checkoutOpenParams.coupon = coupon;
|
|
}
|
|
|
|
Paddle.Checkout.open(checkoutOpenParams);
|
|
setIsWaitingForSubChange(true);
|
|
|
|
return promise.current;
|
|
}
|
|
|
|
async function updatePaymentMethod({ updateUrl }: { updateUrl: string }) {
|
|
const checkoutOpenParams = { override: updateUrl };
|
|
|
|
Paddle.Checkout.open(checkoutOpenParams);
|
|
setIsWaitingForSubChange(true);
|
|
|
|
return promise.current;
|
|
}
|
|
|
|
async function cancelSubscription({ cancelUrl }: { cancelUrl: string }) {
|
|
const checkoutOpenParams = { override: cancelUrl };
|
|
|
|
Paddle.Checkout.open(checkoutOpenParams);
|
|
setIsWaitingForSubChange(true);
|
|
|
|
return promise.current;
|
|
}
|
|
|
|
type ChangePlanParams = {
|
|
planId: number;
|
|
};
|
|
|
|
async function changePlan({ planId }: ChangePlanParams) {
|
|
if (planId === -1) {
|
|
return cancelSubscription({ cancelUrl: subscription!.cancelUrl });
|
|
}
|
|
|
|
try {
|
|
await updateSubscriptionMutation({ planId });
|
|
} catch (error) {
|
|
console.log("error", error);
|
|
}
|
|
}
|
|
|
|
return {
|
|
subscription,
|
|
payments,
|
|
subscribe,
|
|
updatePaymentMethod,
|
|
cancelSubscription,
|
|
changePlan,
|
|
};
|
|
}
|