Function performAutoTopUpForEntity

apps/web/src/lib/autoTopUp.ts:183–342

* Generalized auto-top-up logic that works for both users and organizations.

(
  entity: AutoTopUpEntity,
  traceId: string
)

 * Generalized auto-top-up logic that works for both users and organizations.
 */
async function performAutoTopUpForEntity(
  entity: AutoTopUpEntity,
  traceId: string
): Promise<AutoTopUpResult> {
  const ownerColumn =
    entity.type === 'user'
      ? auto_top_up_configs.owned_by_user_id
      : auto_top_up_configs.owned_by_organization_id;
  const ownerId = entity.type === 'user' ? entity.user.id : entity.organization.id;

  // Atomically check and acquire lock in a single query using SQL NOW()
  const [config] = await db
    .update(auto_top_up_configs)
    .set({ attempt_started_at: sql`NOW()` })
    .where(
      and(
        eq(ownerColumn, ownerId),
        or(
          isNull(auto_top_up_configs.attempt_started_at),
          lt(
            auto_top_up_configs.attempt_started_at,
            sql`NOW() - INTERVAL '${sql.raw(String(ATTEMPT_LOCK_TIMEOUT_SECONDS))} second'`
          )
        )
      )
    )
    .returning({
      id: auto_top_up_configs.id,
      stripe_payment_method_id: auto_top_up_configs.stripe_payment_method_id,
      amount_cents: auto_top_up_configs.amount_cents,
      attempt_started_at: auto_top_up_configs.attempt_started_at,
    });

  if (!config) {
    // Either no config exists, or concurrent attempt in progress
    // Check which case it is
    const existingConfig = await db.query.auto_top_up_configs.findFirst({
      where: eq(ownerColumn, ownerId),
    });

    if (!existingConfig) {
      await disableAutoTopUpForEntity(entity, 'no_payment_method_saved');
      return failureResult('no_payment_method_saved');
    }
    return failureResult('concurrent_attempt_in_progress');
  }

  // Re-check balance after acquiring lock to prevent duplicate top-ups
  // (another request may have completed a top-up while we were waiting for the lock)
  // We fetch fresh data from DB and compute balance directly to avoid
  // calling getBalanceForUser which would create a cycle (it calls maybePerformAutoTopUp)
  const { currentBalance_USD, stripe_customer_id } =
    await getEntityBalanceAndStripeCustomer(entity);
  const threshold =
    entity.type === 'user' ? AUTO_TOP_UP_THRESHOLD_DOLLARS : ORG_AUTO_TOP_UP_THRESHOLD_DOLLARS;
  if (currentBalance_USD >= threshold) {
    // Balance is now sufficient, release lock and exit
    await db