From 1827f5413f39096b58daf18dd9333bfef6295342 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?R=C4=B1dvan=20Akca?= <akca.ridvann@gmail.com>
Date: Thu, 28 Mar 2024 15:29:04 +0300
Subject: [PATCH] refactor(useUserTrial): return hasTrial field from hook

---
 .../components/TrialStatusBadge/index.ee.jsx  |  2 +-
 packages/web/src/hooks/useUserTrial.ee.js     | 34 +++----------------
 2 files changed, 6 insertions(+), 30 deletions(-)

diff --git a/packages/web/src/components/TrialStatusBadge/index.ee.jsx b/packages/web/src/components/TrialStatusBadge/index.ee.jsx
index f4ab263f..d31d9e66 100644
--- a/packages/web/src/components/TrialStatusBadge/index.ee.jsx
+++ b/packages/web/src/components/TrialStatusBadge/index.ee.jsx
@@ -8,7 +8,7 @@ import useUserTrial from 'hooks/useUserTrial.ee';
 export default function TrialStatusBadge() {
   const data = useUserTrial();
 
-  if (!data) return <React.Fragment />;
+  if (!data.hasTrial) return <React.Fragment />;
 
   const { message, status } = data;
 
diff --git a/packages/web/src/hooks/useUserTrial.ee.js b/packages/web/src/hooks/useUserTrial.ee.js
index 4e61ad58..0ff5c80e 100644
--- a/packages/web/src/hooks/useUserTrial.ee.js
+++ b/packages/web/src/hooks/useUserTrial.ee.js
@@ -1,10 +1,8 @@
-import * as React from 'react';
-import { useLocation } from 'react-router-dom';
 import { DateTime } from 'luxon';
+import { useQuery } from '@tanstack/react-query';
 
 import useFormatMessage from './useFormatMessage';
 import api from 'helpers/api';
-import { useQuery } from '@tanstack/react-query';
 
 function getDiffInDays(date) {
   const today = DateTime.now().startOf('day');
@@ -13,6 +11,7 @@ function getDiffInDays(date) {
 
   return roundedDiffInDays;
 }
+
 function getFeedbackPayload(date) {
   const diffInDays = getDiffInDays(date);
 
@@ -41,12 +40,8 @@ function getFeedbackPayload(date) {
 }
 export default function useUserTrial() {
   const formatMessage = useFormatMessage();
-  const location = useLocation();
-  const state = location.state;
-  const checkoutCompleted = state?.checkoutCompleted;
-  const [isPolling, setIsPolling] = React.useState(false);
 
-  const { data, isLoading: isUserTrialLoading } = useQuery({
+  const { data } = useQuery({
     queryKey: ['userTrial'],
     queryFn: async ({ signal }) => {
       const { data } = await api.get('/v1/users/me/trial', {
@@ -55,32 +50,12 @@ export default function useUserTrial() {
 
       return data;
     },
-    refetchInterval: isPolling ? 1000 : false,
   });
+
   const userTrial = data?.data;
 
   const hasTrial = userTrial?.inTrial;
 
-  React.useEffect(
-    function pollDataUntilTrialEnds() {
-      if (checkoutCompleted && hasTrial) {
-        setIsPolling(true);
-      }
-    },
-    [checkoutCompleted, hasTrial, setIsPolling],
-  );
-
-  React.useEffect(
-    function stopPollingWhenTrialEnds() {
-      if (checkoutCompleted && !hasTrial) {
-        setIsPolling(false);
-      }
-    },
-    [checkoutCompleted, hasTrial, setIsPolling],
-  );
-
-  if (isUserTrialLoading || !hasTrial) return null;
-
   const expireAt = DateTime.fromISO(userTrial?.expireAt).startOf('day');
 
   const { translationEntryId, translationEntryValues, status, over } =
@@ -91,5 +66,6 @@ export default function useUserTrial() {
     expireAt,
     over,
     status,
+    hasTrial,
   };
 }