package com.zy.crm.manager.controller;

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.zy.crm.common.web.BaseController;
import com.zy.crm.manager.entity.Item;
import com.zy.crm.manager.entity.PriQuoteBudget;
import com.zy.crm.manager.entity.PriSales;
import com.zy.crm.manager.service.ItemService;
import com.zy.crm.manager.service.PriQuoteBudgetService;
import com.zy.crm.manager.service.PriSalesService;
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 ItemService itemService;

    @Autowired
    private PriSalesService priSalesService;

    @RequestMapping(value = "/priQuoteBudget/{id}/auth")
    @ManagerAuth
    public R get(@PathVariable("id") String id) {
        return R.ok(priQuoteBudgetService.selectById(String.valueOf(id)));
    }

    @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,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);
        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;
        for (Map.Entry<String, Object> entry : map.entrySet()){
            if (entry.getKey().equals("dept_id")){
                signDeptId = 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());
        }
    }

    @RequestMapping(value = "/priQuoteBudget/add/auth")
    @ManagerAuth
    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.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());

        //业务员
        Item item = itemService.selectById(priQuoteBudget.getItemId());
        priQuoteBudget.setMemberId(item.getMember());

        //设置项目流程
        item.setStep(4);
        itemService.updateById(item);

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

    @RequestMapping(value = "/priQuoteBudget/addOther/auth")
    @ManagerAuth
    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());

        //业务员
        Item item = itemService.selectById(priQuoteBudget.getItemId());
        priQuoteBudget.setMemberId(item.getMember());

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

    @RequestMapping(value = "/priQuoteBudget/update/auth")
    @ManagerAuth
    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
    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
    public R delete(Long[] ids){
        if (Cools.isEmpty(ids)){
            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("member_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.getItemId$() + "/" + priQuoteBudget.getMemberId$());
            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();
        }
    }

}