package cn.iocoder.yudao.module.im.dal.mysql.friend;

import cn.hutool.core.date.LocalDateTimeUtil;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
import cn.iocoder.yudao.module.im.controller.admin.manager.friend.vo.ImFriendRequestManagerPageReqVO;
import cn.iocoder.yudao.module.im.dal.dataobject.friend.ImFriendRequestDO;
import cn.iocoder.yudao.module.im.enums.friend.ImFriendRequestHandleResultEnum;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import org.apache.ibatis.annotations.Mapper;

import java.time.LocalDateTime;
import java.util.List;

/**
 * IM 好友申请记录 Mapper
 *
 * @author 芋道源码
 */
@Mapper
public interface ImFriendRequestMapper extends BaseMapperX<ImFriendRequestDO> {

    default ImFriendRequestDO selectByFromUserIdAndToUserId(Long fromUserId, Long toUserId) {
        return selectOne(new LambdaQueryWrapperX<ImFriendRequestDO>()
                .eq(ImFriendRequestDO::getFromUserId, fromUserId)
                .eq(ImFriendRequestDO::getToUserId, toUserId));
    }

    /**
     * 拉取「我相关」的好友申请列表
     *
     * @param userId               当前用户编号
     * @param maxRequestUpdateTime 起始更新时间；首次拉取传 null
     * @param maxId                起始记录编号；首次拉取传 null
     * @param limit                拉取数量
     * @return 好友申请列表
     */
    default List<ImFriendRequestDO> selectMyList(Long userId, LocalDateTime maxRequestUpdateTime,
                                                 Long maxId, int limit) {
        LambdaQueryWrapperX<ImFriendRequestDO> wrapper = new LambdaQueryWrapperX<>();
        wrapper.and(w -> w.eq(ImFriendRequestDO::getFromUserId, userId)
                        .or().eq(ImFriendRequestDO::getToUserId, userId));
        if (maxRequestUpdateTime != null && maxId != null) {
            wrapper.and(w -> w.lt(ImFriendRequestDO::getUpdateTime, maxRequestUpdateTime)
                    .or(n -> n.eq(ImFriendRequestDO::getUpdateTime, maxRequestUpdateTime)
                            .lt(ImFriendRequestDO::getId, maxId)));
        }
        wrapper.orderByDesc(ImFriendRequestDO::getUpdateTime)
                .orderByDesc(ImFriendRequestDO::getId)
                .last("LIMIT " + limit);
        return selectList(wrapper);
    }

    /**
     * 增量拉取「我相关」的好友申请（双向 OR，按 update_time + id 正向游标）
     *
     * @param userId         当前用户编号
     * @param lastUpdateTime 上次拉取到的更新时间；首次拉取传 null
     * @param lastId         上次拉取到的记录编号；首次拉取传 null
     * @param limit          拉取数量
     * @return 好友申请列表
     */
    default List<ImFriendRequestDO> selectPullListByUserId(Long userId, Long lastUpdateTime, Long lastId, Integer limit) {
        LambdaQueryWrapperX<ImFriendRequestDO> query = new LambdaQueryWrapperX<>();
        query.and(w -> w.eq(ImFriendRequestDO::getFromUserId, userId)
                .or().eq(ImFriendRequestDO::getToUserId, userId));
        if (lastUpdateTime != null && lastId != null) {
            LocalDateTime lastTime = LocalDateTimeUtil.of(lastUpdateTime);
            query.and(w -> w.gt(ImFriendRequestDO::getUpdateTime, lastTime)
                    .or(n -> n.eq(ImFriendRequestDO::getUpdateTime, lastTime).gt(ImFriendRequestDO::getId, lastId)));
        }
        return selectList(query.orderByAsc(ImFriendRequestDO::getUpdateTime).orderByAsc(ImFriendRequestDO::getId)
                .last("LIMIT " + limit));
    }

    default int updateByIdAndHandleResult(Long id, Integer handleResult, ImFriendRequestDO updateObj) {
        return update(updateObj, new LambdaUpdateWrapper<ImFriendRequestDO>()
                .eq(ImFriendRequestDO::getId, id).eq(ImFriendRequestDO::getHandleResult, handleResult));
    }

    /**
     * 复用 (fromUserId, toUserId) 旧申请记录：覆盖申请理由 / 备注 / 来源，重置为未处理 + 清空旧处理痕迹
     * <p>
     * handleContent / handleTime 走 LambdaUpdateWrapper.set 显式置 null，updateById 默认会忽略 null 字段
     */
    default int updateByIdReset(Long id, String applyContent, String displayName, Integer addSource,
                                LocalDateTime updateTime) {
        return update(null, new LambdaUpdateWrapper<ImFriendRequestDO>()
                .eq(ImFriendRequestDO::getId, id)
                .set(ImFriendRequestDO::getApplyContent, applyContent)
                .set(ImFriendRequestDO::getDisplayName, displayName)
                .set(ImFriendRequestDO::getAddSource, addSource)
                .set(ImFriendRequestDO::getHandleResult, ImFriendRequestHandleResultEnum.UNHANDLED.getResult())
                .set(ImFriendRequestDO::getHandleContent, null)
                .set(ImFriendRequestDO::getHandleTime, null)
                .set(ImFriendRequestDO::getUpdateTime, updateTime));
    }

    default PageResult<ImFriendRequestDO> selectPage(ImFriendRequestManagerPageReqVO reqVO) {
        return selectPage(reqVO, new LambdaQueryWrapperX<ImFriendRequestDO>()
                .eqIfPresent(ImFriendRequestDO::getFromUserId, reqVO.getFromUserId())
                .eqIfPresent(ImFriendRequestDO::getToUserId, reqVO.getToUserId())
                .eqIfPresent(ImFriendRequestDO::getHandleResult, reqVO.getHandleResult())
                .eqIfPresent(ImFriendRequestDO::getAddSource, reqVO.getAddSource())
                .betweenIfPresent(ImFriendRequestDO::getCreateTime, reqVO.getCreateTime())
                .orderByDesc(ImFriendRequestDO::getId));
    }

}