package com.ruoyi.other.controller;


import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.ruoyi.common.core.domain.R;
import com.ruoyi.other.api.domain.TUserSite;
import com.ruoyi.other.service.TUserSiteService;
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-user-site")
public class TUserSiteController {

	@Resource
	private TUserSiteService userSiteService;
	
	
	/**
	 * 根据用户id获取可访问的站点数据
	 * @param userId 用户id
	 * @return
	 */
	@ResponseBody
	@PostMapping("/getSiteIds/{userId}")
	public R<List<Integer>> getSiteIds(@PathVariable Long userId){
		List<TUserSite> list = userSiteService.list(new LambdaQueryWrapper<TUserSite>().eq(TUserSite::getUserId, userId));
		List<Integer> collect = list.stream().map(TUserSite::getSiteId).collect(Collectors.toList());
		return R.ok(collect);
	}
	
	
	/**
	 * 添加用户站点数据
	 * @param userSite
	 * @return
	 */
	@ResponseBody
	@PostMapping("/addUserSite")
	public R addUserSite(@RequestBody List<TUserSite> userSite){
		userSiteService.saveBatch(userSite);
		return R.ok();
	}
	
	/**
	 * 删除用户对应的站点数据
	 * @param userId
	 * @return
	 */
	@ResponseBody
	@DeleteMapping("/delUserSite/{userId}")
	public R delUserSite(@PathVariable Long userId){
		userSiteService.remove(new LambdaQueryWrapper<TUserSite>().eq(TUserSite::getUserId, userId));
		return R.ok();
	}
}