From 02600880d1042ac69d59fc1856549478219f3c04 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ivar=20Conradi=20=C3=98sthus?= <ivar@getunleash.ai>
Date: Mon, 19 Jun 2023 13:48:26 +0200
Subject: [PATCH] fix: specific actions for enterprise trial messages (#4001)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Co-authored-by: Nuno Góis <github@nunogois.com>
---
 .../common/InstanceStatus/InstanceStatus.tsx  | 20 ++++++++++++++++++-
 .../InstanceStatus/InstanceStatusBar.tsx      | 14 ++++++++-----
 2 files changed, 28 insertions(+), 6 deletions(-)

diff --git a/frontend/src/component/common/InstanceStatus/InstanceStatus.tsx b/frontend/src/component/common/InstanceStatus/InstanceStatus.tsx
index 62f1213dc3..e474004158 100644
--- a/frontend/src/component/common/InstanceStatus/InstanceStatus.tsx
+++ b/frontend/src/component/common/InstanceStatus/InstanceStatus.tsx
@@ -5,7 +5,7 @@ import { ConditionallyRender } from 'component/common/ConditionallyRender/Condit
 import { Dialogue } from 'component/common/Dialogue/Dialogue';
 import { Typography } from '@mui/material';
 import { useNavigate } from 'react-router-dom';
-import { IInstanceStatus } from 'interfaces/instance';
+import { IInstanceStatus, InstancePlan } from 'interfaces/instance';
 import { ADMIN } from 'component/providers/AccessProvider/permissions';
 import AccessContext from 'contexts/AccessContext';
 import useInstanceStatusApi from 'hooks/api/actions/useInstanceStatusApi/useInstanceStatusApi';
@@ -44,6 +44,24 @@ const TrialDialog: VFC<ITrialDialogProps> = ({
         return () => clearInterval(interval);
     }, [expired]);
 
+    if (instanceStatus.plan === InstancePlan.ENTERPRISE) {
+        return (
+            <Dialogue
+                open={dialogOpen}
+                secondaryButtonText="Remind me later"
+                onClose={() => {
+                    setDialogOpen(false);
+                }}
+                title={`Your free ${instanceStatus.plan} trial has expired!`}
+            >
+                <Typography>
+                    Please contact your Unleash sales representative to avoid{' '}
+                    <strong>deletion of your Unleash account.</strong>
+                </Typography>
+            </Dialogue>
+        );
+    }
+
     if (hasAccess(ADMIN)) {
         return (
             <Dialogue
diff --git a/frontend/src/component/common/InstanceStatus/InstanceStatusBar.tsx b/frontend/src/component/common/InstanceStatus/InstanceStatusBar.tsx
index 089c225800..f26a4ed8e4 100644
--- a/frontend/src/component/common/InstanceStatus/InstanceStatusBar.tsx
+++ b/frontend/src/component/common/InstanceStatus/InstanceStatusBar.tsx
@@ -1,5 +1,5 @@
 import { styled, Button, Typography } from '@mui/material';
-import { IInstanceStatus } from 'interfaces/instance';
+import { IInstanceStatus, InstancePlan } from 'interfaces/instance';
 import { INSTANCE_STATUS_BAR_ID } from 'utils/testIds';
 import { InfoOutlined, WarningAmber } from '@mui/icons-material';
 import { useNavigate } from 'react-router-dom';
@@ -86,7 +86,7 @@ const StatusBarExpired = ({ instanceStatus }: IInstanceStatusBarProps) => {
                 has expired. <strong>Upgrade trial</strong> otherwise your{' '}
                 <strong>account will be deleted.</strong>
             </Typography>
-            <BillingLink />
+            <BillingLink instanceStatus={instanceStatus} />
         </StyledWarningBar>
     );
 };
@@ -105,7 +105,7 @@ const StatusBarExpiresSoon = ({ instanceStatus }: IInstanceStatusBarProps) => {
                 <strong>{timeRemaining}</strong> left of your free{' '}
                 {instanceStatus.plan} trial.
             </Typography>
-            <BillingLink />
+            <BillingLink instanceStatus={instanceStatus} />
         </StyledInfoBar>
     );
 };
@@ -118,12 +118,12 @@ const StatusBarExpiresLater = ({ instanceStatus }: IInstanceStatusBarProps) => {
                 <strong>Heads up!</strong> You're currently on a free{' '}
                 {instanceStatus.plan} trial account.
             </Typography>
-            <BillingLink />
+            <BillingLink instanceStatus={instanceStatus} />
         </StyledInfoBar>
     );
 };
 
-const BillingLink = () => {
+const BillingLink = ({ instanceStatus }: IInstanceStatusBarProps) => {
     const { hasAccess } = useContext(AccessContext);
     const navigate = useNavigate();
 
@@ -131,6 +131,10 @@ const BillingLink = () => {
         return null;
     }
 
+    if (instanceStatus.plan === InstancePlan.ENTERPRISE) {
+        return null;
+    }
+
     return (
         <StyledButton
             onClick={() => navigate('/admin/billing')}