Function getHighestPrioritySubscription

apps/sim/lib/billing/core/plan.ts:79–139

(
  userId: string,
  options: GetHighestPrioritySubscriptionOptions = {}
)

 * leak onto the user's row until the next billing cycle.
 */
export async function getHighestPrioritySubscription(
  userId: string,
  options: GetHighestPrioritySubscriptionOptions = {}
) {
  const { onError = 'return-null', executor = db } = options
  try {
    const [personalSubs, memberships] = await Promise.all([
      executor
        .select()
        .from(subscription)
        .where(
          and(
            eq(subscription.referenceId, userId),
            inArray(subscription.status, ENTITLED_SUBSCRIPTION_STATUSES)
          )
        ),
      executor
        .select({ organizationId: member.organizationId })
        .from(member)
        .where(eq(member.userId, userId)),
    ])

    const orgIds = memberships.map((m: { organizationId: string }) => m.organizationId)

    let orgSubs: typeof personalSubs = []
    if (orgIds.length > 0) {
      // Verify orgs exist to filter out orphaned subscriptions
      const existingOrgs = await executor
        .select({ id: organization.id })
        .from(organization)
        .where(inArray(organization.id, orgIds))

      const validOrgIds = existingOrgs.map((o) => o.id)

      if (validOrgIds.length > 0) {
        orgSubs = await executor
          .select()
          .from(subscription)
          .where(
            and(
              inArray(subscription.referenceId, validOrgIds),
              inArray(subscription.status, ENTITLED_SUBSCRIPTION_STATUSES)
            )
          )
      }
    }

    if (personalSubs.length === 0 && orgSubs.length === 0) return null

    return pickHighestPrioritySubscription(
      [...orgSubs, ...personalSubs],
      [checkEnterprisePlan, checkTeamPlan, checkProPlan]
    )
  } catch (error) {
    logger.error('Error getting highest priority subscription', { error, userId })
    if (onError === 'throw') {
      throw error
    }