from typing import Any, Union

from PyQt6.QtCore import *
from PyQt6.QtGui import *
from PyQt6.QtWidgets import *

from ..gui import QXImage
from ._part_QXWidget import _part_QXWidget


class QXLabel(QLabel, _part_QXWidget):
    def __init__(self,  text = None,
                        color = None,
                        image : QXImage = None,
                        movie = None,
                        word_wrap = False, scaled_contents = False,
                        **kwargs):
        super().__init__()
        _part_QXWidget.__init__(self, **kwargs)

        if text is not None:
            self.setText(text)
        if movie is not None:
            self.setMovie(movie)
        if image is not None:
            self.setPixmap(image.as_QXPixmap())
        if word_wrap:
            self.setWordWrap(True)

        self.setScaledContents(scaled_contents)
        
        self._default_pal = QPalette( self.palette() )
        self.set_color(color)

    def _update_color(self):
        if self._color is not None:
            pal = QPalette(self._default_pal)
            pal.setColor( QPalette.ColorRole.WindowText, self._color )
            self.setPalette(pal)
        else:
            self.setPalette(self._default_pal)

    def set_color(self, color : Union[Any,None] ):
        self._color = QColor(color) if color is not None else None
        self._update_color()

    def changeEvent(self, ev : QEvent):
        super().changeEvent(ev)
        if ev.type() == QEvent.Type.EnabledChange:
            self._update_color()

    def focusInEvent(self, ev : QFocusEvent):
        super().focusInEvent(ev)
        _part_QXWidget.focusInEvent(self, ev)

    def resizeEvent(self, ev : QResizeEvent):
        super().resizeEvent(ev)
        _part_QXWidget.resizeEvent(self, ev)