package com.tievd.jyz.controller;

import cn.hutool.core.io.IoUtil;
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.tievd.cube.commons.annotations.AutoLog;
import com.tievd.cube.commons.annotations.DictApi;
import com.tievd.cube.commons.base.CubeController;
import com.tievd.cube.commons.base.Result;
import com.tievd.cube.commons.utils.SystemContextUtil;
import com.tievd.cube.commons.utils.web.HttpServletUtil;
import com.tievd.cube.modules.system.model.LoginUser;
import com.tievd.jyz.constants.SystemConstant;
import com.tievd.jyz.entity.OilEvent;
import com.tievd.jyz.entity.vo.EventReqVo;
import com.tievd.jyz.service.IOilEventService;
import com.tievd.jyz.service.IOiloutEventService;
import com.tievd.jyz.util.BusinessUtil;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.responses.ApiResponse;
import io.swagger.v3.oas.annotations.tags.Tag;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections.CollectionUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.*;

/**
 * OilEvent
 *
 * @author cube
 * @version V2.0.0
 * @since 2023-02-27
 */
@Slf4j
@DictApi
@RestController
@RequestMapping("/jyz/oilEvent")
@Tag(name = "事件列表, 首页事件")
public class OilEventController extends CubeController<OilEvent, IOilEventService> {
    
    @Autowired
    private IOilEventService oilEventService;
    
    @Autowired
    private IOiloutEventService oiloutEventService;
    
    
    /**
     * 分页列表查询
     */
    @GetMapping("/list")
    @Operation(summary = "事件分页查询")
    public Result<IPage<OilEvent>> queryPageList(EventReqVo eventReqVo,
                                                 @RequestParam(defaultValue = "1") Integer pageNo,
                                                 @RequestParam(defaultValue = "10") Integer pageSize,
                                                 HttpServletRequest req) {
        QueryWrapper<OilEvent> wrapper = BusinessUtil.getEventQueryWrapper(eventReqVo, new OilEvent());
        Page<OilEvent> page = new Page<>(pageNo, pageSize);
        IPage<OilEvent> pageList = oilEventService.page(page, wrapper);
        return Result.ok(pageList);
    }
    
    
    @GetMapping("/eventList")
    @Operation(summary = "首页事件列表")
    @ApiResponse(description = "增加字段 phraseDesc: 所属区域/阶段：xxx")
    public Result<List<OilEvent>> eventList(@RequestParam Map param) {
        List eventList = oilEventService.eventMergeList(param);
        currentTime = new Date();
        return Result.ok(eventList);
    }
    
    private static Date currentTime = new Date();
    
    @GetMapping("/eventPopup")
    @Operation(summary = "首页事件弹窗")
    @ApiResponse(ref = "/eventList")
    public Result<?> eventPopupList(Map param) {
        param.put("current", currentTime);
//        param.put("limit", 1);
        List eventList = oilEventService.eventMergeList(param);
        if (CollectionUtils.isNotEmpty(eventList)) {
            currentTime = new Date();
            return Result.ok(eventList.get(new Random().nextInt(eventList.size())));
        }
        return Result.ok();
    }
    
    @GetMapping("/getVideoUrl")
    public Result<?> getVideoUrl(@RequestParam String cameraCode) {
        String url = "https://prod-streaming-video-msn-com.akamaized.net/a8c412fa-f696-4ff2-9c76-e8ed9cdffe0f/604a87fc-e7bc-463e-8d56-cde7e661d690.mp4";
        return Result.ok(url);
    }
    
    /**
     * 添加
     */
    @AutoLog("OilEvent-添加")
    @PostMapping("/add")
    public Result<?> add(@RequestBody OilEvent oilEvent) {
        oilEventService.save(oilEvent);
        return Result.ok();
    }
    
    /**
     * 编辑
     */
    @AutoLog("OilEvent-编辑")
    @PutMapping("/edit")
    public Result<?> edit(@RequestBody OilEvent oilEvent) {
        oilEventService.updateById(oilEvent);
        return Result.ok();
    }
    
    @AutoLog("OilEvent-事件审核")
    @PostMapping("/audit")
    @Operation(summary = "事件审核")
    public Result<?> audit(@RequestBody OilEvent oilEvent) {
        LoginUser sysUser = SystemContextUtil.currentLoginUser();
        oilEvent.setAuditUser(sysUser.getUsername());
        oilEvent.setAuditTime(new Date());
        oilEventService.updateById(oilEvent);
        return Result.ok();
    }
    
    /**
     * 通过id删除
     */
    @AutoLog("OilEvent-通过id删除")
    @DeleteMapping("/delete")
    public Result<?> delete(@RequestParam String id) {
        oilEventService.removeById(id);
        return Result.ok();
    }
    
    /**
     * 批量删除
     */
    @AutoLog("OilEvent-批量删除")
    @DeleteMapping("/deleteBatch")
    public Result<?> deleteBatch(@RequestParam String ids) {
        this.oilEventService.removeByIds(Arrays.asList(ids.split(",")));
        return Result.ok();
    }
    
    /**
     * 通过id查询
     */
    @GetMapping("/queryById")
    public Result<?> queryById(@RequestParam String id) {
        OilEvent oilEvent = oilEventService.getById(id);
        return Result.ok(oilEvent);
    }
    
    /**
     * 导出excel
     */
    @RequestMapping("/exportXls")
    public void exportXls(HttpServletRequest request, HttpServletResponse response, OilEvent oilEvent) throws IOException {
        super.exportXls(request, response, oilEvent, "OilEvent");
    }
    
    @GetMapping("/exportZip")
    @Operation(summary = "导出zip文件（异步）")
    @ApiResponse(description = "0导出中， 1导出成功")
    public Result<Integer> exportZip(EventReqVo eventReqVo) throws IOException {
        LoginUser user = SystemContextUtil.currentLoginUser();
        String fileId = SystemConstant.OIL_EVENT_PREFFIX + user.getUsername();
        Map<String, String> fileMap = oiloutEventService.getFileMap();
        if (fileMap.containsKey(fileId)) {
            String filePath = fileMap.get(fileId);
            if (StringUtils.isNotBlank(filePath)) {
                return Result.ok(1);
            } else {
                return Result.ok(0);
            }
        }
        QueryWrapper<OilEvent> wrapper = BusinessUtil.getEventQueryWrapper(eventReqVo, new OilEvent());;
        List eventList = oilEventService.list(wrapper);
        oiloutEventService.exportZip(eventList, fileId);
        return Result.ok(0);
    }
    
    @GetMapping("/getExportFile")
    @Operation(summary = "下载导出的文件")
    public void getExportFile(HttpServletResponse response, String fileName, @RequestParam(defaultValue = "1") int type) throws IOException {
        LoginUser user = SystemContextUtil.currentLoginUser();
        String fileId = type == 1 ? SystemConstant.OIL_EVENT_PREFFIX + user.getUsername()
                : SystemConstant.OILOUT_EVENT_PREFFIX + user.getUsername();
        //查看文件是否导出成功
        Map<String, String> fileMap = oiloutEventService.getFileMap();
        String zipFile = fileMap.getOrDefault(fileId, "");
        if (StringUtils.isBlank(zipFile)) {
            return;
        }
        //文件下载
        HttpServletUtil.addDownloadHeader(response, fileName + SystemConstant.ZIP_SUFFIX);
        try(InputStream zipStream = new FileInputStream(zipFile)) {
            IoUtil.copy(zipStream, response.getOutputStream());
        } catch (Exception e) {
            log.error("导出失败", e);
            throw new RuntimeException("导出失败" + e.getMessage());
        } finally {
            fileMap.remove(fileId);
        }
        
    }
    
    /**
     * 通过excel导入数据
     */
    @PostMapping("/importExcel")
    public Result<?> importExcel(HttpServletRequest request) throws Exception {
        return super.importExcel(request, OilEvent.class);
    }
}