# -*- coding: utf-8 -*-
"""
Copyright (c) 2025 KenanZhu.
All rights reserved.

This software is provided "as is", without any warranty of any kind.
You may use, modify, and distribute this file under the terms of the MIT License.
See the LICENSE file for details.
"""
from enum import Enum

from PySide6.QtCore import (
    Qt,
    QSize,
    QCoreApplication,
    QRect,
    QPoint
)
from PySide6.QtWidgets import (
    QAbstractScrollArea,
    QAbstractItemView,
    QTreeWidget,
    QTreeWidgetItem
)
from PySide6.QtGui import (
     QDragEnterEvent,
     QDragMoveEvent,
     QDropEvent
)


class ALUserTreeItemType(Enum):

    GROUP = 0
    USER = 1


class ALUserTreeWidget(QTreeWidget):

    def __init__(
        self,
        parent = None
    ):

        super().__init__(parent)

        self.setupUi()
        self.translateUi()

    def setupUi(
        self
    ):

        __QTreeWidgetItem = QTreeWidgetItem()
        __QTreeWidgetItem.setText(0, u"\u5206\u7ec4/\u7528\u6237");
        self.setHeaderItem(__QTreeWidgetItem)
        self.setObjectName(u"UserTreeWidget")
        self.setMinimumSize(QSize(230, 0))
        self.setMaximumSize(QSize(250, 16777215))
        self.setSizeAdjustPolicy(QAbstractScrollArea.SizeAdjustPolicy.AdjustIgnored)
        self.setTabKeyNavigation(True)
        self.setDragEnabled(True)
        self.setAcceptDrops(True)
        self.setDropIndicatorShown(True)
        self.setDragDropMode(QAbstractItemView.DragDropMode.InternalMove)
        self.setDefaultDropAction(Qt.DropAction.IgnoreAction)
        self.setAlternatingRowColors(True)
        self.setSortingEnabled(True)
        self.sortByColumn(0, Qt.SortOrder.AscendingOrder)
        self.setAnimated(True)
        self.setAllColumnsShowFocus(False)
        self.setHeaderHidden(False)
        self.setColumnCount(2)
        self.setColumnWidth(0, 150)
        self.setColumnWidth(1, 20)
        self.header().setCascadingSectionResizes(False)
        self.header().setHighlightSections(False)
        self.header().setProperty(u"showSortIndicator", True)

    def translateUi(
        self
    ):

        ___QTreeWidgetItem = self.headerItem()
        ___QTreeWidgetItem.setText(1, QCoreApplication.translate("ALConfigWidget", u"\u72b6\u6001", None));

    @staticmethod
    def isDragPositionValid(
        target_rect: QRect,
        drag_pos: QPoint,
    ) -> bool:

        y_offset = drag_pos.y() - target_rect.top()
        valid = (y_offset > target_rect.height()*0.2 and
            y_offset < target_rect.height()*0.8)
        return valid

    def dragEnterEvent(
        self,
        event: QDragEnterEvent
    ):

        super().dragEnterEvent(event)

    def dragMoveEvent(
        self,
        event: QDragMoveEvent
    ):

        super().dragMoveEvent(event)

        SourceItem = self.currentItem()
        TargetItem = self.itemAt(event.position().toPoint())
        if SourceItem is None:
            event.ignore()
            return
        if SourceItem.type() == ALUserTreeItemType.GROUP.value:
            if TargetItem is not None:
                event.ignore()
                return
        elif SourceItem.type() == ALUserTreeItemType.USER.value:
            if TargetItem is None:
                event.ignore()
                return
            if TargetItem.type() != ALUserTreeItemType.GROUP.value:
                event.ignore()
                return
            if TargetItem.checkState(1) == Qt.CheckState.Unchecked:
                event.ignore()
                return
            if not self.isDragPositionValid(
                self.visualItemRect(TargetItem),
                event.position().toPoint()
            ):
                event.ignore()
                return
        else:
            event.ignore()
            return
        event.acceptProposedAction()

    def dropEvent(
        self,
        event: QDropEvent
    ):

        super().dropEvent(event)

        for item_index in range(self.topLevelItemCount()):
            self.topLevelItem(item_index).setExpanded(True)
        self.setCurrentItem(None)