git_repository.py

import subprocess, sys
from typing import List


class GitRepository:
    """Handles Git repository operations."""
    
    @staticmethod
    def get_root_path() -> str:
        """
        Get the root path of the current Git repository.
        
        Returns:
            str: Absolute path to the repository root
            
        Raises:
            SystemExit: If not inside a Git repository
        """
        try:
            result = subprocess.run(
                ["git", "rev-parse", "--show-toplevel"],
                capture_output=True,
                text=True,
                check=True,
                encoding='utf-8',
                errors='replace'
            )
            if result and result.stdout:
                return result.stdout.strip()
            return ""
        except subprocess.CalledProcessError:
            print("Error: Not inside a Git repository.", file=sys.stderr)
            sys.exit(1)
    
    @staticmethod
    def get_staged_diff() -> str:
        """
        Get the full diff of staged changes.
        
        Returns:
            str: Git diff output for staged changes
            
        Raises:
            SystemExit: If git diff command fails
        """
        try:
            result = subprocess.run(
                ["git", "diff", "--cached"],
                capture_output=True,
                text=True,
                check=True,
                encoding='utf-8',
                errors='replace'
            )
            if result and result.stdout:
                return result.stdout.strip()
            return ""
        except subprocess.CalledProcessError as e:
            print(f"Error getting diff: {e}", file=sys.stderr)
            sys.exit(1)
        except UnicodeDecodeError as e:
            print(f"Unicode decode error in git diff: {e}", file=sys.stderr)
            # Try again with binary mode and manual decoding
            try:
                result = subprocess.run(
                    ["git", "diff", "--cached"],
                    capture_output=True,
                    check=True
                )
                return result.stdout.decode('utf-8', errors='replace').strip()
            except Exception as e2:
                print(f"Fallback decode also failed: {e2}", file=sys.stderr)
                return ""
    
    @staticmethod
    def get_changed_files() -> List[str]:
        """
        Get list of files that have staged changes.
        
        Returns:
            List[str]: List of file paths with staged changes
            
        Raises:
            SystemExit: If git command fails
        """
        try:
            result = subprocess.run(
                ["git", "diff", "--cached", "--name-only"],
                capture_output=True,
                text=True,
                check=True,
                encoding='utf-8',
                errors='replace'
            )
            if result and result.stdout:
                return result.stdout.strip().splitlines() if result.stdout.strip() else []
            return []
        except subprocess.CalledProcessError as e:
            print(f"Error getting changed files: {e}", file=sys.stderr)
            sys.exit(1)
    
    @staticmethod
    def get_tracked_files() -> List[str]:
        """
        Get all files tracked by Git (including staged and committed).
        
        Returns:
            List[str]: List of all tracked file paths
        """
        try:
            result = subprocess.run(
                ["git", "ls-files", "--exclude-standard", "--others", "--cached"],
                capture_output=True,
                text=True,
                check=True,
                encoding='utf-8',
                errors='replace'
            )
            if result and result.stdout:
                return result.stdout.strip().splitlines() if result.stdout.strip() else []
            return []
        except subprocess.CalledProcessError:
            return []