feat: 新增VIP充值接口并支持通过角色代码为用户分配角色

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

@@ -1,41 +1,73 @@
 using Mapster;
 using Microsoft.AspNetCore.Authorization;
 using Microsoft.AspNetCore.Mvc;
-using Microsoft.Extensions.DependencyInjection;
 using Volo.Abp.Application.Services;
+using Volo.Abp.Users;
 using Yi.Framework.AiHub.Application.Contracts.Dtos.Recharge;
 using Yi.Framework.AiHub.Domain.Entities;
+using Yi.Framework.AiHub.Domain.Shared.Consts;
 using Yi.Framework.Rbac.Application.Contracts.IServices;
-using Yi.Framework.Rbac.Domain.Shared.Dtos;
 using Yi.Framework.SqlSugarCore.Abstractions;
 
-namespace Yi.Framework.AiHub.Application.Services;
-
-/// <summary>
-/// ai 充值表
-/// </summary>
-public class RechargeService : ApplicationService
+namespace Yi.Framework.AiHub.Application.Services
 {
-    private readonly ISqlSugarRepository<AiRechargeAggregateRoot> _repository;
-
-    public RechargeService(ISqlSugarRepository<AiRechargeAggregateRoot> repository)
+    public class RechargeService : ApplicationService
     {
-        _repository = repository;
-    }
+        private readonly ISqlSugarRepository<AiRechargeAggregateRoot> _repository;
+        private readonly ICurrentUser _currentUser;
+        private readonly IUserService _userService;
 
-    /// <summary>
-    /// 查询已登录的账户充值记录
-    /// </summary>
-    /// <returns></returns>
-    [Route("recharge/account")]
-    [Authorize]
-    public async Task<List<RechargeGetListOutput>> GetListByAccountAsync()
-    {
-        var userId = CurrentUser.Id;
-        var entities = await _repository._DbQueryable.Where(x => x.UserId == userId)
-            .OrderByDescending(x => x.CreationTime)
-            .ToListAsync();
-        var output = entities.Adapt<List<RechargeGetListOutput>>();
-        return output;
+        public RechargeService(
+            ISqlSugarRepository<AiRechargeAggregateRoot> repository,
+            ICurrentUser currentUser,
+            IUserService userService)
+        {
+            _repository = repository;
+            _currentUser = currentUser;
+            _userService = userService;
+        }
+
+        /// <summary>
+        /// 查询已登录的账户充值记录
+        /// </summary>
+        /// <returns></returns>
+        [Route("recharge/account")]
+        [Authorize]
+        public async Task<List<RechargeGetListOutput>> GetListByAccountAsync()
+        {
+            var userId = CurrentUser.Id;
+            var entities = await _repository._DbQueryable.Where(x => x.UserId == userId)
+                .OrderByDescending(x => x.CreationTime)
+                .ToListAsync();
+            var output = entities.Adapt<List<RechargeGetListOutput>>();
+            return output;
+        }
+
+        /// <summary>
+        /// 给用户充值VIP
+        /// </summary>
+        /// <param name="input">充值输入参数</param>
+        /// <returns></returns>
+        [HttpPost("recharge/vip")]
+        public async Task RechargeVipAsync(RechargeCreateInput input)
+        {
+            // 创建充值记录
+            var rechargeRecord = new AiRechargeAggregateRoot
+            {
+                UserId = input.UserId,
+                RechargeAmount = input.RechargeAmount,
+                Content = input.Content,
+                ExpireDateTime = input.ExpireDateTime,
+                Remark = input.Remark,
+                ContactInfo = input.ContactInfo
+            };
+
+            // 保存充值记录到数据库
+            await _repository.InsertAsync(rechargeRecord);
+
+            // 使用UserService给用户添加VIP角色
+            await _userService.AddUserRoleByRoleCodeAsync(input.UserId,
+                new List<string>() { AiHubConst.VipRole, "default" });
+        }
     }
 }