import os

import pytest
from PyQt6.QtCore import Qt
from PyQt6.QtWidgets import QComboBox, QPushButton

from buzz.model_loader import get_whisper_file_path, WhisperModelSize, get_local_model_path, TranscriptionModel, \
    ModelType
from buzz.widgets.models_preferences_widget import ModelsPreferencesWidget


class TestModelsPreferencesWidget:
    @pytest.fixture(scope='class')
    def clear_model_cache(self):
        file_path = get_whisper_file_path(size=WhisperModelSize.TINY)
        if os.path.isfile(file_path):
            os.remove(file_path)

    def test_should_show_model_list(self, qtbot):
        widget = ModelsPreferencesWidget()
        qtbot.add_widget(widget)

        first_item = widget.model_list_widget.topLevelItem(0)
        assert first_item.text(0) == 'Downloaded'

        second_item = widget.model_list_widget.topLevelItem(1)
        assert second_item.text(0) == 'Available for Download'

    def test_should_change_model_type(self, qtbot):
        widget = ModelsPreferencesWidget()
        qtbot.add_widget(widget)

        combo_box = widget.findChild(QComboBox)
        assert isinstance(combo_box, QComboBox)
        combo_box.setCurrentText('Faster Whisper')

        first_item = widget.model_list_widget.topLevelItem(0)
        assert first_item.text(0) == 'Downloaded'

        second_item = widget.model_list_widget.topLevelItem(1)
        assert second_item.text(0) == 'Available for Download'

    def test_should_download_model(self, qtbot, clear_model_cache):
        # make progress dialog non-modal to unblock qtbot.wait_until
        widget = ModelsPreferencesWidget(progress_dialog_modality=Qt.WindowModality.NonModal)
        qtbot.add_widget(widget)

        model = TranscriptionModel(model_type=ModelType.WHISPER, whisper_model_size=WhisperModelSize.TINY)

        assert get_local_model_path(model=model) is None

        available_item = widget.model_list_widget.topLevelItem(1)
        assert available_item.text(0) == 'Available for Download'

        tiny_item = available_item.child(0)
        assert tiny_item.text(0) == 'Tiny'
        tiny_item.setSelected(True)

        download_button = widget.findChild(QPushButton)
        assert isinstance(download_button, QPushButton)

        assert download_button.text() == 'Download'
        download_button.click()

        def downloaded_model():
            assert not download_button.isEnabled()

            _downloaded_item = widget.model_list_widget.topLevelItem(0)
            assert _downloaded_item.childCount() > 0
            assert _downloaded_item.child(0).text(0) == 'Tiny'

            _available_item = widget.model_list_widget.topLevelItem(1)
            assert _available_item.childCount() == 0 or _available_item.child(0).text(0) != 'Tiny'

            assert os.path.isfile(get_whisper_file_path(size=model.whisper_model_size))

        qtbot.wait_until(callback=downloaded_model, timeout=60_000)