package com.whyc.controller;

import com.whyc.dto.BattDto;
import com.whyc.dto.InfoDto;
import com.whyc.dto.Response;
import com.whyc.factory.InfoFactory;
import com.whyc.pojo.db_station.BattInf;
import com.whyc.pojo.db_station.PowerInf;
import com.whyc.pojo.db_user.User;
import com.whyc.service.BaojigroupService;
import com.whyc.service.BattInfService;
import com.whyc.service.PowerInfService;
import com.whyc.util.ActionUtil;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

@RestController
@Api(tags = "机房，电源，电池组管理")
@RequestMapping("info")
public class InfoContoller {

    @Autowired
    private PowerInfService pinfService;

    @Autowired
    private BattInfService binfService;

    @Autowired(required = false)
    private BaojigroupService bjService;

    @ApiOperation(value = "新建机房，电源，电池组信息")
    @PostMapping("addInfo")
    public Response addInfo(@RequestBody InfoDto info){
        //先校验当前用户是否存在包机组不存在则不让添加电源
        User user= ActionUtil.getUser();
        int flag=bjService.checkUserBaojiGroup(user.getId());
        String msg="添加成功";
        boolean bl=true;
        if(flag==0){
            bl=false;
            msg="当前用户不存在包机组";
        }
        try {
            PowerInf pinf=InfoFactory.infoToPower(info);
            pinfService.addPower(pinf);
            if(info.getAddBinfFlag()==1){
                BattInf binf=InfoFactory.infoToBatt(info);
                binf.setStationId(pinf.getStationId());
                binf.setPowerId(pinf.getPowerId());
                binfService.addDev(binf);
            }
        } catch (Exception e) {
            e.printStackTrace();
            msg="添加失败";
            bl=false;
        }
        return new Response().set(1,bl,msg);
    }

    @ApiOperation(value = "设备下添加电池组")
    @PostMapping("addBatt")
    public Response addBatt(@RequestBody BattInf binf){
        binfService.addBatt(binf);
        return new Response().set(1,true);
    }

    @ApiOperation(value = "删除电池组")
    @GetMapping("delBatt")
    public Response delBatt(@RequestParam Integer stationId,@RequestParam Integer powerId,@RequestParam(required = false) Integer battgroupId){
        return binfService.delBatt(stationId, powerId, battgroupId);
    }

    @ApiOperation(value = "编辑机房，电源，电池组信息")
    @PostMapping("updateInfo")
    public Response updateInfo(@RequestBody InfoDto info){
        return binfService.updateInfo(info);
    }

    @ApiOperation(value = "查询机房，电源，电池组信息")
    @PostMapping("getInfo")
    public Response getInfo(@RequestBody BattDto dto){
        return binfService.getInfo(dto);
    }
}