sum-cli 3.0.0__py3-none-any.whl

sum/setup/git_ops.py

@@ -0,0 +1,463 @@
+"""Git operations for SUM Platform sites.
+
+Handles git init, initial commit, and remote repository creation.
+Supports GitHub (via gh CLI) and Gitea (via REST API).
+"""
+
+from __future__ import annotations
+
+import os
+import shutil
+import subprocess
+from abc import ABC, abstractmethod
+from pathlib import Path
+
+from sum.exceptions import SetupError
+from sum.system_config import AgencyConfig, get_system_config
+from sum.utils.output import OutputFormatter
+
+# =============================================================================
+# Git Provider Abstraction
+# =============================================================================
+
+
+class GitProvider(ABC):
+    """Abstract base class for git hosting providers."""
+
+    @abstractmethod
+    def is_available(self) -> bool:
+        """Check if the provider is configured and accessible."""
+        ...
+
+    @abstractmethod
+    def create_repo(self, app_dir: Path, repo_name: str) -> str:
+        """Create a remote repository and push the code.
+
+        Args:
+            app_dir: Path to the local git repository.
+            repo_name: Name for the remote repository.
+
+        Returns:
+            URL of the created repository.
+
+        Raises:
+            SetupError: If repo creation or push fails.
+        """
+        ...
+
+    @abstractmethod
+    def get_repo_url(self, org: str, repo_name: str) -> str:
+        """Get the HTTPS URL for a repository."""
+        ...
+
+    @abstractmethod
+    def get_clone_url(self, org: str, repo_name: str) -> str:
+        """Get the SSH clone URL for a repository."""
+        ...
+
+    @property
+    @abstractmethod
+    def name(self) -> str:
+        """Human-readable provider name."""
+        ...
+
+
+class GitHubProvider(GitProvider):
+    """GitHub provider using gh CLI."""
+
+    def __init__(self, config: AgencyConfig) -> None:
+        self.org = config.github_org or ""
+
+    @property
+    def name(self) -> str:
+        return "GitHub"
+
+    def is_available(self) -> bool:
+        """Check if gh CLI is available and authenticated."""
+        if not shutil.which("gh"):
+            return False
+
+        try:
+            result = subprocess.run(
+                ["gh", "auth", "status"],
+                capture_output=True,
+                text=True,
+            )
+            return result.returncode == 0
+        except subprocess.SubprocessError:
+            return False
+
+    def get_repo_url(self, org: str, repo_name: str) -> str:
+        return f"https://github.com/{org}/{repo_name}"
+
+    def get_clone_url(self, org: str, repo_name: str) -> str:
+        return f"git@github.com:{org}/{repo_name}.git"
+
+    def create_repo(self, app_dir: Path, repo_name: str) -> str:
+        """Create a private GitHub repository and push the code."""
+        if not self.is_available():
+            raise SetupError(
+                "GitHub CLI (gh) is not available or not authenticated. "
+                "Run 'gh auth login' to authenticate, or use --no-git to skip."
+            )
+
+        try:
+            subprocess.run(
+                [
+                    "gh",
+                    "repo",
+                    "create",
+                    f"{self.org}/{repo_name}",
+                    "--private",
+                    "--source",
+                    str(app_dir),
+                    "--push",
+                ],
+                cwd=app_dir,
+                check=True,
+                capture_output=True,
+                text=True,
+            )
+
+            return self.get_repo_url(self.org, repo_name)
+
+        except subprocess.CalledProcessError as exc:
+            stderr = exc.stderr or ""
+            if "already exists" in stderr.lower():
+                OutputFormatter.warning(
+                    f"Repository {self.org}/{repo_name} already exists, "
+                    "attempting to push..."
+                )
+                return self._push_to_existing_repo(app_dir, repo_name)
+            raise SetupError(f"Failed to create GitHub repository: {stderr}") from exc
+
+    def _push_to_existing_repo(self, app_dir: Path, repo_name: str) -> str:
+        """Push to an existing GitHub repository."""
+        repo_url = self.get_repo_url(self.org, repo_name)
+
+        try:
+            result = subprocess.run(
+                ["git", "remote", "get-url", "origin"],
+                cwd=app_dir,
+                capture_output=True,
+                text=True,
+            )
+
+            if result.returncode != 0:
+                subprocess.run(
+                    [
+                        "git",
+                        "remote",
+                        "add",
+                        "origin",
+                        self.get_clone_url(self.org, repo_name),
+                    ],
+                    cwd=app_dir,
+                    check=True,
+                    capture_output=True,
+                )
+
+            subprocess.run(
+                ["git", "push", "-u", "origin", "main"],
+                cwd=app_dir,
+                check=True,
+                capture_output=True,
+            )
+
+            return repo_url
+
+        except subprocess.CalledProcessError as exc:
+            raise SetupError(
+                f"Failed to push to existing repository: {exc.stderr}"
+            ) from exc
+
+
+class GiteaProvider(GitProvider):
+    """Gitea provider using REST API."""
+
+    def __init__(self, config: AgencyConfig) -> None:
+        self.org = config.gitea_org or ""
+        self.base_url = (config.gitea_url or "").rstrip("/")
+        self.token_env = config.gitea_token_env
+
+    @property
+    def name(self) -> str:
+        return "Gitea"
+
+    def _get_token(self) -> str | None:
+        """Get the API token from environment."""
+        return os.environ.get(self.token_env)
+
+    def is_available(self) -> bool:
+        """Check if Gitea is configured with a valid token."""
+        if not self.base_url or not self.org:
+            return False
+        return self._get_token() is not None
+
+    def get_repo_url(self, org: str, repo_name: str) -> str:
+        return f"{self.base_url}/{org}/{repo_name}"
+
+    def get_clone_url(self, org: str, repo_name: str) -> str:
+        # Extract host from base_url for SSH
+        # e.g., https://gitea.example.com -> gitea.example.com
+        from urllib.parse import urlparse
+
+        parsed = urlparse(self.base_url)
+        host = parsed.netloc
+        return f"git@{host}:{org}/{repo_name}.git"
+
+    def create_repo(self, app_dir: Path, repo_name: str) -> str:
+        """Create a private Gitea repository and push the code."""
+        token = self._get_token()
+        if not token:
+            raise SetupError(
+                f"Gitea API token not found. Set the {self.token_env} environment "
+                "variable, or use --no-git to skip repository creation."
+            )
+
+        try:
+            import httpx
+        except ImportError:
+            raise SetupError(
+                "httpx is required for Gitea support. "
+                "Install with: pip install sum-cli[gitea]"
+            ) from None
+
+        # Create repository via API
+        api_url = f"{self.base_url}/api/v1/orgs/{self.org}/repos"
+        headers = {
+            "Authorization": f"token {token}",
+            "Content-Type": "application/json",
+        }
+        payload = {
+            "name": repo_name,
+            "private": True,
+            "auto_init": False,
+        }
+
+        try:
+            response = httpx.post(api_url, json=payload, headers=headers, timeout=30.0)
+
+            if response.status_code == 409:
+                # Repository already exists
+                OutputFormatter.warning(
+                    f"Repository {self.org}/{repo_name} already exists, "
+                    "attempting to push..."
+                )
+                return self._push_to_existing_repo(app_dir, repo_name)
+
+            if response.status_code not in (200, 201):
+                error_msg = response.text
+                try:
+                    error_data = response.json()
+                except ValueError:
+                    # Response body is not JSON; fall back to raw text message.
+                    pass
+                else:
+                    error_msg = error_data.get("message", response.text)
+                raise SetupError(
+                    f"Failed to create Gitea repository: {error_msg} "
+                    f"(HTTP {response.status_code})"
+                )
+
+            # Add remote and push
+            return self._setup_remote_and_push(app_dir, repo_name)
+
+        except httpx.RequestError as exc:
+            raise SetupError(
+                f"Failed to connect to Gitea at {self.base_url}: {exc}"
+            ) from exc
+
+    def _setup_remote_and_push(self, app_dir: Path, repo_name: str) -> str:
+        """Set up git remote and push to Gitea."""
+        clone_url = self.get_clone_url(self.org, repo_name)
+
+        try:
+            # Add remote
+            subprocess.run(
+                ["git", "remote", "add", "origin", clone_url],
+                cwd=app_dir,
+                check=True,
+                capture_output=True,
+            )
+
+            # Push
+            subprocess.run(
+                ["git", "push", "-u", "origin", "main"],
+                cwd=app_dir,
+                check=True,
+                capture_output=True,
+            )
+
+            return self.get_repo_url(self.org, repo_name)
+
+        except subprocess.CalledProcessError as exc:
+            raise SetupError(f"Failed to push to Gitea: {exc.stderr}") from exc
+
+    def _push_to_existing_repo(self, app_dir: Path, repo_name: str) -> str:
+        """Push to an existing Gitea repository."""
+        try:
+            result = subprocess.run(
+                ["git", "remote", "get-url", "origin"],
+                cwd=app_dir,
+                capture_output=True,
+                text=True,
+            )
+
+            if result.returncode != 0:
+                subprocess.run(
+                    [
+                        "git",
+                        "remote",
+                        "add",
+                        "origin",
+                        self.get_clone_url(self.org, repo_name),
+                    ],
+                    cwd=app_dir,
+                    check=True,
+                    capture_output=True,
+                )
+
+            subprocess.run(
+                ["git", "push", "-u", "origin", "main"],
+                cwd=app_dir,
+                check=True,
+                capture_output=True,
+            )
+
+            return self.get_repo_url(self.org, repo_name)
+
+        except subprocess.CalledProcessError as exc:
+            raise SetupError(
+                f"Failed to push to existing repository: {exc.stderr}"
+            ) from exc
+
+
+# =============================================================================
+# Provider Factory
+# =============================================================================
+
+
+def get_git_provider(config: AgencyConfig | None = None) -> GitProvider:
+    """Get the configured git provider instance.
+
+    Args:
+        config: Optional AgencyConfig. If not provided, loads from system config.
+
+    Returns:
+        GitProvider instance for the configured provider.
+    """
+    if config is None:
+        config = get_system_config().agency
+
+    if config.git_provider == "gitea":
+        return GiteaProvider(config)
+    return GitHubProvider(config)
+
+
+# =============================================================================
+# Public API
+# =============================================================================
+
+
+def init_git_repo(app_dir: Path) -> None:
+    """Initialize a git repository in the app directory."""
+    try:
+        subprocess.run(
+            ["git", "init"],
+            cwd=app_dir,
+            check=True,
+            capture_output=True,
+        )
+    except subprocess.CalledProcessError as exc:
+        raise SetupError(f"Failed to initialize git repository: {exc.stderr}") from exc
+
+
+def create_initial_commit(
+    app_dir: Path, message: str = "Initial scaffold from sum-platform init"
+) -> None:
+    """Create the initial git commit with all files.
+
+    Uses the system's git configuration for author identity.
+    """
+    try:
+        subprocess.run(
+            ["git", "add", "-A"],
+            cwd=app_dir,
+            check=True,
+            capture_output=True,
+        )
+
+        subprocess.run(
+            ["git", "commit", "-m", message],
+            cwd=app_dir,
+            check=True,
+            capture_output=True,
+        )
+    except subprocess.CalledProcessError as exc:
+        raise SetupError(f"Failed to create initial commit: {exc.stderr}") from exc
+
+
+def setup_git_for_site(
+    app_dir: Path,
+    site_slug: str,
+    skip_remote: bool = False,
+) -> str | None:
+    """Set up git repository for a new site.
+
+    Args:
+        app_dir: Path to the app directory.
+        site_slug: The site slug (used as repo name).
+        skip_remote: If True, only do local git init, skip remote repo creation.
+
+    Returns:
+        Remote repo URL if created, None if skipped.
+    """
+    # Initialize git repo
+    init_git_repo(app_dir)
+
+    # Create initial commit
+    create_initial_commit(app_dir)
+
+    if skip_remote:
+        OutputFormatter.info("Skipping remote repository creation (--no-git)")
+        return None
+
+    # Get the configured provider
+    provider = get_git_provider()
+
+    # Check if provider is available
+    if not provider.is_available():
+        OutputFormatter.warning(
+            f"{provider.name} is not available or not configured. "
+            "Skipping repository creation."
+        )
+        return None
+
+    # Create remote repo
+    repo_url = provider.create_repo(app_dir, site_slug)
+    return repo_url
+
+
+# =============================================================================
+# Backward Compatibility
+# =============================================================================
+
+
+# Keep old function names for backward compatibility
+def check_gh_cli() -> bool:
+    """Check if gh CLI is available and authenticated.
+
+    Deprecated: Use get_git_provider().is_available() instead.
+    """
+    provider = GitHubProvider(get_system_config().agency)
+    return provider.is_available()
+
+
+def create_github_repo(app_dir: Path, repo_name: str) -> str:
+    """Create a private GitHub repository and push the code.
+
+    Deprecated: Use get_git_provider().create_repo() instead.
+    """
+    provider = GitHubProvider(get_system_config().agency)
+    return provider.create_repo(app_dir, repo_name)