@windrun-huaiin/backend-core 10.0.1

package/dist/services/database/credit.service.js

@@ -0,0 +1,515 @@
+'use strict';
+
+var tslib_es6 = require('../../node_modules/.pnpm/@rollup_plugin-typescript@12.1.4_rollup@4.46.2_tslib@2.8.1_typescript@5.9.3/node_modules/tslib/tslib.es6.js');
+var client = require('@prisma/client');
+var constants = require('./constants.js');
+var creditInit = require('../../lib/credit-init.js');
+var prisma = require('../../prisma/prisma.js');
+var creditAuditLog_service = require('./creditAuditLog.service.js');
+
+const CREDIT_PURGE_CONFIG = {
+    [constants.CreditType.FREE]: {
+        amountKey: 'free',
+        balanceField: 'balanceFree',
+        limitField: 'totalFreeLimit',
+        startField: 'freeStart',
+        endField: 'freeEnd',
+    },
+    [constants.CreditType.PAID]: {
+        amountKey: 'paid',
+        balanceField: 'balancePaid',
+        limitField: 'totalPaidLimit',
+        startField: 'paidStart',
+        endField: 'paidEnd',
+    },
+    [constants.CreditType.ONE_TIME_PAID]: {
+        amountKey: 'oneTimePaid',
+        balanceField: 'balanceOneTimePaid',
+        limitField: 'totalOneTimePaidLimit',
+        startField: 'oneTimePaidStart',
+        endField: 'oneTimePaidEnd',
+    },
+};
+class CreditService {
+    normalizeAmounts(amounts) {
+        var _a, _b, _c;
+        return {
+            free: Math.trunc((_a = amounts === null || amounts === void 0 ? void 0 : amounts.free) !== null && _a !== void 0 ? _a : 0),
+            paid: Math.trunc((_b = amounts === null || amounts === void 0 ? void 0 : amounts.paid) !== null && _b !== void 0 ? _b : 0),
+            oneTimePaid: Math.trunc((_c = amounts === null || amounts === void 0 ? void 0 : amounts.oneTimePaid) !== null && _c !== void 0 ? _c : 0),
+        };
+    }
+    hasAnyChange(amounts) {
+        return amounts.free !== 0 || amounts.paid !== 0 || amounts.oneTimePaid !== 0;
+    }
+    ensureNonNegative(amounts, context) {
+        if (amounts.free < 0 || amounts.paid < 0 || amounts.oneTimePaid < 0) {
+            throw new Error(`${context}: negative credit adjustments are not allowed`);
+        }
+    }
+    ensureSufficientBalance(current, deduction) {
+        if (deduction.free > current.balanceFree) {
+            throw new Error('Insufficient free credits');
+        }
+        if (deduction.paid > current.balancePaid) {
+            throw new Error('Insufficient paid credits');
+        }
+        if (deduction.oneTimePaid > current.balanceOneTimePaid) {
+            throw new Error('Insufficient one-time paid credits');
+        }
+    }
+    ensureSufficientLimits(current, deduction) {
+        if (deduction.free > current.totalFreeLimit) {
+            throw new Error('Insufficient free credit limit');
+        }
+        if (deduction.paid > current.totalPaidLimit) {
+            throw new Error('Insufficient paid credit limit');
+        }
+        if (deduction.oneTimePaid > current.totalOneTimePaidLimit) {
+            throw new Error('Insufficient one-time paid credit limit');
+        }
+    }
+    buildIncrementData(amounts, limitAdjustments) {
+        const data = {};
+        if (amounts.free !== 0) {
+            data.balanceFree = { increment: amounts.free };
+            if (limitAdjustments && limitAdjustments.free !== 0) {
+                data.totalFreeLimit = { increment: limitAdjustments.free };
+            }
+        }
+        if (amounts.paid !== 0) {
+            data.balancePaid = { increment: amounts.paid };
+            if (limitAdjustments && limitAdjustments.paid !== 0) {
+                data.totalPaidLimit = { increment: limitAdjustments.paid };
+            }
+        }
+        if (amounts.oneTimePaid !== 0) {
+            data.balanceOneTimePaid = { increment: amounts.oneTimePaid };
+            if (limitAdjustments && limitAdjustments.oneTimePaid !== 0) {
+                data.totalOneTimePaidLimit = { increment: limitAdjustments.oneTimePaid };
+            }
+        }
+        return data;
+    }
+    buildDecrementData(amounts, limitAdjustments) {
+        const data = {};
+        if (amounts.free !== 0) {
+            data.balanceFree = { decrement: amounts.free };
+            if (limitAdjustments && limitAdjustments.free !== 0) {
+                data.totalFreeLimit = { decrement: limitAdjustments.free };
+            }
+        }
+        if (amounts.paid !== 0) {
+            data.balancePaid = { decrement: amounts.paid };
+            if (limitAdjustments && limitAdjustments.paid !== 0) {
+                data.totalPaidLimit = { decrement: limitAdjustments.paid };
+            }
+        }
+        if (amounts.oneTimePaid !== 0) {
+            data.balanceOneTimePaid = { decrement: amounts.oneTimePaid };
+            if (limitAdjustments && limitAdjustments.oneTimePaid !== 0) {
+                data.totalOneTimePaidLimit = { decrement: limitAdjustments.oneTimePaid };
+            }
+        }
+        return data;
+    }
+    executeCreditOperation(userId, amounts, options, tx) {
+        return tslib_es6.__awaiter(this, void 0, void 0, function* () {
+            var _a;
+            const normalized = this.normalizeAmounts(amounts);
+            this.ensureNonNegative(normalized, options.context);
+            if (!this.hasAnyChange(normalized)) {
+                throw new Error(`${options.context}: no credit change specified`);
+            }
+            let normalizedLimitAdjustments;
+            if (options.limitAdjustments || options.defaultLimitAdjustmentsToAmounts) {
+                const raw = (_a = options.limitAdjustments) !== null && _a !== void 0 ? _a : amounts;
+                normalizedLimitAdjustments = this.normalizeAmounts(raw);
+                this.ensureNonNegative(normalizedLimitAdjustments, `${options.context} limitAdjustments`);
+            }
+            const client = prisma.checkAndFallbackWithNonTCClient(tx);
+            const currentCredit = yield client.credit.findUnique({
+                where: { userId },
+            });
+            if (!currentCredit) {
+                throw new Error('User credits not found');
+            }
+            if (options.ensureSufficientBalance) {
+                this.ensureSufficientBalance(currentCredit, normalized);
+            }
+            if (options.ensureSufficientLimits && normalizedLimitAdjustments) {
+                this.ensureSufficientLimits(currentCredit, normalizedLimitAdjustments);
+            }
+            const data = options.updateMode === 'increment'
+                ? this.buildIncrementData(normalized, normalizedLimitAdjustments)
+                : this.buildDecrementData(normalized, normalizedLimitAdjustments);
+            const credit = yield client.credit.update({
+                where: { userId },
+                data,
+            });
+            const usage = yield this.recordCreditAuditLog(client, userId, options.operationType, normalized, {
+                feature: options.feature,
+                operationReferId: options.operationReferId,
+            });
+            return { credit, usage };
+        });
+    }
+    recordCreditAuditLog(client, userId, operationType, amounts, options) {
+        return tslib_es6.__awaiter(this, void 0, void 0, function* () {
+            const auditPayload = [];
+            if (amounts.free > 0) {
+                auditPayload.push({
+                    userId,
+                    feature: options.feature,
+                    operationReferId: options.operationReferId,
+                    creditType: constants.CreditType.FREE,
+                    operationType,
+                    creditsChange: amounts.free,
+                });
+            }
+            if (amounts.paid > 0) {
+                auditPayload.push({
+                    userId,
+                    feature: options.feature,
+                    operationReferId: options.operationReferId,
+                    creditType: constants.CreditType.PAID,
+                    operationType,
+                    creditsChange: amounts.paid,
+                });
+            }
+            if (amounts.oneTimePaid > 0) {
+                auditPayload.push({
+                    userId,
+                    feature: options.feature,
+                    operationReferId: options.operationReferId,
+                    creditType: constants.CreditType.ONE_TIME_PAID,
+                    operationType,
+                    creditsChange: amounts.oneTimePaid,
+                });
+            }
+            if (auditPayload.length === 0) {
+                return [];
+            }
+            const audits = [];
+            for (const payload of auditPayload) {
+                const auditlog = yield client.creditAuditLog.create({ data: payload });
+                audits.push(auditlog);
+            }
+            return audits;
+        });
+    }
+    // Initialize User Credits, use upsert for easy handle anonymous upgrade to register
+    initializeCreditWithFree(init, tx) {
+        return tslib_es6.__awaiter(this, void 0, void 0, function* () {
+            const now = new Date();
+            const freeStart = now;
+            const freeEnd = new Date(now);
+            freeEnd.setDate(freeEnd.getDate() + creditInit.freeExpiredDays);
+            freeEnd.setHours(23, 59, 59, 999);
+            const normalized = this.normalizeAmounts({ free: init.creditsChange });
+            this.ensureNonNegative(normalized, 'initializeCredit');
+            const client = prisma.checkAndFallbackWithNonTCClient(tx);
+            // 这里使用upsert语义是为了代码复用，处理匿名初始化和匿名->注册的初始化
+            const credit = yield client.credit.upsert({
+                where: {
+                    userId: init.userId
+                },
+                update: {
+                    balanceFree: normalized.free,
+                    totalFreeLimit: normalized.free,
+                    freeStart: freeStart,
+                    freeEnd: freeEnd,
+                },
+                create: {
+                    userId: init.userId,
+                    balanceFree: normalized.free,
+                    totalFreeLimit: normalized.free,
+                    freeStart: freeStart,
+                    freeEnd: freeEnd,
+                },
+            });
+            yield creditAuditLog_service.creditAuditLogService.recordCreditOperation(init, tx);
+            return credit;
+        });
+    }
+    payFailedWatcher(data, tx) {
+        return tslib_es6.__awaiter(this, void 0, void 0, function* () {
+            yield creditAuditLog_service.creditAuditLogService.recordAuditLog(data, tx);
+            console.warn('payFailedWatcher completed');
+        });
+    }
+    // Get User Credits
+    getCredit(userId, tx) {
+        return tslib_es6.__awaiter(this, void 0, void 0, function* () {
+            const client = prisma.checkAndFallbackWithNonTCClient(tx);
+            const credit = yield client.credit.findUnique({
+                where: { userId },
+            });
+            if (!credit) {
+                return null;
+            }
+            // Guard query result: if a credit block has no end time or is already expired, treat its balance as 0
+            const now = new Date();
+            const protectedCredit = Object.assign({}, credit);
+            if (!credit.freeEnd || now >= credit.freeEnd) {
+                protectedCredit.balanceFree = 0;
+            }
+            if (!credit.paidEnd || now >= credit.paidEnd) {
+                protectedCredit.balancePaid = 0;
+            }
+            if (!credit.oneTimePaidEnd || now >= credit.oneTimePaidEnd) {
+                protectedCredit.balanceOneTimePaid = 0;
+            }
+            return protectedCredit;
+        });
+    }
+    // Get Total Credit Balance
+    getTotalBalance(userId, tx) {
+        return tslib_es6.__awaiter(this, void 0, void 0, function* () {
+            const credits = yield this.getCredit(userId, tx);
+            if (!credits)
+                return 0;
+            return credits.balanceFree + credits.balancePaid + credits.balanceOneTimePaid;
+        });
+    }
+    // Recharge Credits (Transactional)
+    rechargeCredit(userId, amounts, options, tx) {
+        return tslib_es6.__awaiter(this, void 0, void 0, function* () {
+            return this.executeCreditOperation(userId, amounts, {
+                context: 'rechargeCredit',
+                operationType: constants.OperationType.RECHARGE,
+                updateMode: 'increment',
+                feature: options.feature,
+                operationReferId: options.operationReferId,
+                limitAdjustments: options.limitAdjustments,
+                defaultLimitAdjustmentsToAmounts: options.limitAdjustments === undefined,
+            }, tx);
+        });
+    }
+    // Consume Credits (Transactional)
+    consumeCredit(userId, amounts, options, tx) {
+        return tslib_es6.__awaiter(this, void 0, void 0, function* () {
+            return this.executeCreditOperation(userId, amounts, {
+                context: 'consumeCredit',
+                operationType: constants.OperationType.CONSUME,
+                updateMode: 'decrement',
+                feature: options.feature,
+                operationReferId: options.operationReferId,
+                ensureSufficientBalance: true,
+            }, tx);
+        });
+    }
+    // Freeze Credits
+    freezeCredit(userId, amounts, reason, operationReferId, tx) {
+        return tslib_es6.__awaiter(this, void 0, void 0, function* () {
+            return this.executeCreditOperation(userId, amounts, {
+                context: 'freezeCredit',
+                operationType: constants.OperationType.FREEZE,
+                operationReferId,
+                updateMode: 'decrement',
+                feature: reason,
+                ensureSufficientBalance: true,
+            }, tx);
+        });
+    }
+    // Unfreeze Credits
+    unfreezeCredit(userId, amounts, reason, operationReferId, tx) {
+        return tslib_es6.__awaiter(this, void 0, void 0, function* () {
+            return this.executeCreditOperation(userId, amounts, {
+                context: 'unfreezeCredit',
+                operationType: constants.OperationType.UNFREEZE,
+                operationReferId,
+                updateMode: 'increment',
+                feature: reason,
+            }, tx);
+        });
+    }
+    // Refund Credits
+    refundCredit(userId_1, amounts_1, operationReferId_1) {
+        return tslib_es6.__awaiter(this, arguments, void 0, function* (userId, amounts, operationReferId, options = {}, tx) {
+            var _a;
+            return this.executeCreditOperation(userId, amounts, {
+                context: 'refundCredit',
+                operationType: constants.OperationType.CONSUME,
+                updateMode: 'decrement',
+                feature: (_a = options.feature) !== null && _a !== void 0 ? _a : 'Refund',
+                operationReferId,
+                limitAdjustments: options.limitAdjustments,
+                defaultLimitAdjustmentsToAmounts: options.limitAdjustments === undefined,
+                ensureSufficientBalance: true,
+                ensureSufficientLimits: true,
+            }, tx);
+        });
+    }
+    // Batch Update Credits (Admin Operation)
+    adjustCredit(userId, operationReferId, adjustments, tx) {
+        return tslib_es6.__awaiter(this, void 0, void 0, function* () {
+            var _a, _b, _c, _d, _e, _f;
+            const client = prisma.checkAndFallbackWithNonTCClient(tx);
+            const currentCredit = yield client.credit.findUnique({
+                where: { userId },
+            });
+            if (!currentCredit) {
+                throw new Error('User credits not found');
+            }
+            const nextBalanceFree = (_a = adjustments.balanceFree) !== null && _a !== void 0 ? _a : currentCredit.balanceFree;
+            const nextBalancePaid = (_b = adjustments.balancePaid) !== null && _b !== void 0 ? _b : currentCredit.balancePaid;
+            const nextBalanceOneTimePaid = (_c = adjustments.balanceOneTimePaid) !== null && _c !== void 0 ? _c : currentCredit.balanceOneTimePaid;
+            const nextTotalFreeLimit = (_d = adjustments.totalFreeLimit) !== null && _d !== void 0 ? _d : currentCredit.totalFreeLimit;
+            const nextTotalPaidLimit = (_e = adjustments.totalPaidLimit) !== null && _e !== void 0 ? _e : currentCredit.totalPaidLimit;
+            const nextTotalOneTimePaidLimit = (_f = adjustments.totalOneTimePaidLimit) !== null && _f !== void 0 ? _f : currentCredit.totalOneTimePaidLimit;
+            if (nextBalanceFree < 0 ||
+                nextBalancePaid < 0 ||
+                nextBalanceOneTimePaid < 0 ||
+                nextTotalFreeLimit < 0 ||
+                nextTotalPaidLimit < 0 ||
+                nextTotalOneTimePaidLimit < 0) {
+                throw new Error('adjustCredit: credit values cannot be negative');
+            }
+            const increaseDiff = this.normalizeAmounts({
+                free: Math.max(nextBalanceFree - currentCredit.balanceFree, 0),
+                paid: Math.max(nextBalancePaid - currentCredit.balancePaid, 0),
+                oneTimePaid: Math.max(nextBalanceOneTimePaid - currentCredit.balanceOneTimePaid, 0),
+            });
+            const decreaseDiff = this.normalizeAmounts({
+                free: Math.max(currentCredit.balanceFree - nextBalanceFree, 0),
+                paid: Math.max(currentCredit.balancePaid - nextBalancePaid, 0),
+                oneTimePaid: Math.max(currentCredit.balanceOneTimePaid - nextBalanceOneTimePaid, 0),
+            });
+            const credit = yield client.credit.update({
+                where: { userId },
+                data: {
+                    balanceFree: nextBalanceFree,
+                    balancePaid: nextBalancePaid,
+                    balanceOneTimePaid: nextBalanceOneTimePaid,
+                    totalFreeLimit: nextTotalFreeLimit,
+                    totalPaidLimit: nextTotalPaidLimit,
+                    totalOneTimePaidLimit: nextTotalOneTimePaidLimit,
+                },
+            });
+            if (this.hasAnyChange(increaseDiff)) {
+                yield this.recordCreditAuditLog(client, userId, constants.OperationType.ADJUST_INCREASE, increaseDiff, {
+                    feature: 'admin_adjust',
+                    operationReferId
+                });
+            }
+            if (this.hasAnyChange(decreaseDiff)) {
+                yield this.recordCreditAuditLog(client, userId, constants.OperationType.ADJUST_DECREASE, decreaseDiff, {
+                    feature: 'admin_adjust',
+                    operationReferId
+                });
+            }
+            return credit;
+        });
+    }
+    purgeCreditsByTypes(userId, reason, operationReferId, types, tx) {
+        return tslib_es6.__awaiter(this, void 0, void 0, function* () {
+            const uniqueTypes = Array.from(new Set(types));
+            if (uniqueTypes.length === 0) {
+                throw new Error('purgeCreditsByTypes: no credit types specified');
+            }
+            const client = prisma.checkAndFallbackWithNonTCClient(tx);
+            const currentCredit = yield client.credit.findUnique({ where: { userId }, });
+            if (!currentCredit) {
+                throw new Error('User credits not found');
+            }
+            const deduction = {};
+            const updateData = {};
+            for (const type of uniqueTypes) {
+                const config = CREDIT_PURGE_CONFIG[type];
+                if (!config) {
+                    throw new Error(`Unsupported credit type: ${type}`);
+                }
+                deduction[config.amountKey] = currentCredit[config.balanceField];
+                updateData[config.balanceField] = 0;
+                updateData[config.limitField] = 0;
+                updateData[config.startField] = null;
+                updateData[config.endField] = null;
+            }
+            const normalizedDeduction = this.normalizeAmounts(deduction);
+            const credit = yield client.credit.update({
+                where: { userId },
+                data: updateData,
+            });
+            // 强制留痕，即使是积分变化为0也记录，操作留痕
+            const usage = yield this.recordCreditAuditLog(client, userId, constants.OperationType.PURGE, normalizedDeduction, { feature: reason, operationReferId });
+            return { credit, usage };
+        });
+    }
+    purgePaidCredit(userId, reason, operationReferId, tx) {
+        return tslib_es6.__awaiter(this, void 0, void 0, function* () {
+            return this.purgeCreditsByTypes(userId, reason, operationReferId, [constants.CreditType.PAID], tx);
+        });
+    }
+    purgeFreeCredit(userId, reason, operationReferId, tx) {
+        return tslib_es6.__awaiter(this, void 0, void 0, function* () {
+            return this.purgeCreditsByTypes(userId, reason, operationReferId, [constants.CreditType.FREE], tx);
+        });
+    }
+    purgeCredit(userId, reason, operationReferId, tx) {
+        return tslib_es6.__awaiter(this, void 0, void 0, function* () {
+            return this.purgeCreditsByTypes(userId, reason, operationReferId, [constants.CreditType.FREE, constants.CreditType.PAID, constants.CreditType.ONE_TIME_PAID], tx);
+        });
+    }
+    // Get Users with Low Credit Balance
+    getLowBalanceUsers() {
+        return tslib_es6.__awaiter(this, arguments, void 0, function* (threshold = 10, tx) {
+            const client$1 = prisma.checkAndFallbackWithNonTCClient(tx);
+            const query = client.Prisma.sql `
+      SELECT * FROM credits 
+      WHERE (balance_free + balance_paid + balance_onetime_paid) < ${threshold}
+      ORDER BY (balance_free + balance_paid + balance_onetime_paid) ASC
+    `;
+            return yield client$1.$queryRaw(query);
+        });
+    }
+    // Get Credit Statistics
+    getCreditStats(tx) {
+        return tslib_es6.__awaiter(this, void 0, void 0, function* () {
+            const client = prisma.checkAndFallbackWithNonTCClient(tx);
+            const stats = yield client.credit.aggregate({
+                _count: true,
+                _sum: {
+                    balanceFree: true,
+                    balancePaid: true,
+                    balanceOneTimePaid: true,
+                },
+                _avg: {
+                    balanceFree: true,
+                    balancePaid: true,
+                    balanceOneTimePaid: true,
+                },
+            });
+            const zeroBalanceUsers = yield client.credit.count({
+                where: {
+                    AND: [
+                        { balanceFree: 0 },
+                        { balancePaid: 0 },
+                        { balanceOneTimePaid: 0 },
+                    ],
+                },
+            });
+            return {
+                totalUsers: stats._count,
+                totalFreeBalance: stats._sum.balanceFree || 0,
+                totalPaidBalance: stats._sum.balancePaid || 0,
+                totalOneTimePaidBalance: stats._sum.balanceOneTimePaid || 0,
+                avgFreeBalance: Math.round(stats._avg.balanceFree || 0),
+                avgPaidBalance: Math.round(stats._avg.balancePaid || 0),
+                avgOneTimePaidBalance: Math.round(stats._avg.balanceOneTimePaid || 0),
+                zeroBalanceUsers,
+            };
+        });
+    }
+    // Check if User has Enough Credits
+    hasEnoughCredits(userId, amount, tx) {
+        return tslib_es6.__awaiter(this, void 0, void 0, function* () {
+            const totalBalance = yield this.getTotalBalance(userId, tx);
+            return totalBalance >= amount;
+        });
+    }
+}
+const creditService = new CreditService();
+
+exports.CreditService = CreditService;
+exports.creditService = creditService;