package com.ruoyi.system.service.impl;

import cn.hutool.core.bean.BeanUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ruoyi.common.utils.SecurityUtils;
import com.ruoyi.system.dto.asset.AssetAdDTO;
import com.ruoyi.system.mapper.AssetAdMapper;
import com.ruoyi.system.mapper.AssetAdRentalRecordMapper;
import com.ruoyi.system.model.AssetAd;
import com.ruoyi.system.model.AssetAdRentalRecord;
import com.ruoyi.system.query.AssetAdQuery;
import com.ruoyi.system.service.AssetAdService;
import com.ruoyi.system.service.AssetTypeService;
import com.ruoyi.system.vo.asset.AssetAdDetailVO;
import com.ruoyi.system.vo.asset.AssetAdRentalRecordVO;
import com.ruoyi.system.vo.asset.AssetAdVO;
import lombok.RequiredArgsConstructor;
import org.apache.commons.lang3.StringUtils;
import org.springframework.context.annotation.Lazy;
import org.springframework.stereotype.Service;

import java.time.LocalDate;
import java.time.format.DateTimeFormatter;
import java.time.temporal.ChronoUnit;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
 * <p>
 * 广告无形资产表 服务实现类
 * </p>
 *
 * @author WuGuanFengYue
 * @since 2025-09-15
 */
@Service
@RequiredArgsConstructor(onConstructor_ = {@Lazy})
public class AssetAdServiceImpl extends ServiceImpl<AssetAdMapper, AssetAd> implements AssetAdService {
    private final AssetTypeService assetTypeService;
    private final AssetAdRentalRecordMapper rentalRecordMapper;

    @Override
    public void addAssetAd(AssetAdDTO dto) {
        AssetAd assetAd = BeanUtil.copyProperties(dto, AssetAd.class);
        assetAd.setAssetTypeId(2);
        assetAd.setAssetCode(generateCode());
        assetAd.setCreateBy(SecurityUtils.getLoginUser().getUser().getNickName());
        save(assetAd);
    }

    private String generateCode() {
        // 根据资产类型ID获取资产编码前缀
        String typeCodePrefix = assetTypeService.getAssetCodePrefix(2);
        LocalDate today = LocalDate.now();
        String dateStr = today.format(DateTimeFormatter.ofPattern("yyyyMMdd"));
        String prefix = typeCodePrefix + dateStr + "-";

        // 查询当天该类型资产的数量
        Long count = this.lambdaQuery()
                .like(AssetAd::getAssetCode, prefix)
                .ge(AssetAd::getCreateTime, today.atStartOfDay())
                .lt(AssetAd::getCreateTime, today.plusDays(1).atStartOfDay())
                .count();

        int sequence = (count != null ? count.intValue() : 0) + 1;
        String sequenceStr = String.format("%04d", sequence);

        return prefix + sequenceStr;
    }

    @Override
    public void editAssetAd(AssetAdDTO dto) {
        AssetAd assetAd = BeanUtil.copyProperties(dto, AssetAd.class);
        assetAd.setUpdateBy(SecurityUtils.getLoginUser().getUser().getNickName());
        updateById(assetAd);
    }

    @Override
    public IPage<AssetAdVO> getPageList(AssetAdQuery query) {
        // 1. 分页查询 AssetAd
        Page<AssetAd> page = new Page<>(query.getPageNum(), query.getPageSize());
        LambdaQueryWrapper<AssetAd> queryWrapper = new LambdaQueryWrapper<AssetAd>()
            .like(StringUtils.isNotBlank(query.getKeyword()), AssetAd::getAssetName, query.getKeyword())
            .or()
            .like(StringUtils.isNotBlank(query.getKeyword()), AssetAd::getAssetCode, query.getKeyword())
            .orderByDesc(AssetAd::getCreateTime);

        Page<AssetAd> assetPage = this.page(page, queryWrapper);

        // 2. 如果没有数据，返回空分页
        if (assetPage.getRecords().isEmpty()) {
            return new Page<>(query.getPageNum(), query.getPageSize());
        }

        // 3. 获取所有资产ID
        List<Integer> assetAdIds = assetPage.getRecords().stream()
            .map(AssetAd::getId)
            .collect(Collectors.toList());

        // 4. 查询相关租赁记录
        LocalDate today = LocalDate.now();
        List<AssetAdRentalRecord> allRentalRecords = rentalRecordMapper.selectList(
            new LambdaQueryWrapper<AssetAdRentalRecord>()
                .in(AssetAdRentalRecord::getAssetAdId, assetAdIds)
                    .ge(AssetAdRentalRecord::getStartDate, today)
                    .le(AssetAdRentalRecord::getEndDate,today)
                .orderByDesc(AssetAdRentalRecord::getCreateTime)
        );

        // 5. 转换为Map
        Map<Integer, AssetAdRentalRecord> currentRentalMap = allRentalRecords.stream()
                .collect(Collectors.toMap(AssetAdRentalRecord::getAssetAdId, Function.identity()));

        // 6. 组装VO数据
        List<AssetAdVO> voList = new ArrayList<>();
        for (AssetAd asset : assetPage.getRecords()) {
            AssetAdVO vo = BeanUtil.copyProperties(asset, AssetAdVO.class);

            // 设置租赁相关信息
            AssetAdRentalRecord currentRental = currentRentalMap.get(asset.getId());
            if (currentRental != null) {
                vo.setCurrentUseBy(currentRental.getUseBy());
                vo.setRentalPrice(currentRental.getRentalPrice());
                vo.setUseDuration(calculateUseDuration(currentRental.getStartDate(), currentRental.getEndDate()));
            }

            voList.add(vo);
        }

        // 7. 返回分页结果
        Page<AssetAdVO> resultPage = new Page<>(query.getPageNum(), query.getPageSize());
        resultPage.setRecords(voList);
        resultPage.setTotal(assetPage.getTotal());
        resultPage.setSize(assetPage.getSize());
        resultPage.setCurrent(assetPage.getCurrent());
        resultPage.setPages(assetPage.getPages());

        return resultPage;
    }

    /**
     * 计算使用期限（月）
     */
    private Integer calculateUseDuration(LocalDate startDate, LocalDate endDate) {
        if (startDate == null || endDate == null) {
            return null;
        }
        // 计算月份差值
        long months = ChronoUnit.MONTHS.between(startDate, endDate);
        return (int) months;
    }

    @Override
    public AssetAdDetailVO getDetail(Integer id) {
        // 1. 查询资产基础信息
        AssetAd assetAd = getById(id);
        if (assetAd == null) {
            return null;
        }

        // 2. 查询租赁记录
        List<AssetAdRentalRecord> rentalRecords = rentalRecordMapper.selectList(
            new LambdaQueryWrapper<AssetAdRentalRecord>()
                .eq(AssetAdRentalRecord::getAssetAdId, id)
                .orderByDesc(AssetAdRentalRecord::getStartDate)
        );

        // 3. 转换为AssetAdDetailVO
        AssetAdDetailVO detailVO = BeanUtil.copyProperties(assetAd, AssetAdDetailVO.class);

        // 4. 转换租赁记录为VO列表
        List<AssetAdRentalRecordVO> rentalRecordVOList = BeanUtil.copyToList(rentalRecords, AssetAdRentalRecordVO.class);
        
        detailVO.setRentalRecordVOList(rentalRecordVOList);

        return detailVO;
    }

    @Override
    public void importAssetAd(List<AssetAdDTO> list) {
        List<AssetAd> collect = list.stream().map(item -> {
            AssetAd assetAd = BeanUtil.copyProperties(item, AssetAd.class);
            assetAd.setAssetTypeId(2);
            assetAd.setAssetCode(generateCode());
            assetAd.setCreateBy(SecurityUtils.getLoginUser().getUser().getNickName());
            return assetAd;
        }).collect(Collectors.toList());
        saveBatch(collect);
    }
}