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

using Mapster;
using Microsoft.AspNetCore.Authorization;
using Microsoft.AspNetCore.Mvc;
using Volo.Abp.Application.Services;
using Volo.Abp.Users;
using Yi.Framework.AiHub.Application.Contracts.Dtos;
using Yi.Framework.AiHub.Domain.Entities;
using Yi.Framework.Rbac.Application.Contracts.IServices;
using Yi.Framework.Rbac.Domain.Shared.Dtos;
using Yi.Framework.SqlSugarCore.Abstractions;
using Yi.Framework.AiHub.Domain.Extensions;

namespace Yi.Framework.AiHub.Application.Services;

public class AiAccountService : ApplicationService
{
    private IAccountService _accountService;
    private ISqlSugarRepository<AiUserExtraInfoEntity> _userRepository;
    private ISqlSugarRepository<AiRechargeAggregateRoot> _rechargeRepository;
    private ISqlSugarRepository<PremiumPackageAggregateRoot> _premiumPackageRepository;

    public AiAccountService(
        IAccountService accountService,
        ISqlSugarRepository<AiUserExtraInfoEntity> userRepository,
        ISqlSugarRepository<AiRechargeAggregateRoot> rechargeRepository,
        ISqlSugarRepository<PremiumPackageAggregateRoot> premiumPackageRepository)
    {
        _accountService = accountService;
        _userRepository = userRepository;
        _rechargeRepository = rechargeRepository;
        _premiumPackageRepository = premiumPackageRepository;
    }

    /// <summary>
    /// 获取ai用户信息
    /// </summary>
    /// <returns></returns>
    [Authorize]
    [HttpGet("account/ai")]
    public async Task<AiUserRoleMenuDto> GetAsync()
    {
        var userId = CurrentUser.GetId();
        var userAccount = await _accountService.GetAsync(null, null, userId: CurrentUser.GetId());
        var output = userAccount.Adapt<AiUserRoleMenuDto>();

        // 是否绑定服务号
        output.IsBindFuwuhao = await _userRepository.IsAnyAsync(x => userId == x.UserId);

        // 是否为VIP用户
        output.IsVip = CurrentUser.IsAiVip();

        // 获取VIP到期时间
        if (output.IsVip)
        {
            var recharges = await _rechargeRepository._DbQueryable
                .Where(x => x.UserId == userId)
                .ToListAsync();

            if (recharges.Any())
            {
                // 如果有任何一个充值记录的过期时间为null，说明是永久VIP
                if (recharges.Any(x => !x.ExpireDateTime.HasValue))
                {
                    output.VipExpireTime = null; // 永久VIP
                }
                else
                {
                    // 取最大的过期时间
                    output.VipExpireTime = recharges
                        .Where(x => x.ExpireDateTime.HasValue)
                        .Max(x => x.ExpireDateTime);
                }
            }
        }

        return output;
    }

    /// <summary>
    /// 获取利润统计数据
    /// </summary>
    /// <param name="currentCost">当前成本(RMB)</param>
    /// <returns></returns>
    [Authorize]
    [HttpGet("account/profit-statistics")]
    public async Task<string> GetProfitStatisticsAsync([FromQuery] decimal currentCost)
    {
        if (CurrentUser.UserName != "Guo" && CurrentUser.UserName != "cc")
        {
            throw new UserFriendlyException("您暂无权限访问");
        }

        // 1. 获取尊享包总消耗和剩余库存
        var premiumPackages = await _premiumPackageRepository._DbQueryable.ToListAsync();
        long totalUsedTokens = premiumPackages.Sum(p => p.UsedTokens);
        long totalRemainingTokens = premiumPackages.Sum(p => p.RemainingTokens);

        // 2. 计算1亿Token成本
        decimal costPerHundredMillion = totalUsedTokens > 0
            ? currentCost / (totalUsedTokens / 100000000m)
            : 0;

        // 3. 计算总成本(剩余+已使用的总成本)
        long totalTokens = totalUsedTokens + totalRemainingTokens;
        decimal totalCost = totalTokens > 0
            ? (totalTokens / 100000000m) * costPerHundredMillion
            : 0;

        // 4. 获取总收益(RechargeType=PremiumPackage的充值金额总和)
        decimal totalRevenue = await _rechargeRepository._DbQueryable
            .Where(x => x.RechargeType == Domain.Shared.Enums.RechargeTypeEnum.PremiumPackage)
            .SumAsync(x => x.RechargeAmount);

        // 5. 计算利润率
        decimal profitRate = totalCost > 0
            ? (totalRevenue / totalCost - 1) * 100
            : 0;

        // 6. 按200售价计算成本
        decimal costAt200Price = totalRevenue > 0
            ? (totalCost / totalRevenue) * 200
            : 0;

        // 7. 格式化输出
        var today = DateTime.Now;
        string dayOfWeek = today.ToString("dddd", new System.Globalization.CultureInfo("zh-CN"));
        string weekDay = dayOfWeek switch
        {
            "星期一" => "周1",
            "星期二" => "周2",
            "星期三" => "周3",
            "星期四" => "周4",
            "星期五" => "周5",
            "星期六" => "周6",
            "星期日" => "周日",
            _ => dayOfWeek
        };

        var result = $@"{today:M月d日} {weekDay}
尊享包已消耗({totalUsedTokens / 100000000m:F2}亿){totalUsedTokens}
尊享包剩余库存({totalRemainingTokens / 100000000m:F2}亿){totalRemainingTokens}
当前成本:{currentCost:F2}RMB
1亿Token成本:{costPerHundredMillion:F2} RMB=1亿 Token
总成本:{totalCost:F2} RMB
总收益:{totalRevenue:F2}RMB
利润率: {profitRate:F1}%
按200售价来算,成本在{costAt200Price:F2}";

        return result;
    }
}