from contextlib import suppress
from pathlib import Path
from typing import List

from PyQt5.QtCore import Qt
from PyQt5.QtGui import QColor
from PyQt5.QtWidgets import QFileDialog, QListWidgetItem

with suppress(ImportError):
    from finalcif.appwindow import AppWindow
from finalcif.tools.settings import FinalCifSettings


class ReportTemplates:
    """
    Displays the list of report templates in the options menu.
    """

    def __init__(self, app: 'AppWindow', settings: FinalCifSettings):
        self.app = app
        self.settings = settings
        self.lw = self.app.ui.TemplatesListWidget
        self.load_templates_list()
        self.app.ui.AddNewTemplPushButton.clicked.connect(self.add_new_template)
        self.app.ui.RemoveTemplPushButton.clicked.connect(self.remove_current_template)
        self.app.ui.TemplatesListWidget.currentItemChanged.connect(self.template_changed)
        self.app.ui.TemplatesListWidget.itemChanged.connect(self.template_changed)
        self.app.ui.TemplatesListWidget.setCurrentItem(
            self.app.ui.TemplatesListWidget.item(self.app.options.current_template))

    def add_new_template(self, templ_path: str = '') -> None:
        if not templ_path:
            templ_path, _ = QFileDialog.getOpenFileName(filter="DOCX file (*.docx)", initialFilter="DOCX file (*.docx)",
                                                        caption='Open a Report Template File')
        itemslist = self.get_templates_list_from_widget()
        self.app.status_bar.show_message('')
        if templ_path in itemslist:
            self.app.status_bar.show_message('This templates is already in the list.', 10)
            print('This templates is already in the list.')
            return
        if not Path(templ_path).exists() or not Path(templ_path).is_file() \
            or not Path(templ_path).name.endswith('.docx'):
            self.app.status_bar.show_message('This template does not exist or is unreadable.', 10)
            print('This template does not exist or is unreadable.', Path(templ_path).resolve())
            return
        item = QListWidgetItem(templ_path)
        item.setCheckState(Qt.Unchecked)
        self.app.ui.TemplatesListWidget.addItem(item)
        self.settings.save_template_list('report_templates_list', self.get_templates_list_from_widget())

    def load_templates_list(self):
        templates = self.settings.load_template('report_templates_list')
        if not templates:
            return
        for text in templates:
            if text.startswith('Use'):
                continue
            with suppress(Exception):
                if not Path(text).exists():
                    item = QListWidgetItem(text)
                    item.setForeground(QColor(220, 12, 34))
                else:
                    item = QListWidgetItem(str(Path(text).resolve(strict=True)))
            self.app.ui.TemplatesListWidget.addItem(item)
            item.setCheckState(Qt.Unchecked)

    def get_templates_list_from_widget(self) -> List:
        itemslist = []
        for num in range(self.lw.count()):
            itemtext = self.lw.item(num).text()
            if not itemtext in itemslist:
                itemslist.append(itemtext)
        return itemslist

    def remove_current_template(self) -> None:
        if self.lw.currentRow() == 0:
            return
        self.lw.takeItem(self.lw.row(self.lw.currentItem()))
        self.settings.save_template_list('report_templates_list', self.get_templates_list_from_widget())

    def template_changed(self, current_item: QListWidgetItem):
        # Blocking signal in order to avoid infinitive recursion:
        self.app.ui.TemplatesListWidget.blockSignals(True)
        options = self.settings.load_options()
        options.update({'current_report_template': self.lw.row(current_item)})
        self.uncheck_all_templates()
        if not current_item:
            self.app.ui.TemplatesListWidget.blockSignals(False)
            return
        current_item.setCheckState(Qt.Checked)
        self.settings.save_options(options)
        self.app.ui.TemplatesListWidget.blockSignals(False)

    def uncheck_all_templates(self):
        for num in range(self.lw.count()):
            self.lw.item(num).setCheckState(Qt.Unchecked)
