From b625f3502c1bdb7434d3a7f1ee64e3ba685c91da Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Kevin=20Gr=C3=BCneberg?= <k.grueneberg1994@gmail.com>
Date: Wed, 25 Mar 2026 16:46:38 +0800
Subject: [PATCH] fix: use entitlement check for project pausing (#44172)

Project pausing is available on platform plan, so we should not do a
check based on Free Plan
---
 .../Infrastructure/PauseProjectButton.tsx     | 29 ++++++++++++-------
 1 file changed, 18 insertions(+), 11 deletions(-)

diff --git a/apps/studio/components/interfaces/Settings/General/Infrastructure/PauseProjectButton.tsx b/apps/studio/components/interfaces/Settings/General/Infrastructure/PauseProjectButton.tsx
index b689f02bd1d..e59a59a036d 100644
--- a/apps/studio/components/interfaces/Settings/General/Infrastructure/PauseProjectButton.tsx
+++ b/apps/studio/components/interfaces/Settings/General/Infrastructure/PauseProjectButton.tsx
@@ -4,11 +4,7 @@ import { useSetProjectStatus } from 'data/projects/project-detail-query'
 import { useProjectPauseMutation } from 'data/projects/project-pause-mutation'
 import { useAsyncCheckPermissions } from 'hooks/misc/useCheckPermissions'
 import { useSelectedOrganizationQuery } from 'hooks/misc/useSelectedOrganization'
-import {
-  useIsAwsK8sCloudProvider,
-  useIsProjectActive,
-  useSelectedProjectQuery,
-} from 'hooks/misc/useSelectedProject'
+import { useIsProjectActive, useSelectedProjectQuery } from 'hooks/misc/useSelectedProject'
 import { PROJECT_STATUS } from 'lib/constants'
 import { Pause } from 'lucide-react'
 import { useRouter } from 'next/router'
@@ -16,6 +12,8 @@ import { useState } from 'react'
 import { toast } from 'sonner'
 import ConfirmationModal from 'ui-patterns/Dialogs/ConfirmationModal'
 
+import { useCheckEntitlements } from '@/hooks/misc/useCheckEntitlements'
+
 const PauseProjectButton = () => {
   const router = useRouter()
   const { data: project } = useSelectedProjectQuery()
@@ -32,10 +30,12 @@ const PauseProjectButton = () => {
     'queue_jobs.projects.pause'
   )
 
-  const isAwsK8s = useIsAwsK8sCloudProvider()
   const isFreePlan = organization?.plan.id === 'free'
   const isBranch = Boolean(project?.parent_project_ref)
-  const isPaidAndNotAwsK8s = !isBranch && !isFreePlan && !isAwsK8s
+  const { hasAccess: projectPausingAllowedInOrg } = useCheckEntitlements(
+    'project_pausing',
+    organization?.slug
+  )
 
   const { mutate: pauseProject, isPending: isPausing } = useProjectPauseMutation({
     onSuccess: (_, variables) => {
@@ -53,7 +53,12 @@ const PauseProjectButton = () => {
   }
 
   const buttonDisabled =
-    isPaidAndNotAwsK8s || project === undefined || isPaused || !canPauseProject || !isProjectActive
+    isBranch ||
+    !projectPausingAllowedInOrg ||
+    project === undefined ||
+    isPaused ||
+    !canPauseProject ||
+    !isProjectActive
 
   return (
     <>
@@ -72,9 +77,11 @@ const PauseProjectButton = () => {
                 ? 'You need additional permissions to pause this project'
                 : !isProjectActive
                   ? 'Unable to pause project as project is not active'
-                  : isPaidAndNotAwsK8s
-                    ? 'Projects on a paid plan will always be running'
-                    : undefined,
+                  : isBranch
+                    ? 'Branch projects cannot be paused'
+                    : !projectPausingAllowedInOrg && !isFreePlan
+                      ? 'Projects on a paid plan will always be running'
+                      : undefined,
           },
         }}
       >