package org.jeecg.modules.dnc.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.jeecg.modules.dnc.exception.ExceptionCast;
import org.jeecg.modules.dnc.mapper.ButtonPermissionMapper;
import org.jeecg.modules.dnc.utils.ValidateUtil;
import org.jeecg.modules.dnc.entity.ButtonPermission;
import org.jeecg.modules.dnc.response.RoleCode;
import org.jeecg.modules.dnc.service.IButtonPermissionService;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

@Service
public class ButtonPermissionServiceImpl extends ServiceImpl<ButtonPermissionMapper, ButtonPermission> implements IButtonPermissionService {
    @Override
    @Transactional(rollbackFor = {Exception.class})
    public boolean deleteByRoleId(String roleId) {
        if(!ValidateUtil.validateString(roleId))
            ExceptionCast.cast(RoleCode.ROLE_ID_NONE);
        LambdaQueryWrapper<ButtonPermission> lambdaUpdate = Wrappers.lambdaQuery();
        lambdaUpdate.eq(ButtonPermission::getRoleId, roleId);
        return super.remove(lambdaUpdate);
    }

    @Override
    public Map<String, ButtonPermission> getMapByRoleId(String roleId) {
        if(!ValidateUtil.validateString(roleId))
            return null;
        Map<String, ButtonPermission> map = new HashMap<>();
        List<ButtonPermission> list = super.lambdaQuery().eq(ButtonPermission::getRoleId, roleId).list();
        if(list == null || list.isEmpty())
            return null;
        list.forEach(item -> {
            map.put(item.getButtonId(), item);
        });
        return map;
    }

    @Override
    public List<ButtonPermission> findByButtonId(String buttonId) {
        return super.lambdaQuery().eq(ButtonPermission::getButtonId, buttonId).list();
    }
}