Function getUserStorageLimit

apps/sim/lib/billing/storage/limits.ts:91–124

(
  userId: string,
  prefetchedSub?: HighestPrioritySubscription | null
)

 *   (leave `undefined`) to fetch it here.
 */
export async function getUserStorageLimit(
  userId: string,
  prefetchedSub?: HighestPrioritySubscription | null
): Promise<number> {
  try {
    const sub = prefetchedSub === undefined ? await resolveSub(userId) : prefetchedSub

    const limits = getStorageLimits()

    if (!sub || isFree(sub.plan)) {
      return limits.free
    }

    if (isOrgScopedSubscription(sub, userId)) {
      const metadata = sub.metadata as { customStorageLimitGB?: number } | null
      if (metadata?.customStorageLimitGB) {
        return metadata.customStorageLimitGB * 1024 * 1024 * 1024
      }
      return isEnterprise(sub.plan) ? limits.enterpriseDefault : limits.team
    }

    // Personally-scoped plans use the per-plan default storage cap.
    const effectivePlan = getPlanTypeForLimits(sub.plan)
    const limitByPlan: Record<'free' | 'pro' | 'team', number> = {
      free: limits.free,
      pro: limits.pro,
      team: limits.team,
    }
    return limitByPlan[effectivePlan as 'free' | 'pro' | 'team'] ?? limits.free
  } catch (error) {
    logger.error('Error getting user storage limit:', error)
    return getStorageLimits().free
  }
}

/**
 * Get current storage usage for a user. Returns usage in bytes.