package com.dg.core.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.dg.core.ResultData;
import com.dg.core.db.gen.entity.ClassifyAdministration;
import com.dg.core.db.gen.entity.TransactionEvent;
import com.dg.core.db.gen.mapper.ClassifyAdministrationMapper;
import com.dg.core.db.gen.mapper.TransactionEventMapper;
import com.dg.core.service.IClassifyAdministrationService;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.List;

@Service
public class ClassifyAdministrationImpl extends ServiceImpl<ClassifyAdministrationMapper, ClassifyAdministration>
        implements IClassifyAdministrationService
{


    @Resource
    private TransactionEventMapper transactionEventMapper;

    @Override
    public List<ClassifyAdministration> selectConfigList(String classifyName, String classifyGrade) {
        return baseMapper.selectConfigList( classifyName, classifyGrade);
    }

    @Override
    public List<ClassifyAdministration> selectConfigList(IPage<ClassifyAdministration> page,
                                                         Integer state, String classifyName, String classifyGrade) {
        return baseMapper.selectConfigList(page,state, classifyName, classifyGrade);
    }

    @Override
    public List<ClassifyAdministration> selectConfigListByParentId(String parentId,Integer departmentId){
        List<ClassifyAdministration> classifyAdministrationEntities = baseMapper.selectRootList(parentId ,departmentId);
        for (ClassifyAdministration classifyAdministrationEntitie: classifyAdministrationEntities) {
            classifyAdministrationEntitie.setClassifyAdministrationEntityList(selectConfigListByParentId(classifyAdministrationEntitie.getId().toString(),departmentId));
        }
        return  classifyAdministrationEntities;
    }

    @Override
    public List<ClassifyAdministration> selectRootList(IPage<ClassifyAdministration> page, Integer stat,String parentId,Integer departmentId) {
        List<ClassifyAdministration> classifyAdministrations = baseMapper.selectRootList(page, stat,"0" ,departmentId);
        for (ClassifyAdministration classifyAdministrationEntitie: classifyAdministrations) {
            classifyAdministrationEntitie.setClassifyAdministrationEntityList(selectConfigListByParentId(classifyAdministrationEntitie.getId().toString(),departmentId));
        }
        return  classifyAdministrations;
    }

    @Override
    public ResultData queryListById(Integer id){
        if (id.equals(0)){
            return ResultData.success(baseMapper.selectList(new QueryWrapper<ClassifyAdministration>().lambda()
                    .eq(ClassifyAdministration::getParentId,0)));
        }
        ClassifyAdministration classifyAdministration = baseMapper.selectOne(new QueryWrapper<ClassifyAdministration>().lambda()
                .eq(ClassifyAdministration::getId, id));
        if (classifyAdministration.getClassifyGrade().equals("1")){
            return ResultData.success(baseMapper.selectList(new QueryWrapper<ClassifyAdministration>().lambda()
                    .eq(ClassifyAdministration::getParentId,classifyAdministration.getId())));
        }else if (classifyAdministration.getClassifyGrade().equals("2")){
            return ResultData.success(transactionEventMapper.queryByClassifyIdList(id.toString(),null));
        }
        return  ResultData.error();
    }

    @Override
    public List<String> selectParent(String parentId) {
        return baseMapper.selectParent(parentId);
    }

    @Override
    public List<ClassifyAdministration> selectParentData(String parentId) {
        return baseMapper.selectParentData(parentId);
    }


    @Override
    public int insertConfig(ClassifyAdministration entity) {
        return baseMapper.insertConfig(entity);
    }

    @Override
    public int updateConfig(ClassifyAdministration entity) {
        return baseMapper.updateConfig(entity);
    }

    @Override
    public int deleteConfigById(String Id) {
        return baseMapper.deleteConfigById(Id);
    }

    @Override
    public int countNum(String classifyName,String classifyGrade) {
        return baseMapper.countNum(classifyName,classifyGrade);
    }
}