from PyQt6.QtGui import QIcon
from PyQt6.QtWidgets import QWidget, QLabel, QFormLayout, QSpinBox, QDoubleSpinBox, QLineEdit, QHBoxLayout, QPushButton, QFileDialog, QCheckBox
from PyQt6.QtCore import pyqtSignal
from cpdctrl.utility.config_file import ConfigFile


class FileSelection(QWidget):
    """
    Widget allowing the user to select a file or directory.
    """
    value_changed = pyqtSignal(str)
    def __init__(self, init_file="", filemode=QFileDialog.FileMode.AnyFile, parent=None):
        super().__init__(parent)
        self.setLayout(QHBoxLayout())
        self.w_edit = QLineEdit()
        self.w_edit.setText(init_file)
        self.w_btn = QPushButton()
        self.w_btn.setIcon(QIcon.fromTheme(QIcon.ThemeIcon.FolderOpen))
        self.layout().addWidget(self.w_edit)
        self.layout().addWidget(self.w_btn)
        # open file dialog when button is clicked
        self.w_btn.clicked.connect(self._open_file_dialog)
        # emit value_changed on self when w_edit emits value_changed
        self.w_edit.textChanged.connect(self.value_changed)

        self.file_mode = filemode
        # remove all spacing and padding from the layout
        self.layout().setContentsMargins(0, 0, 0, 0)

    def setValue(self, value: str):
        self.w_edit.setText(value)

    def _open_file_dialog(self):
        dialog = QFileDialog(self)
        # only directories
        dialog.setFileMode(self.file_mode)
        if dialog.exec():
            dirname = dialog.selectedFiles()[0]
            self.w_edit.setText(dirname)

class SettingsForm(QWidget):
    """
    Form that is connected to a config file instance
    """
    def __init__(self, config_file: ConfigFile, parent=None):
        super().__init__(parent)
        self.setLayout(QFormLayout())
        self.ws_form = {}
        self.config_file = config_file

    def __contains__(self, item):
        return item in self.ws_form

    def add_form_row(self, key: str, label: str, default_value, widget: QWidget, tooltip: str = None):
        """
        Add a row to the form. Uses the value from the config file corresponding to <key> or the default value.
        Parameters
        ----------
        key
        label: str
            Label for the form widget
        default_value
        widget
            Widget to add to the form
        tooltip

        Returns
        -------

        """
        if tooltip: widget.setToolTip(tooltip)
        value = self.config_file.get_or(key, default_value)
        # set the value depending on the type of the widget
        if isinstance(widget, QSpinBox) or isinstance(widget, QDoubleSpinBox):
            widget.setValue(value)
            widget.valueChanged.connect(lambda v: self.value_updated(key, v))
        elif isinstance(widget, QCheckBox):
            widget.setChecked(value)
            widget.stateChanged.connect(lambda v: self.value_updated(key, v))
        elif isinstance(widget, FileSelection):
            widget.setValue(value)
            widget.value_changed.connect(lambda v: self.value_updated(key, v))
        else:
            raise ValueError(f"Unknown widget type: {type(widget)}")
        self.layout().addRow(QLabel(label), widget)
        self.ws_form[key] = widget


    def value_updated(self, key, value):
        """
        Update the value in the config file when it is updated in the widget
        Parameters
        ----------
        key
        value
        """
        self.config_file.set(key, value)

    def set_value(self, key, value):
        """
        Set the value of the widget with the given key.
        ""
        Parameters
        ----------
        key
        value

        Returns
        -------
        """
        if key in self.ws_form:
            # set depending on widget type
            if isinstance(self.ws_form[key], QSpinBox) or isinstance(self.ws_form[key], QDoubleSpinBox):
                self.ws_form[key].setValue(value)
            elif isinstance(self.ws_form[key], QCheckBox):
                self.ws_form[key].setChecked(value)
            elif isinstance(self.ws_form[key], FileSelection):
                self.ws_form[key].setValue(value)
            else:
                raise ValueError(f"Unknown widget type: {type(self.ws_form[key])}")
        else:
            raise ValueError(f"Unknown key: {key}")

    def get_value(self, key):
        """
        Get the value of the widget with the given key.
        Parameters
        ----------
        key

        Returns
        -------

        """
        if key in self.ws_form:
            # get depending on widget type
            if isinstance(self.ws_form[key], QSpinBox) or isinstance(self.ws_form[key], QDoubleSpinBox):
                return self.ws_form[key].value()
            elif isinstance(self.ws_form[key], QCheckBox):
                return self.ws_form[key].isChecked()
            else:
                raise ValueError(f"Unknown widget type: {type(self.ws_form[key])}")
        else:
            raise ValueError(f"Unknown key: {key}")