package com.gk.hotwork.Service.ServiceImpl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.OrderItem;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.gk.hotwork.Domain.CompanyInfo;
import com.gk.hotwork.Domain.Do.StatisticCompanyCount;
import com.gk.hotwork.Domain.UserInfo;
import com.gk.hotwork.Domain.Utils.PageInfo;
import com.gk.hotwork.Domain.Utils.StringUtils;
import com.gk.hotwork.Domain.query.CompanyQuery;
import com.gk.hotwork.Domain.query.CountQuery;
import com.gk.hotwork.Mapper.CompanyInfoMapper;
import com.gk.hotwork.Service.CompanyService;
import io.swagger.models.auth.In;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

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

@Service("CompanyService")
public class CompanyServiceImpl extends ServiceImpl<CompanyInfoMapper, CompanyInfo> implements CompanyService {
    @Autowired
    CompanyInfoMapper companyInfoMapper;

    @Override
    public void selectDataGrid(PageInfo pageInfo) {
        Page<CompanyInfo> page = new Page<>(pageInfo.getPageIndex(), pageInfo.getPageSize());
        List<OrderItem> orderItems = new ArrayList<>();
        OrderItem orderItem = new OrderItem();
        if (StringUtils.isNotBlank(pageInfo.getSort()) && StringUtils.isNotBlank(pageInfo.getOrder())) {
            orderItem.setAsc(pageInfo.getOrder().equalsIgnoreCase("ascending"));
            orderItem.setColumn(pageInfo.getSort());
        }else {
            orderItem.setAsc(false);
            orderItem.setColumn("createddate");
        }
        orderItems.add(orderItem);
        page.setOrders(orderItems);
        List<CompanyInfo> list = companyInfoMapper.selectDataGrid(page,pageInfo.getCondition());

        pageInfo.setResult(list);
        pageInfo.setTotalCount(page.getTotal());
    }

    @Override
    public CompanyInfo selectExistByName(Long id, String company) {
        return companyInfoMapper.selectExistByName(id,company);
    }

    @Override
    public CompanyInfo selectByName(String company) {
        LambdaQueryWrapper<CompanyInfo> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(CompanyInfo::getCompany,company);
        return companyInfoMapper.selectOne(wrapper);
    }

    @Override
    public List<CompanyInfo> selectByAll() {
        return companyInfoMapper.selectByAll();
    }

    @Override
    public CompanyInfo selectExistMain(Long id) {
        return companyInfoMapper.selectExistMain(id);
    }

    @Override
    public boolean isMain(String company) {
        if (StringUtils.isBlank(company)){return true; }
        CompanyInfo companyInfo = selectByName(company);
        if (companyInfo == null){return false;}
        return companyInfo.getIsmain() == 1;
    }

    @Override
    public List<CompanyInfo> selectCompanyList(String companyName, UserInfo user) {
        CompanyQuery query = new CompanyQuery();
        query.setCompany(companyName);


        Integer type = user.getType();
        //企业用户
        if (type.equals(3)) {
            Long companyid = user.getCompanyid();
            query.setCompanyid(companyid);
        }
        //监管用户
        if (type.equals(2)) {
            //获取企业信息
            query.setArea(user.getCounty());
            query.setProvince(user.getProvince());
            query.setCity(user.getCity());
        }

        List<CompanyInfo> companyInfos = companyInfoMapper.selectCompanyList(query);
        return companyInfos;
    }

    @Override
    public Integer getCountByArea(Map<String,Object> params) {
        return companyInfoMapper.getCountByArea(params);
    }

    @Override
    public Page<CompanyInfo> getUncheckCompanyByArea(CountQuery query){
        Page<CompanyInfo> page = new Page<>(query.getPageIndex(),query.getPageSize());

        List<CompanyInfo> companyInfos = companyInfoMapper.getUncheckCompanyByArea(page,query);
        page.setRecords(companyInfos);
        return page;
    }


    @Override
    public Page<CompanyInfo> getByArea(CountQuery query){
       Page<CompanyInfo> page = new Page<>(query.getPageIndex(),query.getPageSize());
       CompanyQuery companyQuery = new CompanyQuery();
       companyQuery.setArea(query.getArea());
       List<CompanyInfo> companyInfos = companyInfoMapper.selectCompanyList(page,companyQuery);
       page.setRecords(companyInfos);
       return page;
    }

}