package com.ruoyi.web.controller.api;

import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.ruoyi.common.core.domain.R;
import com.ruoyi.common.utils.DateUtils;
import com.ruoyi.framework.web.service.TokenService;
import com.ruoyi.system.model.*;
import com.ruoyi.system.query.DataStatisticsQuery;
import com.ruoyi.system.service.*;
import com.ruoyi.system.vo.SupplierThirtyMoneyProcurementStatisticsVO;
import com.ruoyi.system.vo.SupplierThirtyProcurementStatisticsVO;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.util.CollectionUtils;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.math.BigDecimal;
import java.text.SimpleDateFormat;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.util.*;
import java.util.stream.Collectors;

@Api(tags = "数据统计")
@RestController
@RequestMapping("/data-statistics")
public class DataStatisticsController {

    private final TErpProcurementService erpProcurementService;
    private final TokenService tokenService;
    private final ISysUserService sysUserService;
    private final TCrmSupplierService crmSupplierService;
    private final TCrmClinicService crmClinicService;
    private final TSysAppUserService sysAppUserService;
    private final TCrmClinicPointsService crmClinicPointsService;

    @Autowired
    public DataStatisticsController(TErpProcurementService erpProcurementService, TokenService tokenService, ISysUserService sysUserService, TCrmSupplierService crmSupplierService, TCrmClinicService crmClinicService, TSysAppUserService sysAppUserService, TCrmClinicPointsService crmClinicPointsService) {
        this.erpProcurementService = erpProcurementService;
        this.tokenService = tokenService;
        this.sysUserService = sysUserService;
        this.crmSupplierService = crmSupplierService;
        this.crmClinicService = crmClinicService;
        this.sysAppUserService = sysAppUserService;
        this.crmClinicPointsService = crmClinicPointsService;
    }

    /**
     * 供应商采购统计
     */
    @ApiOperation(value = "供应商采购统计", notes = "orderTotal:总数;orderQuantity:订单数;orderMoney:订单金额;commissionMoney:分佣金额")
    @PostMapping(value = "/supplierProcurementStatistics")
    public R<Map<String, Object>> supplierProcurementStatistics(@Validated @RequestBody DataStatisticsQuery query) {

        Long userId = tokenService.getLoginUser().getUserId();

        String startTime = null;
        String endTime = null;
        switch (query.getTimeType()){
            case 1:
                startTime = LocalDate.now() + " 00:00:00";
                endTime = LocalDate.now() + " 23:59:59";
                break;
            case 2:
                startTime = LocalDate.now().minusDays(1) + " 00:00:00";
                endTime = LocalDate.now().minusDays(1) + " 23:59:59";
                break;
            case 3:
                startTime = LocalDate.now().minusDays(6) + " 00:00:00";
                endTime = LocalDate.now() + " 23:59:59";
                break;
            case 4:
                startTime = LocalDate.now().minusMonths(6) + " 00:00:00";
                endTime = LocalDate.now() + " 23:59:59";
                break;
            case 5:
                startTime = LocalDate.now().minusYears(1) + " 00:00:00";
                endTime = LocalDate.now() + " 23:59:59";
                break;
        }

        TCrmSupplier crmSupplier = crmSupplierService.getOne(Wrappers.lambdaQuery(TCrmSupplier.class)
                .eq(TCrmSupplier::getUserId, userId));

        Map<String, Object> map = new HashMap<>();

        List<TErpProcurement> orderTotal = erpProcurementService.list(Wrappers.lambdaQuery(TErpProcurement.class)
                .eq(TErpProcurement::getSupplierId, crmSupplier.getId())
                .in(TErpProcurement::getStatus, 3,4,5));
        map.put("orderTotal", orderTotal.size()); // 总数
        List<TErpProcurement> orderQuantity = erpProcurementService.list(Wrappers.lambdaQuery(TErpProcurement.class)
                .eq(TErpProcurement::getSupplierId, crmSupplier.getId())
                .between(TErpProcurement::getPayTime, startTime, endTime)
                .in(TErpProcurement::getStatus, 3,4,5));
        map.put("orderQuantity", orderQuantity.size()); // 订单数

        if(CollectionUtils.isEmpty(orderTotal)){
            map.put("orderMoney", 0); // 订单金额
            map.put("commissionMoney", 0); // 分佣金额
        }else {
            BigDecimal orderMoney = orderQuantity.stream().map(TErpProcurement::getPayMoney).filter(Objects::nonNull).reduce(BigDecimal::add).get();
            map.put("orderMoney", orderMoney);
            BigDecimal commissionMoney = orderQuantity.stream().map(TErpProcurement::getSupplierMoney).filter(Objects::nonNull).reduce(BigDecimal::add).get();
            map.put("commissionMoney", commissionMoney);
        }
        return R.ok(map);
    }

    /**
     * 供应商采购统计
     */
    @ApiOperation(value = "供应商最近额30天采购统计",notes = "supplierThirtyProcurementStatistics:订单，supplierThirtyMoneyProcurementStatistics：金额")
    @PostMapping(value = "/supplierThirtyProcurementStatistics")
    public R<Map<String,Object>> supplierThirtyProcurementStatistics() {
        Long userId = tokenService.getLoginUser().getUserId();
        Integer roleType = tokenService.getLoginUser().getUser().getRoleType();

        String startTime = LocalDate.now().minusDays(29) + " 00:00:00";
        String endTime = LocalDate.now() + " 23:59:59";

        String supplierAndClinicId = null;
        if(roleType == 4){
            TCrmSupplier crmSupplier = crmSupplierService.getOne(Wrappers.lambdaQuery(TCrmSupplier.class)
                    .eq(TCrmSupplier::getUserId, userId));
            supplierAndClinicId = crmSupplier.getId();
        }else {
            TCrmClinic crmClinic = crmClinicService.getOne(Wrappers.lambdaQuery(TCrmClinic.class)
                    .eq(TCrmClinic::getUserId, userId));
            supplierAndClinicId = crmClinic.getId();
        }

        // 订单数
        List<SupplierThirtyProcurementStatisticsVO> supplierThirtyProcurementStatistics = erpProcurementService.supplierThirtyProcurementStatistics(supplierAndClinicId, startTime, endTime,roleType);

        // 金额
        List<SupplierThirtyMoneyProcurementStatisticsVO> supplierThirtyMoneyProcurementStatistics = erpProcurementService.supplierThirtyMoneyProcurementStatistics(supplierAndClinicId, startTime, endTime,roleType);

        LocalDateTime localDateTime = LocalDateTime.now().minusDays(29);
        SimpleDateFormat format = new SimpleDateFormat("MM.dd");
        for (int i = 0; i < 30; i++) {
            LocalDateTime timeNow = localDateTime.plusDays(i);
            List<SupplierThirtyProcurementStatisticsVO> vos = supplierThirtyProcurementStatistics.stream().filter(vo -> timeNow.toLocalDate().isEqual(vo.getPayTime())).collect(Collectors.toList());
            if(CollectionUtils.isEmpty(vos)){
                SupplierThirtyProcurementStatisticsVO vo = new SupplierThirtyProcurementStatisticsVO();
                String time = format.format(DateUtils.localDateTimeToDate(timeNow));
                vo.setTime(time);
                vo.setOrderCount(0);
                supplierThirtyProcurementStatistics.add(vo);
            }
            List<SupplierThirtyMoneyProcurementStatisticsVO> moneyVOS = supplierThirtyMoneyProcurementStatistics.stream().filter(vo -> timeNow.toLocalDate().isEqual(vo.getPayTime())).collect(Collectors.toList());
            if(CollectionUtils.isEmpty(moneyVOS)){
                SupplierThirtyMoneyProcurementStatisticsVO vo = new SupplierThirtyMoneyProcurementStatisticsVO();
                String time = format.format(DateUtils.localDateTimeToDate(timeNow));
                vo.setTime(time);
                vo.setMoneyTotal(BigDecimal.ZERO);
                supplierThirtyMoneyProcurementStatistics.add(vo);
            }
        }
        // 根据时间排序
        supplierThirtyProcurementStatistics.sort(Comparator.comparing(SupplierThirtyProcurementStatisticsVO::getTime));
        supplierThirtyMoneyProcurementStatistics.sort(Comparator.comparing(SupplierThirtyMoneyProcurementStatisticsVO::getTime));
        Map<String,Object> map = new HashMap<>();
        map.put("supplierThirtyProcurementStatistics", supplierThirtyProcurementStatistics);
        map.put("supplierThirtyMoneyProcurementStatistics", supplierThirtyMoneyProcurementStatistics);
        return R.ok(map);
    }

    /**
     * 诊所患者统计
     */
    @ApiOperation(value = "诊所患者统计", notes = "userCount:患者总数;diseaseCount:慢性病患者数")
    @PostMapping(value = "/clinicPatientStatistics")
    public R<Map<String, Integer>> clinicPatientStatistics() {

        Long userId = tokenService.getLoginUser().getUserId();

        TCrmClinic crmClinic = crmClinicService.getOne(Wrappers.lambdaQuery(TCrmClinic.class)
                .eq(TCrmClinic::getUserId, userId));

        List<TSysAppUser> sysAppUsers = sysAppUserService.list(Wrappers.lambdaQuery(TSysAppUser.class)
                .eq(TSysAppUser::getClinicId, crmClinic.getId()));
        Map<String, Integer> map = new HashMap<>();
        map.put("userCount", sysAppUsers.size());
        map.put("diseaseCount", sysAppUsers.stream().filter(sysAppUser -> sysAppUser.getChronicDiseaseId() != null).collect(Collectors.toList()).size());
        return R.ok(map);
    }

    /**
     * 诊所积分统计
     */
    @ApiOperation(value = "诊所积分统计", notes = "remainingPoints:剩余积分;usedPoints:已用积分")
    @PostMapping(value = "/clinicPointsStatistics")
    public R<Map<String, Integer>> clinicPointsStatistics() {

        Long userId = tokenService.getLoginUser().getUserId();

        TCrmClinic crmClinic = crmClinicService.getOne(Wrappers.lambdaQuery(TCrmClinic.class)
                .eq(TCrmClinic::getUserId, userId));

        List<TCrmClinicPoints> crmClinicPoints = crmClinicPointsService.clinicPointsStatistics(crmClinic.getId(),LocalDate.now() + " 23:59:59");
        Map<String, Integer> map = new HashMap<>();
        if(CollectionUtils.isEmpty(crmClinicPoints)){
            map.put("remainingPoints", 0);
            map.put("usedPoints", 0);
        }else {
            int pointsTotal = crmClinicPoints.stream().mapToInt(TCrmClinicPoints::getPoints).sum();
            int usePoints = crmClinicPoints.stream().mapToInt(TCrmClinicPoints::getUsePoints).sum();
            map.put("remainingPoints", pointsTotal-usePoints);
            map.put("usedPoints", usePoints);
        }
        return R.ok(map);
    }

    public static void main(String[] args) {
        System.out.println(LocalDate.now().minusDays(30));
    }

}