meetingroom-netscreen/push_screen.py

import subprocess
import psutil
import os
import sys
import socket
import time
from loguru import logger
from PySide6.QtWidgets import (QApplication, QMainWindow, QPushButton, QMessageBox,
                               QVBoxLayout, QWidget, QLabel, QGroupBox, QGridLayout,
                               QStatusBar, QProgressBar, QFrame)
from PySide6.QtCore import QTimer, Qt, QThread, Signal
from PySide6.QtGui import QFont, QPalette, QColor

logger.add(sys.stderr, format="{time:YYYY-MM-DD HH:mm:ss} | {level} | {message}", level="INFO")

SERVER_IP = "192.168.1.100"
MEDIAMTX_PATH = r"D:\ScreenCast\mediamtx\mediamtx.exe"
FFMPEG_PATH = r"D:\ScreenCast\ffmpeg\bin\ffmpeg.exe"
STREAM_PATH = "screen"

class ConnectionChecker(QThread):
    status_update = Signal(dict)

    def run(self):
        while True:
            status = {
                "mediamtx": self.check_mediamtx(),
                "ffmpeg": self.check_ffmpeg(),
                "server": self.check_server(),
                "server_port": self.check_port(SERVER_IP, 8554),
            }
            self.status_update.emit(status)
            time.sleep(3)

    def check_mediamtx(self):
        for proc in psutil.process_iter(['name']):
            if proc.info['name'] == 'mediamtx.exe':
                return True
        return False

    def check_ffmpeg(self):
        return os.path.exists(FFMPEG_PATH)

    def check_server(self):
        try:
            socket.create_connection((SERVER_IP, 8554), timeout=2)
            return True
        except:
            return False

    def check_port(self, ip, port):
        try:
            sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
            sock.settimeout(2)
            result = sock.connect_ex((ip, port))
            sock.close()
            return result == 0
        except:
            return False

class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("投屏源控制")
        self.setFixedSize(520, 420)

        self.status_bar = QStatusBar()
        self.setStatusBar(self.status_bar)

        central_widget = QWidget()
        self.setCentralWidget(central_widget)
        main_layout = QVBoxLayout(central_widget)

        title_label = QLabel("会议投屏系统 - 主播端")
        title_font = QFont("Microsoft YaHei", 14, QFont.Bold)
        title_label.setFont(title_font)
        title_label.setAlignment(Qt.AlignCenter)
        main_layout.addWidget(title_label)

        config_group = QGroupBox("本地配置")
        config_layout = QGridLayout()

        self.mediamtx_label = QLabel("MediaMTX: 未检测")
        self.ffmpeg_label = QLabel("FFmpeg: 未检测")
        self.server_label = QLabel("服务器连接: 未检测")
        self.port_label = QLabel("RTSP端口: 未检测")

        self.mediamtx_indicator = QLabel("●")
        self.mediamtx_indicator.setFixedWidth(20)
        self.ffmpeg_indicator = QLabel("●")
        self.ffmpeg_indicator.setFixedWidth(20)
        self.server_indicator = QLabel("●")
        self.server_indicator.setFixedWidth(20)
        self.port_indicator = QLabel("●")
        self.port_indicator.setFixedWidth(20)

        config_layout.addWidget(self.mediamtx_indicator, 0, 0)
        config_layout.addWidget(self.mediamtx_label, 0, 1)
        config_layout.addWidget(self.ffmpeg_indicator, 1, 0)
        config_layout.addWidget(self.ffmpeg_label, 1, 1)
        config_layout.addWidget(self.server_indicator, 2, 0)
        config_layout.addWidget(self.server_label, 2, 1)
        config_layout.addWidget(self.port_indicator, 3, 0)
        config_layout.addWidget(self.port_label, 3, 1)

        config_group.setLayout(config_layout)
        main_layout.addWidget(config_group)

        info_group = QGroupBox("流信息")
        info_layout = QGridLayout()
        info_layout.addWidget(QLabel("服务器IP:"), 0, 0)
        info_layout.addWidget(QLabel(SERVER_IP), 0, 1)
        info_layout.addWidget(QLabel("推流地址:"), 1, 0)
        info_layout.addWidget(QLabel(f"rtsp://{SERVER_IP}:8554/{STREAM_PATH}"), 1, 1)
        info_layout.addWidget(QLabel("访问地址:"), 2, 0)
        info_layout.addWidget(QLabel(f"http://{SERVER_IP}:8889/webrtc.html?src=screen"), 2, 1)
        info_group.setLayout(info_layout)
        main_layout.addWidget(info_group)

        self.btn_push = QPushButton("🎬 开始全屏投屏")
        self.btn_push.setFixedHeight(50)
        self.btn_push.setFont(QFont("Microsoft YaHei", 11))
        self.btn_push.clicked.connect(self.push_full_screen)
        main_layout.addWidget(self.btn_push)

        self.log_label = QLabel("日志: 等待启动...")
        self.log_label.setWordWrap(True)
        self.log_label.setFrameShape(QFrame.Shape.StyledPanel)
        self.log_label.setFixedHeight(40)
        main_layout.addWidget(self.log_label)

        self.checker = ConnectionChecker()
        self.checker.status_update.connect(self.update_status)
        self.checker.start()

        logger.info("投屏源控制界面已初始化")

    def update_status(self, status):
        if status["mediamtx"]:
            self.mediamtx_indicator.setStyleSheet("color: green; font-size: 16px;")
            self.mediamtx_label.setText(f"MediaMTX: ✓ 运行中")
        else:
            self.mediamtx_indicator.setStyleSheet("color: red; font-size: 16px;")
            self.mediamtx_label.setText(f"MediaMTX: ✗ 未运行")

        if status["ffmpeg"]:
            self.ffmpeg_indicator.setStyleSheet("color: green; font-size: 16px;")
            self.ffmpeg_label.setText(f"FFmpeg: ✓ 已安装")
        else:
            self.ffmpeg_indicator.setStyleSheet("color: red; font-size: 16px;")
            self.ffmpeg_label.setText(f"FFmpeg: ✗ 未找到")

        if status["server"]:
            self.server_indicator.setStyleSheet("color: green; font-size: 16px;")
            self.server_label.setText(f"服务器连接: ✓ 可连接")
        else:
            self.server_indicator.setStyleSheet("color: orange; font-size: 16px;")
            self.server_label.setText(f"服务器连接: ✗ 连接失败")

        if status["server_port"]:
            self.port_indicator.setStyleSheet("color: green; font-size: 16px;")
            self.port_label.setText(f"RTSP端口: ✓ 端口开放")
        else:
            self.port_indicator.setStyleSheet("color: orange; font-size: 16px;")
            self.port_label.setText(f"RTSP端口: ✗ 端口关闭")

    def start_mediamtx(self):
        if not self.checker.check_mediamtx():
            logger.info("启动MediaMTX服务")
            subprocess.Popen([MEDIAMTX_PATH], cwd=r"D:\ScreenCast\mediamtx", creationflags=subprocess.CREATE_NEW_CONSOLE)
            time.sleep(3)
            logger.info("MediaMTX服务已启动")
            self.log_label.setText("日志: MediaMTX服务已启动")
        else:
            logger.info("MediaMTX已在运行")
            self.log_label.setText("日志: MediaMTX已在运行")

    def push_full_screen(self):
        logger.info("开始全屏投屏")
        self.start_mediamtx()

        if not os.path.exists(FFMPEG_PATH):
            logger.error("FFmpeg未安装")
            QMessageBox.warning(self, "错误", "FFmpeg未安装，请检查配置！")
            return

        cmd = [
            FFMPEG_PATH,
            "-f", "gdigrab", "-framerate", "30", "-i", "desktop",
            "-c:v", "libx264", "-preset", "ultrafast", "-tune", "zerolatency",
            "-f", "rtsp", f"rtsp://{SERVER_IP}:8554/{STREAM_PATH}"
        ]
        logger.info(f"执行FFmpeg命令: {' '.join(cmd)}")
        subprocess.Popen(cmd, creationflags=subprocess.CREATE_NEW_CONSOLE)
        logger.info("FFmpeg推流已启动")
        self.log_label.setText("日志: FFmpeg推流已启动，正在向服务器推送...")
        QMessageBox.information(self, "提示", f"全屏投屏已启动！\n\n接收端可打开浏览器访问：\nhttp://{SERVER_IP}:8889/webrtc.html?src=screen")

    def closeEvent(self, event):
        self.checker.terminate()
        event.accept()

if __name__ == "__main__":
    logger.info("应用程序启动")
    app = QApplication(sys.argv)
    app.setStyle("Fusion")
    window = MainWindow()
    window.show()
    sys.exit(app.exec())