fix(server): pre-check credit balance before health-check AI analysis

The health-check endpoint runs the paid OpenAI analysis before charging
(failed analyses are intentionally not charged). Without a balance
pre-check, a user below HEALTH_CHECK_COST could trigger unlimited free
analyses with fresh idempotency keys.

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
This commit is contained in:
Timo Knuth
2026-07-06 13:02:36 +02:00
parent 3fea520317
commit eda0ac98d9
3 changed files with 31 additions and 0 deletions

View File

@@ -51,6 +51,7 @@ const {
getAccountSnapshot,
getBillingSummary,
getEndpointResponse,
ensureSufficientCredits,
isInsufficientCreditsError,
claimNotificationOnce,
simulatePurchase,
@@ -943,6 +944,10 @@ app.post('/v1/health-check', async (request, response) => {
const accountSnapshot = await getAccountSnapshot(db, userId);
ensureNotGuest(userId, HEALTH_CHECK_COST);
// Balance pre-check: the paid AI analysis below runs BEFORE the charge
// (failed analyses are intentionally not charged), so without this check a
// user with insufficient credits could trigger unlimited free analyses.
ensureSufficientCredits(accountSnapshot, HEALTH_CHECK_COST);
if (!isOpenAiConfigured()) {
const error = new Error('OpenAI health check is unavailable. Please configure OPENAI_API_KEY.');

View File

@@ -259,6 +259,15 @@ const getAvailableCredits = (account) => {
return monthlyRemaining + Math.max(0, account.topupBalance);
};
// Pre-flight balance check for endpoints that must not run paid work
// (e.g. OpenAI analyses) before the actual charge happens.
const ensureSufficientCredits = (account, cost) => {
const available = getAvailableCredits(account);
if (available < cost) {
throw createInsufficientCreditsError(cost, available);
}
};
const buildBillingSummary = (account) => {
return {
entitlement: {
@@ -817,6 +826,7 @@ module.exports = {
getBillingSummary,
getEndpointResponse,
getMonthlyAllowanceForPlan,
ensureSufficientCredits,
isInsufficientCreditsError,
runInTransaction,
simulatePurchase,

View File

@@ -5,6 +5,7 @@ const {
alignAccountToCurrentCycle,
getAvailableCredits,
consumeCredits,
ensureSufficientCredits,
getMonthlyAllowanceForPlan,
} = require('../lib/billing');
@@ -72,3 +73,18 @@ test('monthly cycle rollover resets free usage', () => {
assert.equal(aligned.monthlyAllowance, 3);
assert.equal(getAvailableCredits(aligned), 3);
});
test('ensureSufficientCredits throws 402 when balance is below cost', () => {
const account = freeAccount({ monthlyAllowance: 3, usedThisCycle: 2, topupBalance: 0 });
assert.throws(() => ensureSufficientCredits(account, 2), (error) => {
assert.equal(error.code, 'INSUFFICIENT_CREDITS');
assert.equal(error.status, 402);
assert.deepEqual(error.metadata, { required: 2, available: 1 });
return true;
});
});
test('ensureSufficientCredits passes when balance covers cost', () => {
const account = freeAccount({ monthlyAllowance: 3, usedThisCycle: 1, topupBalance: 0 });
assert.doesNotThrow(() => ensureSufficientCredits(account, 2));
});