package com.klp.ems.controller;

import java.util.List;
import java.util.Arrays;

import com.klp.ems.domain.bo.EmsEnergyConsumptionQueryBo;
import com.klp.ems.domain.vo.EnergyChainRatioVo;
import com.klp.ems.domain.vo.SummaryDailyVo;
import com.klp.ems.domain.vo.SummaryMonthlyVo;
import lombok.RequiredArgsConstructor;
import javax.servlet.http.HttpServletResponse;
import javax.validation.constraints.*;

import org.flywaydb.core.internal.util.StringUtils;
import org.springframework.web.bind.annotation.*;
import org.springframework.validation.annotation.Validated;
import com.klp.common.annotation.RepeatSubmit;
import com.klp.common.annotation.Log;
import com.klp.common.core.controller.BaseController;
import com.klp.common.core.domain.PageQuery;
import com.klp.common.core.domain.R;
import com.klp.common.core.validate.AddGroup;
import com.klp.common.core.validate.EditGroup;
import com.klp.common.enums.BusinessType;
import com.klp.common.utils.poi.ExcelUtil;
import com.klp.ems.domain.vo.EmsEnergyConsumptionVo;
import com.klp.ems.domain.bo.EmsEnergyConsumptionBo;
import com.klp.ems.service.IEmsEnergyConsumptionService;
import com.klp.common.core.page.TableDataInfo;

/**
 * 能耗记录
 *
 * @author Joshi
 * @date 2025-09-28
 */
@Validated
@RequiredArgsConstructor
@RestController
@RequestMapping("/ems/energyConsumption")
public class EmsEnergyConsumptionController extends BaseController {

    private final IEmsEnergyConsumptionService iEmsEnergyConsumptionService;

    /**
     * 近期能耗趋势（测过了）
     */
    @GetMapping("/getEnergySummary")
    public R getEnergySummary(EmsEnergyConsumptionQueryBo queryBo){
        if (StringUtils.hasText(queryBo.getMonth())) {
            List<SummaryDailyVo> data = iEmsEnergyConsumptionService.getEnergySummaryDaily(queryBo.getMonth());
            return R.ok(data);
        }else if (StringUtils.hasText(queryBo.getYear())) {
            List<SummaryMonthlyVo> data = iEmsEnergyConsumptionService.getEnergySummaryMonthly(queryBo.getYear());
            return R.ok(data);
        } else {
            return R.fail("year 或 month 必须提供一个");
        }
    }

    /**
     * 获取能耗环比概况（没写完）
     */
    @GetMapping("/getEnergyChainRatio")
    public R<EnergyChainRatioVo> getEnergyChainRatio() {
        EnergyChainRatioVo energyChainRatioVo = iEmsEnergyConsumptionService.getEnergyChainRatio();
        return R.ok(energyChainRatioVo);
    }

    /**
     * 查询能耗记录列表
     */
    @GetMapping("/list")
    public TableDataInfo<EmsEnergyConsumptionVo> list(EmsEnergyConsumptionBo bo, PageQuery pageQuery) {
        return iEmsEnergyConsumptionService.queryPageList(bo, pageQuery);
    }

    /**
     * 导出能耗记录列表
     */
    @Log(title = "能耗记录", businessType = BusinessType.EXPORT)
    @PostMapping("/export")
    public void export(EmsEnergyConsumptionBo bo, HttpServletResponse response) {
        List<EmsEnergyConsumptionVo> list = iEmsEnergyConsumptionService.queryList(bo);
        ExcelUtil.exportExcel(list, "能耗记录", EmsEnergyConsumptionVo.class, response);
    }

    /**
     * 获取能耗记录详细信息
     *
     * @param energyConsumptionId 主键
     */
    @GetMapping("/{energyConsumptionId}")
    public R<EmsEnergyConsumptionVo> getInfo(@NotNull(message = "主键不能为空")
                                     @PathVariable Long energyConsumptionId) {
        return R.ok(iEmsEnergyConsumptionService.queryById(energyConsumptionId));
    }

    /**
     * 新增能耗记录
     */
    @Log(title = "能耗记录", businessType = BusinessType.INSERT)
    @RepeatSubmit()
    @PostMapping()
    public R<Void> add(@Validated(AddGroup.class) @RequestBody EmsEnergyConsumptionBo bo) {
        return toAjax(iEmsEnergyConsumptionService.insertByBo(bo));
    }

    /**
     * 修改能耗记录
     */
    @Log(title = "能耗记录", businessType = BusinessType.UPDATE)
    @RepeatSubmit()
    @PutMapping()
    public R<Void> edit(@Validated(EditGroup.class) @RequestBody EmsEnergyConsumptionBo bo) {
        return toAjax(iEmsEnergyConsumptionService.updateByBo(bo));
    }

    /**
     * 删除能耗记录
     *
     * @param energyConsumptionIds 主键串
     */
    @Log(title = "能耗记录", businessType = BusinessType.DELETE)
    @DeleteMapping("/{energyConsumptionIds}")
    public R<Void> remove(@NotEmpty(message = "主键不能为空")
                          @PathVariable Long[] energyConsumptionIds) {
        return toAjax(iEmsEnergyConsumptionService.deleteWithValidByIds(Arrays.asList(energyConsumptionIds), true));
    }
}