fix: update usegetpricingdetails hook

push-protocol · Feb 17, 2025 · b12d946 · b12d946
1 parent 9a24495
commit b12d946
Show file tree

Hide file tree

Showing 4 changed files with 358 additions and 361 deletions.
diff --git a/src/components/userPlanAndBillings/UpgradePlanNavigationItem.tsx b/src/components/userPlanAndBillings/UpgradePlanNavigationItem.tsx
@@ -20,7 +20,7 @@ export const UpgradePlanNavigationItem = () => {
   } = useGetPricingPlanStatus({
     channelId: walletAddress,
   });
-  const { selectedPlan, isUserOnFreePlan } = useGetPricingPlanDetails(pricingPlanStatus);
+  const { selectedPlan, isUserOnFreePlan, isUserOnEnterprisePlan } = useGetPricingPlanDetails(pricingPlanStatus);
 
   useMigrateToFreePlan({
     pricingPlanStatus,
@@ -70,14 +70,16 @@ export const UpgradePlanNavigationItem = () => {
           >
             {(pricingPlanStatus && !isUserOnFreePlan && `Push ${selectedPlan?.name}`) || 'Free Plan'}
           </Text>
-          <Button
-            leadingIcon={<Sale />}
-            size="extraSmall"
-            variant="secondary"
-            onClick={handleGoToPricing}
-          >
-            Upgrade
-          </Button>
+          {!isUserOnEnterprisePlan && (
+            <Button
+              leadingIcon={<Sale />}
+              size="extraSmall"
+              variant="secondary"
+              onClick={handleGoToPricing}
+            >
+              Upgrade
+            </Button>
+          )}
         </Box>
 
         <Box

diff --git a/src/components/userPlanAndBillings/UserPlanAndBillings.tsx b/src/components/userPlanAndBillings/UserPlanAndBillings.tsx
@@ -15,7 +15,8 @@ const UserPlanAndBillings = () => {
     channelId: walletAddress,
   });
 
-  const { selectedPlan, isUserOnFreePlan, pricingListDescriptions } = useGetPricingPlanDetails(pricingPlanStatus);
+  const { selectedPlan, isUserOnFreePlan, isUserOnEnterprisePlan, pricingListDescriptions } =
+    useGetPricingPlanDetails(pricingPlanStatus);
 
   const planNotifications = [
     {
@@ -109,13 +110,15 @@ const UserPlanAndBillings = () => {
           justifyContent="space-between"
           alignItems="center"
         >
-          <Button
-            variant={!isUserOnFreePlan ? 'tertiary' : 'primary'}
-            size="small"
-            onClick={navigateToPricing}
-          >
-            Upgrade Plan
-          </Button>
+          {!isUserOnEnterprisePlan && (
+            <Button
+              variant={!isUserOnFreePlan ? 'tertiary' : 'primary'}
+              size="small"
+              onClick={navigateToPricing}
+            >
+              Upgrade Plan
+            </Button>
+          )}
         </Box>
       </Box>
 

diff --git a/src/hooks/useGetPricingPlanDetails.tsx b/src/hooks/useGetPricingPlanDetails.tsx
@@ -27,6 +27,7 @@ export const useGetPricingPlanDetails = (pricingPlanStatus: any) => {
   );
 
   const isUserOnFreePlan = selectedPlan?.id == 1;
+  const isUserOnEnterprisePlan = selectedPlan?.id == 4;
 
-  return { selectedPlan, isUserOnFreePlan, pricingListDescriptions };
+  return { selectedPlan, isUserOnFreePlan, isUserOnEnterprisePlan, pricingListDescriptions };
 };