package cn.lili.modules.lmk.service.impl;

import cn.lili.common.exception.ServiceException;
import cn.lili.modules.lmk.domain.entity.CouponVirtual;
import cn.lili.modules.lmk.domain.entity.StoreCouponSingle;
import cn.lili.modules.lmk.enums.general.GenerateCouponStausEnum;
import cn.lili.modules.lmk.enums.general.StoreCouponStausEnum;
import cn.lili.modules.lmk.service.StoreCouponSingleService;
import cn.lili.modules.order.order.entity.enums.ClaimStatusEnum;
import cn.lili.modules.promotion.entity.dos.Coupon;
import cn.lili.modules.promotion.service.CouponService;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import cn.lili.modules.lmk.domain.entity.StoreCoupon;
import cn.lili.modules.lmk.mapper.StoreCouponMapper;
import cn.lili.modules.lmk.service.StoreCouponService;
import cn.lili.base.Result;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import cn.lili.modules.lmk.domain.form.StoreCouponForm;
import cn.lili.modules.lmk.domain.vo.StoreCouponVO;
import cn.lili.modules.lmk.domain.query.StoreCouponQuery;
import org.redisson.api.RLock;
import org.redisson.api.RedissonClient;
import org.springframework.stereotype.Service;
import lombok.RequiredArgsConstructor;
import cn.lili.utils.PageUtil;
import org.springframework.beans.BeanUtils;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.transaction.support.TransactionSynchronization;
import org.springframework.transaction.support.TransactionSynchronizationManager;
import org.springframework.util.Assert;

import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

/**
 * 店铺优惠卷对应关系 服务实现类
 *
 * @author peng
 * @since 2025-09-25
 */
@Service
@RequiredArgsConstructor
public class StoreCouponServiceImpl extends ServiceImpl<StoreCouponMapper, StoreCoupon> implements StoreCouponService {

    private final StoreCouponMapper storeCouponMapper;
    private final RedissonClient redissonClient;
    private static final String STORE_COUPON_GENERATE = "store_coupon_generate:";
    private final CouponService couponService;
    private final StoreCouponSingleService storeCouponSingleService;

    /**
     * 添加
     * @param form
     * @return
     */
    @Override
    public Result add(StoreCouponForm form) {
        StoreCoupon entity = StoreCouponForm.getEntityByForm(form, null);
        baseMapper.insert(entity);
        return Result.ok("添加成功");
    }

    /**
     * 修改
     * @param form
     * @return
     */
    @Override
    public Result update(StoreCouponForm form) {
        StoreCoupon entity = baseMapper.selectById(form.getId());

        // 为空抛IllegalArgumentException，做全局异常处理
        Assert.notNull(entity, "记录不存在");
        BeanUtils.copyProperties(form, entity);
        baseMapper.updateById(entity);
        return Result.ok("修改成功");
    }

    /**
     * 批量删除
     * @param ids
     * @return
     */
    @Override
    public Result remove(List<String> ids) {
        baseMapper.deleteBatchIds(ids);
        return Result.ok("删除成功");
    }

    /**
     * id删除
     * @param id
     * @return
     */
    @Override
    public Result removeById(String id) {
        baseMapper.deleteById(id);
        return Result.ok("删除成功");
    }

    /**
     * 分页查询
     * @param query
     * @return
     */
    @Override
    public Result page(StoreCouponQuery query) {
        IPage<StoreCouponVO> page = PageUtil.getPage(query, StoreCouponVO.class);
        baseMapper.getPage(page, query);
        return Result.ok().data(page.getRecords()).total(page.getTotal());
    }

    /**
     * 根据id查找
     * @param id
     * @return
     */
    @Override
    public Result detail(String id) {
        StoreCouponVO vo = baseMapper.getById(id);
        Assert.notNull(vo, "记录不存在");
        return Result.ok().data(vo);
    }

    /**
     * 列表
     * @return
     */
    @Override
    public Result all() {
        List<StoreCoupon> entities = baseMapper.selectList(null);
        List<StoreCouponVO> vos = entities.stream()
                .map(entity -> StoreCouponVO.getVoByEntity(entity, null))
                .collect(Collectors.toList());
        return Result.ok().data(vos);
    }

    @Override
    public Result getCoupon(String id) {
        StoreCouponSingle storeCouponSingle = storeCouponSingleService.getById(id);
        StoreCoupon storeCoupon = this.getById(storeCouponSingle.getStoreCoupRef());
        if (storeCoupon == null) {
            throw new ServiceException("店铺礼品卡包不存在");
        }
        if (!StoreCouponStausEnum.ENABLE.name().equals(storeCoupon.getStatus())) {
            throw new ServiceException("店铺礼品卡包没有启用无法领取");
        }
        if (ClaimStatusEnum.CLAIM.name().equals(storeCouponSingle.getClaimStatus())) {
            throw new ServiceException("当前礼品卡包已被领取");
        }
        return Result.ok().data(storeCouponSingle);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public Result addStoreCoupon(StoreCouponForm storeCoupon) {
        StoreCoupon entity = StoreCouponForm.getEntityByForm(storeCoupon, null);
        entity.setStatus(StoreCouponStausEnum.ENABLE.name());
        entity.setCouponClaimNum(storeCoupon.getCouponNum());
        entity.setGenerateStatus(GenerateCouponStausEnum.NOT_GENERATE.name());
        Coupon coupon = couponService.getById(entity.getCouponId());
        Integer publishNum = coupon.getPublishNum();
        //表示不限制领取数量
        if (!Integer.valueOf(0).equals(publishNum)) {
            if (publishNum < storeCoupon.getCouponNum()) {
                throw new ServiceException("优惠卷发行数量小于生成二维码数量");
            }
        }
        this.save(entity);
        return Result.ok();
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public Result generateStoreCoupon(String id) {
        RLock redissonLock = redissonClient.getLock(STORE_COUPON_GENERATE + id);
        try {
            redissonLock.lock();
            LambdaQueryWrapper<StoreCoupon> forUpdate = Wrappers.<StoreCoupon>lambdaQuery().eq(StoreCoupon::getId, id).last("FOR UPDATE");
            StoreCoupon storeCoupon = this.getOne(forUpdate);
            if (storeCoupon == null) {
                throw new ServiceException("当前店铺优惠卷不存在无法生成");
            }
            if (!GenerateCouponStausEnum.NOT_GENERATE.name().equals(  storeCoupon.getGenerateStatus())) {
                throw new ServiceException("当前店铺优惠卷状态异常无法生成");
            }
            //生成优惠卷
            Integer couponNum = storeCoupon.getCouponNum();
            if (couponNum == null) {
                throw new ServiceException("发行数量为空不能生成");
            }
            List<StoreCouponSingle> storeCouponSingles = new ArrayList<>();
            for (int i = 1; i <= couponNum; i++) {
                StoreCouponSingle storeCouponSingle = getStoreCouponSingle(storeCoupon, i);
                storeCouponSingles.add(storeCouponSingle);
            }
            if (!storeCouponSingles.isEmpty()) {
                storeCouponSingleService.saveBatch(storeCouponSingles);
            }
            //更新状态生成状态
            storeCoupon.setGenerateStatus(GenerateCouponStausEnum.GENERATE.name());
            this.updateById(storeCoupon);
            return Result.ok();
        }finally {
            TransactionSynchronizationManager.registerSynchronization(new TransactionSynchronization() {
                @Override
                public void afterCommit() {
                    if (redissonLock.isHeldByCurrentThread()) {
                        redissonLock.unlock();
                    }
                }
                @Override
                public void afterCompletion(int status) {
                    // 确保即使在事务回滚的情况下也能释放锁
                    if (redissonLock.isHeldByCurrentThread()) {
                        redissonLock.unlock();
                    }
                }
            });
        }
    }

    private static StoreCouponSingle getStoreCouponSingle(StoreCoupon storeCoupon, int i) {
        StoreCouponSingle storeCouponSingle = new StoreCouponSingle();
        storeCouponSingle.setStoreCoupRef(storeCoupon.getId());
        storeCouponSingle.setStoreId(storeCoupon.getStoreId());
        storeCouponSingle.setStoreName(storeCoupon.getStoreName());
        storeCouponSingle.setCouponName(storeCoupon.getCouponName());
        storeCouponSingle.setCouponId(storeCoupon.getCouponId());
        storeCouponSingle.setClaimStatus(ClaimStatusEnum.NOT_CLAIM.name());
        storeCouponSingle.setCouponNo(String.format("%08d", i));
        return storeCouponSingle;
    }
}