"""
测试聊天服务
"""
import sys
import importlib.util
from pathlib import Path

# 直接加载chat_service模块，不通过__init__.py
spec = importlib.util.spec_from_file_location(
    "chat_service",
    Path(__file__).parent / "backend" / "services" / "chat_service.py"
)
chat_service_module = importlib.util.module_from_spec(spec)
spec.loader.exec_module(chat_service_module)
ChatService = chat_service_module.ChatService

import json

# 初始化聊天服务
data_path = Path("data")
chat_service = ChatService(data_path)

print("=" * 60)
print("测试1: 获取所有聊天列表")
print("=" * 60)
all_chats = chat_service.list_all_chats()
print(json.dumps(all_chats, ensure_ascii=False, indent=2))

print("\n" + "=" * 60)
print("测试2: 获取指定聊天的完整内容")
print("=" * 60)
try:
    chat_data = chat_service.get_chat("testRole1", "111")
    print(f"Metadata: {json.dumps(chat_data['metadata'], ensure_ascii=False, indent=2)}")
    print(f"\nMessages count: {len(chat_data['messages'])}")
    for msg in chat_data['messages']:
        print(f"  Floor {msg['floor']}: {msg['name']} - {msg['mes'][:50]}...")
except Exception as e:
    print(f"Error: {e}")

print("\n" + "=" * 60)
print("测试3: 获取聊天的所有消息")
print("=" * 60)
try:
    messages_data = chat_service.get_chat("testRole2", "222")
    print(f"Messages count: {len(messages_data['messages'])}")
    for msg in messages_data['messages']:
        print(f"  Floor {msg['floor']}: {msg['name']} ({'User' if msg['is_user'] else 'AI'})")
        print(f"    Content: {msg['mes'][:80]}...")
        if msg.get('swipes') and len(msg['swipes']) > 0:
            print(f"    Swipes: {len(msg['swipes'])} versions")
except Exception as e:
    print(f"Error: {e}")

print("\n" + "=" * 60)
print("测试4: 创建新聊天")
print("=" * 60)
try:
    new_chat = chat_service.create_chat("testRole4", "test_chat", {
        "user_name": "TestUser",
        "character_name": "TestAI"
    })
    print(f"Created: {json.dumps(new_chat, ensure_ascii=False, indent=2)}")
    
    # 添加测试消息
    msg1 = chat_service.add_message("testRole4", "test_chat", {
        "name": "TestUser",
        "is_user": True,
        "mes": "这是一条测试消息"
    })
    print(f"Added message 1: Floor {msg1['floor']}")
    
    msg2 = chat_service.add_message("testRole4", "test_chat", {
        "name": "TestAI",
        "is_user": False,
        "mes": "这是AI的回复"
    })
    print(f"Added message 2: Floor {msg2['floor']}")
    
    # 读取验证
    verify_chat = chat_service.get_chat("testRole4", "test_chat")
    print(f"\nVerification - Messages count: {len(verify_chat['messages'])}")
    for msg in verify_chat['messages']:
        print(f"  Floor {msg['floor']}: {msg['name']} - {msg['mes']}")
        
except Exception as e:
    print(f"Error: {e}")
    import traceback
    traceback.print_exc()

print("\n" + "=" * 60)
print("测试5: 更新消息")
print("=" * 60)
try:
    updated = chat_service.update_message("testRole4", "test_chat", 1, {
        "mes": "这是更新后的测试消息"
    })
    print(f"Updated message at floor 1: {updated['mes']}")
    
    # 验证更新
    verify_chat = chat_service.get_chat("testRole4", "test_chat")
    for msg in verify_chat['messages']:
        if msg['floor'] == 1:
            print(f"Verified - Floor 1: {msg['mes']}")
            
except Exception as e:
    print(f"Error: {e}")
    import traceback
    traceback.print_exc()

print("\n" + "=" * 60)
print("测试6: 删除消息")
print("=" * 60)
try:
    deleted = chat_service.delete_message("testRole4", "test_chat", 1)
    print(f"Deleted message from floor 1: {deleted['mes']}")
    
    # 验证删除并检查floor重新编号
    verify_chat = chat_service.get_chat("testRole4", "test_chat")
    print(f"\nAfter deletion - Messages count: {len(verify_chat['messages'])}")
    for msg in verify_chat['messages']:
        print(f"  Floor {msg['floor']}: {msg['name']} - {msg['mes']}")
        
except Exception as e:
    print(f"Error: {e}")
    import traceback
    traceback.print_exc()

print("\n" + "=" * 60)
print("所有测试完成！")
print("=" * 60)