Function triggerMonthlyResetAndGrant

packages/billing/src/grant-credits.ts:443–539

(params: {
  userId: string
  logger: Logger
})

}

export async function triggerMonthlyResetAndGrant(params: {
  userId: string
  logger: Logger
}): Promise<MonthlyResetResult> {
  const { userId, logger } = params

  const { result } = await withAdvisoryLockTransaction({
    callback: async (tx) => {
      const now = new Date()

      // Get user's current reset date and auto top-up status
      const user = await tx.query.user.findFirst({
        where: eq(schema.user.id, userId),
        columns: {
          next_quota_reset: true,
          auto_topup_enabled: true,
        },
      })

      if (!user) {
        throw new Error(`User ${userId} not found`)
      }

      const autoTopupEnabled = user.auto_topup_enabled ?? false
      const currentResetDate = user.next_quota_reset

      // If reset date is in the future, no action needed
      if (currentResetDate && currentResetDate > now) {
        return { quotaResetDate: currentResetDate, autoTopupEnabled }
      }

      // Calculate new reset date
      const newResetDate = getNextQuotaReset(currentResetDate)

      const [freeGrantAmount, referralBonus] = await Promise.all([
        getGrandfatheredFreeGrantAmount(params),
        calculateTotalLegacyReferralBonus(params),
      ])

      // Generate a deterministic operation ID based on userId and reset date to minute precision
      const timestamp = generateOperationIdTimestamp(newResetDate)
      const freeOperationId = `free-${userId}-${timestamp}`
      const referralOperationId = `referral-${userId}-${timestamp}`

      // Update the user's next reset date
      await tx
        .update(schema.user)
        .set({ next_quota_reset: newResetDate })
        .where(eq(schema.user.id, userId))

      if (freeGrantAmount > 0) {
        await executeGrantCreditOperation({
          ...params,
          amount: freeGrantAmount,
          type: 'free',
          description: 'Monthly free credits (grandfathered)',
          expiresAt: newResetDate,
          operationId: freeOperationId,