package com.by4cloud.platformx.device.controller;

import cn.hutool.core.util.ArrayUtil;
import cn.hutool.core.collection.CollUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.by4cloud.platformx.common.core.util.R;
import com.by4cloud.platformx.common.log.annotation.SysLog;
import com.by4cloud.platformx.device.constant.MaxSizeContant;
import com.by4cloud.platformx.device.entity.*;
import com.by4cloud.platformx.device.service.*;
import com.by4cloud.platformx.device.util.NumUtils;
import com.github.yulichang.adapter.base.ITableInfoAdapter;
import org.springframework.security.access.prepost.PreAuthorize;
import com.by4cloud.platformx.common.excel.annotation.ResponseExcel;
import io.swagger.v3.oas.annotations.security.SecurityRequirement;
import org.springdoc.api.annotations.ParameterObject;
import org.springframework.http.HttpHeaders;
import io.swagger.v3.oas.annotations.tags.Tag;
import io.swagger.v3.oas.annotations.Operation;
import lombok.RequiredArgsConstructor;
import org.springframework.web.bind.annotation.*;

import java.util.*;
import java.util.stream.Collectors;

/**
 * 年度投资计划主表
 *
 * @author kdq
 * @date 2025-03-28 11:06:50
 */
@RestController
@RequiredArgsConstructor
@RequestMapping("/investmentPlan" )
@Tag(description = "investmentPlan" , name = "年度投资计划主表管理" )
@SecurityRequirement(name = HttpHeaders.AUTHORIZATION)
public class InvestmentPlanController {

    private final  InvestmentPlanService investmentPlanService;
    private final  InvestmentPlanItemService investmentPlanItemService;
    private final DeviceDemandPlanService deviceDemandPlanService;
    private final DeviceDemandSubService deviceDemandSubService;
    private final JcMaxSizeService maxSizeService;
    private final DeviceService deviceService;

    /**
     * 分页查询
     * @param page 分页对象
     * @param investmentPlan 年度投资计划主表
     * @return
     */
    @Operation(summary = "分页查询" , description = "分页查询" )
    @GetMapping("/page" )
    @PreAuthorize("@pms.hasPermission('device_investmentPlan_view')" )
    public R getInvestmentPlanPage(@ParameterObject Page page, @ParameterObject InvestmentPlan investmentPlan) {
        LambdaQueryWrapper<InvestmentPlan> wrapper = Wrappers.lambdaQuery();
		wrapper.eq(investmentPlan.getYear()!=null,InvestmentPlan::getYear,investmentPlan.getYear());
		wrapper.like(StringUtils.isNotBlank(investmentPlan.getName()),InvestmentPlan::getName,investmentPlan.getName());
		wrapper.orderByDesc(InvestmentPlan::getCreateTime);
        return R.ok(investmentPlanService.page(page, wrapper));
    }


    /**
     * 通过id查询年度投资计划主表
     * @param id id
     * @return R
     */
    @Operation(summary = "通过id查询" , description = "通过id查询" )
    @GetMapping("/{id}" )
    @PreAuthorize("@pms.hasPermission('device_investmentPlan_view')" )
    public R getById(@PathVariable("id" ) Long id) {
		Map<String,Object> map = new HashMap<>();
		InvestmentPlan plan = investmentPlanService.getById(id);
		QueryWrapper<InvestmentPlanItem> wrapper = new QueryWrapper<>();
		wrapper.lambda()
				.eq(InvestmentPlanItem::getPlanId,plan.getId())
				.orderByAsc(InvestmentPlanItem::getMonth);
		List<InvestmentPlanItem> list = investmentPlanItemService.list(wrapper);
		List<Map<String,Object>> resultList = new ArrayList<>();
		for (Map.Entry<Integer, List<InvestmentPlanItem>> listEntry : list.stream().collect(Collectors.groupingBy(item -> item.getMonth())).entrySet()) {
			Integer month = listEntry.getKey();
			List<InvestmentPlanItem> value = listEntry.getValue();
			Map<String,Object> entry = new HashMap<>();
			entry.put("month",month);
			entry.put("value",value);
			resultList.add(entry);
		}
		map.put("plan",plan);
		map.put("items",resultList);
		return R.ok(map);
    }

    /**
     * 新增年度投资计划主表
     * @param investmentPlan 年度投资计划主表
     * @return R
     */
    @Operation(summary = "新增年度投资计划主表" , description = "新增年度投资计划主表" )
    @SysLog("新增年度投资计划主表" )
    @PostMapping
    @PreAuthorize("@pms.hasPermission('device_investmentPlan_add')" )
    public R save(@RequestBody InvestmentPlan investmentPlan) {
		List<DeviceDemandSub> subs = new ArrayList<>();
		investmentPlan.setNumber(maxSizeService.nextNo(MaxSizeContant.TZPLAN_NUM));
		investmentPlanService.save(investmentPlan);

		String planIds = investmentPlan.getPlanIds();
		String[] split = planIds.split(",");
		for (String s : split) {
			DeviceDemandPlan demandPlan = deviceDemandPlanService.getById(Long.parseLong(s));
			QueryWrapper<DeviceDemandPlan> wrapper = new QueryWrapper<>();
			wrapper.lambda()
					.eq(DeviceDemandPlan::getPlanId,demandPlan.getId());
			List<DeviceDemandPlan> list = deviceDemandPlanService.list(wrapper);
			List<Long> collect = list.stream().map(DeviceDemandPlan::getId).collect(Collectors.toList());
			for (Long l : collect) {
				//合并子项
				List<DeviceDemandSub> byPlanId = deviceDemandSubService.getByPlanId(l);
				subs.addAll(byPlanId);
			}
		}
		List<InvestmentPlanItem> items = new ArrayList<>();
		for (Map.Entry<Integer, List<DeviceDemandSub>> integerListEntry : subs.stream().collect(Collectors.groupingBy(item -> item.getMonth())).entrySet()) {
			Integer month = integerListEntry.getKey();
			List<DeviceDemandSub> subList = integerListEntry.getValue();
			for (Map.Entry<Long, List<DeviceDemandSub>> deviceEntry : subList.stream().collect(Collectors.groupingBy(item -> item.getDeviceId())).entrySet()) {
				Long deviceId = deviceEntry.getKey();
				List<DeviceDemandSub> value = deviceEntry.getValue();
				double total = value.stream().mapToDouble(DeviceDemandSub::getPrice).sum();

				Device device = deviceService.getById(deviceId);
				InvestmentPlanItem item = new InvestmentPlanItem();
				item.setYear(value.get(0).getYear());
				item.setPlanId(investmentPlan.getId());
				item.setMonth(month);
				item.setDeviceId(deviceId);
				item.setNumber(device.getNumber());
				item.setSpecification(device.getSpecification());
				item.setNum(value.size());
				item.setUnit(device.getUnit());
				item.setPrice(device.getPrice());
				item.setAmount(total);
				items.add(item);
			}
		}
		investmentPlanItemService.saveBatch(items);
		return R.ok();
    }

    /**
     * 修改年度投资计划主表
     * @param investmentPlan 年度投资计划主表
     * @return R
     */
    @Operation(summary = "修改年度投资计划主表" , description = "修改年度投资计划主表" )
    @SysLog("修改年度投资计划主表" )
    @PutMapping
    @PreAuthorize("@pms.hasPermission('device_investmentPlan_edit')" )
    public R updateById(@RequestBody InvestmentPlan investmentPlan) {
        return R.ok(investmentPlanService.updateById(investmentPlan));
    }

    /**
     * 通过id删除年度投资计划主表
     * @param ids id列表
     * @return R
     */
    @Operation(summary = "通过id删除年度投资计划主表" , description = "通过id删除年度投资计划主表" )
    @SysLog("通过id删除年度投资计划主表" )
    @DeleteMapping
    @PreAuthorize("@pms.hasPermission('device_investmentPlan_del')" )
    public R removeById(@RequestBody Long[] ids) {
        return R.ok(investmentPlanService.removeBatchByIds(CollUtil.toList(ids)));
    }


    /**
     * 导出excel 表格
     * @param investmentPlan 查询条件
   	 * @param ids 导出指定ID
     * @return excel 文件流
     */
    @ResponseExcel
    @GetMapping("/export")
    @PreAuthorize("@pms.hasPermission('device_investmentPlan_export')" )
    public List<InvestmentPlan> export(InvestmentPlan investmentPlan,Long[] ids) {
        return investmentPlanService.list(Wrappers.lambdaQuery(investmentPlan).in(ArrayUtil.isNotEmpty(ids), InvestmentPlan::getId, ids));
    }
}