Function getUserStorageUsage

apps/sim/lib/billing/storage/limits.ts:132–162

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

 *   to skip the `getHighestPrioritySubscription` lookup on hot paths.
 */
export async function getUserStorageUsage(
  userId: string,
  prefetchedSub?: HighestPrioritySubscription | null
): Promise<number> {
  try {
    const sub = prefetchedSub === undefined ? await resolveSub(userId) : prefetchedSub

    // Org-scoped subs share pooled `organization.storageUsedBytes`;
    // personal plans use `userStats`.
    if (isOrgScopedSubscription(sub, userId) && sub) {
      const orgRecord = await db
        .select({ storageUsedBytes: organization.storageUsedBytes })
        .from(organization)
        .where(eq(organization.id, sub.referenceId))
        .limit(1)

      return orgRecord.length > 0 ? orgRecord[0].storageUsedBytes || 0 : 0
    }

    const stats = await db
      .select({ storageUsedBytes: userStats.storageUsedBytes })
      .from(userStats)
      .where(eq(userStats.userId, userId))
      .limit(1)

    return stats.length > 0 ? stats[0].storageUsedBytes || 0 : 0
  } catch (error) {
    logger.error('Error getting user storage usage:', error)
    return 0
  }
}

/**
 * Check if user has storage quota available