|
|
# P2P Network Communication - Main Window
|
|
|
"""
|
|
|
主窗口模块
|
|
|
实现应用程序主界面，包含联系人列表、聊天窗口和功能按钮
|
|
|
|
|
|
需求: 9.1
|
|
|
"""
|
|
|
|
|
|
import logging
|
|
|
from typing import Optional, Dict, Any
|
|
|
|
|
|
from PyQt6.QtWidgets import (
|
|
|
QMainWindow, QWidget, QVBoxLayout, QHBoxLayout,
|
|
|
QSplitter, QStatusBar, QToolBar, QMenuBar,
|
|
|
QMenu, QMessageBox, QLabel, QFrame
|
|
|
)
|
|
|
from PyQt6.QtCore import Qt, pyqtSignal, QSize
|
|
|
from PyQt6.QtGui import QAction, QIcon, QCloseEvent
|
|
|
|
|
|
from config import ClientConfig
|
|
|
from shared.models import UserInfo, UserStatus, Message, MessageType
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
|
|
|
class MainWindow(QMainWindow):
|
|
|
"""
|
|
|
主窗口
|
|
|
|
|
|
实现应用程序主界面 (需求 9.1)
|
|
|
WHEN P2P_Client 启动 THEN P2P_Client SHALL 显示主界面包含联系人列表、聊天窗口和功能按钮
|
|
|
"""
|
|
|
|
|
|
# 信号定义
|
|
|
message_received = pyqtSignal(Message)
|
|
|
user_status_changed = pyqtSignal(str, UserStatus)
|
|
|
connection_state_changed = pyqtSignal(str)
|
|
|
|
|
|
def __init__(self, config: Optional[ClientConfig] = None, parent: Optional[QWidget] = None):
|
|
|
"""
|
|
|
初始化主窗口
|
|
|
|
|
|
Args:
|
|
|
config: 客户端配置
|
|
|
parent: 父窗口
|
|
|
"""
|
|
|
super().__init__(parent)
|
|
|
|
|
|
self.config = config or ClientConfig()
|
|
|
self._current_user: Optional[UserInfo] = None
|
|
|
self._current_chat_peer: Optional[str] = None
|
|
|
|
|
|
# 子组件引用（延迟导入避免循环依赖）
|
|
|
self._contact_list: Optional[QWidget] = None
|
|
|
self._chat_widget: Optional[QWidget] = None
|
|
|
self._file_transfer_widget: Optional[QWidget] = None
|
|
|
self._media_player_widget: Optional[QWidget] = None
|
|
|
self._voice_call_widget: Optional[QWidget] = None
|
|
|
self._system_tray: Optional[Any] = None
|
|
|
|
|
|
self._setup_ui()
|
|
|
self._setup_menu_bar()
|
|
|
self._setup_tool_bar()
|
|
|
self._setup_status_bar()
|
|
|
self._connect_signals()
|
|
|
|
|
|
logger.info("MainWindow initialized")
|
|
|
|
|
|
def _setup_ui(self) -> None:
|
|
|
"""设置UI布局"""
|
|
|
self.setWindowTitle("P2P 通信应用")
|
|
|
self.setMinimumSize(800, 600)
|
|
|
self.resize(self.config.window_width, self.config.window_height)
|
|
|
|
|
|
# 创建中央部件
|
|
|
central_widget = QWidget()
|
|
|
self.setCentralWidget(central_widget)
|
|
|
|
|
|
# 主布局
|
|
|
main_layout = QHBoxLayout(central_widget)
|
|
|
main_layout.setContentsMargins(0, 0, 0, 0)
|
|
|
main_layout.setSpacing(0)
|
|
|
|
|
|
# 创建分割器
|
|
|
self._splitter = QSplitter(Qt.Orientation.Horizontal)
|
|
|
main_layout.addWidget(self._splitter)
|
|
|
|
|
|
# 左侧面板：联系人列表
|
|
|
self._left_panel = self._create_left_panel()
|
|
|
self._splitter.addWidget(self._left_panel)
|
|
|
|
|
|
# 右侧面板：聊天窗口和功能区
|
|
|
self._right_panel = self._create_right_panel()
|
|
|
self._splitter.addWidget(self._right_panel)
|
|
|
|
|
|
# 设置分割比例
|
|
|
self._splitter.setSizes([250, 750])
|
|
|
self._splitter.setStretchFactor(0, 0)
|
|
|
self._splitter.setStretchFactor(1, 1)
|
|
|
|
|
|
def _create_left_panel(self) -> QWidget:
|
|
|
"""
|
|
|
创建左侧面板（联系人列表）
|
|
|
|
|
|
实现联系人列表面板 (需求 9.1)
|
|
|
"""
|
|
|
panel = QFrame()
|
|
|
panel.setFrameShape(QFrame.Shape.StyledPanel)
|
|
|
panel.setMinimumWidth(200)
|
|
|
panel.setMaximumWidth(400)
|
|
|
|
|
|
layout = QVBoxLayout(panel)
|
|
|
layout.setContentsMargins(0, 0, 0, 0)
|
|
|
layout.setSpacing(0)
|
|
|
|
|
|
# 用户信息区域
|
|
|
self._user_info_widget = self._create_user_info_widget()
|
|
|
layout.addWidget(self._user_info_widget)
|
|
|
|
|
|
# 联系人列表占位符（将由ContactListWidget替换）
|
|
|
self._contact_list_placeholder = QLabel("联系人列表")
|
|
|
self._contact_list_placeholder.setAlignment(Qt.AlignmentFlag.AlignCenter)
|
|
|
self._contact_list_placeholder.setStyleSheet("""
|
|
|
QLabel {
|
|
|
background-color: #f5f5f5;
|
|
|
color: #666;
|
|
|
padding: 20px;
|
|
|
}
|
|
|
""")
|
|
|
layout.addWidget(self._contact_list_placeholder, 1)
|
|
|
|
|
|
return panel
|
|
|
|
|
|
def _create_user_info_widget(self) -> QWidget:
|
|
|
"""创建用户信息显示区域"""
|
|
|
widget = QFrame()
|
|
|
widget.setFrameShape(QFrame.Shape.StyledPanel)
|
|
|
widget.setStyleSheet("""
|
|
|
QFrame {
|
|
|
background-color: #4a90d9;
|
|
|
border: none;
|
|
|
border-bottom: 1px solid #3a80c9;
|
|
|
}
|
|
|
""")
|
|
|
|
|
|
layout = QHBoxLayout(widget)
|
|
|
layout.setContentsMargins(10, 10, 10, 10)
|
|
|
|
|
|
# 用户头像占位
|
|
|
self._avatar_label = QLabel("👤")
|
|
|
self._avatar_label.setFixedSize(40, 40)
|
|
|
self._avatar_label.setAlignment(Qt.AlignmentFlag.AlignCenter)
|
|
|
self._avatar_label.setStyleSheet("""
|
|
|
QLabel {
|
|
|
background-color: white;
|
|
|
border-radius: 20px;
|
|
|
font-size: 20px;
|
|
|
}
|
|
|
""")
|
|
|
layout.addWidget(self._avatar_label)
|
|
|
|
|
|
# 用户名和状态
|
|
|
info_layout = QVBoxLayout()
|
|
|
info_layout.setSpacing(2)
|
|
|
|
|
|
self._username_label = QLabel("未登录")
|
|
|
self._username_label.setStyleSheet("color: white; font-weight: bold; font-size: 14px;")
|
|
|
info_layout.addWidget(self._username_label)
|
|
|
|
|
|
self._status_label = QLabel("离线")
|
|
|
self._status_label.setStyleSheet("color: #ddd; font-size: 12px;")
|
|
|
info_layout.addWidget(self._status_label)
|
|
|
|
|
|
layout.addLayout(info_layout, 1)
|
|
|
|
|
|
return widget
|
|
|
|
|
|
def _create_right_panel(self) -> QWidget:
|
|
|
"""
|
|
|
创建右侧面板（聊天窗口和功能区）
|
|
|
|
|
|
实现聊天窗口面板和功能按钮区域 (需求 9.1)
|
|
|
"""
|
|
|
panel = QFrame()
|
|
|
panel.setFrameShape(QFrame.Shape.StyledPanel)
|
|
|
|
|
|
layout = QVBoxLayout(panel)
|
|
|
layout.setContentsMargins(0, 0, 0, 0)
|
|
|
layout.setSpacing(0)
|
|
|
|
|
|
# 聊天窗口占位符（将由ChatWidget替换）
|
|
|
self._chat_placeholder = QLabel("选择联系人开始聊天")
|
|
|
self._chat_placeholder.setAlignment(Qt.AlignmentFlag.AlignCenter)
|
|
|
self._chat_placeholder.setStyleSheet("""
|
|
|
QLabel {
|
|
|
background-color: #fafafa;
|
|
|
color: #999;
|
|
|
font-size: 16px;
|
|
|
}
|
|
|
""")
|
|
|
layout.addWidget(self._chat_placeholder, 1)
|
|
|
|
|
|
# 功能按钮区域
|
|
|
self._function_bar = self._create_function_bar()
|
|
|
layout.addWidget(self._function_bar)
|
|
|
|
|
|
return panel
|
|
|
|
|
|
def _create_function_bar(self) -> QWidget:
|
|
|
"""
|
|
|
创建功能按钮区域
|
|
|
|
|
|
实现功能按钮区域 (需求 9.1)
|
|
|
"""
|
|
|
bar = QFrame()
|
|
|
bar.setFrameShape(QFrame.Shape.StyledPanel)
|
|
|
bar.setFixedHeight(50)
|
|
|
bar.setStyleSheet("""
|
|
|
QFrame {
|
|
|
background-color: #f0f0f0;
|
|
|
border-top: 1px solid #ddd;
|
|
|
}
|
|
|
""")
|
|
|
|
|
|
layout = QHBoxLayout(bar)
|
|
|
layout.setContentsMargins(10, 5, 10, 5)
|
|
|
layout.setSpacing(10)
|
|
|
|
|
|
# 功能按钮将在后续子任务中添加
|
|
|
layout.addStretch()
|
|
|
|
|
|
return bar
|
|
|
|
|
|
def _setup_menu_bar(self) -> None:
|
|
|
"""设置菜单栏"""
|
|
|
menubar = self.menuBar()
|
|
|
|
|
|
# 文件菜单
|
|
|
file_menu = menubar.addMenu("文件(&F)")
|
|
|
|
|
|
self._login_action = QAction("登录(&L)", self)
|
|
|
self._login_action.setShortcut("Ctrl+L")
|
|
|
self._login_action.triggered.connect(self._on_login)
|
|
|
file_menu.addAction(self._login_action)
|
|
|
|
|
|
self._logout_action = QAction("注销(&O)", self)
|
|
|
self._logout_action.setEnabled(False)
|
|
|
self._logout_action.triggered.connect(self._on_logout)
|
|
|
file_menu.addAction(self._logout_action)
|
|
|
|
|
|
file_menu.addSeparator()
|
|
|
|
|
|
self._exit_action = QAction("退出(&X)", self)
|
|
|
self._exit_action.setShortcut("Ctrl+Q")
|
|
|
self._exit_action.triggered.connect(self.close)
|
|
|
file_menu.addAction(self._exit_action)
|
|
|
|
|
|
# 聊天菜单
|
|
|
chat_menu = menubar.addMenu("聊天(&C)")
|
|
|
|
|
|
self._send_file_action = QAction("发送文件(&F)", self)
|
|
|
self._send_file_action.setShortcut("Ctrl+Shift+F")
|
|
|
self._send_file_action.setEnabled(False)
|
|
|
chat_menu.addAction(self._send_file_action)
|
|
|
|
|
|
self._send_image_action = QAction("发送图片(&I)", self)
|
|
|
self._send_image_action.setShortcut("Ctrl+Shift+I")
|
|
|
self._send_image_action.setEnabled(False)
|
|
|
chat_menu.addAction(self._send_image_action)
|
|
|
|
|
|
chat_menu.addSeparator()
|
|
|
|
|
|
self._voice_call_action = QAction("语音通话(&V)", self)
|
|
|
self._voice_call_action.setShortcut("Ctrl+Shift+V")
|
|
|
self._voice_call_action.setEnabled(False)
|
|
|
chat_menu.addAction(self._voice_call_action)
|
|
|
|
|
|
# 视图菜单
|
|
|
view_menu = menubar.addMenu("视图(&V)")
|
|
|
|
|
|
self._show_contacts_action = QAction("显示联系人(&C)", self)
|
|
|
self._show_contacts_action.setCheckable(True)
|
|
|
self._show_contacts_action.setChecked(True)
|
|
|
self._show_contacts_action.triggered.connect(self._toggle_contacts_panel)
|
|
|
view_menu.addAction(self._show_contacts_action)
|
|
|
|
|
|
# 帮助菜单
|
|
|
help_menu = menubar.addMenu("帮助(&H)")
|
|
|
|
|
|
self._about_action = QAction("关于(&A)", self)
|
|
|
self._about_action.triggered.connect(self._show_about)
|
|
|
help_menu.addAction(self._about_action)
|
|
|
|
|
|
def _setup_tool_bar(self) -> None:
|
|
|
"""设置工具栏"""
|
|
|
toolbar = QToolBar("主工具栏")
|
|
|
toolbar.setMovable(False)
|
|
|
toolbar.setIconSize(QSize(24, 24))
|
|
|
self.addToolBar(toolbar)
|
|
|
|
|
|
# 工具栏按钮将在后续添加图标后完善
|
|
|
toolbar.addAction(self._login_action)
|
|
|
toolbar.addSeparator()
|
|
|
toolbar.addAction(self._send_file_action)
|
|
|
toolbar.addAction(self._send_image_action)
|
|
|
toolbar.addAction(self._voice_call_action)
|
|
|
|
|
|
def _setup_status_bar(self) -> None:
|
|
|
"""设置状态栏"""
|
|
|
self._statusbar = QStatusBar()
|
|
|
self.setStatusBar(self._statusbar)
|
|
|
|
|
|
# 连接状态标签
|
|
|
self._connection_status_label = QLabel("未连接")
|
|
|
self._connection_status_label.setStyleSheet("color: #999;")
|
|
|
self._statusbar.addPermanentWidget(self._connection_status_label)
|
|
|
|
|
|
self._statusbar.showMessage("就绪")
|
|
|
|
|
|
def _connect_signals(self) -> None:
|
|
|
"""连接信号和槽"""
|
|
|
self.message_received.connect(self._on_message_received)
|
|
|
self.user_status_changed.connect(self._on_user_status_changed)
|
|
|
self.connection_state_changed.connect(self._on_connection_state_changed)
|
|
|
|
|
|
# ==================== 事件处理 ====================
|
|
|
|
|
|
def closeEvent(self, event: QCloseEvent) -> None:
|
|
|
"""
|
|
|
窗口关闭事件
|
|
|
"""
|
|
|
# 确认退出
|
|
|
reply = QMessageBox.question(
|
|
|
self,
|
|
|
"确认退出",
|
|
|
"确定要退出应用程序吗？",
|
|
|
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No,
|
|
|
QMessageBox.StandardButton.No
|
|
|
)
|
|
|
|
|
|
if reply == QMessageBox.StandardButton.Yes:
|
|
|
self._cleanup()
|
|
|
event.accept()
|
|
|
else:
|
|
|
event.ignore()
|
|
|
|
|
|
def resizeEvent(self, event) -> None:
|
|
|
"""
|
|
|
窗口大小调整事件
|
|
|
|
|
|
实现自适应调整界面布局 (需求 9.6)
|
|
|
WHEN 用户调整窗口大小 THEN P2P_Client SHALL 自适应调整界面布局
|
|
|
"""
|
|
|
super().resizeEvent(event)
|
|
|
# 布局会自动调整，无需额外处理
|
|
|
|
|
|
# ==================== 槽函数 ====================
|
|
|
|
|
|
def _on_login(self) -> None:
|
|
|
"""处理登录操作"""
|
|
|
from client.ui.login_dialog import LoginDialog
|
|
|
|
|
|
dialog = LoginDialog(self)
|
|
|
if dialog.exec():
|
|
|
user_info = dialog.get_user_info()
|
|
|
if user_info:
|
|
|
self.set_current_user(user_info)
|
|
|
self._statusbar.showMessage(f"已登录: {user_info.username}")
|
|
|
|
|
|
def _on_logout(self) -> None:
|
|
|
"""处理注销操作"""
|
|
|
reply = QMessageBox.question(
|
|
|
self,
|
|
|
"确认注销",
|
|
|
"确定要注销当前账户吗？",
|
|
|
QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No,
|
|
|
QMessageBox.StandardButton.No
|
|
|
)
|
|
|
|
|
|
if reply == QMessageBox.StandardButton.Yes:
|
|
|
self._current_user = None
|
|
|
self._update_user_display()
|
|
|
self._login_action.setEnabled(True)
|
|
|
self._logout_action.setEnabled(False)
|
|
|
self._statusbar.showMessage("已注销")
|
|
|
|
|
|
def _toggle_contacts_panel(self, checked: bool) -> None:
|
|
|
"""切换联系人面板显示"""
|
|
|
self._left_panel.setVisible(checked)
|
|
|
|
|
|
def _show_about(self) -> None:
|
|
|
"""显示关于对话框"""
|
|
|
QMessageBox.about(
|
|
|
self,
|
|
|
"关于 P2P 通信应用",
|
|
|
"P2P 网络通信应用程序\n\n"
|
|
|
"版本: 0.1.0\n\n"
|
|
|
"支持功能:\n"
|
|
|
"- 文本消息通信\n"
|
|
|
"- 文件传输\n"
|
|
|
"- 图片传输与显示\n"
|
|
|
"- 音视频播放\n"
|
|
|
"- 语音聊天"
|
|
|
)
|
|
|
|
|
|
def _on_message_received(self, message: Message) -> None:
|
|
|
"""处理接收到的消息"""
|
|
|
logger.debug(f"Message received: {message.msg_type.value}")
|
|
|
# 消息处理将在ChatWidget中实现
|
|
|
|
|
|
def _on_user_status_changed(self, user_id: str, status: UserStatus) -> None:
|
|
|
"""处理用户状态变化"""
|
|
|
logger.debug(f"User {user_id} status changed to {status.value}")
|
|
|
# 状态更新将在ContactListWidget中实现
|
|
|
|
|
|
def _on_connection_state_changed(self, state: str) -> None:
|
|
|
"""处理连接状态变化"""
|
|
|
self._connection_status_label.setText(state)
|
|
|
|
|
|
if state == "已连接":
|
|
|
self._connection_status_label.setStyleSheet("color: green;")
|
|
|
elif state == "连接中...":
|
|
|
self._connection_status_label.setStyleSheet("color: orange;")
|
|
|
else:
|
|
|
self._connection_status_label.setStyleSheet("color: red;")
|
|
|
|
|
|
# ==================== 公共方法 ====================
|
|
|
|
|
|
def set_current_user(self, user_info: UserInfo) -> None:
|
|
|
"""
|
|
|
设置当前用户
|
|
|
|
|
|
Args:
|
|
|
user_info: 用户信息
|
|
|
"""
|
|
|
self._current_user = user_info
|
|
|
self._update_user_display()
|
|
|
self._login_action.setEnabled(False)
|
|
|
self._logout_action.setEnabled(True)
|
|
|
|
|
|
logger.info(f"Current user set: {user_info.username}")
|
|
|
|
|
|
def _update_user_display(self) -> None:
|
|
|
"""更新用户显示"""
|
|
|
if self._current_user:
|
|
|
self._username_label.setText(self._current_user.display_name or self._current_user.username)
|
|
|
self._status_label.setText(self._current_user.status.value)
|
|
|
else:
|
|
|
self._username_label.setText("未登录")
|
|
|
self._status_label.setText("离线")
|
|
|
|
|
|
def set_contact_list_widget(self, widget: QWidget) -> None:
|
|
|
"""
|
|
|
设置联系人列表组件
|
|
|
|
|
|
Args:
|
|
|
widget: 联系人列表组件
|
|
|
"""
|
|
|
# 移除占位符
|
|
|
layout = self._left_panel.layout()
|
|
|
layout.removeWidget(self._contact_list_placeholder)
|
|
|
self._contact_list_placeholder.deleteLater()
|
|
|
|
|
|
# 添加实际组件
|
|
|
self._contact_list = widget
|
|
|
layout.addWidget(widget, 1)
|
|
|
|
|
|
def set_chat_widget(self, widget: QWidget) -> None:
|
|
|
"""
|
|
|
设置聊天组件
|
|
|
|
|
|
Args:
|
|
|
widget: 聊天组件
|
|
|
"""
|
|
|
# 移除占位符
|
|
|
layout = self._right_panel.layout()
|
|
|
layout.removeWidget(self._chat_placeholder)
|
|
|
self._chat_placeholder.deleteLater()
|
|
|
|
|
|
# 添加实际组件（在功能栏之前）
|
|
|
self._chat_widget = widget
|
|
|
layout.insertWidget(0, widget, 1)
|
|
|
|
|
|
def set_system_tray(self, tray_manager) -> None:
|
|
|
"""
|
|
|
设置系统托盘管理器
|
|
|
|
|
|
Args:
|
|
|
tray_manager: 系统托盘管理器
|
|
|
"""
|
|
|
self._system_tray = tray_manager
|
|
|
|
|
|
def show_notification(self, title: str, message: str) -> None:
|
|
|
"""
|
|
|
显示通知
|
|
|
|
|
|
实现新消息通知 (需求 9.3)
|
|
|
WHEN 有新消息到达 THEN P2P_Client SHALL 在系统托盘显示通知提醒
|
|
|
|
|
|
Args:
|
|
|
title: 通知标题
|
|
|
message: 通知内容
|
|
|
"""
|
|
|
if self._system_tray:
|
|
|
self._system_tray.show_message(title, message)
|
|
|
|
|
|
def enable_chat_actions(self, enabled: bool = True) -> None:
|
|
|
"""
|
|
|
启用/禁用聊天相关操作
|
|
|
|
|
|
Args:
|
|
|
enabled: 是否启用
|
|
|
"""
|
|
|
self._send_file_action.setEnabled(enabled)
|
|
|
self._send_image_action.setEnabled(enabled)
|
|
|
self._voice_call_action.setEnabled(enabled)
|
|
|
|
|
|
def set_current_chat_peer(self, peer_id: Optional[str]) -> None:
|
|
|
"""
|
|
|
设置当前聊天对象
|
|
|
|
|
|
Args:
|
|
|
peer_id: 对等端ID
|
|
|
"""
|
|
|
self._current_chat_peer = peer_id
|
|
|
self.enable_chat_actions(peer_id is not None)
|
|
|
|
|
|
def _cleanup(self) -> None:
|
|
|
"""清理资源"""
|
|
|
logger.info("MainWindow cleanup")
|
|
|
# 清理将在集成时实现
|
|
|
|
|
|
@property
|
|
|
def current_user(self) -> Optional[UserInfo]:
|
|
|
"""获取当前用户"""
|
|
|
return self._current_user
|
|
|
|
|
|
@property
|
|
|
def current_chat_peer(self) -> Optional[str]:
|
|
|
"""获取当前聊天对象"""
|
|
|
return self._current_chat_peer
|
