package com.ruoyi.system.service.impl;

import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ruoyi.common.basic.PageInfo;
import com.ruoyi.common.core.domain.entity.SysRole;
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.system.mapper.TDeliberationVoteMapper;
import com.ruoyi.system.mapper.TDeliberationVoteOptionMapper;
import com.ruoyi.system.model.TDeliberationVote;
import com.ruoyi.system.model.TDeliberationVote;
import com.ruoyi.system.model.TDeliberationVoteOption;
import com.ruoyi.system.query.TDeliberationVoteQuery;
import com.ruoyi.system.service.TDeliberationVoteService;
import com.ruoyi.system.vo.TDeliberationVoteVO;
import io.jsonwebtoken.lang.Collections;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;

import java.util.Collection;
import java.util.List;
import java.util.stream.Collectors;

/**
 * <p>
 * 议事投票 服务实现类
 * </p>
 *
 * @author xiaochen
 * @since 2025-11-06
 */
@Service
public class TDeliberationVoteServiceImpl extends ServiceImpl<TDeliberationVoteMapper, TDeliberationVote> implements TDeliberationVoteService {

    @Autowired
    private TDeliberationVoteOptionMapper deliberationVoteOptionMapper;

    @Override
    public boolean isExit(TDeliberationVote dto) {
        if(StringUtils.isNotEmpty(dto.getId())){
            // 修改
            return this.count(Wrappers.lambdaQuery(TDeliberationVote.class).ne(TDeliberationVote::getId, dto.getId()).eq(TDeliberationVote::getDiscussionTitle, dto.getDiscussionTitle())) > 0;
        }else {
            // 新增
            return this.count(Wrappers.lambdaQuery(TDeliberationVote.class).eq(TDeliberationVote::getDiscussionTitle, dto.getDiscussionTitle())) > 0;
        }
    }

    @Override
    public PageInfo<TDeliberationVoteVO> pageList(TDeliberationVoteQuery query) {
        PageInfo<TDeliberationVoteVO> pageInfo = new PageInfo<>(query.getPageNum(), query.getPageSize());
        List<TDeliberationVoteVO> list = this.baseMapper.pageList(query,pageInfo);
        if(CollectionUtils.isEmpty(list)){
            return pageInfo;
        }
        List<String> voteIds = list.stream().map(TDeliberationVoteVO::getId).collect(Collectors.toList());
        List<TDeliberationVoteOption> deliberationVoteOptions = deliberationVoteOptionMapper.selectList(Wrappers.lambdaQuery(TDeliberationVoteOption.class).in(TDeliberationVoteOption::getVoteId, voteIds));
        for (TDeliberationVoteVO deliberationVoteVO : list) {
            List<TDeliberationVoteOption> deliberationVoteOptionList = deliberationVoteOptions.stream().filter(deliberationVoteOption -> deliberationVoteVO.getId().equals(deliberationVoteOption.getVoteId())).collect(Collectors.toList());
            int count = 0;
            for (TDeliberationVoteOption tDeliberationVoteOption : deliberationVoteOptionList) {
                String voteCount = tDeliberationVoteOption.getVoteCount();
                if(StringUtils.isNotEmpty(voteCount)){
                    String[] split = voteCount.split(",");
                    count += split.length;
                }
            }
            deliberationVoteVO.setVotersTotal(count);
        }
        pageInfo.setRecords(list);
        return pageInfo;
    }
}