package com.tievd.jyz.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.tievd.cube.commons.base.Result;
import com.tievd.jyz.cache.ResourceCache;
import com.tievd.jyz.constants.SystemConstant;
import com.tievd.jyz.dto.IssueDTO;
import com.tievd.jyz.entity.Device;
import com.tievd.jyz.entity.Upgrade;
import com.tievd.jyz.entity.UpgradeRecord;
import com.tievd.jyz.mapper.DeviceMapper;
import com.tievd.jyz.mapper.UpgradeMapper;
import com.tievd.jyz.mapper.UpgradeRecordMapper;
import com.tievd.jyz.mqtt.command.MqttCommandReceiver;
import com.tievd.jyz.mqtt.command.UpgradeCommand;
import com.tievd.jyz.service.IUpgradeService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;

import java.io.Serializable;
import java.util.Date;
import java.util.List;

/**
 * Upgrade
 * @author      cube
 * @since       2023-02-27
 * @version     V2.0.0
 */
@Service
public class UpgradeServiceImpl extends ServiceImpl<UpgradeMapper, Upgrade> implements IUpgradeService {

    @Autowired
    private DeviceMapper deviceMapper;

    @Autowired
    private UpgradeRecordMapper upgradeRecordMapper;

    @Autowired
    private MqttCommandReceiver mqttCommandReceiver;
    @Value("${init.sendCommandTopic:/ty/aibox/command/}")
    private String sendCommandTopic;

    @Override
    public Result issue(IssueDTO dto) {
        Date now = new Date();
        Upgrade upgrade = baseMapper.selectById(dto.getUpgradeId());
        if (upgrade == null){
            return Result.error("该版本不存在");
        }
        List<String> deviceIds = dto.getDeviceId();
        LambdaQueryWrapper<Device> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.in(Device::getId, deviceIds);
        List<Device> devices = deviceMapper.selectList(queryWrapper);
        for (Device device: devices) {
            //下发升级指令只设备
            UpgradeCommand upgradeCommand = new UpgradeCommand(mqttCommandReceiver,upgrade.getVersion(),upgrade.getPath(),sendCommandTopic,device.getSn());
            upgradeCommand.init();
            upgradeCommand.execute();
            UpgradeRecord record = new UpgradeRecord();
            record.setUpgradeId(upgrade.getId());
            record.setCreateTime(new Date());
            record.setDeviceSn(device.getSn());
            record.setStatus(SystemConstant.ISSUE_UNDERWAY);
            record.setCreateTime(now);
            record.setIssueTime(now);
            upgradeRecordMapper.insert(record);
            ResourceCache.addUpgradeRecord(record);
        }
        return Result.ok();
    }

    @Override
    public void remove(Serializable id) {
        baseMapper.deleteById(id);
        QueryWrapper<UpgradeRecord> wrapper = new QueryWrapper<>();
        wrapper.eq("upgrade_id", id);
        upgradeRecordMapper.delete(wrapper);
    }
}