package com.zt.life.core.service;

import com.baomidou.mybatisplus.core.toolkit.IdWorker;
import com.zt.common.constant.Constant;
import com.zt.common.constant.Status;
import com.zt.common.db.query.QueryFilter;
import com.zt.common.entity.MapData;
import com.zt.common.exception.RenException;
import com.zt.common.service.BaseService;
import com.zt.common.utils.CommonUtils;
import com.zt.common.utils.TreeUtils;
import com.zt.core.context.UserContext;
import com.zt.core.sys.model.SysDept;
import com.zt.core.sys.service.ISysDeptService;
import com.zt.life.core.constant.Cache;
import com.zt.life.core.dao.ZtProductDao;
import com.zt.life.core.dto.SearchNodesDto;
import com.zt.life.core.enums.CompanyType;
import com.zt.life.core.model.ZtProduct;
import org.apache.commons.lang.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.cache.annotation.CacheEvict;
import org.springframework.cache.annotation.Cacheable;
import org.springframework.stereotype.Service;

import java.util.*;
import java.util.stream.Collectors;

/**
 * 产品结构树，id为数据主键，productId为产品主键
 * <p>
 * 本类隐藏id，手工将productId作为id返回给外部使用
 *
 * @author zt generator
 * @since 1.0.0 2020-07-15
 */
@Service
public class ZtProductService extends BaseService<ZtProductDao, ZtProduct> {

    @Value("${zt.oss.local-area}")
    private String localArea;

    @Autowired
    private ISysDeptService sysDeptService;

    public List<ZtProduct> getAllNodes() {
        List<ZtProduct> all = baseDao.getAllNodes();
        return all;
    }

    public List<ZtProduct> getTreeNodes(String pid) {
        Date beginDate = new Date();
        List<ZtProduct> list = baseDao.getTreeNodes(pid);
        Date nowDate = new Date();
        String msg = "产品层级查询时间：" + CommonUtils.getDatePoor(nowDate, beginDate) + "\r\n";
        System.out.println(msg);
        return list;
    }

    public List<ZtProduct> getLevelList(String l1) {
        List<ZtProduct> productList = baseDao.getLevelList(l1);
        return productList;
    }

    public List<ZtProduct> page(QueryFilter queryFilter) {
        return query(queryFilter);
    }


    /**
     * 查询所有，包含历史版本
     *
     * @return
     */
    public List<ZtProduct> getAll() {
        // 这里手动缓存，不然在当前方法内调用缓存无效
        List<ZtProduct> all = null;//(List<ZtProduct>) CacheUtils.get(Cache.ZTPRODUCT, "all");
        if (all == null) {
            all = baseDao.getAll();
            //CacheUtils.put(Cache.ZTPRODUCT, "all", all);
        }
        return all.stream().map(product -> {
            try {
                ZtProduct p = product.clone();
                p.setId(p.getProductId()); // 将productId设置给id
                return p;
            } catch (CloneNotSupportedException e) {
            }
            return null;
        }).collect(Collectors.toList());
    }

    public List<ZtProduct> getShipList(String area) {
//        String area = null;
//        if (localArea.equals("qd") || localArea.equals("sy"))
//            area = localArea;
        List<ZtProduct> list = baseDao.getShipList(area);
        return list;
    }

    public List<ZtProduct> getSubList(String pid) {
        Date beginDate = new Date();
        List<ZtProduct> list = baseDao.getSubList(pid);
        Date nowDate = new Date();
        String msg = "子节点查询时间：" + CommonUtils.getDatePoor(nowDate, beginDate) + "\r\n";
        System.out.println(msg);
        return list;
    }

    public List<ZtProduct> getListByIds(String ids) {
        List<ZtProduct> list = baseDao.getListByIds(ids);
        return list;
    }

    public Map getProductMapByIds(String ids) {
        List<ZtProduct> list = baseDao.getListByIds(ids);
        Map<String, String> map = new Hashtable();
        for (ZtProduct data : list) {
            map.put(data.getId().toString(), data.getName());
        }
        return map;
    }

    /**
     * 获取树，懒加载
     *
     * @param parentId
     * @return
     */
    public List<ZtProduct> getProductTree(Long parentId) {
        List<ZtProduct> list = new ArrayList<>();
        if (parentId == 0L) { // 首先只查询两级
            List<ZtProduct> list2 = this.getWithDescendant(0L);
            list.addAll(TreeUtils.build(list2));
            list.stream().forEach(product -> product.getChildren().stream().forEach(product1 -> {
                product1.setHasChildren(product1.getChildren().size() > 0);
                product1.getChildren().clear();
            }));
        } else {
            list.addAll(this.getChildren(parentId));
            Map<Long, Boolean> childrenMap = new HashMap<>();
            this.getAll().stream().forEach(product -> {
                if (!childrenMap.containsKey(product.getParentProductId())) {
                    childrenMap.put(product.getParentProductId(), true);
                }
            });
            list.stream().forEach(product -> product.setHasChildren(childrenMap.containsKey(product.getId())));
        }
        return list;
    }

    public List<ZtProduct> getAllProduct() {
        List<ZtProduct> all = baseDao.getAll();
        return all;
    }

    public Long getProductIdByNameSql(Long pid, String level, String name) {
        return baseDao.getProductIdByNameSql(pid, level, name);
    }

    public Long getProductIdByName(List<ZtProduct> all, Long pid, String level, String name) {
        List<ZtProduct> list = all.stream().filter(product ->
                product.getLevel().equals(level)
                        && product.getName().equals(name)
                        && ("," + product.getParentProductIds() + ",").contains("," + pid + ",")
        ).collect(Collectors.toList());
        if (list.size() > 0) {
            return list.get(0).getProductId();
        }
        return null;
    }

    public Long getProductIdById(List<ZtProduct> all, String level, String name) {
        List<ZtProduct> list = all.stream().filter(product ->
                product.getLevel().equals(level)
                        && product.getName().equals(name)
        ).collect(Collectors.toList());
        if (list.size() > 0) {
            return list.get(0).getProductId();
        }
        return null;
    }

    /**
     * 产品结构树
     *
     * @param parentIds
     * @param showLevel
     * @return
     */
    public List<ZtProduct> getProductTree(String parentIds, String showLevel) {
        ZtProduct currentUserTZtProduct = null;// 当前用户所属的T
        long companyId = UserContext.getUser().getCompanyId();
//        companyId = 1304648154755698690L;
        if (companyId > 0) {
            SysDept dept = sysDeptService.get(companyId);
            if (dept == null) {
                throw new RenException("用户所属单位不存在！");
            }
            if (CompanyType.T.getValue().equals(dept.getNature())) {// T用户登录
                List<ZtProduct> list = this.getAll().stream().filter(product -> dept.getCode().equals(product.getCode())).collect(Collectors.toList());
                if (list.size() > 0) {
                    currentUserTZtProduct = list.get(0);
                }
            }
        }
        List<ZtProduct> list = new ArrayList<>();
        if (StringUtils.isBlank(parentIds) || "0".equals(parentIds)) {
            final ZtProduct productTmp = currentUserTZtProduct;
            if (productTmp == null) {
                list.addAll(this.getWithDescendant(0L));
            } else {
                list.addAll(this.getAll().stream().filter(product -> productTmp.getParentProductId().equals(product.getProductId())).collect(Collectors.toList()));
                list.addAll(this.getWithDescendant(productTmp.getProductId()));
            }
        } else {
            String[] ids = parentIds.split(",");
            for (String id : ids) {
                try {
                    list.addAll(this.getWithDescendant(Long.parseLong(id)));
                } catch (NumberFormatException e) {
                }
            }
        }
        if (StringUtils.isBlank(showLevel)) {
            showLevel = "equipment";
        }
//        int level = Integer.parseInt(showLevel.substring(1));
//        list = list
//                .stream()
//                .filter(product -> Status.ENABLE.getValue().equals(product.getStatus())
//                        && Integer.parseInt(product.getLevel().substring(1)) <= level).collect(Collectors.toList());
        List<ZtProduct> trees = TreeUtils.build(list);
        return trees;
    }

    @CacheEvict(value = Cache.ZTPRODUCT, allEntries = true)
    public void insert(ZtProduct entity) {
        if (entity.getId() == null || entity.getId() == 0L) {
            entity.setId(IdWorker.getId());
            entity.setProductId(entity.getId());// 初始productId和id一致
        }
        if (entity.getPid() == null) {
            entity.setPid(Constant.Sys.DEPT_ROOT);
        }
        ZtProduct parent = getByProductId(entity.getPid());

        if (parent != null) {
            entity.setParentProductIds(parent.getParentProductIds() + "," + entity.getId());
        } else {
            entity.setParentProductIds(Constant.Sys.DEPT_ROOT + "");
        }
        entity.setStatus(Status.ENABLE.getValue());
        entity.setVersion(1);
        setProductId(entity, parent);
        baseDao.insert(entity);
    }

    @Override
    @CacheEvict(value = Cache.ZTPRODUCT, allEntries = true)
    public void update(ZtProduct entity) {
        // 这里id为productId，需要转换一下
        ZtProduct p = getByProductId(entity.getId());
        if (p == null) {
            throw new RenException("产品节点不存在");
        }
        entity.setId(p.getId());// 设置为原始id，通过id保存
        ZtProduct parent = getByProductId(entity.getPid());
        setProductId(entity, parent);
        super.update(entity);
    }

    @CacheEvict(value = Cache.ZTPRODUCT, allEntries = true)
    public void deleteLogic(Long[] ids) {
        for (int i = 0; i < ids.length; i++) {
            Long id = ids[i];
            List<ZtProduct> subList = getChildren(id);
            if (subList.size() > 0) {
                throw new RenException("请先删除下级节点!");
            }
            ZtProduct p = getByProductId(id);
            deleteLogic(p);
        }
    }

    @Override
    @Cacheable(value = Cache.ZTPRODUCT, key = "'id:' + #id")
    public ZtProduct get(Long id) {
        ZtProduct product = this.getByProductId(id);
        //product.setId(product.getProductId());
        return product;
    }

    @Cacheable(value = Cache.ZTPRODUCT, key = "'product:' + #id")
    public ZtProduct getByProductId(Long productId) {
        String msg = "";
        Date beginDate = new Date();
        if (productId == null || productId == 0) {
            return new ZtProduct();
        }
        //ZtProduct bean = baseDao.getByProductId(productId);
        ZtProduct bean = baseDao.getOneByProductId(productId);
        if (bean != null) {
            List<ZtProduct> list = baseDao.getByProductIdList(bean.getParentProductIds());
            list.add(bean);
            for (ZtProduct ztProduct : list) {
                if ("model".equals(ztProduct.getLevel())) {
                    bean.setModelName(ztProduct.getName());
                } else if ("side".equals(ztProduct.getLevel())) {
                    bean.setShipName(ztProduct.getName());
                    bean.setShipId(ztProduct.getDataId());
                    bean.setShipProductId(ztProduct.getProductId());
                } else if ("system1".equals(ztProduct.getLevel())) {
                    bean.setSysName(ztProduct.getName());
                    bean.setSysId(ztProduct.getDataId());
                    bean.setSysProductId(ztProduct.getProductId());
                } else if ("system2".equals(ztProduct.getLevel())) {
                    bean.setSubSysName(ztProduct.getName());
                    bean.setSubSysId(ztProduct.getDataId());
                    bean.setSubSysProductId(ztProduct.getProductId());
                } else if ("equipment".equals(ztProduct.getLevel())) {
                    bean.setDeviceName(ztProduct.getName());
                    bean.setDeviceId(ztProduct.getDataId());
                    bean.setDeviceProductId(ztProduct.getProductId());
                }
            }
        }
        if (bean == null) {
            bean = new ZtProduct();
        }
        msg = "产品节点路徑查询时间：" + CommonUtils.getDatePoor(new Date(), beginDate) + "\r\n";
        System.out.println(msg);
        return bean;
    }

    /**
     * 通过id查找所有祖先，按照祖先的层级排序
     *
     * @param id
     * @return
     */
    public List<ZtProduct> getAncestor(Long id) {
        return TreeUtils.getAncestor(getAll(), id);
    }

    /**
     * 获取自己和所有下级
     *
     * @param id
     * @return
     */
    public List<ZtProduct> getWithDescendant(Long id) {
        return TreeUtils.getWithDescendant(getAll(), id);
    }

    @Cacheable(value = Cache.ZTPRODUCT, key = "'withdescendant:' + #id")
    public List<Long> getWithDescendantIds(Long id) {
        return this.getWithDescendant(id).stream().map(dept -> dept.getId()).collect(Collectors.toList());
    }

    /**
     * 获取下级
     *
     * @param id
     * @return
     */
    public List<ZtProduct> getDescendant(Long id) {
        return TreeUtils.getDescendant(getAll(), id);
    }

    /**
     * 获取某产品的子节点
     *
     * @param id
     * @return
     */
    public List<ZtProduct> getChildren(Long id) {
        return getAll().stream().filter(dept -> dept.getPid().equals(id)).collect(Collectors.toList());
    }

    /**
     * 改换装，升级产品节点版本
     *
     * @param product
     */
    public void replace(ZtProduct product) {
        if (product.getProductId() == null) {
            throw new RenException("缺少产品ZtProductId");
        }
        ZtProduct db = this.getByProductId(product.getProductId());
        if (db == null) {
            throw new RenException("产品节点不存在");
        }
        // 设置原来的版本无效
        db.setStatus(Status.DISABLE.getValue());
        baseDao.updateById(db);

        // 添加新版本
        product.setVersion(db.getVersion() + 1);
        product.setStatus(Status.ENABLE.getValue());
        baseDao.insert(product);
    }

    public void setProductId(ZtProduct product, ZtProduct parent) {
        String level = product.getLevel();
        String parentLevel = parent.getLevel();
        // 型号
        if (StringUtils.equals("L1", level)) {

        }
        // X号
        if (StringUtils.equals("L2", level)) {
            product.setShipId(product.getId());
            product.setShipProductId(product.getId());
        }
        // 系统
        if (StringUtils.equals("L3", level)) {
            // 系统
            if (StringUtils.equals("L2", parentLevel)) {
                product.setShipId(parent.getShipId());
                product.setShipProductId(parent.getShipProductId());
                product.setSysId(product.getId());
                product.setSysProductId(product.getId());
            }
            // 子系统
            if (StringUtils.equals("L3", parentLevel)) {
                product.setShipId(parent.getShipId());
                product.setShipProductId(parent.getShipProductId());
                product.setSysId(parent.getId());
                product.setSysProductId(parent.getId());
                product.setSubSysId(product.getId());
                product.setSubSysProductId(product.getId());
            }
        }
        // 设备
        if (StringUtils.equals("L4", level)) {
            product.setDeviceId(product.getId());
            product.setDeviceProductId(product.getId());
            // 父节点为系统
            if (StringUtils.equals("L3", parentLevel)) {
                ZtProduct sup_parent = getByProductId(parent.getPid());
                String supParentLevel = sup_parent.getLevel();
                // 父节点的节点为X号,则父节点为系统
                if (StringUtils.equals("L2", supParentLevel)) {
                    product.setShipId(sup_parent.getShipId());
                    product.setShipProductId(sup_parent.getShipProductId());
                    product.setSysId(sup_parent.getId());
                    product.setSysProductId(sup_parent.getId());
                }
                // 父节点的父节点为系统,则父节点为子系统
                if (StringUtils.equals("L3", supParentLevel)) {
                    product.setShipId(sup_parent.getShipId());
                    product.setShipProductId(sup_parent.getShipProductId());
                    product.setSysId(sup_parent.getSysId());
                    product.setSysProductId(parent.getSysProductId());
                    product.setSubSysId(sup_parent.getId());
                    product.setSubSysProductId(sup_parent.getId());
                }

            }
            // 父节点为设备,沿用父节点系统/子系统编号
            if (StringUtils.equals("L4", parentLevel)) {
                product.setShipId(parent.getShipId());
                product.setShipProductId(parent.getShipProductId());
                product.setSysId(parent.getSysId());
                product.setSysProductId(parent.getSysProductId());
                product.setSubSysId(product.getSubSysId());
                product.setSubSysProductId(product.getSubSysProductId());
            }
        }
    }

    public Map<String, String> getMapAllNodeByShipId(Boolean idToNmae, Long shipId) {
        Map<String, String> result = new HashMap<>();
        List<MapData> list = baseDao.getAllNodeByShipId(shipId.toString());
        for (MapData item : list) {
            if (idToNmae)
                result.put(item.getId(), item.getName());
            else
                result.put(item.getName(), item.getId());
        }
        return result;
    }

    public Long getIdByName(String name) {
        return baseDao.getIdByName(name);
    }

    public String getNameById(String id) {
        return baseDao.getNameById(id);
    }

    public String getNameListById(Long id) {
        return baseDao.getNameListById(id);
    }

    public void deleteTable(String tableName) {
        baseDao.deleteTable(tableName);

    }

    public List<SearchNodesDto> searchNodes(String pid, String name) {
        List<SearchNodesDto> list = baseDao.searchNodes(pid, name);
        String ids = "'0'";
        if (list.size() > 0) {
            for (SearchNodesDto searchNodesDto : list) {
                ids = ids + ",'" + searchNodesDto.getGroups().replace(",","','")+"'";
            }
            if (!"0".equals(ids)) {

                List<MapData> list2 = baseDao.getNamesByIds(ids);
                for (SearchNodesDto searchNodesDto : list) {
                    String names = "";
                    String[] idArr = searchNodesDto.getGroups().split(",");
                    for (String s : idArr) {
                        Optional<MapData> map = list2.stream().filter(item -> item.getId().equals(s)).findFirst();
                        if (map != null && map.isPresent()) {
                            names = names + "/" + map.get().getName();
                        }
                    }
                    searchNodesDto.setName(names+"/"+searchNodesDto.getName());
                }
            }
        }
        return list;
    }

    public List<MapData> getNameAll() {
        return baseDao.getNameAll();
    }

}