Stanko K.R.
23 lines
463 B
Ruby
23 lines
463 B
Ruby
module User::AiQuota
|
|
extend ActiveSupport::Concern
|
|
|
|
DEFAULT_QUOTA = Ai::Quota::Money.wrap("$100").in_microcents
|
|
|
|
included do
|
|
has_one :ai_quota, class_name: "Ai::Quota"
|
|
end
|
|
|
|
def spend_ai_quota(cost)
|
|
fetch_or_create_ai_quota.spend(cost)
|
|
end
|
|
|
|
def ensure_ai_quota_not_depleted
|
|
fetch_or_create_ai_quota.ensure_not_depleted
|
|
end
|
|
|
|
private
|
|
def fetch_or_create_ai_quota
|
|
ai_quota || create_ai_quota!(limit: DEFAULT_QUOTA)
|
|
end
|
|
end
|