package com.zy.asrs.controller;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.mapper.EntityWrapper;
import com.core.annotations.ManagerAuth;
import com.core.common.Cools;
import com.core.common.R;
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.TvLocDataDto;
import com.zy.asrs.entity.dto.WcsStationDto;
import com.zy.asrs.enums.RedisKeyType;
import com.zy.asrs.service.BasStationTvService;
import com.zy.asrs.service.TvDeviceService;
import com.zy.asrs.utils.StationUtils;
import com.zy.asrs.utils.Utils;
import com.zy.common.utils.RedisUtil;
import com.zy.system.entity.Announcement;
import com.zy.system.service.AnnouncementService;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.beans.factory.annotation.Autowired;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.*;

@Slf4j
@RestController
@RequestMapping("/openapi")
public class OpenController {

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

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

    @Value("${app.version:1.0.0}")
    private String appVersion;
    @Value("${app.version-type:stable}")
    private String appVersionType;

    @Value("${file.upload-path}")
    private String uploadPath;

    @PostMapping("/uploadLogo")
    public R uploadLogo(@RequestParam("file") MultipartFile file) {
        if (file.isEmpty()) {
            return R.error("上传文件不能为空");
        }
        try {
            File dir = new File(uploadPath);
            if (!dir.exists()) {
                dir.mkdirs();
            }

            String originalFilename = file.getOriginalFilename();
            String extension = "";
            if (originalFilename != null && originalFilename.lastIndexOf(".") > 0) {
                extension = originalFilename.substring(originalFilename.lastIndexOf("."));
            }

            // 删除旧的logo文件
            File[] files = dir.listFiles((d, name) -> name.startsWith("logo."));
            if (files != null) {
                for (File f : files) {
                    f.delete();
                }
            }

            String fileName = "logo" + extension;
            File dest = new File(dir, fileName);
            file.transferTo(dest);

            return R.ok();
        } catch (IOException e) {
            log.error("上传Logo失败", e);
            return R.error("上传失败: " + e.getMessage());
        }
    }

    @GetMapping("/getLogo")
    public void getLogo(HttpServletResponse response) {
        try {
            File dir = new File(uploadPath);
            if (!dir.exists()) {
                response.sendError(HttpServletResponse.SC_NOT_FOUND, "Logo directory not found");
                return;
            }

            File[] files = dir.listFiles((d, name) -> name.startsWith("logo."));
            if (files == null || files.length == 0) {
                response.sendError(HttpServletResponse.SC_NOT_FOUND, "Logo not found");
                return;
            }

            File logoFile = files[0];
            String filename = logoFile.getName();
            String contentType = "image/jpeg";
            if (filename.toLowerCase().endsWith(".png")) {
                contentType = "image/png";
            } else if (filename.toLowerCase().endsWith(".gif")) {
                contentType = "image/gif";
            }
            response.setContentType(contentType);

            try (FileInputStream fis = new FileInputStream(logoFile);
                 OutputStream os = response.getOutputStream()) {
                byte[] buffer = new byte[1024];
                int len;
                while ((len = fis.read(buffer)) != -1) {
                    os.write(buffer, 0, len);
                }
                os.flush();
            }
        } catch (Exception e) {
            log.error("获取Logo失败", e);
        }
    }

    @GetMapping("/systemStatus")
    public R systemStatus() {
        return R.ok();
    }

    @GetMapping("/getSystemVersion")
    public R getSystemVersion() {
        HashMap<String, Object> map = new HashMap<>();
        map.put("version", appVersion);
        map.put("versionType", appVersionType);
        return R.ok().add(map);
    }

    @GetMapping("/getLedInfos")
    public R getLedInfos(HttpServletRequest request) {
        HashMap<String, Object> map = new HashMap<>();
        String remoteAddr = request.getRemoteAddr();
        TvDevice tvDevice = tvDeviceService.selectOne(
                new EntityWrapper<TvDevice>().eq("ip", remoteAddr));
        if (tvDevice == null) {
            return R.error("未找到IP对应的电视机设备: " + remoteAddr);
        }
        map.put("deviceName", tvDevice.getName());

        List<BasStationTv> relations = basStationTvService
                .selectList(new EntityWrapper<BasStationTv>().eq("tv_id", tvDevice.getId()));
        if (relations == null || relations.isEmpty()) {
            map.put("data", new ArrayList<>());
            return R.ok().add(map);
        }


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

        List<TvDataDto> list = new ArrayList<>();
        for (BasStationTv relation : relations) {
            WcsStationDto wcsStationDto = stationUtils.stationMap.get(relation.getStationId());

            if (wcsStationDto == null) {
                continue;
            }

            if (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(manualError)) {
                if (!Cools.isEmpty(errorMsg)) {
                    errorMsg += "；";
                }
                errorMsg += manualError;
            }

            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());

            if (Cools.isEmpty(errorMsg)) {
                tvDataDto.setErrorCode(0);
            }else {
                tvDataDto.setErrorCode(1);
            }

            list.add(tvDataDto);
        }


        map.put("data", list);
        return R.ok().add(map);
    }

    @GetMapping("/getError")
    public R getError(HttpServletRequest request) {
        Set<String> errors = new LinkedHashSet<>();
        Object manualErrorObj = redisUtil.get(RedisKeyType.TV_MANUAL_ERROR_MSG.key);
        String manualError = manualErrorObj == null ? "" : String.valueOf(manualErrorObj);
        if (!Cools.isEmpty(manualError)) {
            errors.add(manualError);
        }

        String remoteAddr = request.getRemoteAddr();
        TvDevice tvDevice = tvDeviceService.selectOne(
                new EntityWrapper<TvDevice>().eq("ip", remoteAddr));
        if (tvDevice != null) {
            List<BasStationTv> relations = basStationTvService
                    .selectList(new EntityWrapper<BasStationTv>().eq("tv_id", tvDevice.getId()));
            if (relations != null && !relations.isEmpty()) {
                for (BasStationTv relation : relations) {
                    WcsStationDto wcsStationDto = stationUtils.stationMap.get(relation.getStationId());
                    if (wcsStationDto == null) {
                        continue;
                    }
                    if (wcsStationDto.getLoading() != 1) {
                        continue;
                    }
                    String deviceError = "";
                    if (!Cools.isEmpty(wcsStationDto.getErrorMsg())) {
                        deviceError += wcsStationDto.getErrorMsg();
                    }
                    if (!Cools.isEmpty(wcsStationDto.getSystemWarning())) {
                        if (!Cools.isEmpty(deviceError)) {
                            deviceError += "；";
                        }
                        deviceError += wcsStationDto.getSystemWarning();
                    }
                    if (!Cools.isEmpty(deviceError)) {
                        errors.add(deviceError);
                    }
                }
            }
        }

        String errorMsg = String.join("；", errors);
        Map<String, Object> result = new HashMap<>();
        result.put("errorMsg", errorMsg);
        return R.ok(result);
    }

    @GetMapping("/manualError/auth")
    @ManagerAuth
    public R getManualErrorAuth() {
        Object manualErrorObj = redisUtil.get(RedisKeyType.TV_MANUAL_ERROR_MSG.key);
        String manualError = manualErrorObj == null ? "" : String.valueOf(manualErrorObj);
        Map<String, Object> result = new HashMap<>();
        result.put("manualError", manualError);
        return R.ok(result);
    }

    @PostMapping("/manualError/auth")
    @ManagerAuth
    public R setManualError(@RequestBody JSONObject param) {
        String manualError = param.getString("manualError");
        if (Cools.isEmpty(manualError)) {
            redisUtil.del(RedisKeyType.TV_MANUAL_ERROR_MSG.key);
            return R.ok();
        }
        redisUtil.set(RedisKeyType.TV_MANUAL_ERROR_MSG.key, manualError);
        return R.ok();
    }

    @GetMapping("/announcement")
    public R top5(){
        EntityWrapper<Announcement> wrapper = new EntityWrapper<>();
        wrapper.eq("status", 1);
        wrapper.orderBy("create_time", false);
        wrapper.last("limit 5");
        return R.ok(announcementService.selectList(wrapper));
    }

    /**
     * 获取当前时间
     */
    @GetMapping("/monitor/date")
    public R monitorDate() {
        Date now = new Date();
        Calendar calendar = Calendar.getInstance();
        calendar.setTime(now);
        return R.ok(
                Cools.add("year", calendar.get(Calendar.YEAR))
                        .add("month", Utils.zerofill(String.valueOf(calendar.get(Calendar.MONTH)+1), 2))
                        .add("day", Utils.zerofill(String.valueOf(calendar.get(Calendar.DATE)), 2))
                        .add("hour", Utils.zerofill(String.valueOf(calendar.get(Calendar.HOUR_OF_DAY)), 2))
                        .add("minute", Utils.zerofill(String.valueOf(calendar.get(Calendar.MINUTE)), 2))
                        .add("second", Utils.zerofill(String.valueOf(calendar.get(Calendar.SECOND)) , 2))
                        .add("week", WEEK[calendar.get(Calendar.DAY_OF_WEEK)-1])
        );
    }

    /**
     * 获取库位数据
     */
    @GetMapping("/queryLoc")
    public R queryLoc() {
        TvLocDataDto tvLocDataDto = new TvLocDataDto();
        Object o = redisUtil.get(RedisKeyType.TV_LOC_DATA_DTO.key);
        if(o == null){
            return R.ok().add(tvLocDataDto);
        }

        tvLocDataDto = JSON.parseObject(o.toString(), TvLocDataDto.class);
        return R.ok().add(tvLocDataDto);
    }

    /**
     * WMS入出库折线图
     */
    @GetMapping("/inOutLineCharts")
    public R inOutLineCharts() {
        Object o = redisUtil.get(RedisKeyType.TV_LINE_CHARTS.key);
        if(o == null){
            return R.error();
        }

        return R.ok().add(JSON.parseObject(o.toString()));
    }

    /**
     * WMS库存数据统计
     */
    @GetMapping("/locDetlStatistics")
    public R locDetlStatistics() {
        Object o = redisUtil.get(RedisKeyType.TV_LOC_DETL_STATISTICS.key);
        if(o == null){
            return R.error();
        }

        return R.ok().add(JSON.parseArray(o.toString()));
    }

}