89 lines
2.3 KiB
TypeScript
89 lines
2.3 KiB
TypeScript
import * as React from 'react';
|
|
import { useQuery } from '@apollo/client';
|
|
import { useLocation } from 'react-router-dom';
|
|
import { DateTime } from 'luxon';
|
|
|
|
import { GET_TRIAL_STATUS } from 'graphql/queries/get-trial-status.ee';
|
|
import useFormatMessage from './useFormatMessage';
|
|
|
|
type UseTrialStatusReturn = {
|
|
expireAt: DateTime;
|
|
message: string;
|
|
over: boolean;
|
|
status: 'error' | 'warning';
|
|
} | null;
|
|
|
|
function getDiffInDays(date: DateTime) {
|
|
const today = DateTime.now().startOf('day');
|
|
const diffInDays = date.diff(today, 'days').days;
|
|
const roundedDiffInDays = Math.round(diffInDays);
|
|
|
|
return roundedDiffInDays;
|
|
}
|
|
|
|
function getFeedbackPayload(date: DateTime) {
|
|
const diffInDays = getDiffInDays(date);
|
|
|
|
if (diffInDays <= -1) {
|
|
return {
|
|
translationEntryId: 'trialBadge.over',
|
|
status: 'error' as const,
|
|
over: true,
|
|
};
|
|
} else if (diffInDays <= 0) {
|
|
return {
|
|
translationEntryId: 'trialBadge.endsToday',
|
|
status: 'warning' as const,
|
|
over: false,
|
|
}
|
|
} else {
|
|
return {
|
|
translationEntryId: 'trialBadge.xDaysLeft',
|
|
translationEntryValues: {
|
|
remainingDays: diffInDays
|
|
},
|
|
status: 'warning' as const,
|
|
over: false,
|
|
}
|
|
}
|
|
}
|
|
|
|
export default function useTrialStatus(): UseTrialStatusReturn {
|
|
const formatMessage = useFormatMessage();
|
|
const location = useLocation();
|
|
const state = location.state as { checkoutCompleted: boolean };
|
|
const checkoutCompleted = state?.checkoutCompleted;
|
|
const { data, loading, startPolling, stopPolling } = useQuery(GET_TRIAL_STATUS);
|
|
const hasTrial = !!data?.getTrialStatus?.expireAt;
|
|
|
|
React.useEffect(function pollDataUntilTrialEnds() {
|
|
if (checkoutCompleted && hasTrial) {
|
|
startPolling(1000);
|
|
}
|
|
}, [checkoutCompleted, hasTrial, startPolling]);
|
|
|
|
React.useEffect(function stopPollingWhenTrialEnds() {
|
|
if (checkoutCompleted && !hasTrial) {
|
|
stopPolling();
|
|
}
|
|
}, [checkoutCompleted, hasTrial, stopPolling]);
|
|
|
|
if (loading || !data.getTrialStatus) return null;
|
|
|
|
const expireAt = DateTime.fromMillis(Number(data.getTrialStatus.expireAt)).startOf('day');
|
|
|
|
const {
|
|
translationEntryId,
|
|
translationEntryValues,
|
|
status,
|
|
over,
|
|
} = getFeedbackPayload(expireAt);
|
|
|
|
return {
|
|
message: formatMessage(translationEntryId, translationEntryValues),
|
|
expireAt,
|
|
over,
|
|
status
|
|
};
|
|
}
|