Function resetUsageForSubscription

apps/sim/lib/billing/webhooks/invoices.ts:413–629

(sub: {
  plan: string | null
  referenceId: string
  periodStart?: Date | null
  periodEnd?: Date | null
})

}

export async function resetUsageForSubscription(sub: {
  plan: string | null
  referenceId: string
  periodStart?: Date | null
  periodEnd?: Date | null
}) {
  const billingPeriod =
    sub.periodStart && sub.periodEnd ? { start: sub.periodStart, end: sub.periodEnd } : null

  if (await isSubscriptionOrgScoped(sub)) {
    const ledgerUsageByUser = billingPeriod
      ? await getBillingPeriodUsageCostByUser(
          { type: 'organization', id: sub.referenceId },
          billingPeriod
        )
      : new Map<string, number>()
    // Copilot-family ledger per user, so last-period copilot mirrors last-period
    // cost (baseline + usage_log) instead of capturing the baseline alone.
    const copilotLedgerByUser = billingPeriod
      ? await getBillingPeriodUsageCostByUser(
          { type: 'organization', id: sub.referenceId },
          billingPeriod,
          COPILOT_USAGE_SOURCES
        )
      : new Map<string, number>()

    await db.transaction(async (tx) => {
      await tx.execute(sql.raw(`SET LOCAL lock_timeout = '${BILLING_LOCK_TIMEOUT_MS}ms'`))

      const ownerRows = await tx
        .select({ userId: member.userId })
        .from(member)
        .where(and(eq(member.organizationId, sub.referenceId), eq(member.role, 'owner')))
        .for('update')
        .limit(1)

      const ownerId = ownerRows[0]?.userId
      if (ownerId) {
        await tx
          .select({ userId: userStats.userId })
          .from(userStats)
          .where(eq(userStats.userId, ownerId))
          .for('update')
          .limit(1)
      }

      const membersRows = await tx
        .select({ userId: member.userId })
        .from(member)
        .where(eq(member.organizationId, sub.referenceId))

      const memberIds = membersRows.map((row) => row.userId)

      // Lock every member's userStats before the organization row so this path
      // follows the canonical userStats → organization order shared by the
      // join, remove, threshold-billing, and storage-transfer paths. Locking
      // organization first would invert against them and risk an AB-BA
      // deadlock. The per-member UPDATE below re-locks these rows (no-op).