package com.javaweb.geo.service.impl;

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

import com.javaweb.common.utils.IdGenerate;
import com.javaweb.common.utils.StringUtils;
import com.javaweb.framework.util.ShiroUtils;
import com.javaweb.geotdp.enums.RecordType;
import com.javaweb.system.domain.SysRole;
import com.javaweb.system.domain.SysUser;
import com.javaweb.system.mapper.SysUserMapper;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import com.javaweb.geo.mapper.CompanyMapper;
import com.javaweb.geo.domain.Company;
import com.javaweb.geo.service.ICompanyService;
import com.javaweb.common.core.text.Convert;
import org.springframework.util.ObjectUtils;

/**
 * 勘察单位Service业务层处理
 * 
 * @author zmk
 * @date 2022-10-14
 */
@Service
public class CompanyServiceImpl implements ICompanyService 
{
    @Autowired
    private CompanyMapper companyMapper;

    @Autowired
    private SysUserMapper sysUserMapper;

    /**
     * 查询勘察单位
     * 
     * @param ids 勘察单位ID
     * @return 勘察单位
     */
    @Override
    public Company selectCompanyById(String ids)
    {
        return companyMapper.selectCompanyById(ids);
    }

    /**
     * 查询勘察单位列表
     * 
     * @param company 勘察单位
     * @return 勘察单位
     */
    @Override
    public List<Company> selectCompanyList(Company company)
    {
          // return companyMapper.selectCompanyList(company);
        List<Company> resultList = new ArrayList<>();

        SysUser sysUser = ShiroUtils.getSysUser();
        List<SysRole> roles = sysUser.getRoles();

        if (!ObjectUtils.isEmpty(roles)){
            LinkedHashSet<String> collect = roles.stream().map(SysRole::getRoleKey).collect(Collectors.toCollection(LinkedHashSet::new));

            // 管理员角色
            if (collect.contains(RecordType.role_admin.getId())){
                resultList = companyMapper.selectCompanyList(company);
            }
            // 普通角色
            if (collect.contains(RecordType.role_gongsi.getId())){
                SysUser u = sysUserMapper.selectUserById(sysUser.getUserId());
                company.setIds(u.getCompanyId());
                resultList = companyMapper.selectCompanyList(company);
            }

        }

        return resultList;
    }

    /**
     * 新增勘察单位
     * 
     * @param company 勘察单位
     * @return 结果
     */
    @Override
    public int insertCompany(Company company)
    {
        if(StringUtils.isEmpty(company.getIds())){
            company.setIds(IdGenerate.nextId());
        }
        if(company.getAddTime()==null){
            company.setAddTime(new Date());
        }
        company.setCity(1);//北京
        company.setProvince(1);//北京
        return companyMapper.insertCompany(company);
    }

    /**
     * 修改勘察单位
     * 
     * @param company 勘察单位
     * @return 结果
     */
    @Override
    public int updateCompany(Company company)
    {
        return companyMapper.updateCompany(company);
    }

    /**
     * 删除勘察单位对象
     * 
     * @param ids 需要删除的数据ID
     * @return 结果
     */
    @Override
    public int deleteCompanyByIds(String ids)
    {
        return companyMapper.deleteCompanyByIds(Convert.toStrArray(ids));
    }

    /**
     * 删除勘察单位信息
     * 
     * @param ids 勘察单位ID
     * @return 结果
     */
    @Override
    public int deleteCompanyById(String ids)
    {
        return companyMapper.deleteCompanyById(ids);
    }
}