package com.example.erp.service.userInfo;

import com.baomidou.dynamic.datasource.annotation.DS;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.example.erp.entity.userInfo.UserRole;
import com.example.erp.mapper.userInfo.UserRoleMapper;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

@Service
@DS("user_info")
public class UserRoleService {
    private final UserRoleMapper userRoleMapper;

    public UserRoleService(UserRoleMapper userRoleMapper) {
        this.userRoleMapper = userRoleMapper;
    }

    public  Boolean updateUserRole(Map<String, Object> params) {
        Integer userId = (Integer) params.get("userId");
        List<Integer> list = (List<Integer>) params.get("roles");
        List<UserRole> userRoleList = new ArrayList<>();
        if(!list.isEmpty()) {
            for (Integer roleId : list) {
                UserRole userRole = new UserRole();
                userRole.setUserId(userId);
                userRole.setRoleId(roleId);
                userRoleList.add(userRole);
            }
        }
        userRoleMapper.delete(new QueryWrapper<UserRole>().eq("user_id", userId));
        userRoleList.forEach(userRole -> {
            userRoleMapper.insert(userRole);
        });
        return true;
    }
}