feat: 支付完成后自动为用户充值VIP并支持按商品类型计算有效期

Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application/Services/RechargeService.cs

@@ -13,7 +13,7 @@ using Yi.Framework.SqlSugarCore.Abstractions;
 
 namespace Yi.Framework.AiHub.Application.Services
 {
-    public class RechargeService : ApplicationService,IRechargeService
+    public class RechargeService : ApplicationService, IRechargeService
     {
         private readonly ISqlSugarRepository<AiRechargeAggregateRoot> _repository;
         private readonly ICurrentUser _currentUser;
@@ -57,13 +57,33 @@ namespace Yi.Framework.AiHub.Application.Services
         [HttpPost("recharge/vip")]
         public async Task RechargeVipAsync(RechargeCreateInput input)
         {
+            DateTime? expireDateTime = null;
+
+            // 如果传入了月数，计算过期时间
+            if (input.Months.HasValue && input.Months.Value > 0)
+            {
+                // 直接查询该用户最大的过期时间
+                var maxExpireTime = await _repository._DbQueryable
+                    .Where(x => x.UserId == input.UserId && x.ExpireDateTime.HasValue)
+                    .MaxAsync(x => x.ExpireDateTime);
+
+                // 如果最大过期时间大于现在时间，从最大过期时间开始计算
+                // 否则从当天开始计算
+                DateTime baseDateTime = maxExpireTime.HasValue && maxExpireTime.Value > DateTime.Now
+                    ? maxExpireTime.Value
+                    : DateTime.Now;
+                // 计算新的过期时间
+                expireDateTime = baseDateTime.AddMonths(input.Months.Value);
+            }
+            // 如果月数为空或0，表示永久VIP，ExpireDateTime保持为null
+
             // 创建充值记录
             var rechargeRecord = new AiRechargeAggregateRoot
             {
                 UserId = input.UserId,
                 RechargeAmount = input.RechargeAmount,
                 Content = input.Content,
-                ExpireDateTime = input.ExpireDateTime,
+                ExpireDateTime = expireDateTime,
                 Remark = input.Remark,
                 ContactInfo = input.ContactInfo
             };