plan upgrade to refresh session

Author: icecrasher321

apps/sim/app/_shell/providers/session-provider.tsx

@@ -2,6 +2,7 @@
 
 import type React from 'react'
 import { createContext, useCallback, useEffect, useMemo, useState } from 'react'
+import { useQueryClient } from '@tanstack/react-query'
 import posthog from 'posthog-js'
 import { client } from '@/lib/auth/auth-client'
 
@@ -35,12 +36,15 @@ export function SessionProvider({ children }: { children: React.ReactNode }) {
   const [data, setData] = useState<AppSession>(null)
   const [isPending, setIsPending] = useState(true)
   const [error, setError] = useState<Error | null>(null)
+  const queryClient = useQueryClient()
 
-  const loadSession = useCallback(async () => {
+  const loadSession = useCallback(async (bypassCache = false) => {
     try {
       setIsPending(true)
       setError(null)
-      const res = await client.getSession()
+      const res = bypassCache
+        ? await client.getSession({ query: { disableCookieCache: true } })
+        : await client.getSession()
       setData(res?.data ?? null)
     } catch (e) {
       setError(e instanceof Error ? e : new Error('Failed to fetch session'))
@@ -50,8 +54,25 @@ export function SessionProvider({ children }: { children: React.ReactNode }) {
   }, [])
 
   useEffect(() => {
-    loadSession()
-  }, [loadSession])
+    // Check if user was redirected after plan upgrade
+    const params = new URLSearchParams(window.location.search)
+    const wasUpgraded = params.get('upgraded') === 'true'
+
+    if (wasUpgraded) {
+      params.delete('upgraded')
+      const newUrl = params.toString()
+        ? `${window.location.pathname}?${params.toString()}`
+        : window.location.pathname
+      window.history.replaceState({}, '', newUrl)
+    }
+
+    loadSession(wasUpgraded).then(() => {
+      if (wasUpgraded) {
+        queryClient.invalidateQueries({ queryKey: ['organizations'] })
+        queryClient.invalidateQueries({ queryKey: ['subscription'] })
+      }
+    })
+  }, [loadSession, queryClient])
 
   useEffect(() => {
     if (isPending || typeof posthog.identify !== 'function') {

apps/sim/app/workspace/[workspaceId]/w/components/sidebar/components/settings-modal/components/subscription/subscription.tsx

@@ -8,6 +8,7 @@ import { Skeleton } from '@/components/ui'
 import { useSession } from '@/lib/auth/auth-client'
 import { useSubscriptionUpgrade } from '@/lib/billing/client/upgrade'
 import { USAGE_THRESHOLDS } from '@/lib/billing/client/usage-visualization'
+import { getEffectiveSeats } from '@/lib/billing/subscriptions/utils'
 import { cn } from '@/lib/core/utils/cn'
 import { getBaseUrl } from '@/lib/core/utils/urls'
 import { getUserRole } from '@/lib/workspaces/organization/utils'
@@ -191,7 +192,13 @@ export function Subscription() {
   const [upgradeError, setUpgradeError] = useState<'pro' | 'team' | null>(null)
   const usageLimitRef = useRef<UsageLimitRef | null>(null)
 
-  const isLoading = isSubscriptionLoading || isUsageLimitLoading || isWorkspaceLoading
+  const isOrgPlan =
+    subscriptionData?.data?.plan === 'team' || subscriptionData?.data?.plan === 'enterprise'
+  const isLoading =
+    isSubscriptionLoading ||
+    isUsageLimitLoading ||
+    isWorkspaceLoading ||
+    (isOrgPlan && isOrgBillingLoading)
 
   const subscription = {
     isFree: subscriptionData?.data?.plan === 'free' || !subscriptionData?.data?.plan,
@@ -204,7 +211,7 @@ export function Subscription() {
       subscriptionData?.data?.status === 'active',
     plan: subscriptionData?.data?.plan || 'free',
     status: subscriptionData?.data?.status || 'inactive',
-    seats: organizationBillingData?.totalSeats ?? 0,
+    seats: getEffectiveSeats(subscriptionData?.data),
   }
 
   const usage = {
@@ -445,16 +452,10 @@ export function Subscription() {
             ? `${subscription.seats} seats`
             : undefined
         }
-        current={
-          subscription.isEnterprise || subscription.isTeam
-            ? (organizationBillingData?.totalCurrentUsage ?? usage.current)
-            : usage.current
-        }
+        current={usage.current}
         limit={
           subscription.isEnterprise || subscription.isTeam
-            ? organizationBillingData?.totalUsageLimit ||
-              organizationBillingData?.minimumBillingAmount ||
-              usage.limit
+            ? organizationBillingData?.data?.totalUsageLimit
             : !subscription.isFree &&
                 (permissions.canEditUsageLimit || permissions.showTeamMemberView)
               ? usage.current // placeholder; rightContent will render UsageLimit
@@ -468,19 +469,31 @@ export function Subscription() {
             <UsageLimit
               ref={usageLimitRef}
               currentLimit={
-                subscription.isTeam && isTeamAdmin
-                  ? organizationBillingData?.totalUsageLimit || usage.limit
+                (subscription.isTeam || subscription.isEnterprise) &&
+                isTeamAdmin &&
+                organizationBillingData?.data
+                  ? organizationBillingData.data.totalUsageLimit
                   : usageLimitData.currentLimit || usage.limit
               }
               currentUsage={usage.current}
               canEdit={permissions.canEditUsageLimit}
               minimumLimit={
-                subscription.isTeam && isTeamAdmin
-                  ? organizationBillingData?.minimumBillingAmount || (subscription.isPro ? 20 : 40)
+                (subscription.isTeam || subscription.isEnterprise) &&
+                isTeamAdmin &&
+                organizationBillingData?.data
+                  ? organizationBillingData.data.minimumBillingAmount
                   : usageLimitData.minimumLimit || (subscription.isPro ? 20 : 40)
               }
-              context={subscription.isTeam && isTeamAdmin ? 'organization' : 'user'}
-              organizationId={subscription.isTeam && isTeamAdmin ? activeOrgId : undefined}
+              context={
+                (subscription.isTeam || subscription.isEnterprise) && isTeamAdmin
+                  ? 'organization'
+                  : 'user'
+              }
+              organizationId={
+                (subscription.isTeam || subscription.isEnterprise) && isTeamAdmin
+                  ? activeOrgId
+                  : undefined
+              }
               onLimitUpdated={() => {
                 logger.info('Usage limit updated')
               }}

apps/sim/app/workspace/[workspaceId]/w/components/sidebar/components/settings-modal/settings-modal.tsx

@@ -174,6 +174,7 @@ export function SettingsModal({ open, onOpenChange }: SettingsModalProps) {
 
   const userEmail = session?.user?.email
   const userId = session?.user?.id
+
   const userRole = getUserRole(activeOrganization, userEmail)
   const isOwner = userRole === 'owner'
   const isAdmin = userRole === 'admin'

apps/sim/hooks/queries/organization.ts

@@ -47,6 +47,22 @@ export function useOrganizations() {
   })
 }
 
+/**
+ * Fetch organizations where user is owner/admin directly from DB
+ * This bypasses session's activeOrganizationId for reliable role checking
+ */
+export function useUserOwnedOrganizations() {
+  return useQuery({
+    queryKey: ['user-owned-organizations'],
+    queryFn: async () => {
+      const res = await fetch('/api/organizations')
+      if (!res.ok) return { organizations: [] }
+      return res.json()
+    },
+    staleTime: 0, // Always refetch to ensure fresh org membership data
+  })
+}
+
 /**
  * Fetch a specific organization by ID
  */

apps/sim/lib/billing/client/upgrade.ts

@@ -81,12 +81,13 @@ export function useSubscriptionUpgrade() {
       }
 
       const currentUrl = `${window.location.origin}${window.location.pathname}`
+      const successUrl = `${currentUrl}?upgraded=true`
 
       try {
         const upgradeParams = {
           plan: targetPlan,
           referenceId,
-          successUrl: currentUrl,
+          successUrl,
           cancelUrl: currentUrl,
           ...(targetPlan === 'team' && { seats: CONSTANTS.INITIAL_TEAM_SEATS }),
         } as const