package com.juedui.equipbook.service.impl;


import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.juedui.equipbook.entity.College;
import com.juedui.equipbook.entity.Equipment;
import com.juedui.equipbook.entity.LabTable;
import com.juedui.equipbook.entity.User;
import com.juedui.equipbook.mapper.EquipmentMapper;
import com.juedui.equipbook.service.CollegeService;
import com.juedui.equipbook.service.EquipmentService;
import com.juedui.equipbook.service.LabTableService;
import com.juedui.equipbook.service.UserService;
import com.juedui.equipbook.vo.AddEquipmentVo;
import com.juedui.equipbook.vo.EquipmentVo;
import com.juedui.equipbook.vo.GetEquipmentVo;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

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

@Service
public class EquipmentServiceImpl extends ServiceImpl<EquipmentMapper, Equipment> implements EquipmentService {
    @Autowired
    private LabTableService labTableService;
    @Autowired
    private CollegeService collegeService;
    @Autowired
    private UserService userService;
    @Override
    public Page<EquipmentVo> getEquipment(GetEquipmentVo getEquipmentVo) {
        if (getEquipmentVo.getPageNumber() == null) {
            getEquipmentVo.setPageNumber(1);
        }
        if (getEquipmentVo.getPageLimit() == null) {
            getEquipmentVo.setPageLimit(10);
        }
        LambdaQueryWrapper<Equipment> wrapper = new LambdaQueryWrapper<>();
        if (StringUtils.isNotBlank(getEquipmentVo.getName())) {
            wrapper.like(Equipment::getName, getEquipmentVo.getName());
        }
        if (getEquipmentVo.getCollegeId() != null) {
            wrapper.eq(Equipment::getCollegeId, getEquipmentVo.getCollegeId());
        }
        if (StringUtils.isNotBlank(getEquipmentVo.getStatus())) {
            wrapper.eq(Equipment::getStatus, getEquipmentVo.getStatus());
        }

        Page equipmentPage = new Page(getEquipmentVo.getPageNumber(), getEquipmentVo.getPageLimit());
        Page page = this.baseMapper.selectPage(equipmentPage, wrapper);
        List<Equipment> equipments = page.getRecords();
        List<EquipmentVo> equipmentVos = new ArrayList<>();
        for (Equipment equipment : equipments) {
            EquipmentVo equipmentVo = new EquipmentVo();
            College college = collegeService.getById(equipment.getCollegeId());
            LabTable labTable = labTableService.getById(equipment.getLabId());
            User user = userService.getById(labTable.getAdminId());
            equipmentVo.setCollegeName(college.getName());
            equipmentVo.setLabName(labTable.getName());
            equipmentVo.setAdminName(user.getUsername());
            BeanUtils.copyProperties(equipment, equipmentVo);
            equipmentVos.add(equipmentVo);
        }
        page.setRecords(equipmentVos);
        return page;
    }


    @Override
    public Boolean addEquipment(AddEquipmentVo addEquipmentVo) {
        Equipment equipment = new Equipment();
        BeanUtils.copyProperties(addEquipmentVo, equipment);
        int insert = this.baseMapper.insert(equipment);
        if (insert == 1) {
            return true;
        }
        return false;
    }

    @Override
    public Boolean updateEquipment(Equipment equipment) {
        int update = this.baseMapper.updateById(equipment);
        if (update == 1) {
            return true;
        }
        return false;
    }

    @Override
    public Boolean removeByEquipmentIds(List<String> ids) {
        int i = this.baseMapper.deleteBatchIds(ids);
        if (i == ids.size()) {
            return true;
        }
        return false;
    }

    @Override
    public Boolean updateEquipmentStatus(String equipmentId, String status) {
        LambdaQueryWrapper<Equipment> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(Equipment::getId, equipmentId);
        Equipment equipment = this.baseMapper.selectOne(wrapper);
        equipment.setStatus(status);
        return this.updateById(equipment);
    }
    @Override
    public List<Equipment> getEquipmentByLabTableId(Integer labTableId) {
        LambdaQueryWrapper<Equipment> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(Equipment::getLabId, labTableId);
        return this.baseMapper.selectList(wrapper);
    }
    @Override
    public List<Equipment> getEquipmentByLabTableIds(List<Integer> labTableIds) {
        LambdaQueryWrapper<Equipment> wrapper = new LambdaQueryWrapper<>();
        wrapper.in(Equipment::getLabId, labTableIds);
        return this.baseMapper.selectList(wrapper);
    }


}