block_controller.py

"""Controller for handling block-related business logic."""

from __future__ import annotations

from typing import TYPE_CHECKING, List

from PyQt6.QtCore import QCoreApplication, QObject, QRunnable, QThreadPool, pyqtSignal

from block_events import AnalysisResultEvent, RhymeSyllableResult
from rhyme import rhyme_marks_for_block
from syllables import count_syllables

if TYPE_CHECKING:
    from model import Block, Project


class AnalysisWorker(QRunnable):
    """Worker to perform text analysis in a background thread."""

    def __init__(self, block_id: str, lines: List[str], target_widget: QObject) -> None:
        super().__init__()
        self.block_id = block_id
        self.lines = lines
        self.target_widget = target_widget

    def run(self) -> None:
        """Execute the analysis."""
        syllable_counts = [count_syllables(line) for line in self.lines]
        rhyme_marks = rhyme_marks_for_block(self.lines)
        result = RhymeSyllableResult(syllable_counts, rhyme_marks)
        event = AnalysisResultEvent(self.block_id, result)
        QCoreApplication.postEvent(self.target_widget, event)


class BlockController(QObject):
    """Manages business logic for blocks, decoupled from widgets."""

    def __init__(self, project: Project, parent: QObject | None = None) -> None:
        """Initialize the controller.

        Args:
            project: The main project model.
            parent: The parent QObject.
        """
        super().__init__(parent)
        self._project = project
        self._thread_pool = QThreadPool()
        self._thread_pool.setMaxThreadCount(2)

    def get_verse_number(self, block: Block) -> int:
        """Calculates the sequential index of a verse block within the project.

        Args:
            block: The block to get the verse number for.

        Returns:
            The 1-based index of the verse.
        """
        if block.section != "Verse" or block.kind == "instrumental":
            return 1

        verse_index = 0
        for b in self._project.blocks:
            if b.section == "Verse" and b.kind != "instrumental":
                verse_index += 1
            if b.id == block.id:
                return verse_index
        return verse_index

    def request_analysis(self, block: Block, target_widget: QObject) -> None:
        """Request an asynchronous analysis of a block's text.

        Args:
            block: The block to analyze.
            target_widget: The widget to post the result event to.
        """
        worker = AnalysisWorker(block.id, list(block.lines), target_widget)
        self._thread_pool.start(worker)