package com.zksy.controller.info;

import cn.hutool.core.date.DateTime;
import com.zksy.info.domain.EQuarterlyElectricityBill;
import com.zksy.info.domain.EQuarterlyWaterUsage;
import com.zksy.info.service.EQuarterlyElectricityBillService;
import com.zksy.info.service.EQuarterlyWaterUsageService;
import com.zksy.utils.AjaxResult;
import com.zksy.utils.ExcelExportUtil;
import com.zksy.utils.ExcelImportUtil;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletResponse;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.util.Arrays;

/**
 * @author Administrator
 * @version 1.0
 * @project enterprise-assets-service
 * @description 用电控制层
 * @date 2025/6/24 14:33:13
 */
@RestController
@RequestMapping("/equarterlyElectricityBill")
@Api(tags = "用电信息",description = "用电信息desc")
public class EQuarterlyElectricityBillController {
    @Autowired
    private EQuarterlyElectricityBillService service;
    @GetMapping("/findByPage")
    @ApiOperation(value = "用电信息查询分页")
    public AjaxResult findByPage(long pageNum, long pageSize, String enterpriseName,String unifiedSocialCreditCode,String gridAccountNumber,String year,String arrearExists){
        return AjaxResult.success(service.findByPage(pageNum, pageSize, enterpriseName,unifiedSocialCreditCode,gridAccountNumber,year,arrearExists));
    }
    @GetMapping("/getEQuarterlyWaterUsageList")
    @ApiOperation(value = "用电信息查询")
    public AjaxResult getEQuarterlyWaterUsageList(String enterpriseName,String unifiedSocialCreditCode,String gridAccountNumber,String year,String arrearExists){
        return AjaxResult.success(service.getEQuarterlyElectricityBillList(enterpriseName,unifiedSocialCreditCode,gridAccountNumber,year,arrearExists));
    }
    @GetMapping("/getById/{id}")
    @ApiOperation(value = "根据Id查询用电信息")
    public AjaxResult getById(@PathVariable String id){
        return AjaxResult.success(service.getById(id));
    }
    @PostMapping("/save")
    @ApiOperation(value = "用电信息保存")
    public AjaxResult save(@RequestBody EQuarterlyElectricityBill equarterlyElectricityBill) {
        return service.save(equarterlyElectricityBill) ? AjaxResult.success(equarterlyElectricityBill) : AjaxResult.error("保存失败");
    }
    @PostMapping("/update")
    @ApiOperation(value = "用电信息修改")
    public AjaxResult update(@RequestBody EQuarterlyElectricityBill equarterlyElectricityBill) {
        equarterlyElectricityBill.setUpdateTime(LocalDateTime.now());
        return service.updateById(equarterlyElectricityBill) ? AjaxResult.success(equarterlyElectricityBill) : AjaxResult.error("修改失败");
    }
    @PostMapping("/deleteBatch")
    @ApiOperation(value = "用电信息删除")
    public AjaxResult delete(@RequestBody String[] ids) {
        return service.removeBatchByIds(Arrays.asList(ids)) ? AjaxResult.success("删除成功") : AjaxResult.error("删除失败");
    }
    @PostMapping("/importData")
    @ApiOperation(value = "导入数据")
    public AjaxResult importData(MultipartFile file) {
        return AjaxResult.success(service.saveOrUpdateBatch(ExcelImportUtil.importExcel(file, EQuarterlyElectricityBill.class)));
    }
    @GetMapping("/exportData")
    @ApiOperation(value = "导出数据")
    public void exportData(HttpServletResponse response,String enterpriseName,String unifiedSocialCreditCode,String gridAccountNumber,String year,String arrearExists) {
        ExcelExportUtil.exportExcel(response,service.getEQuarterlyElectricityBillList(enterpriseName,unifiedSocialCreditCode,gridAccountNumber,year,arrearExists), EQuarterlyElectricityBill.class, "用电信息","用电信息");
    }

}
