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.annotations.ManagerAuth;
import com.core.common.BaseRes;
import com.core.common.Cools;
import com.core.common.DateUtils;
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 PriQuoteBudgetController extends BaseController {

    @Autowired
    private PriQuoteBudgetService priQuoteBudgetService;

    @Autowired
    private PlanService planService;

    @Autowired
    private PriSalesService priSalesService;

    @Autowired
    private PriQuoteService priQuoteService;

    @Autowired
    private UserService userService;

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

    @RequestMapping(value = "/priQuoteBudget/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<PriQuoteBudget> 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());
        excludeTrash(param);
        convert(param, wrapper);
        allLike(PriQuoteBudget.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(priQuoteBudgetService.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 = "/priQuoteBudget/add/auth")
    @ManagerAuth(memo = "添加报价预算单")
    public R add(@RequestBody Map<String,Object> map) {
        PriSales priSales = priSalesService.selectById(Long.parseLong(map.get("priSalesId").toString()));

        PriQuoteBudget priQuoteBudget = new PriQuoteBudget();
        priQuoteBudget.setCreateTime(new Date());
        priQuoteBudget.setTitle(map.get("title").toString());
        priQuoteBudget.setTemplateName(priSales.getTemplateName());
        priQuoteBudget.setSheetData(map.get("sheetData").toString());
        priQuoteBudget.setPriSalesId(Long.parseLong(map.get("priSalesId").toString()));
        priQuoteBudget.setItemId(priSales.getItemId());
        //保存产品费用明细订单号
        priQuoteBudget.setOrderNum(priSales.getInOrderNum());
        SimpleDateFormat format = new SimpleDateFormat("yyyyMMddHHmmss");
        priQuoteBudget.setInOrderNum(format.format(new Date()));

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

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

        priQuoteBudget.setSettle(0);

        User manager = userService.getDeptManager(getHostId(), getUser().getDeptId());        // 获取部门领导
        List<String> initNames = new ArrayList<>();
        initNames.add("创建报价预算单模板");
        initNames.add("提交报价预算单");
        initNames.add("部门经理审核");
        initNames.add("业务员确认");
        priQuoteBudget.setSettleMsg(JSON.toJSONString(SettleDto.initSalesPricing(manager,getUser(),getUser(),initNames,3)));

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

        priQuoteBudgetService.insert(priQuoteBudget);
        return R.ok();
    }

    @RequestMapping(value = "/priQuoteBudget/addOther/auth")
    @ManagerAuth(memo = "另存报价预算单")
    public R addOther(@RequestBody Map<String,Object> map) {
        PriQuoteBudget quote = priQuoteBudgetService.selectById(Long.parseLong(map.get("id").toString()));

        PriQuoteBudget priQuoteBudget = new PriQuoteBudget();
        priQuoteBudget.setCreateTime(new Date());
        priQuoteBudget.setTitle(map.get("title").toString());
        priQuoteBudget.setTemplateName(quote.getTemplateName());
        priQuoteBudget.setSheetData(map.get("sheetData").toString());
        priQuoteBudget.setItemId(quote.getItemId());
        priQuoteBudget.setOrderNum(quote.getOrderNum());

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

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

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

        priQuoteBudgetService.insert(priQuoteBudget);
        return R.ok();
    }

    @RequestMapping(value = "/priQuoteBudget/update/auth")
    @ManagerAuth(memo = "更新报价预算单")
    public R update(@RequestBody Map<String,Object> map){
        PriQuoteBudget priQuoteBudget = priQuoteBudgetService.selectById(Long.parseLong(map.get("id").toString()));
        priQuoteBudget.setTitle(map.get("title").toString());
        priQuoteBudget.setSheetData(map.get("sheetData").toString());
        //更新时间
        priQuoteBudget.setUpdateTime(new Date());
        //更新人员
        priQuoteBudget.setUpdateUserId(getUserId());
        priQuoteBudgetService.updateById(priQuoteBudget);
        return R.ok();
    }

    @RequestMapping(value = "/priQuoteBudget/updateForm/auth")
    @ManagerAuth(memo = "更新报价预算单")
    public R updateForm(Long id,String templateName,String title){
        PriQuoteBudget priQuoteBudget = priQuoteBudgetService.selectById(id);
        priQuoteBudget.setTemplateName(templateName);
        priQuoteBudget.setTitle(title);
        //更新时间
        priQuoteBudget.setUpdateTime(new Date());
        //更新人员
        priQuoteBudget.setUpdateUserId(getUserId());
        priQuoteBudgetService.updateById(priQuoteBudget);
        return R.ok();
    }

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

    @RequestMapping(value = "/priQuoteBudget/export/auth")
    @ManagerAuth
    public R export(@RequestBody JSONObject param){
        EntityWrapper<PriQuoteBudget> 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<PriQuoteBudget> list = priQuoteBudgetService.selectList(wrapper);
        return R.ok(exportSupport(list, fields));
    }

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

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

    @RequestMapping("/priQuoteBudget/all/get/kv")
    @ManagerAuth
    public R getDataKV(@RequestParam(required = false) String condition) {
        List<KeyValueVo> vos = new ArrayList<>();
        Wrapper<PriQuoteBudget> wrapper = new EntityWrapper<PriQuoteBudget>().andNew().like("id", condition).orderBy("create_time", false);
        priQuoteBudgetService.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("/priQuoteBudget/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());

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

    @GetMapping("/priQuoteBudget/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) {
        //解析对象，可以参照官方文档: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() + "/priQuoteBudgetTemplate.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);

        //此循环主要目的是进行数据的填充
        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"));

            String value = "";
            if (jsonObjectValue != null && jsonObjectValue.get("v") != null){
                //获取单元格值
                value = jsonObjectValue.get("v").toString();
            }

            int c = Integer.parseInt(object.get("c").toString());
            switch (Integer.parseInt(object.get("r").toString())) {
                //第一行
                case 1:
                    if (c == 1) {
                        //填充客户类别
                        sheet.getRow((int) object.get("r")).getCell((int) object.get("c")).setCellValue(value);
                    }
                    break;
                case 2:
                    if (c == 6) {
                        //目标销售价
                        sheet.getRow((int) object.get("r")).getCell((int) object.get("c")).setCellValue(value);
                    }
                    break;
                case 4:
                    if (c == 1 || c == 3) {
                        //填充产品费用 || 填充费用明细
                        sheet.getRow((int) object.get("r")).getCell((int) object.get("c")).setCellValue(value);
                    }
                    break;
                case 5://运输费用 || 填充费用明细
                case 6://安装费用 || 填充费用明细
                    if (c == 2 || c == 3) {
                        sheet.getRow((int) object.get("r")).getCell((int) object.get("c")).setCellValue(value);
                    }
                    break;
                case 7:
                    if (c == 3) {
                        //项目管理费
                        sheet.getRow((int) object.get("r")).getCell((int) object.get("c")).setCellValue(value);
                    }
                    break;
                case 8://其他费用
                case 9://营销费用
                    if (c == 1 || c == 3) {
                        sheet.getRow((int) object.get("r")).getCell((int) object.get("c")).setCellValue(value);
                    }
                    break;
                case 13://付款周期
                case 16://付款周期
                case 19://付款周期
                case 22://付款周期
                    if (c == 3 || c == 4) {
                        sheet.getRow((int) object.get("r")).getCell((int) object.get("c")).setCellValue(value);
                    }
                    break;
            }

        }

        // 判断路径是否存在
        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 = "/priQuoteBudget/approval/auth")
    @ManagerAuth
    public R approvalBusinessTrip(@RequestParam Long priQuoteBudgetId,
                                  @RequestParam(required = false) Long plannerId){
        PriQuoteBudget priQuoteBudget = priQuoteBudgetService.selectById(priQuoteBudgetId);
        assert priQuoteBudget != null;
        Date now = new Date();
        switch (priQuoteBudget.getSettle()) {
            case 0:
                User user2 = userService.selectById(priQuoteBudget.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(priQuoteBudget.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;
                    }
                }
                priQuoteBudget.setSettleMsg(JSON.toJSONString(list1));
                // 修改规划单状态
                priQuoteBudget.setSettle(1);
                priQuoteBudget.setUpdateUserId(getUserId());
                priQuoteBudget.setUpdateTime(now);
                priQuoteBudget.setMemberId(manager1.getId());

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

                    // 修改 settle 步骤数据
                    List<SettleDto> list = JSON.parseArray(priQuoteBudget.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;
                        }
                    }
                    priQuoteBudget.setSettleMsg(JSON.toJSONString(list));

                    // 修改规划单状态
                    priQuoteBudget.setSettle(2);  // 申请通过
                    priQuoteBudget.setUpdateTime(now);
                    if (!priQuoteBudgetService.updateById(priQuoteBudget)) {
                        throw new CoolException("审核失败，请联系管理员");
                    }
                } else {
                    return R.error("抱歉，您没有审核的权限！！！");
                }
                break;
            case 2:
                // 业务员
                User salesman0 = userService.selectById(priQuoteBudget.getUserId());
                if (!getUserId().equals(salesman0.getId())) {
                    return R.error("抱歉，您无需确认！！！");
                }
                // 修改 settle 步骤数据
                List<SettleDto> list2 = JSON.parseArray(priQuoteBudget.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;
                    }
                }
                priQuoteBudget.setSettleMsg(JSON.toJSONString(list2));
                // 修改规划单状态
                priQuoteBudget.setSettle(3);  // 审批通过
                priQuoteBudget.setUpdateTime(new Date());

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

}