package cc.mrbird.febs.mall.service.impl;

import cc.mrbird.febs.common.entity.FebsResponse;
import cc.mrbird.febs.common.utils.LoginUserUtil;
import cc.mrbird.febs.mall.dto.sale.ApiSaleListInfoDto;
import cc.mrbird.febs.mall.dto.sale.ApiSaleRecordInfoDto;
import cc.mrbird.febs.mall.dto.sale.ApiTeamListInfoDto;
import cc.mrbird.febs.mall.entity.*;
import cc.mrbird.febs.mall.mapper.*;
import cc.mrbird.febs.mall.service.ApiSaleService;
import cc.mrbird.febs.mall.vo.memberLevel.ApiMemberLevelListInfoVo;
import cc.mrbird.febs.mall.vo.sale.*;
import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.date.DateTime;
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.util.ObjectUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;

import java.math.BigDecimal;
import java.util.*;
import java.util.stream.Collectors;

@Slf4j
@Service
@RequiredArgsConstructor
public class ApiSaleServiceImpl extends ServiceImpl<MallMemberMapper, MallMember> implements ApiSaleService {

    private final HappySaleLevelMapper happySaleLevelMapper;
    private final HappyMemberLevelMapper happyMemberLevelMapper;
    private final MallAchieveRecordMapper mallAchieveRecordMapper;
    private final MallMemberWalletMapper mallMemberWalletMapper;
    @Override
    public FebsResponse saleInfo() {
        ApiSaleInfoVo apiSaleInfoVo = new ApiSaleInfoVo();

        Long id = LoginUserUtil.getLoginUser().getId();
        MallMember mallMember = this.baseMapper.selectById(id);
        apiSaleInfoVo.setId(mallMember.getId());
        apiSaleInfoVo.setName(mallMember.getName());
        apiSaleInfoVo.setAvatar(mallMember.getAvatar());
        apiSaleInfoVo.setInviteId(mallMember.getInviteId());
        apiSaleInfoVo.setStoreMaster(mallMember.getStoreMaster());

        HappySaleLevel happySaleLevel = happySaleLevelMapper.selectOne(
                new LambdaQueryWrapper<HappySaleLevel>()
                        .eq(HappySaleLevel::getCode, mallMember.getStoreMaster())
                        .last("limit 1")
        );
        if (happySaleLevel != null) {
            apiSaleInfoVo.setStoreMasterName(happySaleLevel.getName());
            apiSaleInfoVo.setStoreMasterPng(happySaleLevel.getIconPng());
        }

        List<MallAchieveRecord> mallAchieveRecords = mallAchieveRecordMapper.selectList(
                new LambdaQueryWrapper<MallAchieveRecord>()
                        .eq(MallAchieveRecord::getMemberId, id)
        );
        if (CollUtil.isNotEmpty(mallAchieveRecords)) {
            apiSaleInfoVo.setTotalSaleAmount(
                    CollUtil.isNotEmpty(mallAchieveRecords)
                            ? mallAchieveRecords.stream().map(MallAchieveRecord::getCostAmount).reduce(BigDecimal.ZERO, BigDecimal::add)
                            : BigDecimal.ZERO);
            apiSaleInfoVo.setOrderCnt(CollUtil.isNotEmpty(mallAchieveRecords) ? mallAchieveRecords.size() : 0);
        }

        List<MallMember> mallMembers = this.baseMapper.selectAllChildAgentListByInviteId(mallMember.getInviteId());
        apiSaleInfoVo.setTotalCnt(CollUtil.isNotEmpty(mallMembers) ? mallMembers.size() : 0);

        // todo 提现部分暂无

        return new FebsResponse().success().data(apiSaleInfoVo);
    }

    @Override
    public FebsResponse teamHeader() {
        ApiTeamHeaderInfoVo apiTeamHeaderInfoVo = new ApiTeamHeaderInfoVo();

        Long id = LoginUserUtil.getLoginUser().getId();
        MallMember mallMember = this.baseMapper.selectById(id);

        List<MallMember> mallMembers = this.baseMapper.selectAllChildAgentListByInviteId(mallMember.getInviteId());
        if(CollUtil.isEmpty(mallMembers)){
            apiTeamHeaderInfoVo.setTotalCnt(0);
            apiTeamHeaderInfoVo.setTotalAmount(BigDecimal.ZERO);
        }else {
            apiTeamHeaderInfoVo.setTotalCnt(mallMembers.size());
            List<MallAchieveRecord> mallAchieveRecords = mallAchieveRecordMapper.selectList(
                    new LambdaQueryWrapper<MallAchieveRecord>()
                            .eq(MallAchieveRecord::getMemberId, id)
            );
            apiTeamHeaderInfoVo.setTotalAmount(
                    CollUtil.isNotEmpty(mallAchieveRecords)
                            ? mallAchieveRecords.stream().map(MallAchieveRecord::getAmount).reduce(BigDecimal.ZERO, BigDecimal::add)
                            : BigDecimal.ZERO);
        }
        return new FebsResponse().success().data(apiTeamHeaderInfoVo);
    }

    @Override
    public FebsResponse teamList(ApiTeamListInfoDto dto) {
        Long id = LoginUserUtil.getLoginUser().getId();
        MallMember mallMember = this.baseMapper.selectById(id);
        dto.setInviteId(mallMember.getInviteId());

        Page<ApiTeamListInfoVo> page = new Page<>(dto.getPageNow(), dto.getPageSize());
        IPage<ApiTeamListInfoVo> apiTeamListInfoVoIPage = this.baseMapper.selectTeamPage(page, dto);
        return new FebsResponse().success().data(apiTeamListInfoVoIPage);
    }

    @Override
    public FebsResponse saleHeader() {

        ApiSaleHeaderInfoVo apiSaleHeaderInfoVo = new ApiSaleHeaderInfoVo();

        Long id = LoginUserUtil.getLoginUser().getId();
        List<MallAchieveRecord> mallAchieveRecords = mallAchieveRecordMapper.selectList(
                new LambdaQueryWrapper<MallAchieveRecord>()
                        .eq(MallAchieveRecord::getMemberId, id)
        );
        apiSaleHeaderInfoVo.setTotalSaleAmount(
                CollUtil.isNotEmpty(mallAchieveRecords)
                        ? mallAchieveRecords.stream().map(MallAchieveRecord::getCostAmount).reduce(BigDecimal.ZERO, BigDecimal::add)
                        : BigDecimal.ZERO);
        apiSaleHeaderInfoVo.setOrderCnt(CollUtil.isNotEmpty(mallAchieveRecords) ? mallAchieveRecords.size() : 0);
        return new FebsResponse().success().data(apiSaleHeaderInfoVo);
    }

    @Override
    public FebsResponse saleList(ApiSaleListInfoDto dto) {

        Long id = LoginUserUtil.getLoginUser().getId();
        Page<ApiSaleListInfoVo> objectPage = new Page<>(dto.getPageNow(), dto.getPageSize());
        dto.setMemberId(id);
        Page<ApiSaleListInfoVo> mallAchieveRecordPage = this.baseMapper.selectSalePage(objectPage, dto);
        return new FebsResponse().success().data(mallAchieveRecordPage);
    }

    @Override
    public FebsResponse levelInfoList() {
        ArrayList<ApiLevelListInfoVo> apiLevelListInfoVos = new ArrayList<>();

        // 获取当前登录用户的ID
        Long memberId = LoginUserUtil.getLoginUser().getId();
        MallMember mallMember = this.baseMapper.selectById(memberId);

        List<HappySaleLevel> happySaleLevels = happySaleLevelMapper.selectList(
                new LambdaQueryWrapper<HappySaleLevel>()
                        .orderByAsc(HappySaleLevel::getCode)
        );
        if (CollUtil.isNotEmpty(happySaleLevels)) {
            happySaleLevels.forEach(happySaleLevel -> {
                ApiLevelListInfoVo apiLevelListInfoVo = new ApiLevelListInfoVo();
                apiLevelListInfoVo.setName(happySaleLevel.getName());
                apiLevelListInfoVo.setMemberCode(mallMember.getStoreMaster());
                apiLevelListInfoVo.setCode(happySaleLevel.getCode());
                apiLevelListInfoVo.setContent(happySaleLevel.getContent());
                apiLevelListInfoVo.setHeaderPng(happySaleLevel.getIconPng());
                apiLevelListInfoVos.add(apiLevelListInfoVo);
            });
        }
        return new FebsResponse().success().data(apiLevelListInfoVos);
    }

    /**
     * 处理销售记录信息
     *
     * 根据提供的类型参数，查询并统计指定时间范围内的销售记录，并返回相关会员的销售总额信息
     * 类型参数为1时，查询本月的销售记录；类型参数为2时，查询本周的销售记录
     *
     * @param dto 包含类型信息的ApiSaleRecordInfoDto对象，用于指定查询类型
     * @return 返回包含销售记录信息的FebsResponse对象，包括会员名称、头像和消费金额
     */
    @Override
    public FebsResponse saleRecord(ApiSaleRecordInfoDto dto) {
        // 初始化结果列表
        ApiSaleRecordInfoVo apiSaleRecordInfoVo = new ApiSaleRecordInfoVo();
        List<ApiSaleRecordInfoVo> apiSaleRecordInfoVos = new ArrayList<>();

        // 获取当前登录用户的ID
        Long memberId = LoginUserUtil.getLoginUser().getId();
        // 根据ID查询会员信息
        MallMember mallMember = this.baseMapper.selectById(memberId);
        // 设置会员名称、头像、排序号和消费金额初始值
        apiSaleRecordInfoVo.setName(mallMember.getName());
        apiSaleRecordInfoVo.setAvatar(mallMember.getAvatar());
        apiSaleRecordInfoVo.setSortCnt(0);
        apiSaleRecordInfoVo.setCostAmount(BigDecimal.ZERO);

        // 获取类型参数
        Integer type = dto.getType();
        // 校验类型参数，确保其为1（月）或2（周）
        if (type == null || (type != 1 && type != 2)) {
            return new FebsResponse().fail().message("非法的类型参数");
        }

        // 根据类型参数确定开始时间
        DateTime startTime = type == 1 ? DateUtil.beginOfMonth(new Date()) : DateUtil.beginOfWeek(new Date());

        // 查询指定时间范围内的商城销售记录
        List<MallAchieveRecord> mallAchieveRecords = mallAchieveRecordMapper.selectList(
                new LambdaQueryWrapper<MallAchieveRecord>()
                        .select(MallAchieveRecord::getMemberId, MallAchieveRecord::getCostAmount)
                        .ge(MallAchieveRecord::getCreatedTime, startTime)
        );

        // 如果查询到销售记录，则进行处理
        if (CollUtil.isNotEmpty(mallAchieveRecords)) {
            // 提取 memberId 集合，避免重复 stream
            Set<Long> memberIds = mallAchieveRecords.stream()
                    .map(MallAchieveRecord::getMemberId)
                    .collect(Collectors.toSet());

            // 分组求和
            Map<Long, BigDecimal> memberIdToCostAmount = mallAchieveRecords.stream()
                    .collect(Collectors.groupingBy(
                            MallAchieveRecord::getMemberId,
                            Collectors.mapping(
                                    MallAchieveRecord::getCostAmount,
                                    Collectors.reducing(BigDecimal.ZERO, BigDecimal::add)
                            )
                    ));

            // 查询会员信息
            List<MallMember> mallMembers = this.baseMapper.selectList(
                    new LambdaQueryWrapper<MallMember>()
                            .select(MallMember::getId, MallMember::getName, MallMember::getAvatar)
                            .in(MallMember::getId, memberIds)
            );

            // 构建会员信息映射表
            Map<Long, MallMember> mallMemberMap = mallMembers.stream()
                    .collect(Collectors.toMap(MallMember::getId, m -> m));

            // 构建结果列表并避免空指针
            for (Map.Entry<Long, BigDecimal> entry : memberIdToCostAmount.entrySet()) {
                MallMember member = mallMemberMap.get(entry.getKey());
                if (member == null) {
                    continue; // 忽略不存在的会员
                }
                ApiSaleRecordInfoVo vo = new ApiSaleRecordInfoVo();
                vo.setId(member.getId());
                vo.setName(member.getName());
                vo.setAvatar(member.getAvatar());
                vo.setCostAmount(entry.getValue());
                apiSaleRecordInfoVos.add(vo);
            }

            // 排序
            apiSaleRecordInfoVos.sort(Comparator.comparing(ApiSaleRecordInfoVo::getCostAmount).reversed());

            //stream流操作apiSaleRecordInfoVos获取id等于memberId的数据，并且返回第一个数据和在数组中的下标索引
            ApiSaleRecordInfoVo memberSale = apiSaleRecordInfoVos.stream()
                    .filter(vo -> vo.getId().equals(memberId))
                    .findFirst()
                    .orElse(null);
            if (memberSale != null) {
                apiSaleRecordInfoVo.setSortCnt(apiSaleRecordInfoVos.indexOf(memberSale)+1);
                apiSaleRecordInfoVo.setCostAmount(memberSale.getCostAmount());
            }
        }
        apiSaleRecordInfoVo.setSaleRecord(apiSaleRecordInfoVos);

        // 返回成功响应，包含销售记录信息
        return new FebsResponse().success().data(apiSaleRecordInfoVo);
    }

    @Override
    public FebsResponse memberLevelList() {
        ArrayList<ApiMemberLevelListInfoVo> apiLevelListInfoVos = new ArrayList<>();

        // 获取当前登录用户的ID
        Long memberId = LoginUserUtil.getLoginUser().getId();
        MallMember mallMember = this.baseMapper.selectById(memberId);
        MallMemberWallet mallMemberWallet = mallMemberWalletMapper.selectWalletByMemberId(memberId);

        List<HappyMemberLevel> happyMemberLevels = happyMemberLevelMapper.selectList(
                new LambdaQueryWrapper<HappyMemberLevel>()
                        .orderByAsc(HappyMemberLevel::getCode)
        );
        //Stream  操作 happyMemberLevels，返回一个Map<code,HappyMemberLevel>
        Map<Integer, HappyMemberLevel> happyMemberLevelMap = happyMemberLevels.stream()
                .collect(Collectors.toMap(HappyMemberLevel::getCode, happyMemberLevel -> happyMemberLevel));
        if (CollUtil.isNotEmpty(happyMemberLevels)) {
            happyMemberLevels.forEach(happyMemberLevel -> {
                ApiMemberLevelListInfoVo vo = new ApiMemberLevelListInfoVo();
                BeanUtil.copyProperties(happyMemberLevel, vo);
                vo.setMemberCode(mallMember.getDirector());
                vo.setMemberScore(mallMemberWallet.getScore());
                vo.setUpgradeScore(
                        ObjectUtil.isEmpty(happyMemberLevelMap.get(happyMemberLevel.getCode() + 1)) ?
                                 0 : happyMemberLevelMap.get(happyMemberLevel.getCode() + 1).getUpgradeScore()
            );
                apiLevelListInfoVos.add(vo);
            });
        }
        return new FebsResponse().success().data(apiLevelListInfoVos);
    }


}