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

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import com.alibaba.fastjson.JSONObject;
import com.ruoyi.common.utils.DateUtils;
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.framework.web.domain.Ztree;
import com.ruoyi.project.dc.securityType.domain.SecurityType;
import com.ruoyi.project.system.dept.domain.Dept;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import com.ruoyi.project.system.company.mapper.CompanyMapper;
import com.ruoyi.project.system.company.domain.Company;
import com.ruoyi.project.system.company.service.ICompanyService;
import com.ruoyi.common.utils.text.Convert;

/**
 * 公司信息Service业务层处理
 * 
 * @author ruoyi
 * @date 2020-05-04
 */
@Service
public class CompanyServiceImpl implements ICompanyService 
{
    @Autowired
    private CompanyMapper companyMapper;

    /**
     * 查询公司信息
     * 
     * @param companyId 公司信息ID
     * @return 公司信息
     */
    @Override
    public Company selectCompanyById(Long companyId)
    {
        return companyMapper.selectCompanyById(companyId);
    }

    /**
     * 查询公司信息列表
     * 
     * @param company 公司信息
     * @return 公司信息
     */
    @Override
    public List<Company> selectCompanyList(Company company)
    {
        return companyMapper.selectCompanyList(company);
    }

    /**
     * 新增公司信息
     * 
     * @param company 公司信息
     * @return 结果
     */
    @Override
    public int insertCompany(Company company)
    {
        company.setCreateTime(DateUtils.getNowDate());

        if (company.getParentId()!=null&&company.getParentId()!=0L){
            Company info = companyMapper.selectCompanyById(company.getParentId());
            company.setAncestors(info.getAncestors() + "," + company.getParentId());
        }else{
            company.setAncestors("0");
            company.setParentId(0L);
        }

        company.setCreateTime(DateUtils.getNowDate());
        return companyMapper.insertCompany(company);
    }

    /**
     * 修改公司信息
     * 
     * @param company 公司信息
     * @return 结果
     */
    @Override
    public int updateCompany(Company company)
    {
        company.setUpdateTime(DateUtils.getNowDate());

        if (company.getParentId()!=null&&company.getParentId()!=0L){
            Company info = companyMapper.selectCompanyById(company.getParentId());
            company.setAncestors(info.getAncestors() + "," + company.getParentId());
        }else{
            company.setAncestors("0");
            company.setParentId(0L);
        }

        return companyMapper.updateCompany(company);
    }

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

    /**
     * 删除公司信息信息
     * 
     * @param companyId 公司信息ID
     * @return 结果
     */
    @Override
    public int deleteCompanyById(Long companyId)
    {
        return companyMapper.deleteCompanyById(companyId);
    }


    @Override
    public List<String> getCompanyImagesById(Long companyId){
        List<String> urls = new ArrayList<>();
        Company company = companyMapper.selectCompanyById(companyId);
        if(StringUtils.isNotEmpty(company.getCompanyFourColorMapUrl())){
            JSONObject object = JSONObject.parseObject(company.getCompanyFourColorMapUrl());
            for (Map.Entry<String, Object> entry : object.entrySet()) {
                urls.add(entry.getValue().toString());
            }
        }
        return urls;
    }


    /**
     * 查询公司信息树列表
     *
     * @return 所有公司信息信息
     */
    @Override
    public List<Ztree> selectCompanyTree()
    {
        List<Company> companyList = companyMapper.selectCompanyList(new Company());
        List<Ztree> ztrees = new ArrayList<Ztree>();
        for (Company company : companyList)
        {
            Ztree ztree = new Ztree();
            ztree.setId(company.getCompanyId());
            ztree.setpId(company.getParentId());
            ztree.setName(company.getCompanyName());
            ztree.setTitle(company.getCompanyName());
            ztrees.add(ztree);
        }
        return ztrees;
    }



    /**
     * 查询公司人数
     *
     * @param parentId 公司ID
     * @return 结果
     */
    @Override
    public int selectCompanyCount(Long parentId)
    {
        Company company = new Company();
        company.setParentId(parentId);
        return companyMapper.selectCompanyCount(company);
    }

}