import gi
gi.require_version("Gtk", "4.0")
from gi.repository import Gtk

import src.models as models

class Filters(Gtk.Box):

    def __init__(self, init_books, init_progress, init_genre, msg):
        Gtk.Box.__init__(self, spacing=10)

        genres = models.get_genres(init_books)
        self._progress = init_progress
        self._genre = init_genre
        self._msg = msg

        self._progress_dropdown = Gtk.DropDown.new_from_strings(models.all_progress)
        self._progress_dropdown.set_selected(models.all_progress.index(self._progress))
        self._progress_dropdown.connect('notify::selected-item', self._on_selected_progress)
        self.append(self._progress_dropdown)

        self._genres_dropdown = self._get_genres_dropdown(genres)
        self.append(self._genres_dropdown)

    def get_progress(self):
        return self._progress

    def get_genre(self):
        return self._genre

    def set_genre(self, genre):
        self._genre = genre

    def select_progress(self, progress):
        self._progress = progress
        self._progress_dropdown.set_selected(models.all_progress.index(progress))

    # Hacky, this replaces the dropdown by a new one
    def actualize_genres(self, books):
        genres = models.get_genres(books)
        self._genre = self._genre if self._genre in genres else models.all_genres

        if self._last_installed_genres != genres:
            new_dropdown = self._get_genres_dropdown(genres)
            self.insert_child_after(new_dropdown, self._genres_dropdown)
            self.remove(self._genres_dropdown)
            self._genres_dropdown = new_dropdown
        else:
            self._genres_dropdown.set_selected(genres.index(self._genre))

    def _get_genres_dropdown(self, genres):
        self._last_installed_genres = genres
        dropdown = Gtk.DropDown.new_from_strings(genres)
        dropdown.set_selected(genres.index(self._genre))
        dropdown.connect('notify::selected-item', self._on_selected_genre)
        return dropdown

    def _on_selected_progress(self, dropdown, _data):
        self._progress = dropdown.get_selected_item().get_string()
        self._msg(['progress-selected', self._progress])

    def _on_selected_genre(self, dropdown, _data):
        self._genre = dropdown.get_selected_item().get_string()
        self._msg(['genre-selected', self._genre])