package com.tievd.jyz.handler;

import cn.hutool.core.lang.Snowflake;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.tievd.jyz.constants.SystemConstant;
import com.tievd.jyz.dto.CaptureVehicleDTO;
import com.tievd.jyz.entity.Camera;
import com.tievd.jyz.entity.Device;
import com.tievd.jyz.mapper.OilEventMapper;
import com.tievd.jyz.mapper.OilPositionMapper;
import com.tievd.jyz.mapper.OilRecordMapper;
import com.tievd.jyz.mapper.SysCarModelMapper;
import com.tievd.jyz.mqtt.MqttMsgReceiver;
import com.tievd.jyz.mqtt.dto.EventInfoDTO;
import com.tievd.jyz.mqtt.dto.MqttParamDTO;
import com.tievd.jyz.service.ICameraService;
import com.tievd.jyz.service.ICarInfoService;
import com.tievd.jyz.service.IDeviceService;
import com.tievd.jyz.util.MultiMinioUtil;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.time.LocalDate;
import java.time.format.DateTimeFormatter;
import java.util.ArrayList;
import java.util.Base64;
import java.util.List;

/**
 * 事件接收处理类
 * @author timi
 */
@Slf4j
@Component
public class CameraIdentifyHandler {
    
    @Autowired
    private OilEventMapper oilEventMapper;
    @Autowired
    private ICameraService cameraService;
    @Autowired
    private IDeviceService deviceService;
    @Autowired
    private OilRecordMapper oilRecordMapper;
    @Autowired
    private ICarInfoService carInfoService;
    @Autowired
    private SysCarModelMapper sysCarModelMapper;
    @Autowired
    private OilPositionMapper oilPositionMapper;
    
    @Autowired
    MultiMinioUtil s3Util;
    
    @Autowired
    EventInfoHandler eventInfoHandler;
    
    public void handle(CaptureVehicleDTO param){
        log.info("====================>开始处理摄像头抓取数据");
        assert param.getSource() != null;
        assert param.getVehicles() != null;
        String cameraCode = param.getSource().getId();
        Camera camera = cameraService.getOne(new LambdaQueryWrapper<Camera>().eq(Camera::getCode, cameraCode), false);
        Device device = deviceService.getById(camera.getDeviceId());
        if (camera == null) return;
    
        List<JSONObject> cars = new ArrayList<>();
        for (CaptureVehicleDTO.Vehicle vehicle : param.getVehicles()) {
            if (StringUtils.isBlank(vehicle.getLicense()) || StringUtils.isBlank(vehicle.getImageData())) {
                continue;
            }
            JSONObject car = converEvent(vehicle);
            cars.add(car);
        }
        if (cars.size() == 0) {
            log.info("====================>摄像头抓取数据车辆信息有缺失， 结束处理");
            return;
        }
        
        log.info("====================>对摄像头抓取数据进行转换");
        //中间层
        EventInfoDTO eventInfoDTO = new EventInfoDTO();
        eventInfoDTO.setAlgCode("C10001");
        eventInfoDTO.setCameraCode(cameraCode);
        String eventCode = new Snowflake().nextIdStr();
        eventInfoDTO.setEventCode(eventCode);
        eventInfoDTO.setExtend(cars);
        List<EventInfoDTO> eventList = new ArrayList<>();
        eventList.add(eventInfoDTO);
        //最外层
        MqttParamDTO mqttDto = new MqttParamDTO();
        mqttDto.setSn(device.getSn());
        mqttDto.setTime(String.valueOf(param.getDate().getTime()));
        mqttDto.setData(eventList);
        mqttDto.setType(MqttMsgReceiver.EVENT_INFO);
        eventInfoHandler.event(mqttDto);
        log.info("====================>摄像头抓取数据载入主流程");
    }
    
    
    JSONObject converEvent(CaptureVehicleDTO.Vehicle vehicle){
        JSONObject convertEvent = new JSONObject();
        //主要数据
        convertEvent.put("number", vehicle.getLicense());
        convertEvent.put("score", "0.9");
        convertEvent.put("type", vehicle.getExtend().getModel());
        convertEvent.put("color", vehicle.getColor());
        
        //多余字段
        convertEvent.put("licenseColor", vehicle.getLicenseColor());
        convertEvent.put("licenseType", vehicle.getLicenseType());
        
        //转s3地址
        byte[] imgByte = Base64.getDecoder().decode(vehicle.getImageData());
        String parentPath = "cameraPhoto/" + LocalDate.now().format(DateTimeFormatter.ISO_LOCAL_DATE) + "/";
        String key = s3Util.upload(imgByte, parentPath + convertEvent.getString("eventCode"));
        String url = s3Util.getBucketName() + SystemConstant.s3_split_char + key;
        convertEvent.put("sourceType", "jpg");
        convertEvent.put("imgPath", url);
        JSONObject extend = new JSONObject();
        extend.putAll(convertEvent);
        convertEvent.put("extend", extend);
        return convertEvent;
        
    }
    
    
    
}