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.system.mapper.TLargeResourceMapper;
import com.ruoyi.system.mapper.TLargeResourceTypeMapper;
import com.ruoyi.system.model.TAppUser;
import com.ruoyi.system.model.TLargeResource;
import com.ruoyi.system.model.TLargeResourceType;
import com.ruoyi.system.query.TLargeResourceQuery;
import com.ruoyi.system.service.TLargeResourceService;
import com.ruoyi.system.vo.TLargeResourceVO;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;

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

/**
 * <p>
 * 大屏-社区资源 服务实现类
 * </p>
 *
 * @author xiaochen
 * @since 2025-11-11
 */
@Service
public class TLargeResourceServiceImpl extends ServiceImpl<TLargeResourceMapper, TLargeResource> implements TLargeResourceService {

    @Autowired
    private TLargeResourceTypeMapper largeResourceTypeMapper;

    @Override
    public PageInfo<TLargeResourceVO> pageList(TLargeResourceQuery query) {
        PageInfo<TLargeResourceVO> pageInfo = new PageInfo<>(query.getPageNum(), query.getPageSize());
        List<TLargeResourceVO> list = this.baseMapper.pageList(query,pageInfo);
        if(CollectionUtils.isEmpty(list)){
            return pageInfo;
        }

        List<String> ids = list.stream().map(TLargeResourceVO::getTypeParentId).collect(Collectors.toList());
        List<String> childrenIds = list.stream().map(TLargeResourceVO::getTypeChildrenId).collect(Collectors.toList());
        ids.addAll(childrenIds);

        List<TLargeResourceType> largeResourceTypes = largeResourceTypeMapper.selectList(Wrappers.lambdaQuery(TLargeResourceType.class).in(TLargeResourceType::getId, ids));

        for (TLargeResourceVO tLargeResourceVO : list) {
            List<TLargeResourceType> resourceTypeParentList = largeResourceTypes.stream().filter(largeResourceType -> largeResourceType.getId().equals(tLargeResourceVO.getTypeParentId())).collect(Collectors.toList());
            if(!CollectionUtils.isEmpty(resourceTypeParentList)){
                tLargeResourceVO.setTypeParentName(resourceTypeParentList.get(0).getTypeName());
            }
            List<TLargeResourceType> resourceTypeChildrenList = largeResourceTypes.stream().filter(largeResourceType -> largeResourceType.getId().equals(tLargeResourceVO.getTypeChildrenId())).collect(Collectors.toList());
            if(!CollectionUtils.isEmpty(resourceTypeChildrenList)){
                tLargeResourceVO.setTypeChildrenName(resourceTypeChildrenList.get(0).getTypeName());
            }
        }
        pageInfo.setRecords( list);
        return pageInfo;
    }
}