package com.ruoyi.system.service.company.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.utils.DateUtils;
import com.ruoyi.system.domain.company.AreaInfo;
import com.ruoyi.system.domain.company.CompanyInfo;
import com.ruoyi.system.domain.dto.CompanyInfoDto;
import com.ruoyi.system.domain.vo.CompanyInfoVO;
import com.ruoyi.system.mapper.company.CompanyInfoMapper;
import com.ruoyi.system.service.company.IAreaInfoService;
import com.ruoyi.system.service.company.ICompanyInfoService;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;


@Service
public class CompanyInfoServiceImpl extends ServiceImpl<CompanyInfoMapper, CompanyInfo> implements ICompanyInfoService {

    @Autowired
    private IAreaInfoService areaInfoService;

    /**
     * 分页查询
     *
     * @param companyInfoVO
     * @return
     */
    @Override
    public AjaxResult findCompanyInfoPage(CompanyInfoVO companyInfoVO) {
        if (companyInfoVO.getPageSize() == null || companyInfoVO.getPageNum() == null) {
            return AjaxResult.error("请求参数【pageSize，pageNum】不能为空！");
        }
        PageHelper.startPage(companyInfoVO.getPageNum(), companyInfoVO.getPageSize());
        QueryWrapper queryWrapper = initQueryWrapper(companyInfoVO);
        queryWrapper.orderByDesc(CompanyInfo.CREATE_TIME);
        List<CompanyInfo> companyInfoList = this.list(queryWrapper);
        PageInfo companyPageInfo = new PageInfo<>(companyInfoList);
        List<CompanyInfoDto> companyInfoDtoList = initCompanyCity(companyPageInfo.getList());
        companyPageInfo.setList(companyInfoDtoList);
        return AjaxResult.success(companyPageInfo);
    }

    @Override
    public List<CompanyInfoDto> findExportData(CompanyInfoVO companyInfoVO) {
        return initCompanyCity(this.list(initQueryWrapper(companyInfoVO)));
    }


    public QueryWrapper initQueryWrapper(CompanyInfoVO companyInfoVO) {
        QueryWrapper queryWrapper = new QueryWrapper();
        if (StringUtils.isNoneEmpty(companyInfoVO.getCompanyName())) {
            queryWrapper.like(CompanyInfo.COMPANY_NAME, companyInfoVO.getCompanyName());
        }
        if (StringUtils.isNoneEmpty(companyInfoVO.getCompanyDirector())) {
            queryWrapper.like(CompanyInfo.COMPANY_DIRECTOR, companyInfoVO.getCompanyDirector());
        }
        if (StringUtils.isNoneEmpty(companyInfoVO.getCertificateNum())) {
            queryWrapper.like(CompanyInfo.CERTIFICATE_NUM, companyInfoVO.getCertificateNum());
        }
        if (companyInfoVO.getIsValid() != null) {
            if (companyInfoVO.getIsValid() == 0) {
                queryWrapper.ge(CompanyInfo.VALIDITY_DATE_END, DateUtils.parseDateToStr(DateUtils.YYYY_MM_DD, new Date()));
            } else {
                queryWrapper.lt(CompanyInfo.VALIDITY_DATE_END, DateUtils.parseDateToStr(DateUtils.YYYY_MM_DD, new Date()));
            }
        }
        if (companyInfoVO.getCompanyCity() != null) {
            queryWrapper.eq(CompanyInfo.COMPANY_CITY, companyInfoVO.getCompanyCity());
        }
        queryWrapper.eq(CompanyInfo.IS_DELETE, 0);
        return queryWrapper;
    }

    @Override
    public Boolean checkIsExist(CompanyInfo companyInfo) {
        QueryWrapper queryWrapper = new QueryWrapper();

        //如果证书编号重复，则提示企业重复
        queryWrapper.eq(CompanyInfo.CERTIFICATE_NUM, companyInfo.getCertificateNum());
        queryWrapper.eq(CompanyInfo.IS_DELETE, 0);

        return null;
    }

    @Override
    public List<CompanyInfoDto> initCompanyCity(List<CompanyInfo> companyInfoDtoList) {
        List<AreaInfo> areaInfos = areaInfoService.list();
        Map<String, List<AreaInfo>> areaInfoMap = areaInfos.stream().collect(Collectors.groupingBy(AreaInfo::getId));
        List<CompanyInfoDto> companyInfoDtoList1 = new ArrayList<>();
        companyInfoDtoList.forEach(companyInfoDto -> {
            CompanyInfoDto newCompanyInfoDto = new CompanyInfoDto();
            BeanUtils.copyProperties(companyInfoDto, newCompanyInfoDto);
            newCompanyInfoDto.setCompanyCityValue(companyInfoDto.getCompanyCity());
            String companyCity = "";
            if (StringUtils.isNotEmpty(newCompanyInfoDto.getCompanyCityValue()) && CollectionUtils.isNotEmpty(areaInfoMap.get(newCompanyInfoDto.getCompanyCityValue()))) {
                companyCity = areaInfoMap.get(newCompanyInfoDto.getCompanyCityValue()).get(0).getCity() + areaInfoMap.get(newCompanyInfoDto.getCompanyCityValue()).get(0).getArea();
                newCompanyInfoDto.setCity( areaInfoMap.get(newCompanyInfoDto.getCompanyCityValue()).get(0).getCity());
                newCompanyInfoDto.setArea( areaInfoMap.get(newCompanyInfoDto.getCompanyCityValue()).get(0).getArea());
            }
            newCompanyInfoDto.setCompanyCity(companyCity);
            companyInfoDtoList1.add(newCompanyInfoDto);
        });
        return companyInfoDtoList1;
    }
}