package com.zy.crm.manager.controller;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.mapper.EntityWrapper;
import com.baomidou.mybatisplus.mapper.Wrapper;
import com.baomidou.mybatisplus.plugins.Page;
import com.core.common.DateUtils;
import com.core.annotations.ManagerAuth;
import com.core.common.BaseRes;
import com.core.common.Cools;
import com.core.common.R;
import com.core.domain.KeyValueVo;
import com.core.exception.CoolException;
import com.zy.crm.common.model.SettleDto;
import com.zy.crm.common.web.BaseController;
import com.zy.crm.manager.entity.*;
import com.zy.crm.manager.service.*;
import com.zy.crm.system.entity.User;
import com.zy.crm.system.service.UserService;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.io.ClassPathResource;
import org.springframework.util.ClassUtils;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.text.SimpleDateFormat;
import java.util.*;

@RestController
public class PriSalesController extends BaseController {

    @Autowired
    private PriSalesService priSalesService;

    @Autowired
    private PriOnline2Service priOnline2Service;

    @Autowired
    private PlanService planService;

    @Autowired
    private PriQuoteBudgetService priQuoteBudgetService;

    @Autowired
    private UserService userService;

    @RequestMapping(value = "/priSales/{id}/auth")
    @ManagerAuth
    public R get(@PathVariable("id") String id) {
        PriSales priSales = priSalesService.selectById(String.valueOf(id));
        assert priSales != null;
        JSONObject resultObj = JSON.parseObject(JSON.toJSONString(priSales));
        // 步骤条相关
        resultObj.put("step", priSales.getSettle() == 4 ? 0 : priSales.getSettle() + 1);
        return R.ok().add(resultObj);
    }

    @RequestMapping(value = "/priSales/list/auth")
    @ManagerAuth
    public R list(@RequestParam(defaultValue = "1")Integer curr,
                  @RequestParam(defaultValue = "10")Integer limit,
                  @RequestParam(required = false)String orderByField,
                  @RequestParam(required = false)String orderByType,
                  @RequestParam(required = false)String condition,
                  @RequestParam Map<String, Object> param){
        EntityWrapper<PriSales> wrapper = new EntityWrapper<>();
        wrapper.setSqlSelect("id,title,create_time as createTime,settle,filepath,item_id as itemId,order_num as orderNum,template_name as templateName,user_id as userId,update_user_id as updateUserId,update_time as updateTime,in_order_num as inOrderNum,member_id as memberId");
//        wrapper.in("member_id", getUserRoleBelongsToUserId("allopen"));
        excludeTrash(param);
        convert(param, wrapper);
        allLike(PriSales.class, param.keySet(), wrapper, condition);
        wrapper.or().eq("member_id",getUserId());
        if (!Cools.isEmpty(orderByField)){wrapper.orderBy(humpToLine(orderByField), "asc".equals(orderByType));}
        return R.ok(priSalesService.selectPage(new Page<>(curr, limit), wrapper));
    }

    private <T> void convert(Map<String, Object> map, EntityWrapper<T> wrapper){
        boolean signUserId = false;
        boolean signDeptId = false;
        boolean signHostId = false;
        for (Map.Entry<String, Object> entry : map.entrySet()){
            if (entry.getKey().equals("dept_id")){
                signDeptId = true;
                if (String.valueOf(entry.getValue()).equals("19")){
                    signHostId = true;
                }
            }
        }
        for (Map.Entry<String, Object> entry : map.entrySet()){
            String val = String.valueOf(entry.getValue());
            if (val.contains(RANGE_TIME_LINK)){
                String[] dates = val.split(RANGE_TIME_LINK);
                wrapper.ge(entry.getKey(), DateUtils.convert(dates[0]));
                wrapper.le(entry.getKey(), DateUtils.convert(dates[1]));
            } else if (entry.getKey().equals("dept_id")){
                if (!val.equals("19")){
                    wrapper.eq(entry.getKey(), val);
                }
            } else if (entry.getKey().equals("user_id") && !signDeptId){
                signUserId = true;
                wrapper.eq(entry.getKey(), val);
            } else {
                wrapper.like(entry.getKey(), val);
            }
        }
        if (!signUserId && !signDeptId){
            wrapper.eq("user_id", getUserId());
        }
        if (signHostId){
            wrapper.or().eq("host_id",1);
        }
    }

    @RequestMapping(value = "/priSales/add/auth")
    @ManagerAuth(memo = "添加产品费用明细")
    public R add(@RequestBody Map<String,Object> map) {
        PriOnline2 priOnline2 = priOnline2Service.selectById(Long.parseLong(map.get("priOnlineId").toString()));

        PriSales priSales = new PriSales();
        priSales.setCreateTime(new Date());
        priSales.setTitle(map.get("title").toString());
        priSales.setTemplateName(priOnline2.getTemplateName());
        priSales.setSheetData(map.get("sheetData").toString());
        priSales.setPriOnline2Id(Long.parseLong(map.get("priOnlineId").toString()));
        priSales.setItemId(priOnline2.getItemId());
        priSales.setOrderNum(priOnline2.getOrderNum());
        SimpleDateFormat format = new SimpleDateFormat("yyyyMMddHHmmss");
        priSales.setInOrderNum(format.format(new Date()));

        //创建人员
        priSales.setUserId(getUserId());
        //更新人员
        priSales.setUpdateUserId(getUserId());
        //更新时间
        priSales.setUpdateTime(new Date());
        priSales.setDeptId(getDeptId());

        //业务员
        Plan plan = planService.selectById(priSales.getItemId());
        priSales.setMemberId(plan.getUserId());

        priSales.setSettle(0);

        User manager = userService.getDeptManager(getHostId(), getUser().getDeptId());        // 获取部门领导
        List<String> initNames = new ArrayList<>();
        initNames.add("创建产品费用明细模板");
        initNames.add("提交产品费用明细");
        initNames.add("部门经理审核");
        initNames.add("业务员确认");
        priSales.setSettleMsg(JSON.toJSONString(SettleDto.initSalesPricing(manager,getUser(),getUser(),initNames,3)));

        //设置项目流程
        plan.setStep(3);
        planService.updateById(plan);

        priSalesService.insert(priSales);
        return R.ok();
    }

    @RequestMapping(value = "/priSales/addOther/auth")
    @ManagerAuth(memo = "另存产品费用明细")
    public R addOther(@RequestBody Map<String,Object> map) {
        PriSales online = priSalesService.selectById(Long.parseLong(map.get("id").toString()));

        PriSales priSales = new PriSales();
        priSales.setCreateTime(new Date());
        priSales.setTitle(map.get("title").toString());
        priSales.setTemplateName(online.getTemplateName());
        priSales.setSheetData(map.get("sheetData").toString());
        priSales.setItemId(online.getItemId());
        priSales.setOrderNum(online.getOrderNum());

        SimpleDateFormat format = new SimpleDateFormat("yyyyMMddHHmmss");
        priSales.setInOrderNum(format.format(new Date()));

        //创建人员
        priSales.setUserId(getUserId());
        //更新人员
        priSales.setUpdateUserId(getUserId());
        //更新时间
        priSales.setUpdateTime(new Date());

        //业务员
        Plan plan = planService.selectById(priSales.getItemId());
        priSales.setMemberId(plan.getUserId());

        priSalesService.insert(priSales);
        return R.ok();
    }

    @RequestMapping(value = "/priSales/update/auth")
    @ManagerAuth(memo = "更新产品费用明细")
    public R update(@RequestBody Map<String,Object> map){
        PriSales priSales = priSalesService.selectById(Long.parseLong(map.get("id").toString()));
        priSales.setTitle(map.get("title").toString());
        priSales.setSheetData(map.get("sheetData").toString());
        //更新时间
        priSales.setUpdateTime(new Date());
        //更新人员
        priSales.setUpdateUserId(getUserId());
        priSalesService.updateById(priSales);
        return R.ok();
    }

    @RequestMapping(value = "/priSales/updateForm/auth")
    @ManagerAuth(memo = "更新产品费用明细模板名和标题")
    public R updateForm(Long id,String templateName,String title){
        PriSales priSales = priSalesService.selectById(id);
        priSales.setTemplateName(templateName);
        priSales.setTitle(title);
        //更新时间
        priSales.setUpdateTime(new Date());
        //更新人员
        priSales.setUpdateUserId(getUserId());
        priSalesService.updateById(priSales);
        return R.ok();
    }

    @RequestMapping(value = "/priSales/delete/auth")
    @ManagerAuth(memo = "删除产品费用明细")
    public R delete(Long[] ids){
        if (Cools.isEmpty(ids)){
            return R.error();
        }
        for (Long id : ids){
            int  priQuoteBudget= priQuoteBudgetService.selectCount(new EntityWrapper<PriQuoteBudget>().eq("pri_sales_id", id));
            if (priQuoteBudget!=0){
                return R.error("存在关联的报价预算单，禁止删除！！！");
            }
        }
        priSalesService.deleteBatchIds(Arrays.asList(ids));
        return R.ok();
    }

    @RequestMapping(value = "/priSales/export/auth")
    @ManagerAuth
    public R export(@RequestBody JSONObject param){
        EntityWrapper<PriSales> wrapper = new EntityWrapper<>();
        List<String> fields = JSONObject.parseArray(param.getJSONArray("fields").toJSONString(), String.class);
        Map<String, Object> map = excludeTrash(param.getJSONObject("priOnline"));
        convert(map, wrapper);
        List<PriSales> list = priSalesService.selectList(wrapper);
        return R.ok(exportSupport(list, fields));
    }

    @RequestMapping(value = "/priSalesQuery/auth")
    @ManagerAuth
    public R query(String condition) {
        EntityWrapper<PriSales> wrapper = new EntityWrapper<>();
        wrapper.like("id", condition);
        wrapper.in("user_id", getUserRoleBelongsToUserId("allopen"));
        Page<PriSales> page = priSalesService.selectPage(new Page<>(0, 10), wrapper);
        List<Map<String, Object>> result = new ArrayList<>();
        for (PriSales priSales : page.getRecords()){
            Map<String, Object> map = new HashMap<>();
            map.put("id", priSales.getId());
            map.put("value", priSales.getInOrderNum() + "/" + priSales.getPlanId$() + "/" + priSales.getMemberId$()+"/" + priSales.getTemplateName());
            result.add(map);
        }
        return R.ok(result);
    }

    @RequestMapping(value = "/priSales/check/column/auth")
    @ManagerAuth
    public R query(@RequestBody JSONObject param) {
        Wrapper<PriSales> wrapper = new EntityWrapper<PriSales>().eq(humpToLine(String.valueOf(param.get("key"))), param.get("val"));
        if (null != priSalesService.selectOne(wrapper)){
            return R.parse(BaseRes.REPEAT).add(getComment(PriSales.class, String.valueOf(param.get("key"))));
        }
        return R.ok();
    }

    @RequestMapping("/priSales/all/get/kv")
    @ManagerAuth
    public R getDataKV(@RequestParam(required = false) String condition) {
        List<KeyValueVo> vos = new ArrayList<>();
        Wrapper<PriSales> wrapper = new EntityWrapper<PriSales>().andNew().like("id", condition).orderBy("create_time", false);
        priSalesService.selectPage(new Page<>(1, 30), wrapper).getRecords().forEach(item -> vos.add(new KeyValueVo(String.valueOf(item.getId()), item.getId())));
        return R.ok().add(vos);
    }

    @PostMapping("/priSales/excel/downfile")
    @ManagerAuth
    public R downExcelFile(@RequestBody Map<String,Object> map) {
        SimpleDateFormat format = new SimpleDateFormat("yyyyMMdd_HHmmss");
        String path =  ClassUtils.getDefaultClassLoader().getResource("excel").getPath();
        String filename = format.format(new Date()) + ".xlsx";
        String filepath = path + "/" + filename;
        exportLuckySheetXlsx(path,"/" + filename,map.get("exceldata").toString(),Integer.parseInt(map.get("endRow").toString()));

        return R.ok("/priSales/excel/" + filename);
    }

    @GetMapping("/priSales/excel/{path}")
    public void download(@PathVariable String path,HttpServletResponse response) {
        try {
            ClassPathResource pathResource = new ClassPathResource("excel/" + path);
            File file = pathResource.getFile();
            InputStream inputStream = pathResource.getInputStream();
            //输出文件
            InputStream fis = new BufferedInputStream(inputStream);
            byte[] buffer = new byte[fis.available()];
            fis.read(buffer);
            fis.close();
            response.reset();

            //获取文件的名字再浏览器下载页面
            String name = file.getName();
            response.addHeader("Content-Disposition", "attachment;filename=" + new String(name.getBytes(), "iso-8859-1"));
            response.addHeader("Content-Length", "" + file.length());
            OutputStream out = new BufferedOutputStream(response.getOutputStream());
            response.setContentType("application/octet-stream");
            out.write(buffer);
            out.flush();
            out.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public void exportLuckySheetXlsx(String newFileDir,String newFileName, String excelData,int excelDataEndRow) {
        //解析对象，可以参照官方文档:https://mengshukeji.github.io/LuckysheetDocs/zh/guide/#%E6%95%B4%E4%BD%93%E7%BB%93%E6%9E%84
        JSONArray jsonArray = (JSONArray) JSONObject.parse(excelData);
        //如果只有一个sheet那就是get(0),有多个那就对应取下标
        JSONObject jsonObject = (JSONObject) jsonArray.get(0);
        JSONArray jsonObjectList = jsonObject.getJSONArray("celldata");
        //excel模板路径
        String filePath =  ClassUtils.getDefaultClassLoader().getResource("excel").getPath() + "/priSalesTemplate.xlsx";
//        String filePath = "/Users/ouyang/Downloads/uploadTestProductFile/生产日报表.xlsx";
        File file = new File(filePath);
        FileInputStream in = null;
        try {
            in = new FileInputStream(file);
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        }
        //读取excel模板
        XSSFWorkbook wb = null;
        try {
            wb = new XSSFWorkbook(in);
        } catch (IOException e) {
            e.printStackTrace();
        }
        //读取了模板内所有sheet内容
        XSSFSheet sheet = wb.getSheetAt(0);
        //如果这行没有了，整个公式都不会有自动计算的效果的
        sheet.setForceFormulaRecalculation(true);

        //表单数据结束行，默认是11
        int endRow = 11;


        if (excelDataEndRow != -1) {
            //需要自行寻找数据行index
            //此循环主要目的是为了找到表单数据结束行位置，从而进行下一步精准的数据填充
            for (int index = 0; index < jsonObjectList.size(); index++) {
                com.alibaba.fastjson.JSONObject object = jsonObjectList.getJSONObject(index);
                String str_ = (int) object.get("r") + "_" + object.get("c") + "=" + ((com.alibaba.fastjson.JSONObject) object.get("v")).get("v") + "\n";
                JSONObject jsonObjectValue = ((com.alibaba.fastjson.JSONObject) object.get("v"));
                //判断是否超过4行，4行之后才是真正数据
                if (Integer.parseInt(object.get("r").toString()) < 4) {
                    continue;
                }

                //超过4行，开始统计数据

                String value = "";
                if (jsonObjectValue != null && jsonObjectValue.get("v") != null){
                    //获取单元格值
                    value = jsonObjectValue.get("v") + "";
                    if (value.equals("合计：")) {
                        //表单结束
                        Object row = jsonObjectValue.getOrDefault("row",null);
                        if (row == null) {
                            System.out.println("未能成功计算到数据行结束");
                            continue;
                        }
                        endRow = Integer.parseInt(jsonObjectValue.get("row").toString()) - 1;
                        break;
                    }
                }
            }
        }else {
            //前端提交数据行结束位置
            endRow = excelDataEndRow;
        }

        //此循环主要目的是进行数据的填充
        for (int index = 0; index < jsonObjectList.size(); index++) {
            com.alibaba.fastjson.JSONObject object = jsonObjectList.getJSONObject(index);
            String str_ = (int) object.get("r") + "_" + object.get("c") + "=" + ((com.alibaba.fastjson.JSONObject) object.get("v")).get("v") + "\n";
            JSONObject jsonObjectValue = ((com.alibaba.fastjson.JSONObject) object.get("v"));

            //判断是否超过4行，或超过数据最大行
            if (Integer.parseInt(object.get("r").toString()) < 4 || Integer.parseInt(object.get("r").toString()) > endRow) {
                continue;
            }

            //开始统计数据

            String value = "";
            switch (Integer.parseInt(object.get("c").toString())) {
                case 1:
                case 2:
                case 3:
                case 4:
                case 5:
                case 6:
                    if (jsonObjectValue != null && jsonObjectValue.get("v") != null){
                        //获取单元格值
                        value = jsonObjectValue.get("v").toString();
                    }
                    if (sheet.getRow((int) object.get("r")) !=null && sheet.getRow((int) object.get("r")).getCell((int) object.get("c")) != null){
                        if (!value.equals("")) {
                            sheet.getRow((int) object.get("r")).getCell((int) object.get("c")).setCellValue(value);
                        }
                    }else{
                        System.out.println("错误的=" + index + ">>>" + str_);
                    }
                    break;
                default:
                    continue;
            }
        }

        // 判断路径是否存在
        File dir = new File(newFileDir);
        if (!dir.exists()) {
            dir.mkdirs();
        }
        //修改模板内容导出新模板
        FileOutputStream out = null;
        try {
            out = new FileOutputStream(newFileDir + newFileName);
            wb.write(out);
            out.close();
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }


    @PostMapping(value = "/priSales/approval/auth")
    @ManagerAuth
    public R approvalBusinessTrip(@RequestParam Long priSalesId,
                                  @RequestParam(required = false) Long plannerId){
        PriSales priSales = priSalesService.selectById(priSalesId);
        assert priSales != null;
        Date now = new Date();
        switch (priSales.getSettle()) {
            case 0:
                User user2 = userService.selectById(priSales.getUserId());
                User manager1 = userService.getDeptManager(getHostId(), user2.getDeptId());

                if (Cools.isEmpty(getUser())) {
                    return R.error("抱歉，您没有提交的权限");
                }
                if (!getUserId().equals(getUser().getId())) {
                    return R.error("抱歉，您没有提交的权限");
                }
                // 修改 settle 步骤数据
                List<SettleDto> list1 = JSON.parseArray(priSales.getSettleMsg(), SettleDto.class);
                for (SettleDto dto : list1) {
                    switch (dto.getStep()) {
                        case 0:
                            dto.setCurr(Boolean.FALSE);
                            break;
                        case 1:
                            dto.setCurr(Boolean.TRUE);
                            dto.setMsg(getUser().getNickname() + "提交完成");
                            dto.setTime(DateUtils.convert(now));
                            break;
                        default:
                            break;
                    }
                }
                priSales.setSettleMsg(JSON.toJSONString(list1));
                // 修改规划单状态
                priSales.setSettle(1);
                priSales.setUpdateUserId(getUserId());
                priSales.setUpdateTime(now);
                priSales.setMemberId(manager1.getId());

                if (!priSalesService.updateById(priSales)) {
                    throw new CoolException("提交失败，请联系管理员");
                }
                break;
            case 1:
                // 本部门经理审核
                User user = userService.selectById(priSales.getUserId());
                User manager = userService.getDeptManager(getHostId(), user.getDeptId());
                if (manager.getId().equals(getUserId())) {

                    // 修改 settle 步骤数据
                    List<SettleDto> list = JSON.parseArray(priSales.getSettleMsg(), SettleDto.class);
                    for (SettleDto dto : list) {
                        switch (dto.getStep()) {
                            case 1:
                                dto.setCurr(Boolean.FALSE);
                                break;
                            case 2:
                                dto.setCurr(Boolean.TRUE);
                                dto.setMsg("部门经理" + manager.getNickname() + "审批通过");
                                dto.setTime(DateUtils.convert(now));
                                break;
                            default:
                                break;
                        }
                    }
                    priSales.setSettleMsg(JSON.toJSONString(list));

                    // 修改规划单状态
                    priSales.setSettle(2);  // 申请通过
                    priSales.setUpdateTime(now);
                    if (!priSalesService.updateById(priSales)) {
                        throw new CoolException("审核失败，请联系管理员");
                    }
                } else {
                    return R.error("抱歉，您没有审核的权限！！！");
                }
                break;
            case 2:
                // 业务员
                User salesman0 = userService.selectById(priSales.getUserId());
                if (!getUserId().equals(salesman0.getId())) {
                    return R.error("抱歉，您无需确认！！！");
                }
                // 修改 settle 步骤数据
                List<SettleDto> list2 = JSON.parseArray(priSales.getSettleMsg(), SettleDto.class);
                for (SettleDto dto : list2) {
                    switch (dto.getStep()) {
                        case 2:
                            dto.setCurr(Boolean.TRUE);
                            break;
                        case 3:
                            dto.setCurr(Boolean.TRUE);
                            dto.setMsg("业务员" + salesman0.getNickname() + "以确认");
                            dto.setTime(DateUtils.convert(new Date()));
                            break;
                        default:
                            break;
                    }
                }
                priSales.setSettleMsg(JSON.toJSONString(list2));
                // 修改规划单状态
                priSales.setSettle(3);  // 审批通过
                priSales.setUpdateTime(new Date());

                if (!priSalesService.updateById(priSales)) {
                    throw new CoolException("确认失败，请联系管理员");
                }
                break;
            default:
                return R.error();
        }
        return R.ok("审批成功");
    }

}