server.py

# server.py
import sqlite3
import asyncio
import websockets
import re
from datetime import datetime

clients = {}  # 修改为字典，存储用户名和websocket

def init_db():
    conn = sqlite3.connect('chat.db')
    cursor = conn.cursor()
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS messages (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            username TEXT NOT NULL,
            message TEXT NOT NULL,
            timestamp DATETIME DEFAULT CURRENT_TIMESTAMP
        )
    ''')
    conn.commit()
    conn.close()

def save_message(username, message):
    conn = sqlite3.connect('chat.db')
    cursor = conn.cursor()
    cursor.execute('INSERT INTO messages (username, message) VALUES (?, ?)', (username, message))
    conn.commit()
    conn.close()

async def broadcast_message(message):
    if clients:
        tasks = [client.send(message) for client in clients.values()]
        await asyncio.gather(*tasks, return_exceptions=True)

async def handle_client(websocket, path):
    username = path.strip('/')
    
    # 检查用户名是否符合要求
    if not re.match(r'^[a-zA-Z0-9_]{3,20}$', username):
        error_message = f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] [√] 系统: 用户名只能包含26字母、数字及下划线（_），且长度必须为3~20"
        await websocket.send(f"\033[33m{error_message}\033[0m")
        await websocket.close()
        return
    
    # 检查是否有重名用户在线
    if username in clients:
        error_message = f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] [√] 系统: 您已经在线了！"
        await websocket.send(f"\033[33m{error_message}\033[0m")
        await websocket.close()
        return

    clients[username] = websocket
    
    # 发送最近的30条聊天记录给新连接的客户端
    conn = sqlite3.connect('chat.db')
    cursor = conn.cursor()
    cursor.execute('SELECT message FROM (SELECT * FROM messages ORDER BY timestamp DESC LIMIT 30) ORDER BY timestamp')
    rows = cursor.fetchall()
    conn.close()
    
    for row in rows:
        await websocket.send(row[0])
    
    # 发送历史记录结束提示信息
    await websocket.send("\033[32m---以上是历史记录---\033[0m")
    
    try:
        join_message = f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] [√] 系统: {username} 加入了聊天室"
        await broadcast_message(f"\033[32m{join_message}\033[0m")
        save_message("系统", join_message)
        async for message in websocket:
            if message == "/list":
                user_list = "、".join(clients.keys())
                list_message = f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] [√] 命令助手: 在线 {len(clients)} 人: {user_list}"
                await websocket.send(f"\033[33m{list_message}\033[0m")
            else:
                formatted_message = f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] {username}: {message}"
                print(formatted_message)
                save_message(username, formatted_message)
                await broadcast_message(formatted_message)
    except websockets.ConnectionClosedError:
        print(f"{username} 断开连接")
    except Exception as e:
        print(f"连接处理失败: {e}")
    finally:
        del clients[username]
        leave_message = f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] [√] 系统: {username} 退出了聊天室"
        await broadcast_message(f"\033[33m{leave_message}\033[0m")
        save_message("系统", leave_message)

async def main():
    init_db()
    async with websockets.serve(handle_client, "localhost", 8765):
        await asyncio.Future()

if __name__ == "__main__":
    print('欢迎使用DLChatting 服务端')
    asyncio.run(main())