# -*- coding: utf-8 -*-
# Copyright: Ankitects Pty Ltd and contributors
# License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html

from aqt import colors
from aqt.qt import *
from aqt.theme import theme_manager


class Switch(QAbstractButton):
    """A horizontal slider to toggle between two states which can be denoted by short strings.
    The left state is the default and corresponds to isChecked=False.
    """

    _margin: int = 2

    def __init__(
        self,
        radius: int = 10,
        left_label: str = "",
        right_label: str = "",
        parent: QWidget = None,
    ) -> None:
        super().__init__(parent=parent)
        self.setCheckable(True)
        super().setChecked(False)
        self.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
        self._left_label = left_label
        self._right_label = right_label
        self._path_radius = radius
        self._knob_radius = radius - self._margin
        self._left_position = self._position = self._path_radius + self._margin
        self._right_position = 3 * self._path_radius + self._margin

    @pyqtProperty(int)  # type: ignore
    def position(self) -> int:
        return self._position

    @position.setter  # type: ignore
    def position(self, position: int) -> None:
        self._position = position
        self.update()

    @property
    def start_position(self) -> int:
        return self._left_position if self.isChecked() else self._right_position

    @property
    def end_position(self) -> int:
        return self._right_position if self.isChecked() else self._left_position

    @property
    def label(self) -> str:
        return self._right_label if self.isChecked() else self._left_label

    def sizeHint(self) -> QSize:
        return QSize(
            4 * self._path_radius + 2 * self._margin,
            2 * self._path_radius + 2 * self._margin,
        )

    def setChecked(self, checked: bool) -> None:
        super().setChecked(checked)
        self._position = self.end_position
        self.update()

    def paintEvent(self, _event: QPaintEvent) -> None:
        painter = QPainter(self)
        painter.setRenderHint(QPainter.Antialiasing, True)
        painter.setPen(Qt.NoPen)
        self._paint_path(painter)
        self._paint_knob(painter)
        self._paint_label(painter)

    def _paint_path(self, painter: QPainter) -> None:
        painter.setBrush(QBrush(theme_manager.qcolor(colors.FRAME_BG)))
        rectangle = QRectF(
            self._margin,
            self._margin,
            self.width() - 2 * self._margin,
            self.height() - 2 * self._margin,
        )
        painter.drawRoundedRect(rectangle, self._path_radius, self._path_radius)

    def _current_knob_rectangle(self) -> QRectF:
        return QRectF(
            self.position - self._knob_radius,  # type: ignore
            2 * self._margin,
            2 * self._knob_radius,
            2 * self._knob_radius,
        )

    def _paint_knob(self, painter: QPainter) -> None:
        painter.setBrush(QBrush(theme_manager.qcolor(colors.HIGHLIGHT_BG)))
        painter.drawEllipse(self._current_knob_rectangle())

    def _paint_label(self, painter: QPainter) -> None:
        painter.setPen(QColor("white"))
        font = painter.font()
        font.setPixelSize(int(1.5 * self._knob_radius))
        painter.setFont(font)
        painter.drawText(self._current_knob_rectangle(), Qt.AlignCenter, self.label)

    def mouseReleaseEvent(self, event: QMouseEvent) -> None:
        super().mouseReleaseEvent(event)
        if event.button() == Qt.LeftButton:
            animation = QPropertyAnimation(self, b"position", self)
            animation.setDuration(100)
            animation.setStartValue(self.start_position)
            animation.setEndValue(self.end_position)
            animation.start()

    def enterEvent(self, event: QEvent) -> None:
        self.setCursor(Qt.PointingHandCursor)
        super().enterEvent(event)