'''
Contains dialogs for notification and prompting.
'''
from PyQt6.QtWidgets import (
    QLabel, QVBoxLayout, QDialog, 
    QDialogButtonBox, QMessageBox, QComboBox
)
from PyQt6.QtWebEngineWidgets import QWebEngineView
from PyQt6.QtCore import Qt
from tools import jinja_template_loading
import logging
from backend.db import models
from typing import Literal

logger = logging.getLogger(f"submissions.{__name__}")

env = jinja_template_loading()


class QuestionAsker(QDialog):
    """
    dialog to ask yes/no questions
    """    
    def __init__(self, title:str, message:str):
        super().__init__()
        self.setWindowTitle(title)
        # NOTE: set yes/no buttons
        QBtn = QDialogButtonBox.StandardButton.Yes | QDialogButtonBox.StandardButton.No
        self.buttonBox = QDialogButtonBox(QBtn)
        self.buttonBox.accepted.connect(self.accept)
        self.buttonBox.rejected.connect(self.reject)
        self.layout = QVBoxLayout()
        # NOTE: Text for the yes/no question
        self.message = QLabel(message)
        self.layout.addWidget(self.message)
        self.layout.addWidget(self.buttonBox)
        self.setLayout(self.layout)


class AlertPop(QMessageBox):
    """
    Dialog to show an alert.
    """    
    def __init__(self, message:str, status:Literal['Information', 'Question', 'Warning', 'Critical'], owner:str|None=None):
        super().__init__()
        # NOTE: select icon by string
        icon = getattr(QMessageBox.Icon, status)
        self.setIcon(icon)
        self.setInformativeText(message)
        self.setWindowTitle(f"{owner} - {status.title()}")

class HTMLPop(QDialog):

    def __init__(self, html:str, owner:str|None=None, title:str="python"):
        super().__init__()

        self.webview = QWebEngineView(parent=self)
        self.layout = QVBoxLayout()
        self.setWindowTitle(title)
        self.webview.setHtml(html)
        self.webview.setMinimumSize(600, 500)
        self.webview.setMaximumSize(600, 500)
        self.layout.addWidget(self.webview)


class ObjectSelector(QDialog):
    """
    dialog to input BaseClass type manually
    """    
    def __init__(self, title:str, message:str, obj_type:str|models.BaseClass):
        super().__init__()
        self.setWindowTitle(title)
        self.widget = QComboBox()
        if isinstance(obj_type, str):
            obj_type: models.BaseClass = getattr(models, obj_type)
        items = [item.name for item in obj_type.query()]
        self.widget.addItems(items)
        self.widget.setEditable(False)
        # NOTE: set yes/no buttons
        QBtn = QDialogButtonBox.StandardButton.Ok | QDialogButtonBox.StandardButton.Cancel
        self.buttonBox = QDialogButtonBox(QBtn)
        self.buttonBox.accepted.connect(self.accept)
        self.buttonBox.rejected.connect(self.reject)
        self.layout = QVBoxLayout()
        # NOTE: Text for the yes/no question
        message = QLabel(message)
        self.layout.addWidget(message)
        self.layout.addWidget(self.widget)
        self.layout.addWidget(self.buttonBox)
        self.setLayout(self.layout)

    def parse_form(self) -> str:
        """
        Get KitType(str) from widget

        Returns:
            str: KitType as str
        """        
        return self.widget.currentText()