aljaz/opencode
1
0
Fork 0
mirror of https://github.com/aljazceru/opencode.git synced 2025-12-23 18:54:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix: no payment method (#2706)

Browse Source
This commit is contained in:
Aiden Cline
2025-09-20 03:15:44 -05:00
committed by GitHub
parent 5acdd70587
commit d63ce40af2
1 changed files with 29 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
packages/console/app/src/routes/zen/handler.ts
View File

@@ -88,6 +88,7 @@ export async function handler(
const authInfo = await authenticate() const authInfo = await authenticate()
const modelInfo = validateModel(body.model, authInfo) const modelInfo = validateModel(body.model, authInfo)
const providerInfo = selectProvider(modelInfo, authInfo) const providerInfo = selectProvider(modelInfo, authInfo)
if (authInfo && !providerInfo.allowAnonymous) validateBilling(authInfo)
logger.metric({ provider: providerInfo.id }) logger.metric({ provider: providerInfo.id })
// Request to model provider // Request to model provider
@@ -250,33 +251,43 @@ export async function handler(
}) })
const isFree = FREE_WORKSPACES.includes(data.workspaceID) const isFree = FREE_WORKSPACES.includes(data.workspaceID)
if (!isFree) {
if (!data.paymentMethodID) throw new CreditsError("No payment method")
if (data.balance <= 0) throw new CreditsError("Insufficient balance")
if (
data.monthlyLimit &&
data.monthlyUsage &&
data.timeMonthlyUsageUpdated &&
data.monthlyUsage >= centsToMicroCents(data.monthlyLimit * 100)
) {
const now = new Date()
const currentYear = now.getUTCFullYear()
const currentMonth = now.getUTCMonth()
const dateYear = data.timeMonthlyUsageUpdated.getUTCFullYear()
const dateMonth = data.timeMonthlyUsageUpdated.getUTCMonth()
if (currentYear === dateYear && currentMonth === dateMonth)
throw new MonthlyLimitError(`You have reached your monthly spending limit of $${data.monthlyLimit}.`)
}
}
return { return {
apiKeyId: data.apiKey, apiKeyId: data.apiKey,
workspaceID: data.workspaceID, workspaceID: data.workspaceID,
dataShare: data.dataShare, dataShare: data.dataShare,
billing: {
paymentMethodID: data.paymentMethodID,
balance: data.balance,
monthlyLimit: data.monthlyLimit,
monthlyUsage: data.monthlyUsage,
timeMonthlyUsageUpdated: data.timeMonthlyUsageUpdated,
},
isFree, isFree,
} }
} }
function validateBilling(authInfo: Awaited<ReturnType<typeof authenticate>>) {
if (!authInfo || authInfo.isFree) return
const billing = authInfo.billing
if (!billing.paymentMethodID) throw new CreditsError("No payment method")
if (billing.balance <= 0) throw new CreditsError("Insufficient balance")
if (
billing.monthlyLimit &&
billing.monthlyUsage &&
billing.timeMonthlyUsageUpdated &&
billing.monthlyUsage >= centsToMicroCents(billing.monthlyLimit * 100)
) {
const now = new Date()
const currentYear = now.getUTCFullYear()
const currentMonth = now.getUTCMonth()
const dateYear = billing.timeMonthlyUsageUpdated.getUTCFullYear()
const dateMonth = billing.timeMonthlyUsageUpdated.getUTCMonth()
if (currentYear === dateYear && currentMonth === dateMonth)
throw new MonthlyLimitError(`You have reached your monthly spending limit of $${billing.monthlyLimit}.`)
}
}
function validateModel(reqModel: string, authInfo: Awaited<ReturnType<typeof authenticate>>) { function validateModel(reqModel: string, authInfo: Awaited<ReturnType<typeof authenticate>>) {
const json = JSON.parse(Resource.ZEN_MODELS.value) const json = JSON.parse(Resource.ZEN_MODELS.value)