package com.zt.life.core.service;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.IdWorker;
import com.zt.common.constant.Constant;
import com.zt.common.constant.Layer;
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.CacheUtils;
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.constant.ProjectStatus;
import com.zt.life.core.dao.ProductDao;
import com.zt.life.core.enums.CompanyType;
import com.zt.life.core.model.Product;
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 org.springframework.ui.Model;

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

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

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

    @Autowired
    private ISysDeptService sysDeptService;



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

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


    /**
     * 查询所有，包含历史版本
     *
     * @return
     */
    public List<Product> getAll() {
        // 这里手动缓存，不然在当前方法内调用缓存无效
        List<Product> all = (List<Product>) CacheUtils.get(Cache.PRODUCT, "all");
        if (all == null) {
            all = baseDao.getAll();
            CacheUtils.put(Cache.PRODUCT, "all", all);
        }
        return all.stream().map(product -> {
            try {
                Product p = product.clone();
                p.setId(p.getProductId()); // 将productId设置给id
                return p;
            } catch (CloneNotSupportedException e) {
            }
            return null;
        }).collect(Collectors.toList());
    }
    public List<Product>  getShipList(){
        String area = null;
        if (localArea.equals("qd") || localArea.equals("sy"))
            area = localArea;
        List<Product> list = baseDao.getShipList(area);
        return list;
    }
    public List<Product>  getSubList(String pid){
        List<Product> list = baseDao.getSubList(pid);
        return list;
    }
    public List<Product>  getListByIds(String ids){
        List<Product> list = baseDao.getListByIds(ids);
        return list;
    }
    /**
     * 获取树，懒加载
     *
     * @param parentId
     * @return
     */
    public List<Product> getProductTree(Long parentId) {
        List<Product> list = new ArrayList<>();
        if (parentId == 0L) { // 首先只查询两级
            list.addAll(TreeUtils.build(this.getWithDescendant(0L)));
            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;
    }

    /**
     * 产品结构树
     *
     * @param parentIds
     * @param showLevel
     * @return
     */
    public List<Product> getProductTree2(String parentIds, String showLevel) {
/*        Product currentUserTProduct = 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<Product> list = this.getAll().stream().filter(product -> dept.getCode().equals(product.getCode())).collect(Collectors.toList());
                if (list.size() > 0) {
                    currentUserTProduct = list.get(0);
                }
            }
        }
        List<Product> list = new ArrayList<>();
        if (StringUtils.isBlank(parentIds) || "0".equals(parentIds)) {
            final Product productTmp = currentUserTProduct;
            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) {
                }
            }
        }*/

        List<Product> list = new ArrayList<>();
        list.addAll(this.getWithDescendant(0L));

        if (StringUtils.isBlank(showLevel)) {
            showLevel = "equipment";
        }
        String levels= "";
/*        if ("side".equals(showLevel)){
            levels = "'model','side'";
        }
        else if ("system1".equals(showLevel)){
            levels = "'model','side','system1'";
        }
        else if ("system2".equals(showLevel)) {
            levels = "'model','side','system1','system2'";
        }
        else if ("equipment".equals(showLevel)) {
            levels = "'model','side','system1','system2','equipment'";
        }
        List<Product> list = baseDao.getProduct(levels);*/
        //int level = Integer.parseInt(showLevel.substring(1));
        Integer level = Layer.get(showLevel);
        list = list
                .stream()
                .filter(product -> Layer.get(product.getLevel()) <= level).collect(Collectors.toList());
        List<Product> trees = TreeUtils.build(list);
        return trees;
    }

    public List<Product> getProductTree(String parentIds, String showLevel, String pageCode) {

   /*     if (StringUtils.isBlank(showLevel)) {
            showLevel = "equipment";
        }*/
  /*       String levels= "";
       if ("side".equals(showLevel)){
            levels = "'model','side'";
        }
        else if ("system1".equals(showLevel)){
            levels = "'model','side','system1'";
        }
        else if ("system2".equals(showLevel)) {
            levels = "'model','side','system1','system2'";
        }
        else if ("equipment".equals(showLevel)) {
            levels = "'model','side','system1','system2','equipment'";
        }
        List<Product> list = baseDao.getProduct(levels);*/
        //int level = Integer.parseInt(showLevel.substring(1));

        String msg = "";
        Date beginDate = new Date();
        List<Product> projectList = getProjectByProductId(null);
        msg = "工程查询时间：" + CommonUtils.getDatePoor(new Date(), beginDate) + "\r\n";
        System.out.println(msg);
        Product newestProjectProduct = getProductIdAndProjectId();
        msg = "工程查询时间2：" + CommonUtils.getDatePoor(new Date(), beginDate) + "\r\n";
        System.out.println(msg);

        List<Product> modelList = baseDao.getProduct("'model'",null);
        msg = "产品节点查询时间：" + CommonUtils.getDatePoor(new Date(), beginDate) + "\r\n";
        System.out.println(msg);
        int i=0;
        for (Product model : modelList) {
            model.setProductIds(model.getId().toString()+","+newestProjectProduct.getProductId().toString());
            model.setProjectId(newestProjectProduct.getId());

            List<Product> sideList = baseDao.getProduct("'side'",model.getId());
            msg = "产品节点查询时间"+i+"：" + CommonUtils.getDatePoor(new Date(), beginDate) + "\r\n";
            System.out.println(msg);
            model.setChildren(sideList);
            if (!"lxProjectList".equals(pageCode)) {
                for (Product side : sideList) {
                    side.setChildren(projectList.stream().filter(item -> side.getId().equals(item.getProductId())).collect(Collectors.toList()));
                }
            }
            i++;
        }
        msg = "产品节点查询时间(总)：" + CommonUtils.getDatePoor(new Date(), beginDate) + "\r\n";
        System.out.println(msg);
        return modelList;
    }

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

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

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

    @Cacheable(value = Cache.PRODUCT, key = "'product:' + #id")
    public Product getByProductId(Long id) {
        if(id == null || id == 0) {
            return new Product();
        }
        Product bean = baseDao.selectOne(new QueryWrapper<Product>().lambda().eq(Product::getProductId, id)
                .eq(Product::getStatus, Status.ENABLE.getValue()));
        if(bean == null) {
            bean = new Product();
        }
        return bean;
    }

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

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

    @Cacheable(value = Cache.PRODUCT, 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<Product> getDescendant(Long id) {
        return TreeUtils.getDescendant(getAll(), id);
    }

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

    /**
     * 改换装，升级产品节点版本
     *
     * @param product
     */
    public void replace(Product product) {
        if (product.getProductId() == null) {
            throw new RenException("缺少产品ProductId");
        }
        Product 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(Product product, Product 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)) {
                Product 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);
        for (MapData item:list) {
            if (idToNmae)
                result.put(item.getId().toString(), item.getName());
            else
                result.put(item.getName(), item.getId().toString());
        }
        return result;
    }

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

    public Long getIdByNameByProductId(String name,Long productId) {
        return baseDao.getIdByNameByProductId(name,productId);
    }

    public List<Product> getProjectByProductId(Long sideId) {
        return baseDao.getProjectByProductId(sideId);
    }

    public Product getProductIdAndProjectId() {
        return baseDao.getProductIdAndProjectId();
    }
}