package com.ruoyi.other.controller;


import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.ruoyi.chargingPile.api.feignClient.SiteClient;
import com.ruoyi.chargingPile.api.model.Site;
import com.ruoyi.common.core.domain.R;
import com.ruoyi.other.api.domain.TRoleSite;
import com.ruoyi.other.service.TRoleSiteService;
import org.springframework.web.bind.annotation.*;

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

/**
 * <p>
 *  前端控制器
 * </p>
 *
 * @author 无关风月
 * @since 2024-08-06
 */
@RestController
@RequestMapping("/t-role-site")
public class TRoleSiteController {

	@Resource
	private TRoleSiteService roleSiteService;
	
	@Resource
	private SiteClient siteClient;
	
	
	
	/**
	 * 查询角色站点数据
	 * @param roleId
	 * @return
	 */
	@ResponseBody
	@PostMapping("/getSiteIds/{roleId}")
	public R<List<Integer>> getSiteIds(@PathVariable Long roleId){
		List<TRoleSite> list = roleSiteService.list(new LambdaQueryWrapper<TRoleSite>().eq(TRoleSite::getRoleId, roleId));
		if(list.size() == 1){
			Integer siteId = list.get(0).getSiteId();
			if(0 == siteId){
				List<Site> data = siteClient.getSiteAll().getData();
				List<Integer> collect = data.stream().map(Site::getId).collect(Collectors.toList());
				return R.ok(collect);
			}
		}
		List<Integer> collect = list.stream().map(TRoleSite::getSiteId).collect(Collectors.toList());
		return R.ok(collect);
	}
	
	/**
	 * 添加角色站点数据
	 * @param roleSites
	 * @return
	 */
	@ResponseBody
	@PostMapping("/addRoleSite")
	public R addRoleSite(@RequestBody List<TRoleSite> roleSites){
		roleSiteService.saveBatch(roleSites);
		return R.ok();
	}
	
	/**
	 * 删除角色站点数据
	 * @param roleId
	 * @return
	 */
	@ResponseBody
	@DeleteMapping("/delRoleSite/{roleId}")
	public R delRoleSite(@PathVariable("roleId") Long roleId){
		roleSiteService.remove(new LambdaQueryWrapper<TRoleSite>().eq(TRoleSite::getRoleId, roleId));
		return R.ok();
	}
}