package com.hxzkmonitor.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.hxzkmonitor.mapper.TbCompanyMapper;
import com.hxzkmonitor.mapper.TbSystemWinMapper;
import com.hxzkmonitor.mapper.TbTankMapper;
import com.hxzkmonitor.mapper.TbUserMapper;
import com.hxzkmonitor.pojo.*;
import com.hxzkmonitor.service.TbSystemWinService;
import com.hxzkmonitor.service.TbTankService;
import lombok.RequiredArgsConstructor;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.List;
import java.util.Objects;

@Service
@RequiredArgsConstructor(onConstructor = @__(@Autowired))
public class TbTankServiceImpl extends ServiceImpl<TbTankMapper, TbTank> implements TbTankService {

    @Autowired
    TbTankMapper tbTankMapper;


    private final TbUserMapper tbUserMapper;

    private final TbCompanyMapper tbCompanyMapper;

    @Override
    public Integer addorupTbTank(TbTank tank) {


        Date date = new Date();
        SimpleDateFormat formatter = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
        tank.setAddtime(formatter.format(date));
        if (tank.getId() != null) {
            return baseMapper.updateById(tank);
        } else {
            QueryWrapper queryWrapper2 = new QueryWrapper<>();
            queryWrapper2.eq("username", tank.getCompanyname());
            TbUser tbUser = tbUserMapper.selectOne(queryWrapper2);
            tank.setCompanyname(tbUser.getCaname());
            QueryWrapper<TbTank> queryWrapper = new QueryWrapper<>();
            queryWrapper.eq("tankname", tank.getTankname());
            queryWrapper.eq("companyname", tank.getCompanyname());
            List<TbTank> tagid1 = baseMapper.selectList(queryWrapper);
            if (tagid1.size()>0){
                return -1;
            }else{
                return baseMapper.insert(tank);
            }

        }
    }

    @Override
    public IPage<TbTank> getTbTankPage(Page page, String username) {

        QueryWrapper queryWrapper2 = new QueryWrapper<>();
        queryWrapper2.eq("username", username);
        TbUser tbUser = tbUserMapper.selectOne(queryWrapper2);
        if (Objects.equals(tbUser.getRole(), "超级管理员")) {
            QueryWrapper queryWrapper = new QueryWrapper<>();
            queryWrapper.orderByDesc("addtime");
            IPage<TbTank> userPage = baseMapper.selectPage(page, queryWrapper);
            return userPage;

        } else {
            QueryWrapper queryWrapper = new QueryWrapper<>();
            queryWrapper.eq("companyname", tbUser.getCaname());
            queryWrapper.orderByDesc("addtime");
            IPage<TbTank> userPage = baseMapper.selectPage(page, queryWrapper);
            return userPage;
        }
    }

    @Override
    public IPage<TbTank> searchTbTank(Page page, String keyword, String username) {

        QueryWrapper queryWrapperu = new QueryWrapper<>();
        queryWrapperu.eq("username",username);
        TbUser tbUser=  tbUserMapper.selectOne(queryWrapperu);
        if (Objects.equals(tbUser.getRole(), "超级管理员")){
            QueryWrapper<TbTank> queryWrapper = new QueryWrapper<>();
            queryWrapper.eq("tankname", keyword);
            queryWrapper.orderByDesc("addtime");
            List<TbTank> tagid1 = baseMapper.selectList(queryWrapper);

            if (tagid1.size() > 0) {
                IPage<TbTank> tagid = baseMapper.selectPage(page, queryWrapper);
                for (int i = 0; i < tagid.getRecords().size(); i++) {
                    QueryWrapper queryWrapper1 = new QueryWrapper<>();
                    queryWrapper1.eq("companyname",tagid.getRecords().get(i).getCompanyname());
                    TbCompany tbCompany =  tbCompanyMapper.selectOne(queryWrapper1);
                    if (tbCompany!=null){
                        tagid.getRecords().get(i).setCompanyname(tbCompany.getCompanyname());
                    }
//                tagid.getRecords().get(i).setCompany(tbCompany.getCompanyname());
                }
                return tagid;
            }  else {
                return null;
            }
        }else{


            QueryWrapper<TbTank> queryWrapper = new QueryWrapper<>();
            queryWrapper.eq("tankname", keyword);
            queryWrapper.eq("companyname",tbUser.getCaname());
            queryWrapper.orderByDesc("addtime");
            List<TbTank> tagid1 = baseMapper.selectList(queryWrapper);

            if (tagid1.size() > 0) {
                IPage<TbTank> tagid = baseMapper.selectPage(page, queryWrapper);
                for (int i = 0; i < tagid.getRecords().size(); i++) {
                    QueryWrapper queryWrapper1 = new QueryWrapper<>();
                    queryWrapper1.eq("companyname",tagid.getRecords().get(i).getTankname());
                    TbCompany tbCompany =  tbCompanyMapper.selectOne(queryWrapper1);
                    if (tbCompany!=null){
                        tagid.getRecords().get(i).setCompanyname(tbCompany.getCompanyname());
                    }
//                tagid.getRecords().get(i).setCompany(tbCompany.getCompanyname());
                }
                return tagid;
            }  else {
                return null;
            }
        }
    }

    @Override
    public TbTank delTbTank(Integer id) {
        QueryWrapper queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("id",id);
        baseMapper.delete(queryWrapper);
        return baseMapper.selectOne(queryWrapper);
    }

}