package com.zy.asrs.service;

import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.core.common.Cools;
import com.zy.system.entity.Announcement;
import com.zy.asrs.entity.BasStationTv;
import com.zy.asrs.entity.TvDevice;
import com.zy.asrs.entity.dto.TvDataDto;
import com.zy.asrs.entity.dto.WcsStationDto;
import com.zy.asrs.enums.RedisKeyType;
import com.zy.asrs.utils.StationUtils;
import com.zy.asrs.websocket.TvWebSocketServer;
import com.zy.common.model.WebSocketMessage;
import com.zy.common.utils.RedisUtil;
import com.zy.system.service.AnnouncementService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Service;

import java.util.*;

@Slf4j
@Service
public class TvDataPushService {

    @Autowired
    private TvWebSocketServer tvWebSocketServer;
    @Autowired
    private TvDeviceService tvDeviceService;
    @Autowired
    private BasStationTvService basStationTvService;
    @Autowired
    private StationUtils stationUtils;
    @Autowired
    private RedisUtil redisUtil;
    @Autowired
    private AnnouncementService announcementService;

    private static final String[] WEEK = {"星期日","星期一","星期二","星期三","星期四","星期五","星期六"};


    @Scheduled(fixedRate = 1000)
    public void pushAll() {
        if (tvWebSocketServer.getOnlineIps().isEmpty()) {
            return;
        }
        pushLedInfosAndError();
        pushLocData();
        pushLineCharts();
        pushLocDetlStatistics();
        pushAnnouncement();
        pushDate();
    }

    private void pushLedInfosAndError() {
        String manualError = "";
        Object manualErrorObj = redisUtil.get(RedisKeyType.TV_MANUAL_ERROR_MSG.key);
        if (manualErrorObj != null) {
            manualError = String.valueOf(manualErrorObj);
        }

        for (String ip : tvWebSocketServer.getOnlineIps()) {
            TvDevice tvDevice = tvDeviceService.getOne(
                    new QueryWrapper<TvDevice>().eq("ip", ip));
            if (tvDevice == null) {
                // 非注册设备（如调试页面），推送空数据和通用错误
                Map<String, Object> ledData = new HashMap<>();
                ledData.put("deviceName", "未注册设备(" + ip + ")");
                ledData.put("data", new ArrayList<>());
                tvWebSocketServer.sendMessageToDevice(ip, buildMessage("ledInfos", ledData));

                Map<String, Object> errorData = new HashMap<>();
                errorData.put("errorMsg", manualError != null ? manualError : "");
                tvWebSocketServer.sendMessageToDevice(ip, buildMessage("error", errorData));
                continue;
            }

            List<BasStationTv> relations = basStationTvService
                    .list(new QueryWrapper<BasStationTv>().eq("tv_id", tvDevice.getId()));

            List<TvDataDto> dataList = new ArrayList<>();
            Set<String> errors = new LinkedHashSet<>();

            if (manualError != null && !manualError.isEmpty()) {
                errors.add(manualError);
            }

            if (relations != null && !relations.isEmpty()) {
                for (BasStationTv relation : relations) {
                    WcsStationDto wcsStationDto = stationUtils.stationMap.get(relation.getStationId());
                    if (wcsStationDto == null || wcsStationDto.getLoading() != 1) {
                        continue;
                    }

                    String errorMsg = "";
                    if (!Cools.isEmpty(wcsStationDto.getErrorMsg())) {
                        errorMsg += wcsStationDto.getErrorMsg();
                    }
                    if (!Cools.isEmpty(wcsStationDto.getSystemWarning())) {
                        errorMsg += wcsStationDto.getSystemWarning();
                    }
                    if (!Cools.isEmpty(errorMsg)) {
                        errors.add(errorMsg);
                    }

                    TvDataDto tvDataDto = new TvDataDto();
                    tvDataDto.setStationId(wcsStationDto.getStationId());
                    tvDataDto.setTaskNo(wcsStationDto.getTaskNo());
                    tvDataDto.setBarcode(wcsStationDto.getBarcode());
                    tvDataDto.setErrorMsg(errorMsg);
                    tvDataDto.setIoType(wcsStationDto.getIoType());
                    tvDataDto.setWrkDetls(wcsStationDto.getWrkDetls());
                    tvDataDto.setErrorCode(Cools.isEmpty(errorMsg) ? 0 : 1);
                    dataList.add(tvDataDto);
                }
            }

            // 推送 ledInfos
            Map<String, Object> ledData = new HashMap<>();
            ledData.put("deviceName", tvDevice.getName());
            ledData.put("data", dataList);
            tvWebSocketServer.sendMessageToDevice(ip, buildMessage("ledInfos", ledData));

            // 推送 error
            Map<String, Object> errorData = new HashMap<>();
            errorData.put("errorMsg", String.join("；", errors));
            tvWebSocketServer.sendMessageToDevice(ip, buildMessage("error", errorData));
        }
    }

    private void pushLocData() {
        Object o = redisUtil.get(RedisKeyType.TV_LOC_DATA_DTO.key);
        String currentData = o != null ? o.toString() : "{}";
        tvWebSocketServer.sendMessageToAll(buildMessage("locData", JSON.parseObject(currentData)));
    }

    private void pushLineCharts() {
        Object o = redisUtil.get(RedisKeyType.TV_LINE_CHARTS.key);
        if (o == null) {
            return;
        }
        String currentData = o.toString();
        tvWebSocketServer.sendMessageToAll(buildMessage("lineCharts", JSON.parseObject(currentData)));
    }

    private void pushLocDetlStatistics() {
        Object o = redisUtil.get(RedisKeyType.TV_LOC_DETL_STATISTICS.key);
        if (o == null) {
            return;
        }
        String currentData = o.toString();
        tvWebSocketServer.sendMessageToAll(buildMessage("locDetlStatistics", JSON.parseArray(currentData)));
    }

    private void pushAnnouncement() {
        QueryWrapper<Announcement> wrapper = new QueryWrapper<>();
        wrapper.eq("status", 1);
        wrapper.orderBy(true, false, "create_time");
        wrapper.last("limit 5");
        List<Announcement> list = announcementService.list(wrapper);
        tvWebSocketServer.sendMessageToAll(buildMessage("announcement", list));
    }

    private void pushDate() {
        Date now = new Date();
        Calendar calendar = Calendar.getInstance();
        calendar.setTime(now);
        Map<String, Object> dateMap = new HashMap<>();
        dateMap.put("year", calendar.get(Calendar.YEAR));
        dateMap.put("month", zerofill(String.valueOf(calendar.get(Calendar.MONTH) + 1), 2));
        dateMap.put("day", zerofill(String.valueOf(calendar.get(Calendar.DATE)), 2));
        dateMap.put("hour", zerofill(String.valueOf(calendar.get(Calendar.HOUR_OF_DAY)), 2));
        dateMap.put("minute", zerofill(String.valueOf(calendar.get(Calendar.MINUTE)), 2));
        dateMap.put("second", zerofill(String.valueOf(calendar.get(Calendar.SECOND)), 2));
        dateMap.put("week", WEEK[calendar.get(Calendar.DAY_OF_WEEK) - 1]);
        tvWebSocketServer.sendMessageToAll(buildMessage("date", dateMap));
    }

    private String buildMessage(String url, Object data) {
        WebSocketMessage message = new WebSocketMessage();
        message.setUrl(url);
        message.setData(JSON.toJSONString(data));
        return JSON.toJSONString(message);
    }

    private String zerofill(String str, int length) {
        while (str.length() < length) {
            str = "0" + str;
        }
        return str;
    }
}