-- ===================================================================
-- 消息系统测试数据生成脚本
-- 适用于运动训练监测与AI评估平台
-- 包含教师端和学生端用户的各种消息场景
-- PostgreSQL 15+ 兼容版本
-- ===================================================================

-- 测试用户ID
-- 教师端用户ID: 7bf7378e-a027-473e-97ac-3460ed3f170a
-- 学生端用户ID: eed3824b-bba1-4309-8048-19d17367c084

-- ===================================================================
-- 1. 清理可能存在的测试数据
-- ===================================================================

-- 删除测试消息接收记录
DELETE FROM ak_message_recipients 
WHERE message_id IN (
    SELECT id FROM ak_messages 
    WHERE metadata::jsonb ? 'test_data' 
    AND metadata->>'test_data' = 'true'
);

-- 删除测试消息
DELETE FROM ak_messages 
WHERE metadata::jsonb ? 'test_data' 
AND metadata->>'test_data' = 'true';

-- 删除测试群组成员
DELETE FROM ak_message_group_members 
WHERE group_id IN (
    SELECT id FROM ak_message_groups 
    WHERE name LIKE '%测试%'
);

-- 删除测试群组
DELETE FROM ak_message_groups 
WHERE name LIKE '%测试%';

-- ===================================================================
-- 2. 创建测试用户消息偏好设置
-- ===================================================================

-- 为教师用户创建消息偏好设置
INSERT INTO ak_user_message_preferences (
    user_id, 
    global_enabled, 
    push_enabled, 
    email_enabled,
    type_preferences
) VALUES (
    '7bf7378e-a027-473e-97ac-3460ed3f170a'::uuid, 
    true, 
    true, 
    true,
    '{
        "system": {"enabled": true, "push": true, "email": true},
        "device": {"enabled": true, "push": true, "email": false},
        "training": {"enabled": true, "push": true, "email": false},
        "social": {"enabled": true, "push": true, "email": false},
        "assignment": {"enabled": true, "push": true, "email": true},
        "achievement": {"enabled": true, "push": true, "email": false},
        "chat": {"enabled": true, "push": true, "email": false},
        "announcement": {"enabled": true, "push": true, "email": true},
        "reminder": {"enabled": true, "push": true, "email": true},
        "alert": {"enabled": true, "push": true, "email": true}
    }'::jsonb
) ON CONFLICT (user_id) DO UPDATE SET
    global_enabled = EXCLUDED.global_enabled,
    push_enabled = EXCLUDED.push_enabled,
    email_enabled = EXCLUDED.email_enabled,
    type_preferences = EXCLUDED.type_preferences;

-- 为学生用户创建消息偏好设置
INSERT INTO ak_user_message_preferences (
    user_id, 
    global_enabled, 
    push_enabled, 
    email_enabled,
    type_preferences
) VALUES (
    'eed3824b-bba1-4309-8048-19d17367c084'::uuid, 
    true, 
    true, 
    false,
    '{
        "system": {"enabled": true, "push": true, "email": false},
        "device": {"enabled": true, "push": true, "email": false},
        "training": {"enabled": true, "push": true, "email": false},
        "social": {"enabled": true, "push": true, "email": false},
        "assignment": {"enabled": true, "push": true, "email": false},
        "achievement": {"enabled": true, "push": true, "email": false},
        "chat": {"enabled": true, "push": true, "email": false},
        "announcement": {"enabled": true, "push": true, "email": false},
        "reminder": {"enabled": true, "push": true, "email": false},
        "alert": {"enabled": true, "push": true, "email": false}
    }'::jsonb
) ON CONFLICT (user_id) DO UPDATE SET
    global_enabled = EXCLUDED.global_enabled,
    push_enabled = EXCLUDED.push_enabled,
    email_enabled = EXCLUDED.email_enabled,
    type_preferences = EXCLUDED.type_preferences;

-- ===================================================================
-- 3. 创建测试消息群组
-- ===================================================================

-- 创建班级群组
INSERT INTO ak_message_groups (
    id,
    name,
    description,
    group_type,
    owner_id,
    is_public,
    member_limit,
    settings
) VALUES (
    'f47ac10b-58cc-4372-a567-0e02b2c3d479'::uuid,
    '高一(1)班运动训练群',
    '高一(1)班运动训练讨论群，用于发布训练任务和交流心得',
    'class',
    '7bf7378e-a027-473e-97ac-3460ed3f170a'::uuid,
    false,
    50,
    '{"allow_file_upload": true, "allow_voice_message": true, "auto_archive_days": 30}'::jsonb
);

-- 创建师生群组
INSERT INTO ak_message_groups (
    id,
    name,
    description,
    group_type,
    owner_id,
    is_public,
    member_limit,
    settings
) VALUES (
    'a1b2c3d4-e5f6-7890-abcd-ef1234567890'::uuid,
    '师生交流群',
    '教师与学生的日常交流群组',
    'normal',
    '7bf7378e-a027-473e-97ac-3460ed3f170a'::uuid,
    false,
    100,
    '{"allow_file_upload": true, "allow_voice_message": true}'::jsonb
);

-- 添加群组成员
INSERT INTO ak_message_group_members (
    group_id,
    user_id,
    role,
    status,
    nickname,
    notification_enabled
) VALUES 
-- 班级群成员
('f47ac10b-58cc-4372-a567-0e02b2c3d479'::uuid, '7bf7378e-a027-473e-97ac-3460ed3f170a'::uuid, 'owner', 'active', '李老师', true),
('f47ac10b-58cc-4372-a567-0e02b2c3d479'::uuid, 'eed3824b-bba1-4309-8048-19d17367c084'::uuid, 'member', 'active', '张同学', true),
-- 师生交流群成员
('a1b2c3d4-e5f6-7890-abcd-ef1234567890'::uuid, '7bf7378e-a027-473e-97ac-3460ed3f170a'::uuid, 'owner', 'active', '李老师', true),
('a1b2c3d4-e5f6-7890-abcd-ef1234567890'::uuid, 'eed3824b-bba1-4309-8048-19d17367c084'::uuid, 'member', 'active', '张同学', true);

-- ===================================================================
-- 4. 创建测试消息数据
-- ===================================================================

-- 4.1 系统欢迎消息
INSERT INTO ak_messages (
    id, message_type_id, sender_type, sender_id, sender_name,
    receiver_type, receiver_id, title, content, content_type,
    priority, is_broadcast, metadata, created_at
) VALUES (
    gen_random_uuid(), 
    (SELECT id FROM ak_message_types WHERE code = 'system'),
    'system', null, '系统',
    'user', 'eed3824b-bba1-4309-8048-19d17367c084'::uuid, 
    '欢迎使用运动训练平台！',
    '欢迎来到运动训练监测平台！您可以在这里记录训练数据、查看AI分析报告、与老师同学交流。祝您运动愉快！',
    'text', 100, false,
    '{"test_data": "true", "welcome_message": true}'::jsonb,
    now() - interval '2 days'
);

-- 为系统消息创建接收记录
INSERT INTO ak_message_recipients (
    message_id, recipient_type, recipient_id, status, delivered_at, read_at
) 
SELECT 
    m.id, 'user', 'eed3824b-bba1-4309-8048-19d17367c084'::uuid, 
    'read', now() - interval '2 days', now() - interval '1 day'
FROM ak_messages m 
WHERE m.title = '欢迎使用运动训练平台！' 
AND m.metadata->>'test_data' = 'true';

-- 4.2 设备连接成功消息
INSERT INTO ak_messages (
    id, message_type_id, sender_type, sender_id, sender_name,
    receiver_type, receiver_id, title, content, content_type,
    priority, device_data, metadata, created_at
) VALUES (
    gen_random_uuid(), 
    (SELECT id FROM ak_message_types WHERE code = 'device'),
    'device', gen_random_uuid(), '智能手环Pro',
    'user', 'eed3824b-bba1-4309-8048-19d17367c084'::uuid, 
    '设备连接成功',
    '您的智能手环Pro已成功连接，电量85%，可以开始记录运动数据了。',
    'text', 80,
    '{"device_name": "智能手环Pro", "battery_level": 85, "firmware_version": "2.1.3", "connection_type": "bluetooth"}'::jsonb,
    '{"test_data": "true", "device_type": "wearable"}'::jsonb,
    now() - interval '1 day'
);

-- 为设备消息创建接收记录
INSERT INTO ak_message_recipients (
    message_id, recipient_type, recipient_id, status, delivered_at, read_at
) 
SELECT 
    m.id, 'user', 'eed3824b-bba1-4309-8048-19d17367c084'::uuid, 
    'read', now() - interval '1 day', now() - interval '20 hours'
FROM ak_messages m 
WHERE m.title = '设备连接成功' 
AND m.metadata->>'test_data' = 'true';

-- 4.3 设备低电量警告
INSERT INTO ak_messages (
    id, message_type_id, sender_type, sender_id, sender_name,
    receiver_type, receiver_id, title, content, content_type,
    priority, is_urgent, device_data, metadata, created_at
) VALUES (
    gen_random_uuid(), 
    (SELECT id FROM ak_message_types WHERE code = 'device'),
    'device', gen_random_uuid(), '智能手环Pro',
    'user', 'eed3824b-bba1-4309-8048-19d17367c084'::uuid, 
    '设备电量不足',
    '您的智能手环Pro电量仅剩15%，请及时充电以确保正常使用。',
    'text', 85, true,
    '{"device_name": "智能手环Pro", "battery_level": 15, "estimated_time_left": "2小时"}'::jsonb,
    '{"test_data": "true", "alert_type": "low_battery"}'::jsonb,
    now() - interval '2 hours'
);

-- 为低电量警告创建未读接收记录
INSERT INTO ak_message_recipients (
    message_id, recipient_type, recipient_id, status, delivered_at
) 
SELECT 
    m.id, 'user', 'eed3824b-bba1-4309-8048-19d17367c084'::uuid, 
    'delivered', now() - interval '2 hours'
FROM ak_messages m 
WHERE m.title = '设备电量不足' 
AND m.metadata->>'test_data' = 'true';

-- 4.4 训练完成通知
INSERT INTO ak_messages (
    id, message_type_id, sender_type, sender_id, sender_name,
    receiver_type, receiver_id, title, content, content_type,
    priority, metadata, created_at
) VALUES (
    gen_random_uuid(), 
    (SELECT id FROM ak_message_types WHERE code = 'training'),
    'system', null, '训练助手',
    'user', 'eed3824b-bba1-4309-8048-19d17367c084'::uuid, 
    '🎉 训练完成！',
    '恭喜完成跑步训练！用时25分钟，消耗180卡路里，步数3250步，平均心率145 bpm。您的表现很棒，继续保持！',
    'text', 70,
    '{"test_data": "true", "activity_type": "running", "duration": 25, "calories": 180, "steps": 3250, "avg_heart_rate": 145}'::jsonb,
    now() - interval '3 hours'
);

-- 为训练消息创建接收记录
INSERT INTO ak_message_recipients (
    message_id, recipient_type, recipient_id, status, delivered_at, read_at
) 
SELECT 
    m.id, 'user', 'eed3824b-bba1-4309-8048-19d17367c084'::uuid, 
    'read', now() - interval '3 hours', now() - interval '2 hours'
FROM ak_messages m 
WHERE m.title = '🎉 训练完成！' 
AND m.metadata->>'test_data' = 'true';

-- 4.5 教师发布作业通知
INSERT INTO ak_messages (
    id, message_type_id, sender_type, sender_id, sender_name,
    receiver_type, receiver_id, title, content, content_type,
    priority, metadata, expires_at, created_at
) VALUES (
    gen_random_uuid(), 
    (SELECT id FROM ak_message_types WHERE code = 'assignment'),
    'user', '7bf7378e-a027-473e-97ac-3460ed3f170a'::uuid, '李老师',
    'user', 'eed3824b-bba1-4309-8048-19d17367c084'::uuid, 
    '新作业：每日运动打卡',
    '同学们好！本周的运动作业已发布：每天完成30分钟有氧运动，记录心率和步数数据。截止时间：本周日23:59。请按时完成并上传数据。',
    'text', 75,
    '{"test_data": "true", "assignment_id": "homework_001", "due_date": "2025-06-22 23:59:00", "subject": "体育"}'::jsonb,
    '2025-06-22 23:59:00'::timestamp,
    now() - interval '1 day'
);

-- 为作业通知创建接收记录
INSERT INTO ak_message_recipients (
    message_id, recipient_type, recipient_id, status, delivered_at, read_at
) 
SELECT 
    m.id, 'user', 'eed3824b-bba1-4309-8048-19d17367c084'::uuid, 
    'read', now() - interval '1 day', now() - interval '18 hours'
FROM ak_messages m 
WHERE m.title = '新作业：每日运动打卡' 
AND m.metadata->>'test_data' = 'true';

-- 4.6 成就解锁通知
INSERT INTO ak_messages (
    id, message_type_id, sender_type, sender_id, sender_name,
    receiver_type, receiver_id, title, content, content_type,
    priority, metadata, created_at
) VALUES (
    gen_random_uuid(), 
    (SELECT id FROM ak_message_types WHERE code = 'achievement'),
    'system', null, '成就系统',
    'user', 'eed3824b-bba1-4309-8048-19d17367c084'::uuid, 
    '🏆 恭喜解锁新成就！',
    '恭喜您解锁了"坚持不懈"成就！连续7天完成运动打卡，获得50积分奖励。继续努力，向更高目标前进！',
    'text', 50,
    '{"test_data": "true", "achievement_name": "坚持不懈", "points": 50, "days_streak": 7}'::jsonb,
    now() - interval '6 hours'
);

-- 为成就通知创建未读接收记录
INSERT INTO ak_message_recipients (
    message_id, recipient_type, recipient_id, status, delivered_at
) 
SELECT 
    m.id, 'user', 'eed3824b-bba1-4309-8048-19d17367c084'::uuid, 
    'delivered', now() - interval '6 hours'
FROM ak_messages m 
WHERE m.title = '🏆 恭喜解锁新成就！' 
AND m.metadata->>'test_data' = 'true';

-- 4.7 教师群组消息
INSERT INTO ak_messages (
    id, message_type_id, sender_type, sender_id, sender_name,
    receiver_type, receiver_id, title, content, content_type,
    priority, conversation_id, metadata, created_at
) VALUES (
    gen_random_uuid(), 
    (SELECT id FROM ak_message_types WHERE code = 'chat'),
    'user', '7bf7378e-a027-473e-97ac-3460ed3f170a'::uuid, '李老师',
    'group', 'f47ac10b-58cc-4372-a567-0e02b2c3d479'::uuid, 
    null,
    '同学们，下周一我们将进行体能测试，请大家提前做好准备。测试项目包括：50米跑、立定跳远、引体向上。有问题随时问我。',
    'text', 90, 'f47ac10b-58cc-4372-a567-0e02b2c3d479'::uuid,
    '{"test_data": "true", "group_message": true}'::jsonb,
    now() - interval '4 hours'
);

-- 为群组消息创建接收记录（教师和学生都收到）
INSERT INTO ak_message_recipients (
    message_id, recipient_type, recipient_id, status, delivered_at, read_at
) 
SELECT 
    m.id, 'user', 'eed3824b-bba1-4309-8048-19d17367c084'::uuid, 
    'read', now() - interval '4 hours', now() - interval '3 hours'
FROM ak_messages m 
WHERE m.content LIKE '%体能测试%' 
AND m.metadata->>'test_data' = 'true'
UNION ALL
SELECT 
    m.id, 'user', '7bf7378e-a027-473e-97ac-3460ed3f170a'::uuid, 
    'read', now() - interval '4 hours', now() - interval '4 hours'
FROM ak_messages m 
WHERE m.content LIKE '%体能测试%' 
AND m.metadata->>'test_data' = 'true';

-- 4.8 学生回复群组消息
WITH parent_msg AS (
    SELECT id as parent_id FROM ak_messages 
    WHERE content LIKE '%体能测试%' 
    AND metadata->>'test_data' = 'true'
    LIMIT 1
)
INSERT INTO ak_messages (
    id, message_type_id, sender_type, sender_id, sender_name,
    receiver_type, receiver_id, title, content, content_type,
    priority, conversation_id, parent_message_id, metadata, created_at
) 
SELECT
    gen_random_uuid(), 
    (SELECT id FROM ak_message_types WHERE code = 'chat'),
    'user', 'eed3824b-bba1-4309-8048-19d17367c084'::uuid, '张同学',
    'group', 'f47ac10b-58cc-4372-a567-0e02b2c3d479'::uuid, 
    null,
    '好的老师，我会认真准备的！请问引体向上有什么技巧吗？',
    'text', 90, 'f47ac10b-58cc-4372-a567-0e02b2c3d479'::uuid, parent_id,
    '{"test_data": "true", "group_message": true, "reply": true}'::jsonb,
    now() - interval '3 hours'
FROM parent_msg;

-- 为学生回复创建接收记录
INSERT INTO ak_message_recipients (
    message_id, recipient_type, recipient_id, status, delivered_at, read_at
) 
SELECT 
    m.id, 'user', '7bf7378e-a027-473e-97ac-3460ed3f170a'::uuid, 
    'read', now() - interval '3 hours', now() - interval '2 hours'
FROM ak_messages m 
WHERE m.content LIKE '%引体向上有什么技巧%' 
AND m.metadata->>'test_data' = 'true'
UNION ALL
SELECT 
    m.id, 'user', 'eed3824b-bba1-4309-8048-19d17367c084'::uuid, 
    'read', now() - interval '3 hours', now() - interval '3 hours'
FROM ak_messages m 
WHERE m.content LIKE '%引体向上有什么技巧%' 
AND m.metadata->>'test_data' = 'true';

-- 4.9 教师私信学生
INSERT INTO ak_messages (
    id, message_type_id, sender_type, sender_id, sender_name,
    receiver_type, receiver_id, title, content, content_type,
    priority, metadata, created_at
) VALUES (
    gen_random_uuid(), 
    (SELECT id FROM ak_message_types WHERE code = 'chat'),
    'user', '7bf7378e-a027-473e-97ac-3460ed3f170a'::uuid, '李老师',
    'user', 'eed3824b-bba1-4309-8048-19d17367c084'::uuid, 
    null,
    '张同学，我看到你最近的运动数据很不错，心率控制得很好。继续保持这个节奏，相信你的体能会有很大提升。',
    'text', 90,
    '{"test_data": "true", "private_message": true}'::jsonb,
    now() - interval '1 hour'
);

-- 为私信创建未读接收记录
INSERT INTO ak_message_recipients (
    message_id, recipient_type, recipient_id, status, delivered_at
) 
SELECT 
    m.id, 'user', 'eed3824b-bba1-4309-8048-19d17367c084'::uuid, 
    'delivered', now() - interval '1 hour'
FROM ak_messages m 
WHERE m.content LIKE '%运动数据很不错%' 
AND m.metadata->>'test_data' = 'true';

-- 4.10 系统公告消息
INSERT INTO ak_messages (
    id, message_type_id, sender_type, sender_id, sender_name,
    receiver_type, receiver_id, title, content, content_type,
    priority, is_broadcast, metadata, created_at
) VALUES (
    gen_random_uuid(), 
    (SELECT id FROM ak_message_types WHERE code = 'announcement'),
    'system', null, '运动平台',
    'broadcast', null, 
    '平台升级通知',
    '各位用户，运动训练平台将于本周六凌晨2:00-4:00进行系统升级，届时将暂停服务。升级后将新增AI运动分析功能，敬请期待！',
    'text', 95, true,
    '{"test_data": "true", "maintenance": true, "upgrade": true}'::jsonb,
    now() - interval '12 hours'
);

-- 为公告创建接收记录（广播消息，所有用户都收到）
INSERT INTO ak_message_recipients (
    message_id, recipient_type, recipient_id, status, delivered_at, read_at
) 
SELECT 
    m.id, 'user', '7bf7378e-a027-473e-97ac-3460ed3f170a'::uuid, 
    'read', now() - interval '12 hours', now() - interval '11 hours'
FROM ak_messages m 
WHERE m.title = '平台升级通知' 
AND m.metadata->>'test_data' = 'true'
UNION ALL
SELECT 
    m.id, 'user', 'eed3824b-bba1-4309-8048-19d17367c084'::uuid, 
    'read', now() - interval '12 hours', now() - interval '10 hours'
FROM ak_messages m 
WHERE m.title = '平台升级通知' 
AND m.metadata->>'test_data' = 'true';

-- 4.11 运动提醒消息
INSERT INTO ak_messages (
    id, message_type_id, sender_type, sender_id, sender_name,
    receiver_type, receiver_id, title, content, content_type,
    priority, metadata, scheduled_at, created_at
) VALUES (
    gen_random_uuid(), 
    (SELECT id FROM ak_message_types WHERE code = 'reminder'),
    'system', null, '运动助手',
    'user', 'eed3824b-bba1-4309-8048-19d17367c084'::uuid, 
    '⏰ 运动时间到了！',
    '根据您的运动计划，现在是跑步时间。建议运动30分钟，注意热身和拉伸。今天天气不错，适合户外运动！',
    'text', 65,
    '{"test_data": "true", "reminder_type": "exercise", "activity": "running", "duration": 30}'::jsonb,
    now() - interval '30 minutes',
    now() - interval '30 minutes'
);

-- 为提醒消息创建未读接收记录
INSERT INTO ak_message_recipients (
    message_id, recipient_type, recipient_id, status, delivered_at
) 
SELECT 
    m.id, 'user', 'eed3824b-bba1-4309-8048-19d17367c084'::uuid, 
    'delivered', now() - interval '30 minutes'
FROM ak_messages m 
WHERE m.title = '⏰ 运动时间到了！' 
AND m.metadata->>'test_data' = 'true';

-- 4.12 异常心率警报
INSERT INTO ak_messages (
    id, message_type_id, sender_type, sender_id, sender_name,
    receiver_type, receiver_id, title, content, content_type,
    priority, is_urgent, device_data, metadata, created_at
) VALUES (
    gen_random_uuid(), 
    (SELECT id FROM ak_message_types WHERE code = 'alert'),
    'device', gen_random_uuid(), '智能手环Pro',
    'user', 'eed3824b-bba1-4309-8048-19d17367c084'::uuid, 
    '⚠️ 心率异常警报',
    '检测到您的心率持续偏高（185 bpm），已超过安全范围。建议立即停止运动，休息并观察。如有不适请及时就医。',
    'text', 99, true,
    '{"device_name": "智能手环Pro", "heart_rate": 185, "safe_range": "60-180", "duration": "5分钟"}'::jsonb,
    '{"test_data": "true", "alert_type": "heart_rate", "severity": "high"}'::jsonb,
    now() - interval '10 minutes'
);

-- 为心率警报创建已读接收记录
INSERT INTO ak_message_recipients (
    message_id, recipient_type, recipient_id, status, delivered_at, read_at
) 
SELECT 
    m.id, 'user', 'eed3824b-bba1-4309-8048-19d17367c084'::uuid, 
    'read', now() - interval '10 minutes', now() - interval '8 minutes'
FROM ak_messages m 
WHERE m.title = '⚠️ 心率异常警报' 
AND m.metadata->>'test_data' = 'true';

-- 4.13 社交点赞消息
INSERT INTO ak_messages (
    id, message_type_id, sender_type, sender_id, sender_name,
    receiver_type, receiver_id, title, content, content_type,
    priority, metadata, created_at
) VALUES (
    gen_random_uuid(), 
    (SELECT id FROM ak_message_types WHERE code = 'social'),
    'user', '7bf7378e-a027-473e-97ac-3460ed3f170a'::uuid, '李老师',
    'user', 'eed3824b-bba1-4309-8048-19d17367c084'::uuid, 
    '👍 老师为您点赞了',
    '李老师为您的运动记录"今日跑步5公里"点赞了，并评论："坚持得很好，继续加油！"',
    'text', 60,
    '{"test_data": "true", "social_action": "like", "post_id": "post_001", "comment": "坚持得很好，继续加油！"}'::jsonb,
    now() - interval '5 hours'
);

-- 为社交消息创建接收记录
INSERT INTO ak_message_recipients (
    message_id, recipient_type, recipient_id, status, delivered_at, read_at
) 
SELECT 
    m.id, 'user', 'eed3824b-bba1-4309-8048-19d17367c084'::uuid, 
    'read', now() - interval '5 hours', now() - interval '4 hours'
FROM ak_messages m 
WHERE m.title = '👍 老师为您点赞了' 
AND m.metadata->>'test_data' = 'true';

-- ===================================================================
-- 5. 更新消息统计
-- ===================================================================

-- 更新所有测试消息的统计数据
UPDATE ak_messages 
SET 
    total_recipients = (
        SELECT COUNT(*) 
        FROM ak_message_recipients mr 
        WHERE mr.message_id = ak_messages.id
    ),
    delivered_count = (
        SELECT COUNT(*) 
        FROM ak_message_recipients mr 
        WHERE mr.message_id = ak_messages.id 
        AND mr.status IN ('delivered', 'read')
    ),
    read_count = (
        SELECT COUNT(*) 
        FROM ak_message_recipients mr 
        WHERE mr.message_id = ak_messages.id 
        AND mr.status = 'read'
    )
WHERE metadata->>'test_data' = 'true';

-- ===================================================================
-- 6. 创建一些历史统计数据
-- ===================================================================

-- 创建过去一个月的消息统计数据
INSERT INTO ak_message_stats (
    date_bucket, hour_bucket, message_type_id, sender_type,
    total_sent, total_delivered, total_read, total_replied
)
SELECT 
    date_bucket::date,
    hour_bucket,
    mt.id,
    sender_types.sender_type,
    (FLOOR(RANDOM() * 50) + 10)::INT as total_sent,
    (FLOOR(RANDOM() * 45) + 8)::INT as total_delivered,
    (FLOOR(RANDOM() * 40) + 5)::INT as total_read,
    (FLOOR(RANDOM() * 10) + 1)::INT as total_replied
FROM (
    SELECT generate_series(
        CURRENT_DATE - INTERVAL '30 days',
        CURRENT_DATE - INTERVAL '1 day',
        INTERVAL '1 day'
    ) as date_bucket,
    generate_series(8, 22) as hour_bucket
) dates
CROSS JOIN (
    SELECT UNNEST(ARRAY['user', 'system', 'device']) as sender_type
) sender_types
CROSS JOIN ak_message_types mt
WHERE mt.is_active = true
  AND RANDOM() > 0.3 -- 随机生成70%的数据
ON CONFLICT (date_bucket, COALESCE(hour_bucket, -1), COALESCE(message_type_id::text, ''), COALESCE(sender_type, '')) DO NOTHING;

-- ===================================================================
-- 7. 输出测试数据统计
-- ===================================================================

-- 统计创建的测试数据
SELECT 
    '测试数据统计' as info,
    (SELECT COUNT(*) FROM ak_messages WHERE metadata->>'test_data' = 'true') as total_messages,
    (SELECT COUNT(*) FROM ak_message_recipients mr 
     JOIN ak_messages m ON mr.message_id = m.id 
     WHERE m.metadata->>'test_data' = 'true') as total_recipients,
    (SELECT COUNT(*) FROM ak_message_recipients mr 
     JOIN ak_messages m ON mr.message_id = m.id 
     WHERE m.metadata->>'test_data' = 'true' 
       AND mr.recipient_id = 'eed3824b-bba1-4309-8048-19d17367c084'
       AND mr.status != 'read') as student_unread_count,
    (SELECT COUNT(*) FROM ak_message_groups WHERE name LIKE '%测试%' OR name LIKE '%群%') as test_groups;

-- ===================================================================
-- 8. 常用测试查询示例
-- ===================================================================

/*
-- 查询学生的所有消息（按时间倒序）
SELECT 
    m.title,
    m.content,
    mt.name as message_type,
    m.sender_name,
    mr.status,
    m.created_at,
    m.is_urgent,
    m.priority
FROM ak_messages m
JOIN ak_message_types mt ON m.message_type_id = mt.id
JOIN ak_message_recipients mr ON m.id = mr.message_id
WHERE mr.recipient_id = 'eed3824b-bba1-4309-8048-19d17367c084'
  AND mr.recipient_type = 'user'
  AND m.metadata->>'test_data' = 'true'
ORDER BY m.created_at DESC;

-- 查询教师发送的所有消息
SELECT 
    m.title,
    m.content,
    mt.name as message_type,
    m.receiver_type,
    m.created_at,
    m.total_recipients,
    m.read_count
FROM ak_messages m
JOIN ak_message_types mt ON m.message_type_id = mt.id
WHERE m.sender_id = '7bf7378e-a027-473e-97ac-3460ed3f170a'
  AND m.metadata->>'test_data' = 'true'
ORDER BY m.created_at DESC;

-- 查询群组消息
SELECT 
    m.content,
    m.sender_name,
    mg.name as group_name,
    m.created_at,
    CASE WHEN m.parent_message_id IS NOT NULL THEN '回复' ELSE '原始消息' END as message_type
FROM ak_messages m
JOIN ak_message_groups mg ON m.receiver_id = mg.id
WHERE m.receiver_type = 'group'
  AND m.metadata->>'test_data' = 'true'
ORDER BY m.created_at DESC;

-- 查询学生未读消息统计
SELECT 
    mt.name as message_type,
    COUNT(*) as unread_count
FROM ak_message_recipients mr
JOIN ak_messages m ON mr.message_id = m.id
JOIN ak_message_types mt ON m.message_type_id = mt.id
WHERE mr.recipient_id = 'eed3824b-bba1-4309-8048-19d17367c084'
  AND mr.recipient_type = 'user'
  AND mr.status != 'read'
  AND m.metadata->>'test_data' = 'true'
GROUP BY mt.name, mt.priority
ORDER BY mt.priority DESC;

-- 查询设备消息
SELECT 
    m.title,
    m.content,
    m.device_data,
    m.is_urgent,
    mr.status,
    m.created_at
FROM ak_messages m
JOIN ak_message_recipients mr ON m.id = mr.message_id
WHERE m.sender_type = 'device'
  AND mr.recipient_id = 'eed3824b-bba1-4309-8048-19d17367c084'
  AND m.metadata->>'test_data' = 'true'
ORDER BY m.created_at DESC;
*/

-- 完成提示
SELECT '====================================================================' as separator
UNION ALL
SELECT '测试数据生成完成！' as info
UNION ALL
SELECT '教师ID: 7bf7378e-a027-473e-97ac-3460ed3f170a' as info
UNION ALL
SELECT '学生ID: eed3824b-bba1-4309-8048-19d17367c084' as info
UNION ALL
SELECT '班级群组ID: f47ac10b-58cc-4372-a567-0e02b2c3d479' as info
UNION ALL
SELECT '师生群组ID: a1b2c3d4-e5f6-7890-abcd-ef1234567890' as info
UNION ALL
SELECT '可以使用注释中的查询语句测试各种消息场景' as info
UNION ALL
SELECT '====================================================================' as separator;