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.entity.ObjectButtonPermission;
import org.jeecg.modules.dnc.exception.ExceptionCast;
import org.jeecg.modules.dnc.service.IObjectButtonPermissionService;
import org.jeecg.modules.dnc.mapper.ObjectButtonPermissionMapper;
import org.jeecg.modules.dnc.utils.ValidateUtil;
import org.jeecg.modules.dnc.response.RoleCode;
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 ObjectButtonPermissionServiceImpl extends ServiceImpl<ObjectButtonPermissionMapper, ObjectButtonPermission> implements IObjectButtonPermissionService {
    @Override
    @Transactional(rollbackFor = {Exception.class})
    public boolean deleteByRoleId(String roleId) {
        if(!ValidateUtil.validateString(roleId))
            ExceptionCast.cast(RoleCode.ROLE_ID_NONE);
        LambdaQueryWrapper<ObjectButtonPermission> lambdaUpdate = Wrappers.lambdaQuery();
        lambdaUpdate.eq(ObjectButtonPermission::getRoleId, roleId);
        return super.remove(lambdaUpdate);
    }

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

    @Override
    public List<ObjectButtonPermission> findByObjectId(String objId) {
        return super.lambdaQuery().eq(ObjectButtonPermission::getObjectId, objId).list();
    }
}