import asyncio import json import logging import sys import threading import io import urllib.request from PIL import Image, ImageDraw import pystray from winotify import Notification, audio import websockets import pythoncom # PyQt5 相关导入 from PyQt5.QtWidgets import (QApplication, QWidget, QLabel, QLineEdit, QRadioButton, QPushButton, QSlider, QVBoxLayout, QHBoxLayout, QGroupBox, QMessageBox, QFrame, QScrollArea, QSizePolicy, QDialog) from PyQt5.QtCore import Qt, QThread, pyqtSignal, QSize, QPropertyAnimation, QPoint, QRect, QTimer from PyQt5.QtGui import QIntValidator, QFont, QResizeEvent, QMouseEvent, QColor, QPainter, QBrush, QPen, QFontMetrics, QIcon, QPixmap, QImage # ==================== 版本号 ==================== VERSION = "0.6.2" # ==================== 配置日志 ==================== logging.basicConfig( level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s' ) logger = logging.getLogger("danmaku_client") # ==================== 全局变量 ==================== exit_event = threading.Event() # 退出标志 active_websockets = [] # 所有活跃的 WebSocket 连接对象 ws_lock = threading.Lock() # 保护 active_websockets 的线程锁 # 图标URL（若无法访问，将自动降级为默认图标） ICON_URL = "https://cdn.rrrosmontis.icu/tsukuyomi/TKYM-Singl.png" # ==================== 获取应用图标（供任务栏和托盘共用） ==================== def create_default_icon(): size = 64 img = Image.new('RGB', (size, size), color='white') draw = ImageDraw.Draw(img) draw.ellipse([8, 8, size-8, size-8], fill='blue', outline='darkblue') return img def get_icon_image(): try: with urllib.request.urlopen(ICON_URL, timeout=5) as response: img_data = response.read() img = Image.open(io.BytesIO(img_data)) if img.mode != 'RGBA': img = img.convert('RGBA') logger.info("成功加载远程图标") return img except Exception as e: logger.warning(f"加载远程图标失败: {e}，使用默认图标") return create_default_icon() def get_qicon(): """将 PIL Image 转换为 QIcon""" img = get_icon_image() img = img.convert("RGBA") data = img.tobytes("raw", "RGBA") qimage = QImage(data, img.width, img.height, QImage.Format_RGBA8888) pixmap = QPixmap.fromImage(qimage) return QIcon(pixmap) # ==================== PyQt5 配置对话框（改为 QDialog） ==================== class ConfigDialog(QDialog): """美观的配置对话框，用于输入房间号、选择节点、显示模式""" def __init__(self): super().__init__() self.room_id = None self.mode = None self.opacity = 0.85 # 默认透明度 self.font_size = 10 # 默认字号（减小） self.aspect_ratio = "9:16" # 默认宽高比改为 9:16 self.display_mode = "window" # 默认悬浮窗模式 ("window" 或 "notification") self.init_ui() def init_ui(self): """初始化界面组件""" self.setWindowTitle(f"TSUKUYOMI DANMAKU - CONFIG") self.resize(500, 520) # 稍微增加高度以容纳新模式选项 self.setMinimumSize(400, 480) self.setWindowFlags(self.windowFlags() & ~Qt.WindowContextHelpButtonHint) # 深色背景样式表（与原来一致） self.setStyleSheet(""" QWidget { background-color: #0a0e17; color: #e0e0e0; font-family: "Microsoft YaHei", "Segoe UI", sans-serif; font-size: 13px; } QLabel { color: #c0c0c0; } QLineEdit { background-color: #1e2434; border: 1px solid #2c3348; border-radius: 5px; padding: 8px 10px; color: #ffffff; font-size: 13px; selection-background-color: #3a425c; } QLineEdit:focus { border: 1px solid #5a6e8a; } QRadioButton { color: #d0d0d0; font-size: 13px; spacing: 10px; padding: 4px 0; } QRadioButton::indicator { width: 16px; height: 16px; } QRadioButton::indicator:unchecked { border: 1px solid #5a6e8a; border-radius: 8px; background-color: #1e2434; } QRadioButton::indicator:checked { border: 1px solid #7c9cbf; border-radius: 8px; background-color: #4c7a9a; } QPushButton { background-color: #2c3348; border: none; border-radius: 6px; padding: 8px 20px; color: #ffffff; font-size: 13px; font-weight: bold; } QPushButton:hover { background-color: #3a425c; } QPushButton:pressed { background-color: #1f2538; } QGroupBox { border: 1px solid #2c3348; border-radius: 8px; margin-top: 12px; font-size: 13px; font-weight: normal; } QGroupBox::title { subcontrol-origin: margin; left: 12px; padding: 0 6px 0 6px; color: #b0b0b0; } """) # 主布局 main_layout = QVBoxLayout() main_layout.setSpacing(18) main_layout.setContentsMargins(25, 20, 25, 20) # 标题区域 title_label = QLabel("TSUKUYOMI DANMAKU") title_font = QFont() title_font.setPointSize(15) title_font.setBold(True) title_label.setFont(title_font) title_label.setAlignment(Qt.AlignCenter) title_label.setStyleSheet("color: #d4d4d4; margin-bottom: 8px;") main_layout.addWidget(title_label) # 房间号输入区域 room_layout = QVBoxLayout() room_label = QLabel("房间号 (1-100)") room_label.setStyleSheet("margin-bottom: 4px; font-weight: bold;") self.room_edit = QLineEdit() self.room_edit.setPlaceholderText("请输入数字房间号") self.room_edit.setValidator(QIntValidator(1, 100)) room_layout.addWidget(room_label) room_layout.addWidget(self.room_edit) main_layout.addLayout(room_layout) # 节点选择区域 node_group = QGroupBox("接入节点选择") node_group.setMinimumHeight(140) node_group.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Minimum) node_layout = QVBoxLayout() node_layout.setSpacing(10) node_layout.setContentsMargins(15, 12, 15, 12) self.radio_main = QRadioButton("[主站] api.rrrosmontis.icu") self.radio_backup = QRadioButton("[副站] tsk.aeolianchi.me") self.radio_both = QRadioButton("[主站+副站] 同时接收") self.radio_main.setChecked(True) node_layout.addWidget(self.radio_main) node_layout.addWidget(self.radio_backup) node_layout.addWidget(self.radio_both) node_layout.addStretch() node_group.setLayout(node_layout) main_layout.addWidget(node_group) # ========== 新增：显示模式选择（悬浮窗 / 通知） ========== mode_group = QGroupBox("显示模式") mode_layout = QVBoxLayout() mode_layout.setSpacing(10) mode_layout.setContentsMargins(15, 12, 15, 12) self.radio_window = QRadioButton("悬浮窗模式") self.radio_notification = QRadioButton("通知模式") self.radio_window.setChecked(True) # 默认悬浮窗 mode_layout.addWidget(self.radio_window) mode_layout.addWidget(self.radio_notification) mode_group.setLayout(mode_layout) main_layout.addWidget(mode_group) # 添加弹性空间 main_layout.addStretch(1) # 按钮区域 btn_layout = QHBoxLayout() btn_layout.setSpacing(25) self.start_btn = QPushButton("启动接收") self.start_btn.setMinimumHeight(36) self.cancel_btn = QPushButton("退出程序") self.cancel_btn.setMinimumHeight(36) btn_layout.addStretch() btn_layout.addWidget(self.start_btn) btn_layout.addWidget(self.cancel_btn) btn_layout.addStretch() main_layout.addLayout(btn_layout) # 版本提示 ver_label = QLabel(f"版本 {VERSION}") ver_label.setAlignment(Qt.AlignCenter) ver_label.setStyleSheet("color: #6a6e7a; font-size: 10px; margin-top: 10px;") main_layout.addWidget(ver_label) self.setLayout(main_layout) # 连接信号 self.start_btn.clicked.connect(self.on_start_clicked) self.cancel_btn.clicked.connect(self.on_cancel_clicked) def on_start_clicked(self): """启动按钮处理：验证房间号并保存配置""" room_text = self.room_edit.text().strip() if not room_text: QMessageBox.warning(self, "输入错误", "请输入房间号") return try: room_val = int(room_text) if not (1 <= room_val <= 100): QMessageBox.warning(self, "输入错误", "房间号必须在 1-100 之间") return except ValueError: QMessageBox.warning(self, "输入错误", "房间号必须为整数") return if self.radio_main.isChecked(): mode = "main" elif self.radio_backup.isChecked(): mode = "backup" else: mode = "both" # 获取显示模式 if self.radio_window.isChecked(): display_mode = "window" else: display_mode = "notification" self.room_id = room_val self.mode = mode self.display_mode = display_mode # 透明度、字号、宽高比使用默认值（注意字号默认10，比例9:16） self.opacity = 0.85 self.font_size = 10 # 默认字号减小 self.aspect_ratio = "9:16" self.accept() # 关闭对话框并返回 Accepted def on_cancel_clicked(self): """退出程序""" self.reject() # ==================== 设置对话框（用于主悬浮窗） ==================== class SettingsDialog(QDialog): """设置对话框：调节透明度、字号、宽高比、窗口大小（等比缩放）""" def __init__(self, parent_window): super().__init__(parent_window) self.parent_window = parent_window # DanmakuWindow 实例 self.setWindowTitle("悬浮窗设置") self.setModal(True) self.setFixedSize(400, 420) self.setStyleSheet(""" QDialog { background-color: #1e1e2e; color: #ffffff; } QLabel { color: #ffffff; } QRadioButton { color: #ffffff; } QSlider::groove:horizontal { height: 4px; background: #2c3348; border-radius: 2px; } QSlider::handle:horizontal { background: #7c9cbf; width: 12px; height: 12px; margin: -4px 0; border-radius: 6px; } QPushButton { background-color: #2c3348; border: none; border-radius: 6px; padding: 6px 12px; color: #ffffff; } QPushButton:hover { background-color: #3a425c; } QGroupBox { border: 1px solid #2c3348; border-radius: 6px; margin-top: 10px; color: #ffffff; } QGroupBox::title { subcontrol-origin: margin; left: 10px; padding: 0 5px; color: #ffffff; } """) layout = QVBoxLayout(self) layout.setSpacing(15) # 透明度 opacity_group = QGroupBox("窗口透明度") opacity_layout = QVBoxLayout() self.opacity_slider = QSlider(Qt.Horizontal) self.opacity_slider.setRange(20, 100) self.opacity_slider.setValue(int(self.parent_window.opacity_value * 100)) self.opacity_slider.valueChanged.connect(self.on_opacity_changed) opacity_layout.addWidget(self.opacity_slider) self.opacity_label = QLabel(f"当前: {self.opacity_slider.value()}%") opacity_layout.addWidget(self.opacity_label) opacity_group.setLayout(opacity_layout) layout.addWidget(opacity_group) # 字号 font_group = QGroupBox("弹幕字号") font_layout = QVBoxLayout() self.font_slider = QSlider(Qt.Horizontal) self.font_slider.setRange(4, 24) self.font_slider.setValue(self.parent_window.current_font_size) self.font_slider.valueChanged.connect(self.on_font_changed) font_layout.addWidget(self.font_slider) self.font_label = QLabel(f"当前: {self.font_slider.value()}px") font_layout.addWidget(self.font_label) font_group.setLayout(font_layout) layout.addWidget(font_group) # 宽高比 ratio_group = QGroupBox("悬浮窗宽高比") ratio_layout = QVBoxLayout() ratio_radio_layout = QHBoxLayout() self.radio_16_9 = QRadioButton("16:9") self.radio_9_16 = QRadioButton("9:16") if self.parent_window.current_aspect_ratio == "16:9": self.radio_16_9.setChecked(True) else: self.radio_9_16.setChecked(True) self.radio_16_9.toggled.connect(self.on_ratio_changed) self.radio_9_16.toggled.connect(self.on_ratio_changed) ratio_radio_layout.addWidget(self.radio_16_9) ratio_radio_layout.addWidget(self.radio_9_16) ratio_radio_layout.addStretch() ratio_layout.addLayout(ratio_radio_layout) ratio_group.setLayout(ratio_layout) layout.addWidget(ratio_group) # 窗口大小调节（等比缩放） size_group = QGroupBox("窗口大小（等比缩放）") size_layout = QVBoxLayout() # 宽度调节滑块 width_layout = QHBoxLayout() width_layout.addWidget(QLabel("宽度:")) self.width_slider = QSlider(Qt.Horizontal) # 宽度范围：最小宽度 ~ 屏幕宽度 min_w = self.parent_window.min_width max_w = QApplication.primaryScreen().geometry().width() - 100 self.width_slider.setRange(min_w, max_w) self.width_slider.setValue(self.parent_window.width()) self.width_slider.valueChanged.connect(self.on_width_changed) width_layout.addWidget(self.width_slider) self.width_value_label = QLabel(f"{self.parent_window.width()}px") width_layout.addWidget(self.width_value_label) size_layout.addLayout(width_layout) # 高度显示（只读） height_layout = QHBoxLayout() height_layout.addWidget(QLabel("高度:")) self.height_value_label = QLabel(f"{self.parent_window.height()}px") height_layout.addWidget(self.height_value_label) height_layout.addStretch() size_layout.addLayout(height_layout) size_group.setLayout(size_layout) layout.addWidget(size_group) # 按钮 btn_layout = QHBoxLayout() btn_layout.addStretch() close_btn = QPushButton("关闭") close_btn.clicked.connect(self.accept) btn_layout.addWidget(close_btn) layout.addLayout(btn_layout) def on_opacity_changed(self, value): opacity = value / 100.0 self.parent_window.set_opacity(opacity) self.opacity_label.setText(f"当前: {value}%") def on_font_changed(self, value): self.parent_window.set_font_size(value) self.font_label.setText(f"当前: {value}px") def on_ratio_changed(self): if self.radio_16_9.isChecked(): new_ratio = "16:9" else: new_ratio = "9:16" self.parent_window.set_aspect_ratio(new_ratio) min_w = self.parent_window.min_width max_w = self.width_slider.maximum() self.width_slider.setRange(min_w, max_w) current_width = self.parent_window.width() if current_width < min_w: current_width = min_w self.parent_window.set_window_width(current_width) self.width_slider.setValue(current_width) self.width_value_label.setText(f"{current_width}px") self.height_value_label.setText(f"{self.parent_window.height()}px") def on_width_changed(self, width): self.parent_window.set_window_width(width) self.width_value_label.setText(f"{width}px") self.height_value_label.setText(f"{self.parent_window.height()}px") def showEvent(self, event): self.opacity_slider.setValue(int(self.parent_window.opacity_value * 100)) self.font_slider.setValue(self.parent_window.current_font_size) if self.parent_window.current_aspect_ratio == "16:9": self.radio_16_9.setChecked(True) else: self.radio_9_16.setChecked(True) self.width_slider.setValue(self.parent_window.width()) self.width_slider.setRange(self.parent_window.min_width, self.width_slider.maximum()) self.width_value_label.setText(f"{self.parent_window.width()}px") self.height_value_label.setText(f"{self.parent_window.height()}px") super().showEvent(event) # 新增：从主窗口同步当前字号并更新UI def refresh_font_display(self): """从主窗口同步当前字号并更新滑块和标签""" current_size = self.parent_window.current_font_size self.font_slider.setValue(current_size) self.font_label.setText(f"当前: {current_size}px") # ==================== 弹幕接收线程（异步） ==================== class DanmakuReceiver(QThread): """在独立线程中运行 asyncio 事件循环，接收弹幕并通过信号发送到主窗口""" new_danmaku = pyqtSignal(str, str) # 用户名, 弹幕内容 def __init__(self, room_id: int, mode: str): super().__init__() self.room_id = room_id self.mode = mode self.exit_event = threading.Event() self.loop = None def stop(self): self.exit_event.set() global active_websockets with ws_lock: for ws in active_websockets: if self.loop and self.loop.is_running(): asyncio.run_coroutine_threadsafe(ws.close(), self.loop) active_websockets.clear() if self.loop and self.loop.is_running(): self.loop.call_soon_threadsafe(self.loop.stop) def run(self): self.loop = asyncio.new_event_loop() asyncio.set_event_loop(self.loop) try: self.loop.run_until_complete(self.main_async()) except asyncio.CancelledError: logger.info("异步任务被取消") except Exception as e: logger.error(f"异步任务异常: {e}") finally: pending = asyncio.all_tasks(self.loop) for task in pending: task.cancel() if pending: self.loop.run_until_complete(asyncio.gather(*pending, return_exceptions=True)) self.loop.close() logger.info("事件循环已关闭") async def handle_message(self, message: str): try: data = json.loads(message) msg_type = data.get("type") if msg_type == "danmaku": danmaku = data.get("data", {}) user = danmaku.get("user", "匿名") text = danmaku.get("text", "") self.new_danmaku.emit(user, text) elif msg_type == "init": logger.info("收到 init 消息（忽略）") else: logger.debug(f"未知消息类型: {msg_type}") except json.JSONDecodeError: logger.error(f"解析消息失败: {message}") except Exception as e: logger.error(f"处理消息异常: {e}") async def websocket_task(self, url: str, name: str): global active_websockets while not self.exit_event.is_set(): websocket = None try: async with websockets.connect(url) as websocket: with ws_lock: active_websockets.append(websocket) logger.info(f"[{name}] 已连接到 {url}") join_msg = { "type": "join", "room": self.room_id, "user": "客户端用户" } await websocket.send(json.dumps(join_msg)) init_msg = await websocket.recv() logger.info(f"[{name}] 收到 init: {init_msg[:100]}") async for message in websocket: if self.exit_event.is_set(): break await self.handle_message(message) except websockets.exceptions.ConnectionClosed: if not self.exit_event.is_set(): logger.warning(f"[{name}] 连接已关闭，5秒后重连...") except Exception as e: if not self.exit_event.is_set(): logger.error(f"[{name}] 连接错误: {e}，5秒后重连...") finally: if websocket: with ws_lock: if websocket in active_websockets: active_websockets.remove(websocket) if not self.exit_event.is_set(): await asyncio.sleep(5) logger.info(f"[{name}] WebSocket 任务已退出") async def main_async(self): MAIN_URL = "wss://api.rrrosmontis.icu/v1/live/danmaku" BACKUP_URL = "wss://tsk.aeolianchi.me/danmaku" tasks = [] if self.mode == "main": tasks.append(self.websocket_task(MAIN_URL, "主站")) elif self.mode == "backup": tasks.append(self.websocket_task(BACKUP_URL, "副站")) elif self.mode == "both": tasks.append(self.websocket_task(MAIN_URL, "主站")) tasks.append(self.websocket_task(BACKUP_URL, "副站")) else: logger.error(f"未知模式: {self.mode}") return await asyncio.gather(*tasks, return_exceptions=True) # ==================== 弹幕条目控件（支持动画） ==================== class DanmakuItem(QLabel): """单个弹幕的显示控件，支持淡入动画""" def __init__(self, user: str, text: str, font_size: int, parent=None): super().__init__(parent) html = f'
{user}: {text}
' self.setText(html) self.setTextFormat(Qt.RichText) self.setWordWrap(True) self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Minimum) font = QFont("Microsoft YaHei", font_size) self.setFont(font) self.setContentsMargins(4, 2, 4, 2) self.setStyleSheet(""" QLabel { background-color: rgba(30, 30, 40, 100); border-radius: 6px; padding: 4px 6px; } """) self.opacity_effect = None self.animation = None def start_animation(self): from PyQt5.QtWidgets import QGraphicsOpacityEffect self.opacity_effect = QGraphicsOpacityEffect(self) self.setGraphicsEffect(self.opacity_effect) self.opacity_effect.setOpacity(0) self.animation = QPropertyAnimation(self.opacity_effect, b"opacity") self.animation.setDuration(300) self.animation.setStartValue(0) self.animation.setEndValue(1) self.animation.start() # ==================== 主悬浮窗（无边框、可拖拽、等比例缩放、半透明） ==================== class DanmakuWindow(QWidget): def __init__(self, room_id: int, mode: str, opacity: float, font_size: int, aspect_ratio: str): super().__init__() self.room_id = room_id self.mode = mode self.opacity_value = opacity self.current_font_size = font_size self.current_aspect_ratio = aspect_ratio # "16:9" 或 "9:16" self.dragging = False self.drag_pos = None self.receiver = None self._updating_scaling = False self.settings_dialog = None # 记录设置对话框实例，用于同步字号 self.update_min_size() if self.current_aspect_ratio == "16:9": self.base_width = 350 # 减小默认宽度 self.base_height = int(350 * 9 / 16) else: self.base_width = 250 # 9:16 默认宽度更小 self.base_height = int(250 * 16 / 9) if self.base_width < self.min_width: self.base_width = self.min_width self.base_height = int(self.base_width * (9/16 if self.current_aspect_ratio == "16:9" else 16/9)) self.init_ui() # 窗口标志：无边框、置顶、工具窗口（隐藏任务栏） self.setWindowFlags(Qt.FramelessWindowHint | Qt.WindowStaysOnTopHint | Qt.Tool) self.setAttribute(Qt.WA_TranslucentBackground) self.setWindowOpacity(self.opacity_value) self.resize(self.base_width, self.base_height) self.center() self._resizing = False self.update_ui_scaling() def update_min_size(self): if self.current_aspect_ratio == "16:9": self.min_width = 150 self.min_height = int(150 * 9 / 16) else: self.min_width = 100 self.min_height = int(100 * 16 / 9) def center(self): screen_geo = QApplication.primaryScreen().geometry() x = (screen_geo.width() - self.width()) // 2 y = (screen_geo.height() - self.height()) // 2 self.move(x, y) def init_ui(self): self.container = QFrame(self) self.container.setObjectName("Container") self.container.setStyleSheet(""" QFrame#Container { background-color: rgba(20, 20, 30, 255); border-radius: 12px; border: 1px solid rgba(255, 255, 255, 30); } QLabel { color: #f0f0f0; background: transparent; } """) main_layout = QVBoxLayout(self.container) main_layout.setContentsMargins(8, 12, 8, 12) main_layout.setSpacing(8) self.title_layout = QVBoxLayout() self.title_layout.setSpacing(4) first_row = QHBoxLayout() self.title_label1 = QLabel("弹幕列表") self.title_label1.setStyleSheet("font-weight: bold; color: #d4d4d4;") first_row.addWidget(self.title_label1) first_row.addStretch() self.settings_btn = QPushButton("⚙") self.settings_btn.setStyleSheet(""" QPushButton { background: rgba(80, 80, 90, 150); border-radius: 12px; font-weight: bold; color: #c0c0c0; } QPushButton:hover { background: rgba(120, 120, 130, 200); color: white; } """) self.settings_btn.clicked.connect(self.open_settings) first_row.addWidget(self.settings_btn) self.close_btn = QPushButton("✕") self.close_btn.setStyleSheet(""" QPushButton { background: rgba(80, 80, 90, 150); border-radius: 12px; font-weight: bold; color: #c0c0c0; } QPushButton:hover { background: rgba(120, 120, 130, 200); color: white; } """) self.close_btn.clicked.connect(self.close_app) first_row.addWidget(self.close_btn) self.title_layout.addLayout(first_row) mode_text = "" if self.mode == "main": mode_text = "SITE 1" elif self.mode == "backup": mode_text = "SITE 2" else: mode_text = "SITE 1 & 2" self.title_label2 = QLabel(f"{mode_text} · ROOM {self.room_id}") self.title_label2.setStyleSheet("color: #a0a0c0;") self.title_layout.addWidget(self.title_label2) main_layout.addLayout(self.title_layout) line = QFrame() line.setFrameShape(QFrame.HLine) line.setStyleSheet("background-color: rgba(255, 255, 255, 40); max-height: 1px;") main_layout.addWidget(line) self.scroll_area = QScrollArea() self.scroll_area.setWidgetResizable(True) self.scroll_area.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff) self.scroll_area.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff) self.scroll_area.setStyleSheet(""" QScrollArea { background: transparent; border: none; } QScrollArea > QWidget { background: transparent; } """) self.scroll_area.viewport().setStyleSheet("background: transparent;") self.danmaku_container = QWidget() self.danmaku_container.setStyleSheet("background: transparent;") self.danmaku_layout = QVBoxLayout(self.danmaku_container) self.danmaku_layout.setAlignment(Qt.AlignTop) self.danmaku_layout.setSpacing(6) self.danmaku_layout.setContentsMargins(2, 4, 2, 4) self.scroll_area.setWidget(self.danmaku_container) main_layout.addWidget(self.scroll_area) self.container.setGeometry(0, 0, self.width(), self.height()) self.container.setLayout(main_layout) def update_ui_scaling(self): if self._updating_scaling: return self._updating_scaling = True base_width = 500 scale = self.width() / base_width scale = max(0.5, min(1.5, scale)) title_font_size = int(16 * scale) title_font = QFont("Microsoft YaHei", title_font_size) title_font.setBold(True) self.title_label1.setFont(title_font) subtitle_font_size = int(11 * scale) subtitle_font = QFont("Microsoft YaHei", subtitle_font_size) self.title_label2.setFont(subtitle_font) btn_size = int(24 * scale) btn_font_size = int(14 * scale) self.settings_btn.setFixedSize(btn_size, btn_size) self.close_btn.setFixedSize(btn_size, btn_size) btn_font = QFont("Segoe UI", btn_font_size) self.settings_btn.setFont(btn_font) self.close_btn.setFont(btn_font) self.title_layout.setSpacing(int(4 * scale)) self._updating_scaling = False def open_settings(self): if self.settings_dialog is None: dlg = SettingsDialog(self) self.settings_dialog = dlg dlg.finished.connect(lambda: setattr(self, 'settings_dialog', None)) dlg.show() else: self.settings_dialog.raise_() self.settings_dialog.activateWindow() def set_opacity(self, value): self.opacity_value = value self.setWindowOpacity(value) def set_font_size(self, size): self.current_font_size = size font = QFont("Microsoft YaHei", size) for i in range(self.danmaku_layout.count()): item = self.danmaku_layout.itemAt(i) if item and item.widget(): widget = item.widget() widget.setFont(font) widget.updateGeometry() self.danmaku_layout.invalidate() self.danmaku_container.updateGeometry() self.scroll_area.updateGeometry() self.scroll_to_bottom() # 如果设置对话框已打开，同步其显示 if self.settings_dialog is not None: self.settings_dialog.refresh_font_display() def set_aspect_ratio(self, ratio): if ratio == self.current_aspect_ratio: return self.current_aspect_ratio = ratio self.update_min_size() self.set_window_width(self.width()) def set_window_width(self, width): if width < self.min_width: width = self.min_width if self.current_aspect_ratio == "16:9": height = int(width * 9 / 16) else: height = int(width * 16 / 9) if height < self.min_height: height = self.min_height width = int(height * (16/9 if self.current_aspect_ratio == "16:9" else 9/16)) self.resize(width, height) def add_danmaku(self, user: str, text: str): """添加一条弹幕，并强制滚动到底部（修复留白和显示不全问题）""" if not self.isVisible(): return item = DanmakuItem(user, text, self.current_font_size) self.danmaku_layout.addWidget(item) item.start_animation() # 延迟滚动，确保布局完全更新（避免留白/显示不全） QTimer.singleShot(30, self.scroll_to_bottom) # 限制最大条目数 if self.danmaku_layout.count() > 200: oldest_item = self.danmaku_layout.takeAt(0) if oldest_item.widget(): oldest_item.widget().deleteLater() def scroll_to_bottom(self): """滚动到底部，强制更新滚动条最大值""" scrollbar = self.scroll_area.verticalScrollBar() # 先强制刷新布局，确保最大值正确 self.danmaku_container.updateGeometry() self.scroll_area.updateGeometry() scrollbar.setValue(scrollbar.maximum()) def close_app(self): if self.receiver: try: self.receiver.new_danmaku.disconnect(self.add_danmaku) except (TypeError, AttributeError): pass exit_event.set() if self.receiver and self.receiver.isRunning(): self.receiver.stop() self.receiver.wait(3000) QApplication.quit() # ========== 无边框窗口拖拽和等比例缩放 ========== def mousePressEvent(self, event: QMouseEvent): if event.button() == Qt.LeftButton: if event.pos().y() < 60: self.dragging = True self.drag_pos = event.globalPos() event.accept() def mouseMoveEvent(self, event: QMouseEvent): if self.dragging and self.drag_pos is not None: delta = event.globalPos() - self.drag_pos self.move(self.x() + delta.x(), self.y() + delta.y()) self.drag_pos = event.globalPos() event.accept() def mouseReleaseEvent(self, event: QMouseEvent): self.dragging = False self.drag_pos = None def mouseDoubleClickEvent(self, event: QMouseEvent): if event.pos().y() < 60: if self.isMaximized(): self.showNormal() else: self.showMaximized() def update_danmaku_max_width(self): """根据当前视口宽度，限制所有弹幕条目的最大宽度""" # 获取视口宽度（滚动区域内部可用宽度） viewport_width = self.scroll_area.viewport().width() # 减去 danmaku_layout 的左右边距（2+2=4）以及容器内边距 max_width = viewport_width - 8 # 左右各2px边距，再留一点余量 if max_width < 50: max_width = 50 for i in range(self.danmaku_layout.count()): item = self.danmaku_layout.itemAt(i) if item and item.widget(): widget = item.widget() widget.setMaximumWidth(max_width) # 强制重新布局，确保换行生效 widget.updateGeometry() def auto_adjust_font_size(self): """根据当前窗口宽度自动计算并调整弹幕字号""" width = self.width() new_size = int(width * 0.04) # 线性关系：宽度250 -> 字号10 new_size = max(4, min(24, new_size)) # 限制字号范围 4~24 if new_size != self.current_font_size: self.set_font_size(new_size) def resizeEvent(self, event: QResizeEvent): if not self._resizing and not self.isMaximized(): self._resizing = True if self.current_aspect_ratio == "16:9": target_ratio = 16 / 9 else: target_ratio = 9 / 16 new_width = event.size().width() new_height = event.size().height() current_ratio = new_width / new_height if abs(current_ratio - target_ratio) > 0.01: if current_ratio > target_ratio: new_width = int(new_height * target_ratio) else: new_height = int(new_width / target_ratio) new_width = max(self.min_width, new_width) new_height = max(self.min_height, new_height) self.resize(new_width, new_height) event.accept() self.container.setGeometry(0, 0, self.width(), self.height()) self._resizing = False elif self.isMaximized(): self.container.setGeometry(0, 0, self.width(), self.height()) self.update_ui_scaling() self.update_danmaku_max_width() self.auto_adjust_font_size() super().resizeEvent(event) # ==================== 系统通知发送函数（静音） ==================== def send_notification(title: str, main_content: str, sub_content: str = ""): """在独立线程中发送 Windows 通知（避免 Qt 事件循环冲突）""" def _show(): pythoncom.CoInitialize() # 初始化 COM 为 STA notification = Notification( app_id="TSUKUYOMI DANMAKU", title=title, msg=main_content, duration="short" ) if sub_content: notification.msg = f"{main_content}\n{sub_content}" # 暂时注释掉 set_audio，避免未知干扰 # notification.set_audio(audio.Silent, loop=False) notification.show() pythoncom.CoUninitialize() threading.Thread(target=_show, daemon=True).start() # ==================== 托盘图标相关 ==================== def setup_tray(quit_callback, icon_image): menu = pystray.Menu(pystray.MenuItem("退出", quit_callback)) icon = pystray.Icon("danmaku_client", icon_image, "TSUKUYOMI DANMAKU", menu) tray_thread = threading.Thread(target=icon.run, daemon=True) tray_thread.start() return icon, tray_thread # ==================== 主函数 ==================== def main(): if sys.platform != "win32": logger.error("此客户端仅在 Windows 上支持系统通知，请使用其他方式") sys.exit(1) # 启用高 DPI 支持 QApplication.setAttribute(Qt.AA_EnableHighDpiScaling, True) QApplication.setAttribute(Qt.AA_UseHighDpiPixmaps, True) app = QApplication(sys.argv) # 设置应用图标（任务栏） app_icon = get_qicon() app.setWindowIcon(app_icon) # 显示配置对话框（模态） config_dialog = ConfigDialog() if config_dialog.exec_() != QDialog.Accepted: logger.info("用户取消配置，退出程序") sys.exit(0) room_id = config_dialog.room_id mode = config_dialog.mode opacity = config_dialog.opacity font_size = config_dialog.font_size aspect_ratio = config_dialog.aspect_ratio display_mode = config_dialog.display_mode # "window" 或 "notification" # 创建弹幕接收线程（两种模式都需要） receiver = DanmakuReceiver(room_id, mode) # 托盘图标图片（提前加载） tray_icon_img = get_icon_image() # 根据显示模式创建不同的 UI 组件 if display_mode == "window": # 悬浮窗模式 main_window = DanmakuWindow(room_id, mode, opacity, font_size, aspect_ratio) main_window.show() receiver.new_danmaku.connect(main_window.add_danmaku) main_window.receiver = receiver # 退出回调 def on_quit(icon=None, item=None): logger.info("正在安全退出...") exit_event.set() try: receiver.new_danmaku.disconnect(main_window.add_danmaku) except TypeError: pass receiver.stop() receiver.wait(3000) if icon: icon.stop() QApplication.quit() # 替换 close_app 方法 main_window.close_app = lambda: on_quit() # 设置 app 在最后一个窗口关闭时退出（悬浮窗关闭即退出） app.setQuitOnLastWindowClosed(True) else: # 通知模式：不创建悬浮窗，只使用系统通知 # 禁止窗口关闭时退出程序（因为没有主窗口） app.setQuitOnLastWindowClosed(False) # 发送启动通知 mode_text = "" if mode == "main": mode_text = "SITE 1" elif mode == "backup": mode_text = "SITE 2" else: mode_text = "SITE 1 & 2" send_notification( title="TSUKUYOMI DANMAKU", main_content="弹幕接收已开启。右键托盘图标可关闭", sub_content=f"{mode_text} · ROOM {room_id}" ) # 连接弹幕信号到通知函数 def on_new_danmaku(user, text): send_notification( title=text, # 弹幕内容作为标题 main_content=f"来自 {user}", # 发送者信息作为正文 sub_content="" # 无需副内容 ) receiver.new_danmaku.connect(on_new_danmaku) # 退出回调 def on_quit(icon=None, item=None): logger.info("正在安全退出...") exit_event.set() receiver.stop() receiver.wait(3000) if icon: icon.stop() QApplication.quit() # 启动接收线程 receiver.start() # 创建托盘图标（两种模式都需要） tray_icon, tray_thread = setup_tray(on_quit, tray_icon_img) # 运行 Qt 主循环 try: sys.exit(app.exec_()) except SystemExit: pass finally: if receiver.isRunning(): receiver.stop() receiver.wait(2000) if tray_thread.is_alive(): tray_thread.join(timeout=1) logger.info("程序已完全退出") if __name__ == "__main__": main()