git_provider.py

import subprocess
from typing import List


class GitError(Exception):
    """Ошибка выполнения git-команды."""
    pass


class AbstractGitProvider(object):
    def amend(self, message: str) -> None:
        raise NotImplementedError()

    def get_last_diff(self) -> str:
        raise NotImplementedError()

    def get_commit_messages_samples(self) -> str:
        raise NotImplementedError()


class ShellGitProvider(AbstractGitProvider):
    @staticmethod
    def _run_command(args: List[str], check: bool = True) -> str:
        """
        Выполняет git-команду и возвращает stdout.
        При ошибке выбрасывает GitError.
        """
        try:
            proc = subprocess.run(
                ["git"] + args,
                capture_output=True,
                text=True,
                encoding="utf-8",
                check=check,
            )
            if proc.returncode != 0:
                raise GitError(f"git {' '.join(args)}: {proc.stderr.strip()}")
            return proc.stdout.strip()
        except subprocess.CalledProcessError as e:
            raise GitError(f"git {' '.join(args)}: {e.stderr.strip()}") from e
        except FileNotFoundError as e:
            raise GitError("Git не установлен или не найден в PATH") from e

    def amend(self, message: str):
        self._run_command(["commit", "--amend", "-m", message])

    def get_last_diff(self) -> str:
        """
        Возвращает diff последнего коммита (изменения относительно родителя).
        Для первого коммита возвращает diff всего коммита.
        """
        # Проверяем, есть ли родительский коммит
        try:
            self._run_command(["rev-parse", "HEAD^"], check=False)
            has_parent = True
        except GitError:
            has_parent = False

        if has_parent:
            diff = self._run_command(["diff", "HEAD^", "HEAD", "--no-color"])
        else:
            # Первый коммит: берём diff из самого коммита
            diff = self._run_command(["show", "--format=", "--no-color", "--patch", "HEAD"])
        return diff

    def get_commit_messages_samples(self) -> str:
        """
        Возвращает форматированные примеры сообщений коммитов из репозитория.
        Получает последние 20 сообщений и форматирует их как список.
        """
        try:
            messages = self._run_command(["log", "--format=%s", "-20"], check=False)
            if not messages:
                return ""

            lines = messages.split("\n")[1:]
            return "\n".join([f"- {msg}" for msg in lines if msg])
        except GitError:
            return ""