package com.whyc.controller;

import com.whyc.dto.Response;
import com.whyc.pojo.db_station.PowerInf;
import com.whyc.pojo.db_user.Baojigroup;
import com.whyc.pojo.db_user.BaojigroupPower;
import com.whyc.pojo.db_user.BaojigroupUsr;
import com.whyc.pojo.db_user.User;
import com.whyc.service.BaojigroupService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

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

@RestController
@Api(tags = "包机组管理")
@RequestMapping("baoji")
public class BaojiGroupController {
    @Autowired
    private BaojigroupService service;

    @PostMapping("updateTeamFlag")
    @ApiOperation("设置包机组标记")
    public Response updateTeamFlag(@RequestParam Integer baoJiGroupId,@RequestParam Integer flag){
        return service.updateTeamFlag(baoJiGroupId,flag);
    }
    @ApiOperation("添加包机组")
    @GetMapping("addBaoji")
    public Response addBaoji(@RequestParam String baojiName) {
        return service.addBaoji(baojiName);
    }
    @ApiOperation("编辑包机组")
    @GetMapping("updateBaoji")
    public Response updateBaoji(@RequestParam Integer id, @RequestParam String baojiName) {
        return service.updateBaoji(id,baojiName);
    }
    @ApiOperation("删除包机组")
    @GetMapping("delBaoji")
    public Response delBaoji(@RequestParam Integer id) {
        return service.delBaoji(id);
    }


    @GetMapping("list")
    @ApiOperation(value = "包机组列表")
    public Response<List<Baojigroup>> getBaoJiGroupList(){
        return service.getBaoJiGroup();
    }

    @GetMapping("userList")
    @ApiOperation(value = "包机组已添加用户和未添加用户列表")
    public Response<Map<String,List<User>>> getUserList(@RequestParam int id ){
        Map<String,List<User>> resultMap = service.getUserList(id);
        return new Response().setII(1,resultMap.size()>0,resultMap,"包机组已添加用户和未添加用户列表");
    }

    @GetMapping("stationList")
    @ApiOperation(value = "已添加机房电源和未添加机房电源")
    public Response<Map<String,List<PowerInf>>> getStationList(@RequestParam int id){
        Map<String,List<PowerInf>> resultMap = service.getStationList(id);
        return new Response().setII(1,resultMap.size()>0,resultMap,"已添加机房锁和未添加机房电源");
    }
    @ApiOperation("查询包机组及包机组对应的用户和对应机房和电源")
    @GetMapping("getAllBaojiInf")
    public Response getAllBaojiInf() {
        return service.getAllBaojiInf();
    }

    @PostMapping("updateStationList")
    @ApiOperation(value = "包机组添加机房和电源|移除机房和电源",notes = "operationFlag为1,添加;-1,移除")
    public Response updateStationAndLockList(@RequestBody List<BaojigroupPower> baoJiGroupLockList, @RequestParam int operationFlag){
        service.updateStationAndLockList(baoJiGroupLockList,operationFlag);
        if(operationFlag==1) {
            return new Response().set(1, true,"添加成功");
        }else{
            return new Response().set(1,true,"移除成功");
        }

    }


    @PostMapping("updateUserList")
    @ApiOperation(value = "包机组添加用户|移除用户",notes = "operationFlag为1,添加;-1,移除")
    public Response updateUserList(@RequestBody List<BaojigroupUsr> baoJiGroupUserList, @RequestParam int operationFlag){
        service.updateUserList(baoJiGroupUserList, operationFlag);
        if (operationFlag == 1) {
            return new Response().set(1,true, "添加成功");
        } else {
            return new Response().set(1,true, "移除成功");
        }
    }
}