from typing import Protocol

from prophet.domain.improvement import Improvement


class ImprovementNotFoundError(Exception):
    pass


class IImprovementRepo(Protocol):
    def add(self, improvement: Improvement) -> None:
        raise NotImplementedError

    def add_all(self, improvements: list[Improvement]) -> None:
        raise NotImplementedError

    def get(self, id: str) -> Improvement:
        raise NotImplementedError

    def get_all(self, last_n: int | None = None) -> list[Improvement]:
        raise NotImplementedError

    def remove(self, id: str) -> Improvement:
        """Returns single deleted improvement"""
        raise NotImplementedError

    def remove_all(self, ids: list[str]) -> list[Improvement]:
        """Returns list of deleted improvements"""
        raise NotImplementedError