Merge branch 'fix-cli-help-text' of https://github.com/Sendi0011/gitingest into fix-cli-help-text

Author: Sendi0011

current_help.txt

@@ -2,10 +2,12 @@
 
 from pathlib import Path
 from typing import Optional
+from urllib.parse import urlparse
 
 from gitingest.config import DEFAULT_TIMEOUT
 from gitingest.schemas import CloneConfig
 from gitingest.utils.git_utils import (
+    _is_github_host,
     check_repo_exists,
     create_git_auth_header,
     create_git_command,
@@ -48,7 +50,7 @@ async def clone_repo(config: CloneConfig, token: Optional[str] = None) -> None:
     partial_clone: bool = config.subpath != "/"
 
     # Validate token if provided
-    if token and url.startswith("https://github.com"):
+    if token and _is_github_host(url):
         validate_github_token(token)
 
     # Create parent directory if it doesn't exist
@@ -59,8 +61,14 @@ async def clone_repo(config: CloneConfig, token: Optional[str] = None) -> None:
         raise ValueError("Repository not found. Make sure it is public or that you have provided a valid token.")
 
     clone_cmd = ["git"]
-    if token and url.startswith("https://github.com"):
-        clone_cmd += ["-c", create_git_auth_header(token)]
+    if token and _is_github_host(url):
+        # Only pass URL if it's not the default github.com to maintain backward compatibility
+
+        parsed = urlparse(url)
+        if parsed.hostname == "github.com":
+            clone_cmd += ["-c", create_git_auth_header(token)]
+        else:
+            clone_cmd += ["-c", create_git_auth_header(token, url)]
 
     clone_cmd += ["clone", "--single-branch"]
     # TODO: Re-enable --recurse-submodules when submodule support is needed

src/gitingest/cloning.py

@@ -4,12 +4,32 @@
 import base64
 import re
 from typing import List, Optional, Tuple
+from urllib.parse import urlparse
 
 from gitingest.utils.exceptions import InvalidGitHubTokenError
 
 GITHUB_PAT_PATTERN = r"^(?:github_pat_|ghp_)[A-Za-z0-9_]{36,}$"
 
 
+def _is_github_host(url: str) -> bool:
+    """
+    Check if a URL is from a GitHub host (github.com or GitHub Enterprise).
+
+    Parameters
+    ----------
+    url : str
+        The URL to check
+
+    Returns
+    -------
+    bool
+        True if the URL is from a GitHub host, False otherwise
+    """
+    parsed = urlparse(url)
+    hostname = parsed.hostname or ""
+    return hostname == "github.com" or hostname.startswith("github.")
+
+
 async def run_command(*args: str) -> Tuple[bytes, bytes]:
     """
     Execute a shell command asynchronously and return (stdout, stderr) bytes.
@@ -80,7 +100,7 @@ async def check_repo_exists(url: str, token: Optional[str] = None) -> bool:
     RuntimeError
         If the curl command returns an unexpected status code.
     """
-    if token and "github.com" in url:
+    if token and _is_github_host(url):
         return await _check_github_repo_exists(url, token)
 
     proc = await asyncio.create_subprocess_exec(
@@ -131,12 +151,18 @@ async def _check_github_repo_exists(url: str, token: Optional[str] = None) -> bo
     RuntimeError
         If the repository is not found, if the provided URL is invalid, or if the token format is invalid.
     """
-    m = re.match(r"https?://github\.com/([^/]+)/([^/]+?)(?:\.git)?/?$", url)
+    m = re.match(r"https?://github\.([^/]*)/([^/]+)/([^/]+?)(?:\.git)?/?$", url)
     if not m:
-        raise ValueError(f"Un-recognised GitHub URL: {url!r}")
-    owner, repo = m.groups()
-
-    api = f"https://api.github.com/repos/{owner}/{repo}"
+        m = re.match(r"https?://github\.com/([^/]+)/([^/]+?)(?:\.git)?/?$", url)
+        if not m:
+            raise ValueError(f"Un-recognised GitHub URL: {url!r}")
+        owner, repo = m.groups()
+        api = f"https://api.github.com/repos/{owner}/{repo}"
+    else:
+        _, owner, repo = m.groups()
+
+        parsed = urlparse(url)
+        api = f"https://{parsed.hostname}/api/v3/repos/{owner}/{repo}"
     cmd = [
         "curl",
         "--silent",
@@ -189,8 +215,14 @@ async def fetch_remote_branch_list(url: str, token: Optional[str] = None) -> Lis
     fetch_branches_command = ["git"]
 
     # Add authentication if needed
-    if token and "github.com" in url:
-        fetch_branches_command += ["-c", create_git_auth_header(token)]
+    if token and _is_github_host(url):
+        # Only pass URL if it's not the default github.com to maintain backward compatibility
+
+        parsed = urlparse(url)
+        if parsed.hostname == "github.com":
+            fetch_branches_command += ["-c", create_git_auth_header(token)]
+        else:
+            fetch_branches_command += ["-c", create_git_auth_header(token, url)]
 
     fetch_branches_command += ["ls-remote", "--heads", url]
 
@@ -225,27 +257,39 @@ def create_git_command(base_cmd: List[str], local_path: str, url: str, token: Op
         The git command with authentication if needed
     """
     cmd = base_cmd + ["-C", local_path]
-    if token and url.startswith("https://github.com"):
+    if token and _is_github_host(url):
         validate_github_token(token)
-        cmd += ["-c", create_git_auth_header(token)]
+        # Only pass URL if it's not the default github.com to maintain backward compatibility
+
+        parsed = urlparse(url)
+        if parsed.hostname == "github.com":
+            cmd += ["-c", create_git_auth_header(token)]
+        else:
+            cmd += ["-c", create_git_auth_header(token, url)]
     return cmd
 
 
-def create_git_auth_header(token: str) -> str:
+def create_git_auth_header(token: str, url: str = "https://github.com") -> str:
     """Create a Basic authentication header for GitHub git operations.
 
     Parameters
     ----------
     token : str
         GitHub personal access token
+    url : str
+        The GitHub URL to create the authentication header for.
+        Defaults to "https://github.com".
 
     Returns
     -------
     str
         The git config command for setting the authentication header
     """
+
+    parsed = urlparse(url)
+    hostname = parsed.hostname or "github.com"
     basic = base64.b64encode(f"x-oauth-basic:{token}".encode()).decode()
-    return f"http.https://github.com/.extraheader=Authorization: Basic {basic}"
+    return f"http.https://{hostname}/.extraheader=Authorization: Basic {basic}"
 
 
 def validate_github_token(token: str) -> None:

src/gitingest/utils/git_utils.py

@@ -85,8 +85,8 @@ def _looks_like_git_host(host: str) -> bool:
     """
     Check if the given host looks like a Git host.
 
-    The current heuristic returns `True` when the host starts with `git.` (e.g. `git.example.com`) or starts with
-    `gitlab.` (e.g. `gitlab.company.com`).
+    The current heuristic returns `True` when the host starts with `git.` (e.g. `git.example.com`), starts with
+    `gitlab.` (e.g. `gitlab.company.com`), or starts with `github.` (e.g. `github.company.com` for GitHub Enterprise).
 
     Parameters
     ----------
@@ -99,7 +99,7 @@ def _looks_like_git_host(host: str) -> bool:
         True if the host looks like a Git host, otherwise False.
     """
     host = host.lower()
-    return host.startswith(("git.", "gitlab."))
+    return host.startswith(("git.", "gitlab.", "github."))
 
 
 def _validate_url_scheme(scheme: str) -> None: