package com.doumee.service.business.impl;

import com.doumee.core.model.LoginUserInfo;
import com.doumee.core.model.PageData;
import com.doumee.core.model.PageWrap;
import com.doumee.core.utils.Constants;
import com.doumee.core.utils.Utils;
import com.doumee.dao.business.CategoryMapper;
import com.doumee.dao.business.model.BjParam;
import com.doumee.dao.business.model.Category;
import com.doumee.dao.system.model.SystemUser;
import com.doumee.service.business.CategoryService;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.github.yulichang.wrapper.MPJLambdaWrapper;
import org.apache.shiro.SecurityUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;

/**
 * 分类信息表Service实现
 * @author 江蹄蹄
 * @date 2024/07/05 17:52
 */
@Service
public class CategoryServiceImpl implements CategoryService {

    @Autowired
    private CategoryMapper categoryMapper;

    @Override
    public Long create(Category category) {
        LoginUserInfo userInfo = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal();
        category.setIsdeleted(Constants.ZERO);
        category.setEditor(userInfo.getId());
        category.setCreator(userInfo.getId());
        category.setCreateDate(new Date());
        category.setEditDate(category.getCreateDate());
        categoryMapper.insert(category);
        return category.getId();
    }

    @Override
    public void deleteById(Long id) {
        LoginUserInfo userInfo = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal();
        categoryMapper.update(null,new UpdateWrapper<Category>().lambda()
                .set(Category::getIsdeleted,Constants.ONE )
                .set(Category::getEditor,userInfo.getId() )
                .set(Category::getEditDate,new Date() )
                .eq(Category::getParentId,id)
                .or().eq(Category::getId,id));

    }

    @Override
    public void delete(Category category) {
        UpdateWrapper<Category> deleteWrapper = new UpdateWrapper<>(category);
        categoryMapper.delete(deleteWrapper);
    }

    @Override
    public void deleteByIdInBatch(List<Long> ids) {
        if (CollectionUtils.isEmpty(ids)) {
            return;
        }
        for(Long id : ids){
            deleteById(id);
        }
    }

    @Override
    public void updateById(Category category) {
        LoginUserInfo userInfo = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal();
        category.setEditor(userInfo.getId());
        category.setEditDate(new Date());
        categoryMapper.updateById(category);
    }

    @Override
    public void updateByIdInBatch(List<Category> categorys) {
        if (CollectionUtils.isEmpty(categorys)) {
            return;
        }
        for (Category category: categorys) {
            this.updateById(category);
        }
    }

    @Override
    public Category findById(Long id) {
        return categoryMapper.selectById(id);
    }

    @Override
    public Category findOne(Category category) {
        QueryWrapper<Category> wrapper = new QueryWrapper<>(category);
        return categoryMapper.selectOne(wrapper);
    }

    @Override
    public List<Category> findList(Category category) {
        QueryWrapper<Category> wrapper = new QueryWrapper<>(category);
        return categoryMapper.selectList(wrapper);
    }
  
    @Override
    public PageData<Category> findPage(PageWrap<Category> pageWrap) {
        IPage<Category> page = new Page<>(pageWrap.getPage(), pageWrap.getCapacity());
        QueryWrapper<Category> queryWrapper = new QueryWrapper<>();
        Utils.MP.blankToNull(pageWrap.getModel());
        if (pageWrap.getModel().getId() != null) {
            queryWrapper.lambda().eq(Category::getId, pageWrap.getModel().getId());
        }
        if (pageWrap.getModel().getIsdeleted() != null) {
            queryWrapper.lambda().eq(Category::getIsdeleted, pageWrap.getModel().getIsdeleted());
        }
        if (pageWrap.getModel().getCreateDate() != null) {
            queryWrapper.lambda().ge(Category::getCreateDate, Utils.Date.getStart(pageWrap.getModel().getCreateDate()));
            queryWrapper.lambda().le(Category::getCreateDate, Utils.Date.getEnd(pageWrap.getModel().getCreateDate()));
        }
        if (pageWrap.getModel().getCreator() != null) {
            queryWrapper.lambda().eq(Category::getCreator, pageWrap.getModel().getCreator());
        }
        if (pageWrap.getModel().getEditDate() != null) {
            queryWrapper.lambda().ge(Category::getEditDate, Utils.Date.getStart(pageWrap.getModel().getEditDate()));
            queryWrapper.lambda().le(Category::getEditDate, Utils.Date.getEnd(pageWrap.getModel().getEditDate()));
        }
        if (pageWrap.getModel().getEditor() != null) {
            queryWrapper.lambda().eq(Category::getEditor, pageWrap.getModel().getEditor());
        }
        if (pageWrap.getModel().getType() != null) {
            queryWrapper.lambda().eq(Category::getType, pageWrap.getModel().getType());
        }
        if (pageWrap.getModel().getInfo() != null) {
            queryWrapper.lambda().eq(Category::getInfo, pageWrap.getModel().getInfo());
        }
        if (pageWrap.getModel().getName() != null) {
            queryWrapper.lambda().eq(Category::getName, pageWrap.getModel().getName());
        }
        if (pageWrap.getModel().getParentId() != null) {
            queryWrapper.lambda().eq(Category::getParentId, pageWrap.getModel().getParentId());
        }
        if (pageWrap.getModel().getSortnum() != null) {
            queryWrapper.lambda().eq(Category::getSortnum, pageWrap.getModel().getSortnum());
        }
        for(PageWrap.SortData sortData: pageWrap.getSorts()) {
            if (sortData.getDirection().equalsIgnoreCase(PageWrap.DESC)) {
                queryWrapper.orderByDesc(sortData.getProperty());
            } else {
                queryWrapper.orderByAsc(sortData.getProperty());
            }
        }
        return PageData.from(categoryMapper.selectPage(page, queryWrapper));
    }

    @Override
    public long count(Category category) {
        QueryWrapper<Category> wrapper = new QueryWrapper<>(category);
        return categoryMapper.selectCount(wrapper);
    }
    @Override
    public List<Category> findTree(Category category ){
        MPJLambdaWrapper<Category> wrapper = new MPJLambdaWrapper<>();
        wrapper.selectAll(Category.class)
                .selectAs(SystemUser::getRealname,Category::getEditorName)
                .leftJoin(SystemUser.class,SystemUser::getId,Category::getEditor)
                .eq(category.getType()!=null,Category::getType,category.getType())
                .eq(Category::getIsdeleted,Constants.ZERO )
                .orderByAsc(Category::getSortnum);
        List<Category> list =  categoryMapper.selectJoinList(Category.class,wrapper);
        List<Category> treeList = new ArrayList<>();
        if(list!=null && list.size()>0){
            for(Category model : list){
                if(model.getParentId() == null){
                    initModelChildData(model,list);
                    treeList.add(model);
                }
            }
        }

        return treeList;
    }

    private void initModelChildData(Category param, List<Category> list) {
        List<Category> clist = new ArrayList<>();
        param.setHasChildren(Boolean.TRUE);
        if(list!=null && list.size()>0){
            for(Category model : list){
                if(model.getParentId() != null &&  param.getId()!=null && param.getId().equals(model.getParentId())){
                    param.setHasChildren(Boolean.FALSE);
                    model.setHasChildren(Boolean.TRUE);
                    clist.add(model);
                }
            }
        }
        param.setChildren(clist);
    }
}