from pathlib import Path
from core.database import DatabaseManager
from core.media import MediaRepository
from ui.widgets.color_list_widget import ColorListWidget
from ui.widgets.thumbnail_list_widget import ThumbnailListWidget
from ui.widgets.split_view_widget import SplitView
from PySide6.QtWidgets import QPushButton


class MainController:
    def __init__(self, view):
        self.db = DatabaseManager()
        self.db.connect()
        self.media_repo = MediaRepository(self.db)
        self.media_repo.sync_media()

        self.view = view
        self.color_list: ColorListWidget = view.color_list_widget
        self.thumbnail_list: ThumbnailListWidget = view.thumbnail_widget
        self.split_view: SplitView = view.preview_widget

        self.photo_button: QPushButton = view.photo_button
        self.photo_button.clicked.connect(self.take_photo)

        self.color_list.colorSelected.connect(self.on_color_selected)
        self.color_list.editColor.connect(self.on_edit_color)
        self.thumbnail_list.selectedThumbnail.connect(self.on_thumbnail_selected)

    def load_colors(self) -> None:
        colors = self.db.get_all_colors()
        print("Loaded colors:", colors)
        self.color_list.set_colors(colors)


    def on_color_selected(self, color_name: str):
        print(f"Wybrano kolor: {color_name}")
        color_id = self.db.get_color_id(color_name)
        if color_id is not None:
            media_items = self.db.get_media_for_color(color_id)
            print(f"Media dla koloru {color_name} (ID: {color_id}):", media_items)

            self.thumbnail_list.list_widget.clear()
            for media in media_items:
                if media['file_type'] == 'photo':
                    file_name = Path(media['media_path']).name
                    self.thumbnail_list.add_thumbnail(media['media_path'], file_name, media['id'])
        else:
            print(f"Nie znaleziono koloru o nazwie: {color_name}")

    def on_edit_color(self, color_name: str):
        print(f"Edycja koloru: {color_name}")

    def on_thumbnail_selected(self, media_id: int):
        media = self.db.get_media(media_id)
        if media:
            print(f"Wybrano miniaturę o ID: {media_id}, ścieżka: {media['media_path']}")
            self.split_view.set_reference_image(media['media_path'])
        else:
            print(f"Nie znaleziono mediów o ID: {media_id}")

    def take_photo(self):
        print("Robienie zdjęcia...")
        self.split_view.toglle_live_view()