genie_tts/GUI/Utils.py

import os
import re
from enum import Enum
from datetime import datetime
import socket
from typing import List

from PySide6.QtWidgets import (
    QWidget, QHBoxLayout, QPushButton, QLineEdit, QFileDialog, QMessageBox, QComboBox, QTextEdit
)
from PySide6.QtCore import (Qt, Signal, QSettings, Property, QObject, QEvent, QMimeData)
from PySide6.QtGui import QFont


def sanitize_filename(filename: str, replacement: str = '') -> str:
    """
    将文本清理为合法的 Windows 文件名。

    Args:
        filename (str): 原始文件名。
        replacement (str): 非法字符的替换字符，默认为空，建议使用 "_"。

    Returns:
        str: 清理后的文件名。
    """
    # 1. 去除非法字符
    # \/:*?"<>| 是标准非法字符
    # \x00-\x1f 是控制字符 (如换行符、制表符等)，Windows 也不允许
    cleaned = re.sub(r'[\\/:*?"<>|\x00-\x1f]', replacement, filename)

    # 2. 去除首尾的空格和点
    # Windows 文件名不能以空格或点结尾，也不能以空格开头(虽然允许但通常不推荐)
    cleaned = cleaned.strip().rstrip('.')

    # 3. 处理 Windows 保留文件名 (CON, PRN, AUX, NUL, COM1-9, LPT1-9)
    # 这些名字不论加什么扩展名都是非法的 (例如 con.txt 也是非法的)
    reserved_names = {
        "CON", "PRN", "AUX", "NUL",
        "COM1", "COM2", "COM3", "COM4", "COM5", "COM6", "COM7", "COM8", "COM9",
        "LPT1", "LPT2", "LPT3", "LPT4", "LPT5", "LPT6", "LPT7", "LPT8", "LPT9"
    }

    # 如果文件名(全大写)是保留字，或者文件名是保留字+扩展名(如 con.txt)，则加下划线前缀
    filename_upper = cleaned.upper()
    file_stem = filename_upper.split('.')[0]  # 获取不带后缀的主文件名

    if filename_upper in reserved_names or file_stem in reserved_names:
        cleaned = "_" + cleaned

    # 4. 处理空文件名 (如果输入全是乱码被删光了)
    if not cleaned:
        cleaned = "unnamed_file"

    # 5. 限制长度 (Windows API 通常限制 255 字符，但在某些路径下更短)
    cleaned = truncate_text(cleaned)

    return cleaned


def is_port_free(port: int) -> bool:
    """返回端口是否可用"""
    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
        try:
            s.bind(('', port))
        except OSError:
            return False
    return True


def find_free_port(preferred: int = 8000) -> int:
    # 先尝试 preferred 端口
    if is_port_free(preferred):
        return preferred

    # 否则用系统自动分配
    s = socket.socket()
    s.bind(('', 0))
    port = s.getsockname()[1]
    s.close()
    return port


def truncate_text(text: str, max_len: int = 30) -> str:
    """
    按指定宽度截断文本
    """
    result = ""
    cur_len = 0
    for ch in text:
        char_len = 2 if ('\u4e00' <= ch <= '\u9fff') else 1
        if cur_len + char_len > max_len:
            break
        result += ch
        cur_len += char_len
    return result


def generate_output_filenames(folder: str, original_texts: List[str]) -> List[str]:
    """
    批量生成文件名：
    输入 original_texts 列表
    输出等长 filenames 列表
    """
    today = datetime.now().strftime("%Y%m%d")
    pattern = re.compile(rf'^\[{today}]\[(\d{{3}})]')

    # ① 查找当天现存最大编号
    max_n = 0
    if os.path.isdir(folder):
        for name in os.listdir(folder):
            m = pattern.match(name)
            if m:
                n = int(m.group(1))
                max_n = max(max_n, n)

    filenames = []
    cur_n = max_n

    # ② 依次生成新文件名
    for text in original_texts:
        cur_n += 1
        n_str = f"{cur_n:03d}"

        cleaned = sanitize_filename(text)

        filename = f"[{today}][{n_str}]{cleaned}.wav"
        filenames.append(filename)

    return filenames


# ==================== 通用组件 ====================

class FileSelectionMode(Enum):
    FILE = 0
    DIRECTORY = 1


class FileSelectorWidget(QWidget):
    """一个包含行编辑和浏览按钮的复合控件，支持文件和文件夹选择。（原样引用并稍作适配）"""
    pathChanged = Signal(str)

    def __init__(
            self,
            setting_key: str,
            selection_mode: FileSelectionMode = FileSelectionMode.DIRECTORY,
            file_filter: str = "All Files (*)",
            parent: QWidget = None,
    ):
        super().__init__(parent)
        self.setting_key: str = setting_key
        self.selection_mode: FileSelectionMode = selection_mode
        self.file_filter: str = file_filter

        # 使用 QSettings 模拟 STUDIO_SETTINGS
        self.settings = QSettings("MyTTS", "GUI")

        layout = QHBoxLayout(self)
        layout.setContentsMargins(0, 0, 0, 0)
        layout.setSpacing(5)

        self.path_edit: QLineEdit = QLineEdit()
        self.path_edit.setReadOnly(True)
        self.path_edit.setPlaceholderText("未选择路径")

        self.browse_button: QPushButton = QPushButton("📁")
        self.browse_button.setCursor(Qt.CursorShape.PointingHandCursor)
        default_font = QFont()
        default_font.setPointSize(10)
        self.browse_button.setFont(default_font)
        self.browse_button.setFixedSize(30, 30)

        self.clear_button: QPushButton = QPushButton("❌")
        self.clear_button.setCursor(Qt.CursorShape.PointingHandCursor)
        self.clear_button.setFont(default_font)
        self.clear_button.setFixedSize(30, 30)

        layout.addWidget(self.path_edit)
        layout.addWidget(self.browse_button)
        layout.addWidget(self.clear_button)

        self.browse_button.clicked.connect(self._open_dialog)
        self.clear_button.clicked.connect(self._clear_path)
        self.path_edit.textChanged.connect(self.pathChanged)

    def _open_dialog(self):
        path = self.path_edit.text()
        if path and os.path.exists(path):
            start_path = path
        else:
            # 【修改点】默认路径改为 Desktop
            desktop_path = os.path.join(os.path.expanduser("~"), "Desktop")
            start_path = str(self.settings.value(
                f"last_path_{self.setting_key}", defaultValue=desktop_path
            ))

        if self.selection_mode == FileSelectionMode.DIRECTORY:
            selected_path = QFileDialog.getExistingDirectory(
                self, "选择文件夹", start_path
            )
        else:
            selected_path, _ = QFileDialog.getOpenFileName(
                self, "选择文件", start_path, self.file_filter
            )

        if selected_path:
            self.set_path(selected_path)
            parent_path = os.path.dirname(selected_path)
            if parent_path:
                self.settings.setValue(
                    f"last_path_{self.setting_key}", parent_path)

    def _clear_path(self):
        if not self.path_edit.text():
            return
        reply = QMessageBox.question(self, '确认', '您确定要清空路径吗？',
                                     QMessageBox.StandardButton.Yes | QMessageBox.StandardButton.No,
                                     QMessageBox.StandardButton.No)
        if reply == QMessageBox.StandardButton.Yes:
            self.set_path("")

    def get_path(self) -> str:
        text = self.path_edit.text()
        # 即使路径不存在（可能是输入时），也返回文本供逻辑判断，或者严格校验
        return text

    def set_path(self, path: str, block_signals: bool = False):
        if block_signals:
            self.path_edit.blockSignals(True)
        self.path_edit.setText(path)
        if block_signals:
            self.path_edit.blockSignals(False)

    path = Property(str, fget=get_path, fset=set_path, notify=pathChanged)  # type: ignore


class WheelEventFilter(QObject):
    def eventFilter(self, obj, event):
        if event.type() == QEvent.Type.Wheel and isinstance(obj, QComboBox):
            return True  # 阻止默认滚轮行为
        return super().eventFilter(obj, event)


class MyComboBox(QComboBox):
    def __init__(self, parent: QWidget = None):
        super().__init__(parent)
        self._wheelFilter = WheelEventFilter()
        self.installEventFilter(self._wheelFilter)


class MyTextEdit(QTextEdit):
    def insertFromMimeData(self, source: QMimeData) -> None:
        # 仅取纯文本
        if source.hasText():
            self.insertPlainText(source.text())
        else:
            super().insertFromMimeData(source)