package com.sercurityControl.proteam.controller;

import com.alibaba.fastjson2.JSONObject;
import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import com.securityControl.common.core.utils.StringUtils;
import com.securityControl.common.core.utils.poi.ExcelUtil;
import com.securityControl.common.log.annotation.Log;
import com.securityControl.common.log.enums.BusinessType;
import com.securityControl.common.log.enums.OperationType;
import com.securityControl.common.security.utils.SecurityUtils;
import com.sercurityControl.proteam.domain.vo.ReturnCodeEntity;
import com.sercurityControl.proteam.domain.vo.SupeConsUserVo;
import com.sercurityControl.proteam.service.ManageService;
import com.sercurityControl.proteam.util.BaseController;
import com.sercurityControl.proteam.util.DateTimeHelper;
import com.sercurityControl.proteam.util.ImportExcelHelper;
import com.sercurityControl.proteam.util.ResultModel;
import lombok.extern.slf4j.Slf4j;
import org.apache.poi.util.IOUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;
import org.springframework.web.multipart.commons.CommonsMultipartResolver;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 监理单位 、施工单位维护信息层
 */
@RestController
@Slf4j
@RequestMapping("/sys/manage/")
public class ManageController extends BaseController {

    @Autowired
    private ManageService service;

    /**
     * 分页查询
     *
     * @param supeConsUserVo
     * @return
     */
    @PostMapping("getPageInfo")
    @Log(title = "单位管理", menu = "施工单位管理->监理单位管理", businessType = BusinessType.QUERY, grade = OperationType.QUERY_BUSINESS, details = "施工单位-监理单位")
    public Map<String, Object> getPageInfo(SupeConsUserVo supeConsUserVo) {
        PageHelper.startPage(supeConsUserVo.getPage(), supeConsUserVo.getLimit());
        List<SupeConsUserVo> deviceList = service.getPageInfo(supeConsUserVo);
        PageInfo<SupeConsUserVo> pageInfo = new PageInfo<SupeConsUserVo>(deviceList);
        System.out.println("输出分页后的数据：" + pageInfo.getList());
        Map<String, Object> map = new HashMap<String, Object>(16);
        map.put("code", 200);
        map.put("msg", "");
        map.put("count", pageInfo.getTotal());
        map.put("curr", supeConsUserVo.getPage());
        map.put("limit", supeConsUserVo.getLimit());
        map.put("data", pageInfo.getList());
        return map;
    }

    @PostMapping("insertData")
    @Log(title = "单位管理", menu = "施工单位管理->监理单位管理", businessType = BusinessType.INSERT, details = "施工单位-监理单位", grade = OperationType.ADD_BUSINESS)
    public ReturnCodeEntity insertData(SupeConsUserVo supeConsUserVo) {
        ReturnCodeEntity entity = new ReturnCodeEntity();
        try {
            entity = service.insertData(supeConsUserVo);
        } catch (Exception e) {
            entity.setCode("202");
            entity.setMsg("解析异常,请联系管理员");
        }
        return entity;
    }

    @PostMapping("updateData")
    @Log(title = "单位管理", menu = "施工单位管理->监理单位管理", businessType = BusinessType.UPDATE, details = "施工单位-监理单位", grade = OperationType.UPDATE_BUSINESS)
    public ReturnCodeEntity updateData(SupeConsUserVo deviceVo) {
        ReturnCodeEntity entity = new ReturnCodeEntity();
        try {
            entity = service.updateData(deviceVo);
        } catch (Exception e) {
            entity.setCode("202");
            entity.setMsg("解析异常,请联系管理员");
        }
        return entity;
    }

    /**
     * 删除角色
     *
     * @param
     * @return
     */
    @PostMapping("deleteData")
    @Log(title = "单位管理", menu = "施工单位管理->监理单位管理", businessType = BusinessType.DELETE, details = "删除单位", grade = OperationType.DELETE_BUSINESS)
    public ReturnCodeEntity deleteData(String keyId) {
        ReturnCodeEntity entity = new ReturnCodeEntity();
        try {
            entity = service.deleteData(keyId);
        } catch (Exception e) {
            entity.setCode("202");
            entity.setMsg("解析异常,请联系管理员");
        }
        return entity;
    }

    /**
     * 查看详情
     *
     * @param
     * @return
     */
    @PostMapping("getDataDetail")
    @Log(title = "单位管理", menu = "施工单位管理->监理单位管理", businessType = BusinessType.QUERY, details = "查看单位详情", grade = OperationType.QUERY_BUSINESS)
    public SupeConsUserVo getDataDetail(String keyId) {
        SupeConsUserVo supeConsUserVo = null;
        try {
            supeConsUserVo = service.getDataDetail(keyId);
            return supeConsUserVo;
        } catch (Exception e) {
            log.error(e.toString(), e);
        }
        return supeConsUserVo;
    }

    /**
     * 导出
     *
     * @param response
     * @param supeConsUserVo
     */
    @GetMapping("exportDevice")
    public void exportDevice(HttpServletResponse response, SupeConsUserVo supeConsUserVo) {
        List<SupeConsUserVo> studentsList = service.getPageInfo(supeConsUserVo);
        ExcelUtil<SupeConsUserVo> util = new ExcelUtil<SupeConsUserVo>(SupeConsUserVo.class);
        util.exportExcel(response, studentsList, "施工单位");
    }


    /**
     * 下拉选查询
     *
     * @param
     * @return
     */
    @PostMapping("getSelected")
    public List<SupeConsUserVo> getSelected(SupeConsUserVo supeConsUserVo) {
        try {
            List<SupeConsUserVo> studentsList = service.getPageInfo(supeConsUserVo);
            return studentsList;
        } catch (Exception e) {
            log.error(e.toString(), e);
        }
        return null;
    }

    /**
     * @param request
     * @param response
     * @description 监理单位/施工单位 模板下载
     * @author cwchen
     * @date 2024/1/23 11:07
     */
    @GetMapping("downLoadExcelModel")
    public void downLoadExcelModel(HttpServletRequest request, HttpServletResponse response) {
        InputStream inputStream = null;
        ServletOutputStream servletOutputStream = null;
        try {
            String path = "download/" + "unit_model.xlsx";
            inputStream = this.getClass().getClassLoader().getResourceAsStream(path);
            response.setContentType("application/vnd.ms-excel");
            response.addHeader("Cache-Control", "no-cache, no-store, must-revalidate");
            response.addHeader("charset", "utf-8");
            response.addHeader("Pragma", "no-cache");
            String encodeName = URLEncoder.encode("gt_model.xlsx", StandardCharsets.UTF_8.toString());
            response.setHeader("Content-Disposition", "attachment; filename=\"" + encodeName + "\"; filename*=utf-8''" + encodeName);
            servletOutputStream = response.getOutputStream();
            IOUtils.copy(inputStream, servletOutputStream);
            response.flushBuffer();
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            try {
                if (servletOutputStream != null) {
                    servletOutputStream.close();
                }
                if (inputStream != null) {
                    inputStream.close();
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }

    @PostMapping("/importExcel")
    @Log(title = "监理/施工单位数据导入", menu = "监理/施工单位数据导入", businessType = BusinessType.IMPORT, details = "监理/施工单位数据导入", grade = OperationType.IMPORT_BUSINESS)
    public void importExcel(MultipartFile file, HttpServletRequest request, HttpServletResponse response) {
        CommonsMultipartResolver multipartResolver = new CommonsMultipartResolver();
        MultipartHttpServletRequest multiReq = multipartResolver.resolveMultipart(request);
        // 1.施工单位 2.监理单位
        String type = multiReq.getParameter("type");
        Long userId = SecurityUtils.getLoginUser().getUserid();
        String nowTime = DateTimeHelper.getNowTime();
        try {
            List<JSONObject> lstObj = (List<JSONObject>) ImportExcelHelper.readExcel(file, SupeConsUserVo.class);
            List<SupeConsUserVo> list = new ArrayList<>();
            List<JSONObject> lstError = new ArrayList<>();
            if (lstObj != null && lstObj.size() > 0) {
                list = new ArrayList<>();
                lstError = new ArrayList<>();
                for (JSONObject obj : lstObj) {
                    if (StringUtils.isBlank(obj.getString("name"))) {
                        JSONObject error = new JSONObject();
                        error.put("errorRow", obj.getString("rowNo"));
                        error.put("errorLine", "单位名称");
                        error.put("errorData", "");
                        error.put("errorMsg", "单位名称不能为空");
                        lstError.add(error);
                        continue;
                    }
                    if (StringUtils.isNotBlank(obj.getString("name")) && obj.getString("name").length() > 200) {
                        JSONObject error = new JSONObject();
                        error.put("errorRow", obj.getString("rowNo"));
                        error.put("errorLine", "单位名称");
                        error.put("errorData", obj.getString("name"));
                        error.put("errorMsg", "单位名称超过200个字符");
                        lstError.add(error);
                        continue;
                    }
                    if (StringUtils.isBlank(obj.getString("code"))) {
                        JSONObject error = new JSONObject();
                        error.put("errorRow", obj.getString("rowNo"));
                        error.put("errorLine", "单位编码");
                        error.put("errorData", "");
                        error.put("errorMsg", "单位编码不能为空");
                        lstError.add(error);
                        continue;
                    }
                    if (StringUtils.isNotBlank(obj.getString("code")) && obj.getString("code").length() > 200) {
                        JSONObject error = new JSONObject();
                        error.put("errorRow", obj.getString("rowNo"));
                        error.put("errorLine", "单位编码");
                        error.put("errorData", obj.getString("code"));
                        error.put("errorMsg", "单位编码超过200个字符");
                        lstError.add(error);
                        continue;
                    }
                    if (StringUtils.isNotBlank(obj.getString("remarks")) && obj.getString("remarks").length() > 200) {
                        JSONObject error = new JSONObject();
                        error.put("errorRow", obj.getString("rowNo"));
                        error.put("errorLine", "备注");
                        error.put("errorData", obj.getString("remarks"));
                        error.put("errorMsg", "备注超过200个字符");
                        lstError.add(error);
                        continue;
                    }
                    SupeConsUserVo vo = new SupeConsUserVo();
                    vo.setName(obj.getString("name"));
                    vo.setCode(obj.getString("code"));
                    vo.setRemarks(obj.getString("remarks"));
                    vo.setType(type);
                    vo.setCreateTime(nowTime);
                    vo.setCreateUser(userId + "");
                    list.add(vo);
                }
            }
            ResultModel resultModel = new ResultModel();
            if (lstError != null && lstError.size() > 0) {
                resultModel.setSuccess(false);
                resultModel.setStatus(200);
                resultModel.setMsg("导入成功");
                resultModel.setData(lstError);
            } else if (lstError.size() == 0) {
                service.addSuperConsData(list);
                resultModel.setSuccess(true);
                resultModel.setStatus(200);
                resultModel.setMsg("导入成功");
                resultModel.setData(null);
            }
            sendJson(response, resultModel);
        } catch (Exception ex) {
            ex.printStackTrace();
            resultModel = new ResultModel();
            resultModel.setSuccess(false);
            resultModel.setStatus(1000);
            resultModel.setMsg("服务异常");
            try {
                sendJson(response, resultModel);
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
}