package com.tievd.jyz.service.impl;

import cn.hutool.core.util.ObjectUtil;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
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.tievd.cube.modules.system.model.SysDepartTreeModel;
import com.tievd.jyz.cache.ResourceCache;
import com.tievd.jyz.constants.SystemConstant;
import com.tievd.jyz.dto.DeviceDTO;
import com.tievd.jyz.entity.Device;
import com.tievd.jyz.entity.UpgradeRecord;
import com.tievd.jyz.mapper.CameraMapper;
import com.tievd.jyz.mapper.DeviceMapper;
import com.tievd.jyz.mapper.UpgradeRecordMapper;
import com.tievd.jyz.mqtt.PojoDataConverter;
import com.tievd.jyz.mqtt.command.InitConfigCommand;
import com.tievd.jyz.mqtt.command.MqttCommandReceiver;
import com.tievd.jyz.mqtt.dto.InitConfigDTO;
import com.tievd.jyz.mqtt.dto.MqttParamDTO;
import com.tievd.jyz.mqtt.dto.RegisterDTO;
import com.tievd.jyz.service.IDeviceService;
import com.tievd.jyz.util.S3Util;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;

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

/**
 * Device
 *
 * @author cube
 * @version V2.0.0
 * @since 2023-02-27
 */
@Slf4j
@Service
public class DeviceServiceImpl extends ServiceImpl<DeviceMapper, Device> implements IDeviceService {
    @Autowired
    private MqttCommandReceiver mqttCommandReceiver;

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

    @Value("${init.heartbeatTime:30}")
    private Integer heartbeatTime;

    @Autowired
    private S3Util s3Util;

    @Autowired
    private CameraMapper cameraMapper;

    @Autowired
    private UpgradeRecordMapper upgradeRecordMapper;

    /**
     * 网关注册处理
     *
     * @param mqttParamDTO
     */
    @Override
    public void register(MqttParamDTO<JSON> mqttParamDTO) {
        String deviceSN = mqttParamDTO.getSn();
        RegisterDTO registerDTO = JSONObject.toJavaObject(mqttParamDTO.getData(), RegisterDTO.class);

        Device device = baseMapper.selectOne(new QueryWrapper<Device>().eq("sn", deviceSN));
        if (ObjectUtil.isNull(device)) {
            log.error("平台未找到网关设备，注册失败，SN:{}", deviceSN);
            return;
        }
        Device tmpDevice = PojoDataConverter.deviceConvert(deviceSN, registerDTO);
        tmpDevice.setId(device.getId());
        baseMapper.updateById(tmpDevice);
        // 升级结果处理
        List<UpgradeRecord> records = ResourceCache.getUpgradeRecord(deviceSN);
        for (UpgradeRecord record : records) {
            record.setStatus(SystemConstant.ISSUE_SUCCESS);
            record.setUpdateTime(new Date());
            upgradeRecordMapper.updateById(record);
            ResourceCache.removeUpgradeRecord(record);
        }
        //返回配置信息
        InitConfigDTO initConfigDTO = PojoDataConverter.InitConfigConvert(s3Util, heartbeatTime);
        InitConfigCommand initConfigCommand = new InitConfigCommand(mqttCommandReceiver, initConfigDTO, sendCommandTopic, deviceSN);
        if (initConfigCommand.init() == SystemConstant.DEAL_FAIL) {
            log.error("初始化指令配置失败");
        }
        if (initConfigCommand.execute() == SystemConstant.DEAL_FAIL) {
            log.error("初始化指令发送失败");
        }
    }

    @Override
    public IPage<DeviceDTO> tables(Page<Device> page, QueryWrapper<Device> queryWrapper) {
        IPage<DeviceDTO> tables = baseMapper.tables(page, queryWrapper);
        return tables;
    }

    @Override
    public void deleteById(String id) {
        baseMapper.deleteById(id);
        cameraMapper.deleteByDeviceId(id);
    }

    @Override
    public void assembleDevice(List<SysDepartTreeModel> list) {
        for (SysDepartTreeModel sysDepartTreeModel : list) {
            LambdaQueryWrapper<Device> queryWrapper = new LambdaQueryWrapper<>();
            queryWrapper.likeRight(Device::getOrgCode, sysDepartTreeModel.getOrgCode());
            //卸油盒子不下发
            queryWrapper.ne(Device::getDeviceType, SystemConstant.OILOUT_DEVICE_TYPE);
            List<Device> devices = baseMapper.selectList(queryWrapper);
            for (Device device : devices) {
                SysDepartTreeModel treeModel = new SysDepartTreeModel();
                treeModel.setKey(device.getId().toString());
                treeModel.setValue(device.getId().toString());
                treeModel.setId(device.getId().toString());
                treeModel.setDepartName(device.getName());
                treeModel.setTitle(device.getName());
                treeModel.setOrgCode(device.getOrgCode());
                treeModel.setStatus(device.getStatus().toString());
                treeModel.setOrgType("99999");
                treeModel.setOrgCategory("99999");
                assemble(treeModel, sysDepartTreeModel);
            }
        }
    }

    private void assemble(SysDepartTreeModel device, SysDepartTreeModel sysDepartTreeModel) {
        List<SysDepartTreeModel> children = sysDepartTreeModel.getChildren();
        if (device.getOrgCode().equals(sysDepartTreeModel.getOrgCode())) {
            if (children == null) {
                children = new ArrayList<>();
            }
            device.setParentId(sysDepartTreeModel.getId());
            children.add(device);
            sysDepartTreeModel.setChildren(children);
        } else {
            if (children == null) {
                return;
            }
            for (SysDepartTreeModel treeModel : children) {
                assemble(device, treeModel);
            }
        }
    }

}