-- 用戶表
CREATE TABLE users (
    user_id INTEGER PRIMARY KEY AUTOINCREMENT,
    username TEXT UNIQUE NOT NULL,
    password TEXT NOT NULL
);
 
-- 好友關系表
CREATE TABLE friendships (
    user_id INTEGER,
    friend_id INTEGER,
    PRIMARY KEY (user_id, friend_id),
    FOREIGN KEY (user_id) REFERENCES users(user_id),
    FOREIGN KEY (friend_id) REFERENCES users(user_id)
);
 
-- 消息表
CREATE TABLE messages (
    message_id INTEGER PRIMARY KEY AUTOINCREMENT,
    sender_id INTEGER,
    receiver_id INTEGER,
    content TEXT NOT NULL,
    timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
    FOREIGN KEY (sender_id) REFERENCES users(user_id),
    FOREIGN KEY (receiver_id) REFERENCES users(user_id)
);

import socket
import sqlite3
import threading
import hashlib
import json
 
# 數(shù)據(jù)庫連接
conn = sqlite3.connect('im.db')
cursor = conn.cursor()
 
# 用戶登錄狀態(tài)
users_online = {}
 
# 處理客戶端連接
def handle_client(client_socket):
    # 接收客戶端消息
    while True:
        try:
            data = client_socket.recv(1024).decode('utf-8')
            if not data:
                break
            
            # 解析消息
            message = json.loads(data)
            action = message['action']
            
            if action == 'register':
                username = message['username']
                password = hashlib.sha256(message['password'].encode('utf-8')).hexdigest()
                cursor.execute('INSERT INTO users (username, password) VALUES (?, ?)', (username, password))
                conn.commit()
                client_socket.sendall(json.dumps({'status': 'success', 'message': '注冊成功'}).encode('utf-8'))
            
            elif action == 'login':
                username = message['username']
                password = hashlib.sha256(message['password'].encode('utf-8')).hexdigest()
                cursor.execute('SELECT * FROM users WHERE username=? AND password=?', (username, password))
                user = cursor.fetchone()
                if user:
                    users_online[client_socket] = user[0]
                    client_socket.sendall(json.dumps({'status': 'success', 'message': '登錄成功'}).encode('utf-8'))
                else:
                    client_socket.sendall(json.dumps({'status': 'fail', 'message': '用戶名或密碼錯誤'}).encode('utf-8'))
            
            elif action == 'send_message':
                sender_id = users_online[client_socket]
                receiver_username = message['receiver_username']
                content = message['content']
                cursor.execute('SELECT user_id FROM users WHERE username=?', (receiver_username,))
                receiver_id = cursor.fetchone()
                if receiver_id:
                    cursor.execute('INSERT INTO messages (sender_id, receiver_id, content) VALUES (?, ?, ?)', (sender_id, receiver_id[0], content))
                    conn.commit()
                    # 廣播消息給接收者（這里簡化處理，只打印消息）
                    print(f'User {sender_id} sent message to {receiver_id[0]}: {content}')
                else:
                    client_socket.sendall(json.dumps({'status': 'fail', 'message': '接收者不存在'}).encode('utf-8'))
            
            # 其他功能（如好友管理等）可以類似實現(xiàn)
            
        except Exception as e:
            print(f'Error: {e}')
            break
    
    client_socket.close()
 
# 啟動服務器
def start_server():
    server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server_socket.bind(('0.0.0.0', 5000))
    server_socket.listen(5)
    print('Server started on port 5000')
    
    while True:
        client_socket, addr = server_socket.accept()
        print(f'Accepted connection from {addr}')
        client_handler = threading.Thread(target=handle_client, args=(client_socket,))
        client_handler.start()
 
if __name__ == '__main__':
    start_server()

import socket
import threading
import json
import tkinter as tk
from tkinter import scrolledtext
import hashlib

# 客戶端UI
class IMClient:
    def __init__(self, root):
        self.root = root
        self.root.title('IM Client')
        
        self.username = tk.StringVar()
        self.password = tk.StringVar()
        self.receiver = tk.StringVar()
        self.message = tk.StringVar()
        
        # UI組件
        self.label_username = tk.Label(root, text='Username:')
        self.label_username.grid(row=0, column=0, padx=10, pady=10)
        
        self.entry_username = tk.Entry(root, textvariable=self.username)
        self.entry_username.grid(row=0, column=1, padx=10, pady=10)
        
        self.label_password = tk.Label(root, text='Password:')
        self.label_password.grid(row=1, column=0, padx=10, pady=10)
        
        self.entry_password = tk.Entry(root, show='*', textvariable=self.password)
        self.entry_password.grid(row=1, column=1, padx=10, pady=10)
        
        self.login_button = tk.Button(root, text='Login', command=self.login)
        self.login_button.grid(row=2, column=0, columnspan=2, pady=20)
        
        self.chat_window = scrolledtext.ScrolledText(root, width=50, height=20)
        self.chat_window.grid(row=3, column=0, columnspan=2, padx=10, pady=10)
        
        self.label_receiver = tk.Label(root, text='Receiver:')
        self.label_receiver.grid(row=4, column=0, padx=10, pady=10)
        
        self.entry_receiver = tk.Entry(root, textvariable=self.receiver)
        self.entry_receiver.grid(row=4, column=1, padx=10, pady=10)
        
        self.label_message = tk.Label(root, text='Message:')
        self.label_message.grid(row=5, column=0, padx=10, pady=10)
        
        self.entry_message = tk.Entry(root, textvariable=self.message)
        self.entry_message.grid(row=5, column=1, padx=10, pady=10)
        
        self.send_button = tk.Button(root, text='Send', command=self.send_message)
        self.send_button.grid(row=6, column=0, columnspan=2, pady=20)
        
        # 初始化socket連接
        self.server_ip = '127.0.0.1'  # 服務器IP地址
        self.server_port = 12345      # 服務器端口號
        self.client_socket = None
        
        # 啟動接收消息線程
        self.receive_thread = threading.Thread(target=self.receive_messages)
        self.receive_thread.daemon = True
        self.receive_thread.start()

    def login(self):
        # 在這里添加登錄邏輯（例如，驗證用戶名和密碼）
        # 由于這個示例代碼僅用于演示，我們直接連接服務器
        try:
            self.client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
            self.client_socket.connect((self.server_ip, self.server_port))
            self.chat_window.insert(tk.END, "Connected to server\n")
        except Exception as e:
            self.chat_window.insert(tk.END, f"Failed to connect to server: {e}\n")

    def send_message(self):
        if self.client_socket and self.receiver.get() and self.message.get():
            message_data = {
                'type': 'message',
                'sender': self.username.get(),
                'receiver': self.receiver.get(),
                'content': self.message.get()
            }
            self.client_socket.sendall(json.dumps(message_data).encode('utf-8'))
            self.chat_window.insert(tk.END, f"You: {self.message.get()}\n")
            self.message.set('')  # 清空消息輸入框

    def receive_messages(self):
        while self.client_socket:
            try:
                data = self.client_socket.recv(1024).decode('utf-8')
                if data:
                    message = json.loads(data)
                    if message['type'] == 'message':
                        self.chat_window.insert(tk.END, f"{message['sender']}: {message['content']}\n")
            except Exception as e:
                self.chat_window.insert(tk.END, f"Error receiving message: {e}\n")
                break

if __name__ == "__main__":
    root = tk.Tk()
    client = IMClient(root)
    root.mainloop()