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

import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
import cn.iocoder.yudao.module.im.dal.dataobject.group.ImGroupDO;
import cn.iocoder.yudao.module.im.dal.dataobject.group.ImGroupMemberDO;
import cn.iocoder.yudao.module.im.dal.dataobject.message.ImGroupMessageDO;
import cn.iocoder.yudao.module.im.dal.dataobject.message.ImPrivateMessageDO;
import cn.iocoder.yudao.module.im.dal.mysql.group.ImGroupMapper;
import cn.iocoder.yudao.module.im.dal.mysql.group.ImGroupMemberMapper;
import cn.iocoder.yudao.module.im.dal.mysql.message.ImGroupMessageMapper;
import cn.iocoder.yudao.module.im.dal.mysql.message.ImPrivateMessageMapper;
import cn.iocoder.yudao.module.im.enums.message.ImMessageStatusEnum;
import cn.iocoder.yudao.module.im.enums.ImContentTypeEnum;
import jakarta.annotation.Resource;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.springframework.jdbc.core.JdbcTemplate;

import javax.sql.DataSource;
import java.time.LocalDateTime;
import java.util.List;
import java.util.Map;

import static org.junit.jupiter.api.Assertions.*;

/**
 * {@link ImStatisticsManagerMapper} 的单元测试
 *
 * @author 芋道源码
 */
public class ImStatisticsManagerMapperTest extends BaseDbUnitTest {

    private static final LocalDateTime WINDOW_BEGIN = LocalDateTime.of(2026, 1, 1, 0, 0);
    private static final LocalDateTime WINDOW_END = LocalDateTime.of(2026, 2, 1, 0, 0);

    @Resource
    private ImStatisticsManagerMapper mapper;

    @Resource
    private ImPrivateMessageMapper privateMessageMapper;
    @Resource
    private ImGroupMessageMapper groupMessageMapper;
    @Resource
    private ImGroupMapper groupMapper;
    @Resource
    private ImGroupMemberMapper groupMemberMapper;
    @Resource
    private DataSource dataSource;

    private JdbcTemplate jdbcTemplate;

    @BeforeEach
    public void initJdbcTemplate() {
        jdbcTemplate = new JdbcTemplate(dataSource);
    }

    // ========== 用户 ==========

    @Test
    public void testSelectTotalUserCount() {
        // 准备：3 个未删除用户
        insertUser(1L, LocalDateTime.now());
        insertUser(2L, LocalDateTime.now());
        insertUser(3L, LocalDateTime.now());

        // 调用 + 断言
        assertEquals(3L, mapper.selectTotalUserCount());
    }

    @Test
    public void testSelectNewUserCount_windowFilter() {
        // 准备：1 个在窗口内 + 2 个在窗口外
        insertUser(1L, WINDOW_BEGIN.plusDays(3));
        insertUser(2L, WINDOW_BEGIN.minusDays(1));
        insertUser(3L, WINDOW_END.plusDays(1));

        // 调用 + 断言：只命中窗口内
        assertEquals(1L, mapper.selectNewUserCount(WINDOW_BEGIN, WINDOW_END));
    }

    @Test
    public void testSelectActiveUserCount_distinctAcrossPrivateAndGroup() {
        // 准备：1 私聊；2 群聊；3 在窗口外；1 既发私聊又发群聊（去重）
        privateMessageMapper.insert(buildPrivate(1L, 2L, WINDOW_BEGIN.plusDays(1)));
        privateMessageMapper.insert(buildPrivate(1L, 2L, WINDOW_BEGIN.plusDays(2)));
        groupMessageMapper.insert(buildGroupMessage(2L, 100L, WINDOW_BEGIN.plusDays(3)));
        groupMessageMapper.insert(buildGroupMessage(1L, 100L, WINDOW_BEGIN.plusDays(4))); // 用户 1 已计一次
        privateMessageMapper.insert(buildPrivate(3L, 4L, WINDOW_BEGIN.minusDays(1)));     // 窗口外

        // 调用 + 断言：去重后 = 2
        assertEquals(2L, mapper.selectActiveUserCount(WINDOW_BEGIN, WINDOW_END));
    }

    @Test
    public void testSelectNewUserDailyCount_groupByDay() {
        // 准备：第 1 天 2 个；第 2 天 1 个；窗口外 1 个
        insertUser(1L, WINDOW_BEGIN.plusDays(1));
        insertUser(2L, WINDOW_BEGIN.plusDays(1));
        insertUser(3L, WINDOW_BEGIN.plusDays(2));
        insertUser(4L, WINDOW_BEGIN.minusDays(1));

        // 调用
        List<Map<String, Object>> list = mapper.selectNewUserDailyCount(WINDOW_BEGIN, WINDOW_END);

        // 断言：返回两个分组、总数 = 3
        assertEquals(2, list.size());
        long sum = list.stream().mapToLong(m -> ((Number) m.get("count")).longValue()).sum();
        assertEquals(3L, sum);
    }

    @Test
    public void testSelectActiveUserDailyCount_distinctPerDay() {
        // 准备：同一天里用户 1 在私聊与群聊都发了；第二天用户 2 发了
        privateMessageMapper.insert(buildPrivate(1L, 9L, WINDOW_BEGIN.plusDays(1)));
        groupMessageMapper.insert(buildGroupMessage(1L, 100L, WINDOW_BEGIN.plusDays(1)));
        groupMessageMapper.insert(buildGroupMessage(2L, 100L, WINDOW_BEGIN.plusDays(2)));

        // 调用
        List<Map<String, Object>> list = mapper.selectActiveUserDailyCount(WINDOW_BEGIN, WINDOW_END);

        // 断言：两天合计 2 名活跃用户（同一天 user 1 去重 + 第二天 user 2）
        assertEquals(2, list.size());
        long sum = list.stream().mapToLong(m -> ((Number) m.get("count")).longValue()).sum();
        assertEquals(2L, sum);
    }

    // ========== 群 ==========

    @Test
    public void testSelectTotalGroupCount_onlyEnabled() {
        // 准备：2 个正常 + 1 个已解散
        groupMapper.insert(buildGroup(CommonStatusEnum.ENABLE));
        groupMapper.insert(buildGroup(CommonStatusEnum.ENABLE));
        groupMapper.insert(buildGroup(CommonStatusEnum.DISABLE));

        // 调用 + 断言
        assertEquals(2L, mapper.selectTotalGroupCount());
    }

    @Test
    public void testSelectNewGroupCount_windowFilter() {
        // 准备：1 个窗口内 + 1 个窗口外
        groupMapper.insert(buildGroup(CommonStatusEnum.ENABLE));
        groupMapper.insert(buildGroup(CommonStatusEnum.ENABLE));
        // 窗口窄到只覆盖 1 秒
        LocalDateTime narrowBegin = LocalDateTime.now().plusYears(10);
        LocalDateTime narrowEnd = narrowBegin.plusSeconds(1);

        // 调用 + 断言：极窄窗口不命中任何群
        assertEquals(0L, mapper.selectNewGroupCount(narrowBegin, narrowEnd));
    }

    @Test
    public void testSelectGroupSizeDistribution_bucketing() {
        // 准备：一个 5 人群、一个 15 人群
        Long groupSmall = insertGroupAndMembers(5);
        Long groupMid = insertGroupAndMembers(15);
        assertNotNull(groupSmall);
        assertNotNull(groupMid);

        // 调用
        List<Map<String, Object>> dist = mapper.selectGroupSizeDistribution();

        // 断言：分桶包含「1-9 人」和「10-49 人」
        Map<Object, Long> byRange = dist.stream().collect(java.util.stream.Collectors.toMap(
                m -> m.get("range"), m -> ((Number) m.get("count")).longValue()));
        assertEquals(1L, byRange.get("1-9 人"));
        assertEquals(1L, byRange.get("10-49 人"));
    }

    // ========== 消息 ==========

    @Test
    public void testSelectPrivateMessageCount_windowFilter() {
        // 准备：1 在窗口内 + 1 在窗口外
        privateMessageMapper.insert(buildPrivate(1L, 2L, WINDOW_BEGIN.plusDays(1)));
        privateMessageMapper.insert(buildPrivate(1L, 2L, WINDOW_BEGIN.minusDays(1)));

        // 调用 + 断言
        assertEquals(1L, mapper.selectPrivateMessageCount(WINDOW_BEGIN, WINDOW_END));
    }

    @Test
    public void testSelectGroupMessageCount_windowFilter() {
        // 准备：1 在窗口内 + 1 在窗口外
        groupMessageMapper.insert(buildGroupMessage(1L, 100L, WINDOW_BEGIN.plusDays(1)));
        groupMessageMapper.insert(buildGroupMessage(1L, 100L, WINDOW_BEGIN.minusDays(1)));

        // 调用 + 断言
        assertEquals(1L, mapper.selectGroupMessageCount(WINDOW_BEGIN, WINDOW_END));
    }

    @Test
    public void testSelectMessageTypeDistribution_mergePrivateAndGroup() {
        // 准备：私聊 type=0 ×1，群聊 type=0 ×2、type=1 ×1
        privateMessageMapper.insert(buildPrivate(1L, 2L, WINDOW_BEGIN.plusDays(1)).setType(ImContentTypeEnum.TEXT.getType()));
        groupMessageMapper.insert(buildGroupMessage(1L, 100L, WINDOW_BEGIN.plusDays(1)).setType(ImContentTypeEnum.TEXT.getType()));
        groupMessageMapper.insert(buildGroupMessage(1L, 100L, WINDOW_BEGIN.plusDays(1)).setType(ImContentTypeEnum.TEXT.getType()));
        groupMessageMapper.insert(buildGroupMessage(1L, 100L, WINDOW_BEGIN.plusDays(1)).setType(ImContentTypeEnum.IMAGE.getType()));

        // 调用
        List<Map<String, Object>> dist = mapper.selectMessageTypeDistribution(WINDOW_BEGIN, WINDOW_END);

        // 断言：type=TEXT(0) → 3；type=IMAGE(1) → 1
        Map<Object, Long> byType = dist.stream().collect(java.util.stream.Collectors.toMap(
                m -> ((Number) m.get("type")).intValue(), m -> ((Number) m.get("count")).longValue()));
        assertEquals(3L, byType.get(ImContentTypeEnum.TEXT.getType()));
        assertEquals(1L, byType.get(ImContentTypeEnum.IMAGE.getType()));
    }

    @Test
    public void testSelectTopSenders_orderByCountDescAndLimit() {
        // 准备：user 1 共 3 条；user 2 共 2 条；user 3 共 1 条
        privateMessageMapper.insert(buildPrivate(1L, 9L, WINDOW_BEGIN.plusDays(1)));
        groupMessageMapper.insert(buildGroupMessage(1L, 100L, WINDOW_BEGIN.plusDays(1)));
        groupMessageMapper.insert(buildGroupMessage(1L, 100L, WINDOW_BEGIN.plusDays(1)));
        privateMessageMapper.insert(buildPrivate(2L, 9L, WINDOW_BEGIN.plusDays(1)));
        groupMessageMapper.insert(buildGroupMessage(2L, 100L, WINDOW_BEGIN.plusDays(1)));
        privateMessageMapper.insert(buildPrivate(3L, 9L, WINDOW_BEGIN.plusDays(1)));

        // 调用：取 TOP 2
        List<Map<String, Object>> tops = mapper.selectTopSenders(WINDOW_BEGIN, WINDOW_END, 2);

        // 断言：返回 2 条，按消息数倒序，user1 > user2
        assertEquals(2, tops.size());
        assertEquals(1L, ((Number) tops.get(0).get("userId")).longValue());
        assertEquals(3L, ((Number) tops.get(0).get("messageCount")).longValue());
        assertEquals(2L, ((Number) tops.get(1).get("userId")).longValue());
        assertEquals(2L, ((Number) tops.get(1).get("messageCount")).longValue());
    }

    // ========== 工具方法 ==========

    private void insertUser(Long id, LocalDateTime createTime) {
        jdbcTemplate.update(
                "INSERT INTO system_users (id, username, password, nickname, status, create_time, update_time, deleted, tenant_id) " +
                        "VALUES (?, ?, '', ?, 0, ?, ?, FALSE, 0)",
                id, "u" + id, "n" + id, createTime, createTime);
    }

    private static ImPrivateMessageDO buildPrivate(Long senderId, Long receiverId, LocalDateTime sendTime) {
        return ImPrivateMessageDO.builder()
                .clientMessageId("uuid-" + System.nanoTime())
                .senderId(senderId).receiverId(receiverId)
                .type(ImContentTypeEnum.TEXT.getType())
                .content("{}")
                .status(ImMessageStatusEnum.NORMAL.getStatus())
                .sendTime(sendTime).build();
    }

    private static ImGroupMessageDO buildGroupMessage(Long senderId, Long groupId, LocalDateTime sendTime) {
        return new ImGroupMessageDO()
                .setClientMessageId("uuid-" + System.nanoTime())
                .setSenderId(senderId).setGroupId(groupId)
                .setType(ImContentTypeEnum.TEXT.getType())
                .setContent("{}")
                .setStatus(ImMessageStatusEnum.NORMAL.getStatus())
                .setSendTime(sendTime);
    }

    private static ImGroupDO buildGroup(CommonStatusEnum status) {
        return ImGroupDO.builder().name("g" + System.nanoTime()).ownerUserId(1L)
                .status(status.getStatus()).build();
    }

    private Long insertGroupAndMembers(int memberCount) {
        ImGroupDO group = buildGroup(CommonStatusEnum.ENABLE);
        groupMapper.insert(group);
        for (int i = 0; i < memberCount; i++) {
            groupMemberMapper.insert(new ImGroupMemberDO()
                    .setGroupId(group.getId()).setUserId((long) (1000 + i))
                    .setStatus(CommonStatusEnum.ENABLE.getStatus()).setRole(3));
        }
        return group.getId();
    }

}