fix(data-retention): bypass enterprise gate when billing is disabled for self-hosted

Author: waleedlatif1

apps/sim/app/api/workspaces/[id]/data-retention/route.ts

@@ -10,6 +10,7 @@ import { CLEANUP_CONFIG } from '@/lib/billing/cleanup-dispatcher'
 import { getHighestPrioritySubscription } from '@/lib/billing/core/plan'
 import { isEnterprisePlan } from '@/lib/billing/core/subscription'
 import { getPlanType, type PlanCategory } from '@/lib/billing/plan-helpers'
+import { isBillingEnabled } from '@/lib/core/config/feature-flags'
 import { getUserEntityPermissions } from '@/lib/workspaces/permissions/utils'
 import { getWorkspaceBilledAccountUserId } from '@/lib/workspaces/utils'
 
@@ -79,7 +80,7 @@ export async function GET(_request: NextRequest, { params }: { params: Promise<{
 
     const plan = await resolveWorkspacePlan(ws.billedAccountUserId)
     const defaults = getPlanDefaults(plan)
-    const isEnterpriseWorkspace = plan === 'enterprise'
+    const isEnterpriseWorkspace = !isBillingEnabled || plan === 'enterprise'
 
     return NextResponse.json({
       success: true,
@@ -135,12 +136,14 @@ export async function PUT(request: NextRequest, { params }: { params: Promise<{
       return NextResponse.json({ error: 'Workspace not found' }, { status: 404 })
     }
 
-    const hasEnterprise = await isEnterprisePlan(billedAccountUserId)
-    if (!hasEnterprise) {
-      return NextResponse.json(
-        { error: 'Data Retention configuration is available on Enterprise plans only' },
-        { status: 403 }
-      )
+    if (isBillingEnabled) {
+      const hasEnterprise = await isEnterprisePlan(billedAccountUserId)
+      if (!hasEnterprise) {
+        return NextResponse.json(
+          { error: 'Data Retention configuration is available on Enterprise plans only' },
+          { status: 403 }
+        )
+      }
     }
 
     const body = await request.json()

apps/sim/ee/data-retention/components/data-retention-settings.tsx

@@ -5,6 +5,7 @@ import { createLogger } from '@sim/logger'
 import { toError } from '@sim/utils/errors'
 import { useParams } from 'next/navigation'
 import { Button, Combobox, toast } from '@/components/emcn'
+import { isBillingEnabled } from '@/lib/core/config/feature-flags'
 import { useUserPermissionsContext } from '@/app/workspace/[workspaceId]/providers/workspace-permissions-provider'
 import { SettingRow } from '@/ee/components/setting-row'
 import { DataRetentionSkeleton } from '@/ee/data-retention/components/data-retention-skeleton'
@@ -132,7 +133,7 @@ export function DataRetentionSettings() {
     )
   }
 
-  if (!data.isEnterprise) {
+  if (isBillingEnabled && !data.isEnterprise) {
     return (
       <div className='flex h-full items-center justify-center text-[var(--text-muted)] text-sm'>
         Data retention is available on Enterprise plans only.