From 93214efc54f980fe2a7014c4650c3ae5c1124037 Mon Sep 17 00:00:00 2001
From: Conor Macpherson <conor.macpherson@mattermost.com>
Date: Tue, 28 Mar 2023 14:41:28 -0400
Subject: [PATCH] add further check for ability to expand.

---
 .../common/hooks/useControlSelfHostedExpansionModal.ts      | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/webapp/channels/src/components/common/hooks/useControlSelfHostedExpansionModal.ts b/webapp/channels/src/components/common/hooks/useControlSelfHostedExpansionModal.ts
index bda0099ca8..42ef497e86 100644
--- a/webapp/channels/src/components/common/hooks/useControlSelfHostedExpansionModal.ts
+++ b/webapp/channels/src/components/common/hooks/useControlSelfHostedExpansionModal.ts
@@ -16,6 +16,7 @@ import {STORAGE_KEY_EXPANSION_IN_PROGRESS} from 'components/self_hosted_expansio
 import SelfHostedExpansionModal from 'components/self_hosted_expansion_modal';
 
 import {useControlModal, ControlModal} from './useControlModal';
+import useCanSelfHostedExpand from './useCanSelfHostedExpand';
 
 interface HookOptions{
     onClick?: () => void;
@@ -25,6 +26,7 @@ interface HookOptions{
 export default function useControlSelfHostedExpansionModal(options: HookOptions): ControlModal {
     const dispatch = useDispatch();
     const currentUser = useSelector(getCurrentUser);
+    const canExpand = useCanSelfHostedExpand();
     const controlModal = useControlModal({
         modalId: ModalIdentifiers.SELF_HOSTED_EXPANSION,
         dialogType: SelfHostedExpansionModal,
@@ -34,6 +36,10 @@ export default function useControlSelfHostedExpansionModal(options: HookOptions)
         return {
             ...controlModal,
             open: async () => {
+                if (!canExpand) {
+                    return;
+                }
+
                 const purchaseInProgress = localStorage.getItem(STORAGE_KEY_EXPANSION_IN_PROGRESS) === 'true';
 
                 // check if user already has an open purchase modal in current browser.