from tofu_api.common.rest import BaseBlueprint
from .task_views import TaskCollectionView, TaskItemView


class TaskBlueprint(BaseBlueprint):
    """
    Blueprint for the tasks REST API.
    """

    # Blueprint settings
    name = 'rest_api_tasks'
    import_name = __name__
    url_prefix = '/tasks'

    def init_blueprint(self) -> None:
        """
        Register URL rules.
        """
        task_handler = self.app.dependencies.get_task_handler()

        self.add_url_rule(
            '',
            view_func=TaskCollectionView.as_view(task_handler=task_handler),
            methods=['GET', 'POST'],
        )
        self.add_url_rule(
            '/<int:task_id>',
            view_func=TaskItemView.as_view(task_handler=task_handler),
            methods=['GET', 'PATCH', 'DELETE'],
        )