package com.zzg.web.controller.state;

import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.zzg.common.constant.state.UrlConstants;
import com.zzg.common.core.domain.AjaxResult;
import com.zzg.common.core.domain.entity.state.StateProjectCompensateStandard;
import com.zzg.system.domain.vo.StateProjectCompensateStandardVO;
import com.zzg.system.service.state.StateProjectCompensateStandardService;
import lombok.RequiredArgsConstructor;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

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

@RestController
@RequiredArgsConstructor
public class StateProjectCompensateStandardController {

    private final StateProjectCompensateStandardService stateProjectCompensateStandardService;
    //获取补偿标准列表
    @GetMapping(UrlConstants.STATE_PROJECT_COMPENSATE_STANDARD_LIST)
    public AjaxResult<Map<String, List<StateProjectCompensateStandardVO>>> getAll(
            @RequestParam(required = false) String standardName,
            @RequestParam(required = false) Integer stopFlag,
            @RequestParam(required = false) Integer compensateType
    ) {
        return AjaxResult.success(stateProjectCompensateStandardService.getCompensateStandardMap(standardName, stopFlag, compensateType));
    }
    //创建补偿标准
    @PostMapping(UrlConstants.STATE_PROJECT_COMPENSATE_STANDARD_ADD)
    public AjaxResult<Boolean> create(@RequestBody StateProjectCompensateStandard stateProjectCompensateStandard) {
        return AjaxResult.success(stateProjectCompensateStandardService.createCompensatesStandard(stateProjectCompensateStandard));
    }
    //修改补偿标准
    @PostMapping(UrlConstants.STATE_PROJECT_COMPENSATE_STANDARD_UPDATE)
    public AjaxResult<Boolean> update(@RequestBody StateProjectCompensateStandard stateProjectCompensateStandard) {
        return AjaxResult.success(stateProjectCompensateStandardService.saveOrUpdate(stateProjectCompensateStandard));
    }
    //修改补偿标准名字（同时更新项目中的标准名称）
    @PostMapping(UrlConstants.STATE_PROJECT_COMPENSATE_STANDARD_UPDATE_NAME)
    public AjaxResult<Boolean> updateName(@RequestBody StateProjectCompensateStandard stateProjectCompensateStandard) {
        return AjaxResult.success(stateProjectCompensateStandardService.updateName(stateProjectCompensateStandard));
    }
    //停用补偿标准
    @PostMapping(UrlConstants.STATE_PROJECT_COMPENSATE_STANDARD_STOP)
    public AjaxResult<Boolean> stop(@RequestParam String standardName, @RequestParam Integer compensateType) {
        LambdaUpdateWrapper<StateProjectCompensateStandard> wrapper = new LambdaUpdateWrapper<>();
        wrapper.eq(StateProjectCompensateStandard::getCompensateType, compensateType)
                .eq(StateProjectCompensateStandard::getStandardName, standardName)
                .set(StateProjectCompensateStandard::getStopFlag, 0);
        return AjaxResult.success(stateProjectCompensateStandardService.update(wrapper));
    }
    //启用补偿标准
    @PostMapping(UrlConstants.STATE_PROJECT_COMPENSATE_STANDARD_START)
    public AjaxResult<Boolean> start(@RequestParam String standardName, @RequestParam Integer compensateType) {
        LambdaUpdateWrapper<StateProjectCompensateStandard> wrapper = new LambdaUpdateWrapper<>();
        wrapper.eq(StateProjectCompensateStandard::getCompensateType, compensateType)
                .eq(StateProjectCompensateStandard::getStandardName, standardName)
                .set(StateProjectCompensateStandard::getStopFlag, 1);
        return AjaxResult.success(stateProjectCompensateStandardService.update(wrapper));
    }
    //导入补偿标准
    @PostMapping(UrlConstants.STATE_PROJECT_COMPENSATE_STANDARD_IMPORT)
    public AjaxResult<Boolean> importExcelFile(@RequestParam("file") MultipartFile file) {
        return AjaxResult.success(stateProjectCompensateStandardService.importByExcel(file));
    }
}