package doumeemes.service.system.impl;

import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import doumeemes.core.model.PageData;
import doumeemes.core.model.PageWrap;
import doumeemes.core.utils.DateUtil;
import doumeemes.dao.ext.dto.ActivityLoginDTO;
import doumeemes.dao.ext.dto.CompanyLoginDTO;
import doumeemes.dao.ext.vo.ActivityLoginVO;
import doumeemes.dao.ext.vo.CompanyLoginStatisticsVO;
import doumeemes.dao.system.SystemUserMapper;
import doumeemes.dao.system.dto.QuerySystemUserDTO;
import doumeemes.dao.system.model.SystemUser;
import doumeemes.dao.system.vo.SystemUserListVO;
import doumeemes.service.system.SystemDepartmentService;
import doumeemes.service.system.SystemPositionService;
import doumeemes.service.system.SystemRoleService;
import doumeemes.service.system.SystemUserService;
import doumeemes.dao.system.vo.SystemDepartmentListVO;
import doumeemes.service.aware.DepartmentDataPermissionAware;
import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.CollectionUtils;

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

/**
 * 系统用户Service实现
 * @author Eva.Caesar Liu
 * @date 2022/04/18 18:12
 */
@Service
public class SystemUserServiceImpl implements SystemUserService {

    @Autowired
    private SystemUserMapper systemUserMapper;

    @Autowired
    private SystemRoleService systemRoleService;

    @Autowired
    private SystemDepartmentService systemDepartmentService;

    @Autowired
    private SystemPositionService systemPositionService;

    @Autowired
    private DepartmentDataPermissionAware departmentDataPermissionAware;

    @Override
    public Integer create(SystemUser systemUser) {
        systemUserMapper.insert(systemUser);
        return systemUser.getId();
    }

    @Override
    public void deleteById(Integer id) {
        SystemUser systemUser = new SystemUser();
        systemUser.setId(id);
        systemUser.setDeleted(Boolean.TRUE);
        this.updateById(systemUser);
    }

    @Override
    @Transactional
    public void deleteByIdInBatch(List<Integer> ids) {
        if (CollectionUtils.isEmpty(ids)) return;
        for (Integer id : ids) {
            this.deleteById(id);
        }
    }

    @Override
    public void updateById(SystemUser systemUser) {
        systemUserMapper.updateById(systemUser);
    }

    @Override
    @Transactional
    public void updateByIdInBatch(List<SystemUser> systemUsers) {
        if (CollectionUtils.isEmpty(systemUsers)) return;
        for (SystemUser systemUser: systemUsers) {
            this.updateById(systemUser);
        }
    }

    @Override
    public SystemUser findById(Integer id) {
        return systemUserMapper.selectById(id);
    }

    @Override
    public SystemUser findOne(SystemUser systemUser) {
        QueryWrapper<SystemUser> wrapper = new QueryWrapper<>(systemUser);
        return systemUserMapper.selectOne(wrapper.last("limit 1"));
    }

    @Override
    public List<SystemUser> findList(SystemUser systemUser) {
        Wrapper<SystemUser> wrapper = new QueryWrapper<>(systemUser);
        return systemUserMapper.selectList(wrapper);
    }
  
    @Override
    public PageData<SystemUserListVO> findPage(PageWrap<QuerySystemUserDTO> pageWrap) {
        // 根部门条件处理（需查询根部门下所有部门的用户）
        if (pageWrap.getModel().getRootDeptId() != null) {
            pageWrap.getModel().setDepartmentIds(getAllowedDeptIds(pageWrap.getModel().getRootDeptId()));
        } else {
            pageWrap.getModel().setDepartmentIds(getAllowedDeptIds(null));
        }
        // 执行查询
        PageHelper.startPage(pageWrap.getPage(), pageWrap.getCapacity());
        List<SystemUserListVO> userList = systemUserMapper.selectManageList(pageWrap.getModel(), pageWrap.getOrderByClause());
        for (SystemUserListVO user : userList) {
            // 查询用户角色列表
            user.setRoles(systemRoleService.findByUserId(user.getId()));
            // 查询用户岗位列表
            user.setPositions(systemPositionService.findByUserId(user.getId()));
        }
        return PageData.from(new PageInfo<>(userList));
    }


    @Override
    public List<SystemUserListVO> findSystemUsers(Integer type) {
        QueryWrapper<SystemUser> wrapper = new QueryWrapper<>();
        wrapper.lambda().eq(SystemUser::getType,type);
//                        .like(StringUtils.isEmpty(querySystemUserDTO.getUsername()),SystemUser::getUsername,querySystemUserDTO.getUsername());

        List<SystemUser> systemUsers = systemUserMapper.selectList(wrapper);
        List<SystemUserListVO> collect = systemUsers.stream().map(s -> {
            SystemUserListVO systemUserListVO = new SystemUserListVO();
            systemUserListVO.setId(s.getId());
            systemUserListVO.setUsername(s.getUsername());
            systemUserListVO.setRealname(s.getRealname());
            systemUserListVO.setEmpNo(s.getEmpNo());
            systemUserListVO.setBirthday(s.getBirthday());
            systemUserListVO.setSex(s.getSex());
            systemUserListVO.setEmail(s.getEmail());
            systemUserListVO.setMobile(s.getMobile());
            return systemUserListVO;

        }).collect(Collectors.toList());
        return collect;
    }

    @Override
    public long count(SystemUser systemUser) {
        Wrapper<SystemUser> wrapper = new QueryWrapper<>(systemUser);
        return systemUserMapper.selectCount(wrapper);
    }

    /**
     * 获取用户权限内允许查询的部门ID
     */
    private List<Integer> getAllowedDeptIds(Integer rootDeptId) {
        List<SystemDepartmentListVO> allowedDepartments = departmentDataPermissionAware.execute();
        List<Integer> allowedDeptIds = new ArrayList<>();
        for (SystemDepartmentListVO listVO : allowedDepartments) {
            injectIds(allowedDeptIds, listVO);
        }
        // 没有允许的部门
        if (allowedDeptIds.size() == 0) {
            allowedDeptIds.add(-1);
            return allowedDeptIds;
        }
        if (rootDeptId == null) {
            return allowedDeptIds;
        }
        List<Integer> departmentIds = systemDepartmentService.findChildren(rootDeptId);
        departmentIds.add(rootDeptId);
        departmentIds.removeIf(deptId -> !allowedDeptIds.contains(deptId));
        return departmentIds;
    }

    /**
     * 递归注入用户权限内的部门ID
     */
    private void injectIds (List<Integer> pool, SystemDepartmentListVO listVO) {
        pool.add(listVO.getId());
        if (CollectionUtils.isEmpty(listVO.getChildren())) {
            return;
        }
        for (SystemDepartmentListVO child : listVO.getChildren()) {
            injectIds(pool, child);
        }
    }



    @Override
    public List<ActivityLoginVO> activityLoginVOList(ActivityLoginDTO activityLoginDTO) {
        if(StringUtils.isNotBlank(activityLoginDTO.getEndDate())){
            activityLoginDTO.setEndDate(DateUtil.afterDate(activityLoginDTO.getEndDate(),1));
        }
        return systemUserMapper.activityLoginVOList(activityLoginDTO);
    }



    @Override
    public PageData<CompanyLoginStatisticsVO> companyLoginStatisticsPage(PageWrap<CompanyLoginDTO> pageWrap) {
        // 执行查询
        PageHelper.startPage(pageWrap.getPage(), pageWrap.getCapacity());
        List<CompanyLoginStatisticsVO> companyLoginStatisticsList = systemUserMapper.companyLoginStatisticsList(pageWrap.getModel());
        return PageData.from(new PageInfo<>(companyLoginStatisticsList));
    }
}