Function workspaceQuotaDetails

coderd/x/chatd/chattool/quotaerror.go:94–147

(
	ctx context.Context,
	logger slog.Logger,
	db database.Store,
	ownerID uuid.UUID,
	organizationID uuid.UUID,
)

}

func workspaceQuotaDetails(
	ctx context.Context,
	logger slog.Logger,
	db database.Store,
	ownerID uuid.UUID,
	organizationID uuid.UUID,
) *quotaErrorDetails {
	if db == nil || ownerID == uuid.Nil || organizationID == uuid.Nil {
		return nil
	}

	quotaCtx := ctx
	if actor, ok := dbauthz.ActorFromContext(ctx); !ok || actor.ID != ownerID.String() {
		ownerCtx, err := asOwner(ctx, db, ownerID)
		if err != nil {
			logger.Debug(ctx, "failed to load owner authorization for quota lookup",
				slog.F("owner_id", ownerID),
				slog.F("organization_id", organizationID),
				slog.Error(err),
			)
			return nil
		}
		quotaCtx = ownerCtx
	}

	consumed, err := db.GetQuotaConsumedForUser(quotaCtx, database.GetQuotaConsumedForUserParams{
		OwnerID:        ownerID,
		OrganizationID: organizationID,
	})
	if err != nil {
		logger.Debug(ctx, "failed to load consumed workspace quota",
			slog.F("owner_id", ownerID),
			slog.F("organization_id", organizationID),
			slog.Error(err),
		)
		return nil
	}
	budget, err := db.GetQuotaAllowanceForUser(quotaCtx, database.GetQuotaAllowanceForUserParams{
		UserID:         ownerID,
		OrganizationID: organizationID,
	})
	if err != nil {
		logger.Debug(ctx, "failed to load workspace quota allowance",
			slog.F("owner_id", ownerID),
			slog.F("organization_id", organizationID),
			slog.Error(err),
		)
		return nil
	}
	return &quotaErrorDetails{
		CreditsConsumed: consumed,
		Budget:          budget,
	}
}

func quotaErrorToolResponse(
	ctx context.Context,