from PySide6.QtGui import QAction, QIcon
from PySide6.QtWidgets import QWidget, QVBoxLayout, QMenu, QMessageBox, QTabWidget, QTableView
from PySide6.QtWidgets import QLabel, QMainWindow

from database import KontorDB
from database.media import MediaFile
from database.comic import Comic
from gui.dialogs import ExportKontorDialog, ImportKontorDialog
from gui.model_config import KontorModelConfig
from gui.table_model import KontorTableModel


class MainWindow(QMainWindow):

    def __init__(self, config):
        super().__init__()

        self.tick = QIcon('res/tick.png')
        self.cross = QIcon('res/cross.png')
        self.import_icon = QIcon("res/application-import.png")
        self.export_icon = QIcon("res/application-export.png")
        self.circle_icon = QIcon("res/arrow-circle-double.png")

        self.setWindowTitle("Kontor")
        self.setMinimumSize(800, 500)
        self._create_actions()
        self._create_menubar()
        self._create_toolbars()
        self._create_statusbar()

        self.data = []
        self.filter = {}
        self.kontor_db = KontorDB(config)
        self.central_widget = QWidget()
        parent_layout = QVBoxLayout()
        self.central_widget.setLayout(parent_layout)
        self.tabs = QTabWidget()
        self.tabs.addTab(self.generate_data_tab("comic", Comic), "Comics")
        self.tabs.addTab(self.generate_data_tab("media_file", MediaFile), "MediaFile")
        self.tabs.currentChanged.connect(self._tab_changed)
        #label.setAlignment(Qt.AlignmentFlag.AlignCenter)
        parent_layout.addWidget(self.tabs)

        self.setCentralWidget(self.central_widget)

    def _create_actions(self):
        self.newAction = QAction("&New", self)
        self.aboutAction = QAction("&Über...", self)
        self.aboutAction.triggered.connect(self.about)
        self.importAction = QAction(self.import_icon, "&Import", self)
        self.importAction.triggered.connect(self.import_from_file)
        self.exportAction = QAction(self.export_icon, "&Export", self)
        self.exportAction.triggered.connect(self.export_to_file)
        self.refreshAction = QAction(self.circle_icon, "&Refresh", self)
        self.refreshAction.triggered.connect(self.refresh)
        self.updateTitleAction = QAction("&Update Titles", self)
        self.downloadAction = QAction("&Download Videos", self)
        self.exitAction = QAction("&Beenden", self)
        self.exitAction.setShortcut("Alt+F4")
        self.exitAction.triggered.connect(self.close)

    def _create_menubar(self):
        menu_bar = self.menuBar()
        # File menu
        file_menu = QMenu("&Datei")
        menu_bar.addMenu(file_menu)
        file_menu.addAction(self.exitAction)
        # Kontor menu
        kontor_menu = QMenu("&Kontor")
        menu_bar.addMenu(kontor_menu)
        kontor_menu.addAction(self.importAction)
        kontor_menu.addAction(self.exportAction)
        comic_menu = QMenu("&Comic")
        tysc_menu = QMenu("&TradeYourSportCards")
        media_file_menu = QMenu("&MediaFile")
        media_file_menu.addAction(self.updateTitleAction)
        media_file_menu.addAction(self.downloadAction)
        kontor_menu.addMenu(comic_menu)
        kontor_menu.addMenu(tysc_menu)
        kontor_menu.addMenu(media_file_menu)
        # Help menu
        help_menu = QMenu("&Hilfe")
        menu_bar.addMenu(help_menu)
        help_menu.addAction(self.aboutAction)

    def _create_toolbars(self):
        # Kontor toolbar
        kontor_tool_bar = self.addToolBar("Kontor")
        kontor_tool_bar.addAction(self.importAction)
        kontor_tool_bar.addAction(self.exportAction)
        kontor_tool_bar.addAction(self.refreshAction)

    def _create_statusbar(self):
        self.statusBar = self.statusBar()
        self.statusBar.showMessage("Kontor ready", 6000)
        self.status_label = QLabel("")
        self.statusBar.addPermanentWidget(self.status_label)

    def about(self):
        QMessageBox.about(self.central_widget, "Über Kontor", f"Python: 3.11\nKontor: 0.1.0")

    def import_from_file(self):
        import_dlg = ImportKontorDialog(self)
        if import_dlg.exec():
            print(f"import DB from file {import_dlg.file_name}")
        else:
            print("no nothing for import")
        pass

    def export_to_file(self):
        export_dlg = ExportKontorDialog(self, self.kontor_db)
        if export_dlg.exec():
            print(export_dlg.get_tables_to_export())
            print(f"export DB to {export_dlg.file_name}")
            self.statusBar.showMessage(f"export DB to {export_dlg.file_name}", 3000)
            self.kontor_db.export_db(export_dlg.current_export_type, export_dlg.file_name, export_dlg.get_tables_to_export())
        else:
            self.statusBar.showMessage("Export cancelled", 3000)

    def refresh(self):
        self.data[self.tabs.currentIndex()].refresh()

    def _tab_changed(self, tab_index):
        self.data[tab_index].refresh()

    def generate_data_tab(self, table_name, table):
        data_tab = QWidget()
        table_config = KontorModelConfig(self.kontor_db, self, table_name, table)
        model = KontorTableModel(table_config)
        layout = QVBoxLayout()
        self.data.append(model)
        data_tab.setLayout(layout)
        table_view = QTableView()
        table_view.setModel(model)
        layout.addLayout(table_config.get_filter_layout())
        layout.addWidget(table_view)
        model.refresh()
        return data_tab