package com.ruoyi.system.controller;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.ruoyi.common.core.domain.R;
import com.ruoyi.system.api.model.UserShop;
import com.ruoyi.system.service.UserShopService;
import org.springframework.web.bind.annotation.*;

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

/**
 * @author zhibing.pu
 * @date 2025/1/2 18:04
 */
@RestController
@RequestMapping("/userShop")
public class UserShopController {

    @Resource
    private UserShopService userShopService;


    /**
     * 获取用户门店关系数据
     * @param userShop
     * @return
     */
    @PostMapping("/getUserShop")
    public R<List<UserShop>> getUserShop(@RequestBody UserShop userShop){
        LambdaUpdateWrapper<UserShop> wrapper = new LambdaUpdateWrapper<UserShop>();
        if(null != userShop.getUserId()){
            wrapper.eq(UserShop::getUserId, userShop.getUserId());
        }
        if(null != userShop.getShopId()){
            wrapper.eq(UserShop::getShopId, userShop.getShopId());
        }
        if(null != userShop.getRoleType()){
            wrapper.eq(UserShop::getRoleType, userShop.getRoleType());
        }
        List<UserShop> list = userShopService.list(wrapper);
        return R.ok(list);
    }


    /**
     * 保存数据
     * @param userShop
     * @return
     */
    @PostMapping("/saveUserShop")
    public R saveUserShop(@RequestBody UserShop userShop){
        userShopService.save(userShop);
        return R.ok();
    }
    
    
    /**
     * 删除后台账号和门店关系数据
     * @param userShop
     * @return
     */
    @PostMapping("/delUserShop")
    public R delUserShop(@RequestBody UserShop userShop){
        userShopService.remove(new LambdaQueryWrapper<UserShop>().eq(UserShop::getUserId, userShop.getUserId()).eq(UserShop::getShopId, userShop.getShopId()));
        return R.ok();
    }
    
    
    /**
     * 删除门店所有关系数据
     * @param shopId
     * @return
     */
    @PostMapping("/delShopAllUserShop")
    public R delShopAllUserShop(@RequestParam("shopId") Integer shopId){
        userShopService.remove(new LambdaQueryWrapper<UserShop>().eq(UserShop::getShopId, shopId));
        return R.ok();
    }
}