"""
sudo 密码输入对话框
"""
from PyQt5.QtWidgets import (
    QDialog, QVBoxLayout, QHBoxLayout, QLabel,
    QLineEdit, QPushButton, QMessageBox
)
from PyQt5.QtCore import Qt
from PyQt5.QtGui import QFont, QIcon


class SudoDialog(QDialog):
    """sudo 密码输入对话框"""

    def __init__(self, parent=None):
        super().__init__(parent)
        self.password = None
        self.init_ui()

    def init_ui(self):
        """初始化界面"""
        self.setWindowTitle("需要 sudo 权限")
        self.setFixedSize(400, 200)
        self.setWindowFlags(self.windowFlags() & ~Qt.WindowContextHelpButtonHint)

        # 主布局
        layout = QVBoxLayout(self)
        layout.setSpacing(15)
        layout.setContentsMargins(20, 20, 20, 20)

        # 图标和提示
        icon_label = QLabel("🔒")
        icon_label.setAlignment(Qt.AlignCenter)
        font = QFont()
        font.setPointSize(36)
        icon_label.setFont(font)
        layout.addWidget(icon_label)

        # 提示文本
        hint_label = QLabel(
            "本工具需要 sudo 权限来访问磁盘分区和删除文件。\n"
            "请输入 sudo 密码："
        )
        hint_label.setAlignment(Qt.AlignCenter)
        hint_label.setWordWrap(True)
        layout.addWidget(hint_label)

        # 密码输入框
        self.password_input = QLineEdit()
        self.password_input.setEchoMode(QLineEdit.Password)
        self.password_input.setPlaceholderText("请输入 sudo 密码")
        self.password_input.returnPressed.connect(self.on_ok_clicked)
        layout.addWidget(self.password_input)

        # 按钮
        button_layout = QHBoxLayout()
        button_layout.addStretch()

        self.cancel_button = QPushButton("取消")
        self.cancel_button.clicked.connect(self.on_cancel_clicked)
        button_layout.addWidget(self.cancel_button)

        self.ok_button = QPushButton("确定")
        self.ok_button.setDefault(True)
        self.ok_button.clicked.connect(self.on_ok_clicked)
        button_layout.addWidget(self.ok_button)

        layout.addLayout(button_layout)

        # 设置焦点
        self.password_input.setFocus()

    def on_ok_clicked(self):
        """确定按钮点击"""
        password = self.password_input.text()

        if not password:
            QMessageBox.warning(
                self,
                "警告",
                "请输入密码！"
            )
            return

        self.password = password
        self.accept()

    def on_cancel_clicked(self):
        """取消按钮点击"""
        self.password = None
        self.reject()

    def get_password(self) -> str:
        """获取输入的密码"""
        return self.password


class SudoVerifyDialog(QDialog):
    """sudo 密码验证对话框（显示验证进度）"""

    def __init__(self, parent=None):
        super().__init__(parent)
        self.password = None
        self.init_ui()

    def init_ui(self):
        """初始化界面"""
        self.setWindowTitle("验证 sudo 密码")
        self.setFixedSize(300, 150)
        self.setWindowFlags(self.windowFlags() & ~Qt.WindowContextHelpButtonHint)

        # 主布局
        layout = QVBoxLayout(self)
        layout.setSpacing(15)
        layout.setContentsMargins(20, 20, 20, 20)

        # 提示文本
        self.status_label = QLabel("正在验证 sudo 密码...")
        self.status_label.setAlignment(Qt.AlignCenter)
        layout.addWidget(self.status_label)

        # 进度指示
        self.progress_label = QLabel("⏳")
        self.progress_label.setAlignment(Qt.AlignCenter)
        font = QFont()
        font.setPointSize(24)
        self.progress_label.setFont(font)
        layout.addWidget(self.progress_label)

        # 取消按钮
        button_layout = QHBoxLayout()
        button_layout.addStretch()

        self.cancel_button = QPushButton("取消")
        self.cancel_button.clicked.connect(self.reject)
        button_layout.addWidget(self.cancel_button)

        layout.addLayout(button_layout)

    def set_status(self, status: str):
        """设置状态文本"""
        self.status_label.setText(status)

    def set_success(self):
        """设置验证成功状态"""
        self.progress_label.setText("✅")
        self.status_label.setText("验证成功！")
        self.cancel_button.setText("关闭")

    def set_failure(self, message: str):
        """设置验证失败状态"""
        self.progress_label.setText("❌")
        self.status_label.setText(f"验证失败：{message}")
        self.cancel_button.setText("关闭")