from PySide6.QtCore import QAbstractListModel, QModelIndex, Qt
from PySide6.QtWidgets import QWidget, QFormLayout, QVBoxLayout
from qfluentwidgets import (
    MessageBoxBase, LineEdit, ModelComboBox, TextEdit, TeachingTip,
    InfoBarIcon, TeachingTipTailPosition
)

from common.utils import SAFE, SAFE_MAP, SAFE_MAP_ICON


class SafeStatusListModel(QAbstractListModel):

    def __init__(self, parent: QWidget = None):
        super().__init__(parent)
        self.safe_status = SAFE

    def rowCount(self, /, parent = ...):
        return len(self.safe_status)

    def columnCount(self, parent, /):
        return 1

    def data(self, index: QModelIndex, /, role: int = ...):
        row = index.row()
        if role == Qt.ItemDataRole.EditRole:
            return SAFE_MAP[self.safe_status[row]]
        elif role == Qt.ItemDataRole.DecorationRole:
            return SAFE_MAP_ICON[self.safe_status[row]]
        elif role == Qt.ItemDataRole.UserRole:
            return self.safe_status[row]
        return None

# --- 添加/编辑 对话框 ---
class ExtensionDialog(MessageBoxBase):
    """用于添加或编辑插件条目的对话框。"""

    def __init__(self, extension_data: dict[str, str | int] = None, parent=None):
        super().__init__(parent)
        self.is_edit_mode = extension_data is not None

        self.setMinimumWidth(400)

        # 创建控件
        self.lne_id = LineEdit(self)
        self.lne_name = LineEdit(self)
        self.safe_combo = ModelComboBox(self)
        self.notes_edit = TextEdit(self)

        # 填充 SAFE 下拉框 (使用新的 MAP)
        safe_list_model = SafeStatusListModel(self)
        self.safe_combo.setModel(safe_list_model)

        # 布局
        form_layout = QFormLayout()
        form_layout.addRow("ID:", self.lne_id)
        form_layout.addRow("名称:", self.lne_name)
        form_layout.addRow("安全状态:", self.safe_combo)
        form_layout.addRow("备注:", self.notes_edit)

        self.cw = QWidget(self)
        main_layout = QVBoxLayout()
        main_layout.addLayout(form_layout)
        self.cw.setLayout(main_layout)

        self.viewLayout.addWidget(self.cw)

        if self.is_edit_mode:
            self.lne_id.setText(extension_data.get("ID", ""))
            self.lne_id.setReadOnly(True)  # ID 不允许修改
            self.lne_name.setText(extension_data.get("NAME", ""))
            safe_status: int = extension_data.get("SAFE", -2)
            self.safe_combo.setCurrentIndex(SAFE.index(safe_status))
            self.notes_edit.setText(extension_data.get("NOTES", ""))
        else:
            self.safe_combo.setCurrentIndex(SAFE.index(-2))

    def validate(self) -> bool:
        if len(self.lne_id.text()) == 0:
            TeachingTip.create(
                target=self.lne_id,
                title="错误",
                content="ID 不能为空！",
                icon=InfoBarIcon.ERROR,
                isClosable=True,
                duration=2000,
                tailPosition=TeachingTipTailPosition.BOTTOM,
                parent=self
            )
            return False
        if len(self.lne_name.text()) == 0:
            TeachingTip.create(
                target=self.lne_name,
                title="错误",
                content="名称不能为空！",
                icon=InfoBarIcon.ERROR,
                isClosable=True,
                duration=2000,
                tailPosition=TeachingTipTailPosition.BOTTOM,
                parent=self,
            )
            return False
        return True

    def get_data(self) -> dict[str, str | int]:
        """获取对话框中用于 'add_one' 的数据"""
        return {
            "ID": self.lne_id.text(),
            "NAME": self.lne_name.text(),
            "SAFE": self.safe_combo.currentData(),
            "NOTES": self.notes_edit.toPlainText()
        }

    def get_update_payload(self) -> dict[str, str | int]:
        """获取对话框中用于 'update_one' 的数据 (不含ID)"""
        return {
            "NAME": self.lne_name.text(),
            "SAFE": self.safe_combo.currentData(),
            "NOTES": self.notes_edit.toPlainText()
        }