Function calculateUsageAndBalance

packages/billing/src/balance-calculator.ts:295–427

(
  params: OptionalFields<
    {
      userId: string
      quotaResetDate: Date
      now: Date
      conn: DbConn
      isPersonalContext: boolean
      includeSubscriptionCredits: boolean
      logger: Logger
    } & ParamsOf<typeof getOrderedActiveGrants>,
    'now' | 'conn' | 'isPersonalContext' | 'includeSubscriptionCredits'
  >,
)

 * This is more efficient than calculating them separately.
 */
export async function calculateUsageAndBalance(
  params: OptionalFields<
    {
      userId: string
      quotaResetDate: Date
      now: Date
      conn: DbConn
      isPersonalContext: boolean
      includeSubscriptionCredits: boolean
      logger: Logger
    } & ParamsOf<typeof getOrderedActiveGrants>,
    'now' | 'conn' | 'isPersonalContext' | 'includeSubscriptionCredits'
  >,
): Promise<CreditUsageAndBalance> {
  const withDefaults = {
    now: new Date(),
    conn: db, // Add optional conn parameter to pass transaction
    isPersonalContext: false, // Add flag to exclude organization credits for personal usage
    includeSubscriptionCredits: false,
    ...params,
  }
  const {
    userId,
    quotaResetDate,
    now,
    isPersonalContext,
    includeSubscriptionCredits,
    logger,
  } = withDefaults

  // Get all relevant grants in one query, using the provided connection
  const grants = await getOrderedActiveGrants(withDefaults)

  // Initialize breakdown and principals with all grant types set to 0
  const initialBreakdown: Record<GrantType, number> = {} as Record<
    GrantType,
    number
  >
  const initialPrincipals: Record<GrantType, number> = {} as Record<
    GrantType,
    number
  >

  for (const type of GrantTypeValues) {
    initialBreakdown[type] = 0
    initialPrincipals[type] = 0
  }

  // Initialize balance structure
  const balance: CreditBalance = {
    totalRemaining: 0,
    totalDebt: 0,
    netBalance: 0,
    breakdown: initialBreakdown,
    principals: initialPrincipals,
  }

  // Calculate both metrics in one pass