package com.doumee.service.business.impl;

import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.doumee.core.model.PageData;
import com.doumee.core.model.PageWrap;
import com.doumee.core.utils.Constants;
import com.doumee.core.utils.Utils;
import com.doumee.dao.business.model.Company;
import com.doumee.dao.business.model.dto.CompanyUserDTO;
import com.doumee.dao.business.model.dto.CompanyUserQueryDTO;
import com.doumee.dao.system.SystemUserJoinMapper;
import com.doumee.dao.system.model.SystemRole;
import com.doumee.dao.system.model.SystemUser;
import com.doumee.dao.system.model.SystemUserRole;
import com.doumee.dao.system.vo.SystemUserListVO;
import com.doumee.service.business.CompanyUserService;
import com.doumee.service.system.SystemRoleService;
import com.github.yulichang.base.mapper.MPJJoinMapper;
import com.github.yulichang.wrapper.MPJLambdaWrapper;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.List;
import java.util.Objects;

/**
 * @author T14
 */
@Service
public class CompanyUserServiceImpl implements CompanyUserService {

    @Autowired
    private SystemUserJoinMapper systemUserJoinMapper;

    @Autowired
    private SystemRoleService systemRoleService;


    @Override
    public PageData<CompanyUserDTO> findPage(PageWrap<CompanyUserQueryDTO> pageWrap) {
        IPage<Company> page = new Page<>(pageWrap.getPage(), pageWrap.getCapacity());

        MPJLambdaWrapper<SystemUser> mpjJoinMapper = new MPJLambdaWrapper<>();
        Utils.MP.blankToNull(pageWrap.getModel());
        mpjJoinMapper.selectAll(SystemUser.class)
                    .selectAs(Company::getName,CompanyUserDTO::getCompanyName)
//                    .selectAs(SystemRole::getId,CompanyUserDTO::getRoleId)
//                    .selectAs(SystemRole::getCode,CompanyUserDTO::getRoleCode)
//                    .selectAs(SystemRole::getName,CompanyUserDTO::getRoleName)
//                    .selectAs(SystemRole::getType,CompanyUserDTO::getRoleType)
//                    .leftJoin(SystemUserRole.class,SystemUserRole::getUserId,SystemUser::getId)
//                    .leftJoin(SystemRole.class,SystemRole::getId,SystemUserRole::getRoleId)
                    .leftJoin(Company.class,Company::getId,SystemUser::getCompanyId)
                    .eq(SystemUser::getType, pageWrap.getModel().getType())
                    .eq(Objects.nonNull(pageWrap.getModel().getStatus()),SystemUser::getStatus,pageWrap.getModel().getStatus())
                    .like(StringUtils.isNotBlank(pageWrap.getModel().getCompanyName()),Company::getName,pageWrap.getModel().getCompanyName())
                    .and(StringUtils.isNotBlank(pageWrap.getModel().getKeyWords()),wapper ->{wapper.like(SystemUser::getUsername,pageWrap.getModel().getKeyWords())
                            .or().like(SystemUser::getRealname,pageWrap.getModel().getKeyWords());})
                    .orderByDesc(SystemUser::getCreateTime);

        IPage<CompanyUserDTO> companyUserDTOIPage = systemUserJoinMapper.selectJoinPage(page, CompanyUserDTO.class, mpjJoinMapper);
        if(Constants.equalsInteger(pageWrap.getModel().getType(),Constants.ONE) && companyUserDTOIPage.getRecords()!=null){
            for (CompanyUserDTO user : companyUserDTOIPage.getRecords()) {
                // 查询用户角色列表
                user.setRoles(systemRoleService.findByUserId(user.getId()));
            }
        }

        return PageData.from(companyUserDTOIPage);
    }

    @Override
    public CompanyUserDTO findOne(CompanyUserDTO companyUserDTO) {
        MPJLambdaWrapper<SystemUser> mpjJoinMapper = new MPJLambdaWrapper<>();
        mpjJoinMapper.selectAll(SystemUser.class)
                .selectAs(Company::getName,CompanyUserDTO::getCompanyName)
//                .selectAs(SystemRole::getId,CompanyUserDTO::getRoleId)
//                .selectAs(SystemRole::getCode,CompanyUserDTO::getRoleCode)
//                .selectAs(SystemRole::getName,CompanyUserDTO::getRoleName)
//                .selectAs(SystemRole::getType,CompanyUserDTO::getRoleType)
//                .leftJoin(SystemUserRole.class,SystemUserRole::getUserId,SystemUser::getId)
//                .leftJoin(SystemRole.class,SystemRole::getId,SystemUserRole::getRoleId)
                .leftJoin(Company.class,Company::getId,SystemUser::getCompanyId)
                .eq(SystemUser::getType, Constants.ONE)
                .eq(SystemUser::getId,companyUserDTO.getId());

        CompanyUserDTO companyUserDTOIPage = systemUserJoinMapper.selectJoinOne(CompanyUserDTO.class, mpjJoinMapper);
        if(companyUserDTOIPage!=null && Constants.equalsInteger(companyUserDTOIPage.getType(),Constants.ONE) ){
                // 查询用户角色列表
            companyUserDTOIPage.setRoles(systemRoleService.findByUserId(companyUserDTOIPage.getId()));
        }
        return companyUserDTOIPage;
    }

    @Override
    public List<CompanyUserDTO> findListByCompanyID(CompanyUserDTO companyUserDTO) {
        MPJLambdaWrapper<SystemUser> mpjJoinMapper = new MPJLambdaWrapper<>();
        mpjJoinMapper.selectAll(SystemUser.class)
                .selectAs(Company::getName,CompanyUserDTO::getCompanyName)
                .selectAs(SystemRole::getId,CompanyUserDTO::getRoleId)
                .selectAs(SystemRole::getCode,CompanyUserDTO::getRoleCode)
                .selectAs(SystemRole::getName,CompanyUserDTO::getRoleName)
                .selectAs(SystemRole::getType,CompanyUserDTO::getRoleType)
                .leftJoin(SystemUserRole.class,SystemUserRole::getUserId,SystemUser::getId)
                .leftJoin(SystemRole.class,SystemRole::getId,SystemUserRole::getRoleId)
                .leftJoin(Company.class,Company::getId,SystemUser::getCompanyId)
                .eq(SystemUser::getType, Constants.ONE)
                .eq(SystemUser::getId,companyUserDTO.getCompanyId());

        List<CompanyUserDTO> companyUserDTOS = systemUserJoinMapper.selectJoinList(CompanyUserDTO.class, mpjJoinMapper);
        return companyUserDTOS;
    }
}