gitwit 0.1.0__py3-none-any.whl

gitwit/git.py

@@ -0,0 +1,312 @@
+"""Git operations for GitWit."""
+
+import subprocess
+from dataclasses import dataclass
+from pathlib import Path
+
+
+class GitError(Exception):
+    """Exception raised for git-related errors."""
+    pass
+
+
+class NotAGitRepoError(GitError):
+    """Exception raised when not in a git repository."""
+    pass
+
+
+class NoStagedChangesError(GitError):
+    """Exception raised when there are no staged changes."""
+    pass
+
+
+@dataclass
+class DiffStats:
+    """Statistics about a git diff."""
+    files_changed: list[str]
+    additions: int
+    deletions: int
+    total_lines: int
+
+
+def run_git_command(args: list[str], cwd: Path | None = None) -> str:
+    """
+    Run a git command and return its output.
+
+    Args:
+        args: List of arguments to pass to git.
+        cwd: Working directory for the command.
+
+    Returns:
+        The command output as a string.
+
+    Raises:
+        GitError: If the command fails.
+        NotAGitRepoError: If not in a git repository.
+    """
+    try:
+        result = subprocess.run(
+            ["git"] + args,
+            capture_output=True,
+            text=True,
+            cwd=cwd,
+            timeout=30,
+        )
+
+        if result.returncode != 0:
+            stderr = result.stderr.strip()
+            if "not a git repository" in stderr.lower():
+                raise NotAGitRepoError("Not a git repository. Run 'git init' first.")
+            raise GitError(f"Git command failed: {stderr}")
+
+        return result.stdout
+
+    except subprocess.TimeoutExpired:
+        raise GitError("Git command timed out")
+    except FileNotFoundError:
+        raise GitError("Git is not installed or not in PATH")
+
+
+def is_git_repo(path: Path | None = None) -> bool:
+    """
+    Check if the current directory is a git repository.
+
+    Args:
+        path: Path to check. Defaults to current directory.
+
+    Returns:
+        True if in a git repository, False otherwise.
+    """
+    try:
+        run_git_command(["rev-parse", "--git-dir"], cwd=path)
+        return True
+    except GitError:
+        return False
+
+
+def get_repo_root(path: Path | None = None) -> Path:
+    """
+    Get the root directory of the git repository.
+
+    Args:
+        path: Path to start from. Defaults to current directory.
+
+    Returns:
+        Path to the repository root.
+
+    Raises:
+        NotAGitRepoError: If not in a git repository.
+    """
+    output = run_git_command(["rev-parse", "--show-toplevel"], cwd=path)
+    return Path(output.strip())
+
+
+def get_staged_diff(path: Path | None = None) -> str:
+    """
+    Get the diff of staged changes.
+
+    Args:
+        path: Repository path. Defaults to current directory.
+
+    Returns:
+        The staged diff as a string.
+
+    Raises:
+        NoStagedChangesError: If there are no staged changes.
+    """
+    diff = run_git_command(["diff", "--cached"], cwd=path)
+
+    if not diff.strip():
+        raise NoStagedChangesError(
+            "No staged changes. Stage changes with 'git add' first, "
+            "or use 'gitwit commit --all' to stage all changes."
+        )
+
+    return diff
+
+
+def get_staged_files(path: Path | None = None) -> list[str]:
+    """
+    Get a list of staged files.
+
+    Args:
+        path: Repository path. Defaults to current directory.
+
+    Returns:
+        List of staged file paths.
+    """
+    output = run_git_command(["diff", "--cached", "--name-only"], cwd=path)
+    return [f for f in output.strip().split("\n") if f]
+
+
+def has_staged_changes(path: Path | None = None) -> bool:
+    """
+    Check if there are any staged changes.
+
+    Args:
+        path: Repository path. Defaults to current directory.
+
+    Returns:
+        True if there are staged changes, False otherwise.
+    """
+    try:
+        get_staged_diff(path)
+        return True
+    except NoStagedChangesError:
+        return False
+
+
+def stage_all_changes(path: Path | None = None) -> None:
+    """
+    Stage all changes in the repository.
+
+    Args:
+        path: Repository path. Defaults to current directory.
+    """
+    run_git_command(["add", "-A"], cwd=path)
+
+
+def commit(message: str, path: Path | None = None) -> str:
+    """
+    Create a commit with the given message.
+
+    Args:
+        message: Commit message.
+        path: Repository path. Defaults to current directory.
+
+    Returns:
+        The commit output.
+    """
+    return run_git_command(["commit", "-m", message], cwd=path)
+
+
+def get_branch_name(path: Path | None = None) -> str:
+    """
+    Get the current branch name.
+
+    Args:
+        path: Repository path. Defaults to current directory.
+
+    Returns:
+        The current branch name.
+    """
+    output = run_git_command(["branch", "--show-current"], cwd=path)
+    return output.strip()
+
+
+def get_branch_diff(base_branch: str = "main", path: Path | None = None) -> str:
+    """
+    Get the diff between the current branch and a base branch.
+
+    Args:
+        base_branch: The base branch to compare against.
+        path: Repository path. Defaults to current directory.
+
+    Returns:
+        The diff as a string.
+    """
+    # Try the specified branch, fall back to 'master' if 'main' doesn't exist
+    try:
+        return run_git_command(["diff", f"{base_branch}...HEAD"], cwd=path)
+    except GitError:
+        if base_branch == "main":
+            return run_git_command(["diff", "master...HEAD"], cwd=path)
+        raise
+
+
+def parse_diff_stats(diff: str) -> DiffStats:
+    """
+    Parse a diff string and extract statistics.
+
+    Args:
+        diff: The git diff string.
+
+    Returns:
+        DiffStats with information about the changes.
+    """
+    lines = diff.split('\n')
+    files = []
+    additions = 0
+    deletions = 0
+
+    for line in lines:
+        if line.startswith('diff --git'):
+            # Extract filename from "diff --git a/path b/path"
+            parts = line.split(' ')
+            if len(parts) >= 4:
+                files.append(parts[2][2:])  # Remove 'a/' prefix
+        elif line.startswith('+') and not line.startswith('+++'):
+            additions += 1
+        elif line.startswith('-') and not line.startswith('---'):
+            deletions += 1
+
+    return DiffStats(
+        files_changed=files,
+        additions=additions,
+        deletions=deletions,
+        total_lines=len(lines)
+    )
+
+
+def get_current_branch(path: Path | None = None) -> str:
+    """
+    Get the current branch name.
+
+    Args:
+        path: Repository path. Defaults to current directory.
+
+    Returns:
+        The current branch name.
+    """
+    return get_branch_name(path)
+
+
+def get_default_branch(path: Path | None = None) -> str:
+    """
+    Detect the default branch (main or master).
+
+    Args:
+        path: Repository path. Defaults to current directory.
+
+    Returns:
+        'main' if it exists, otherwise 'master'.
+
+    Raises:
+        GitError: If neither main nor master exists.
+    """
+    # Check if 'main' branch exists
+    try:
+        run_git_command(["rev-parse", "--verify", "main"], cwd=path)
+        return "main"
+    except GitError:
+        pass
+
+    # Check if 'master' branch exists
+    try:
+        run_git_command(["rev-parse", "--verify", "master"], cwd=path)
+        return "master"
+    except GitError:
+        pass
+
+    raise GitError("Could not find default branch (main or master)")
+
+
+def get_recent_commits(count: int = 5, path: Path | None = None) -> list[str]:
+    """
+    Get recent commit messages for style reference.
+
+    Args:
+        count: Number of commits to retrieve.
+        path: Repository path. Defaults to current directory.
+
+    Returns:
+        List of recent commit messages.
+    """
+    try:
+        output = run_git_command(
+            ["log", f"-{count}", "--pretty=format:%s"],
+            cwd=path
+        )
+        return [msg for msg in output.strip().split("\n") if msg]
+    except GitError:
+        return []

gitwit/license.py

@@ -0,0 +1,124 @@
+"""License validation for GitWit (LemonSqueezy integration stub)."""
+
+from dataclasses import dataclass
+from typing import Any
+
+
+@dataclass
+class LicenseInfo:
+    """Information about a validated license."""
+    valid: bool
+    license_key: str | None = None
+    email: str | None = None
+    plan: str = "free"  # free, pro, team
+    features: list[str] | None = None
+
+    def __post_init__(self):
+        if self.features is None:
+            self.features = self._get_plan_features()
+
+    def _get_plan_features(self) -> list[str]:
+        """Get features for the current plan."""
+        free_features = ["commit"]
+        pro_features = free_features + ["pr", "changelog", "release", "custom-prompts"]
+        team_features = pro_features + ["team-config", "priority-support"]
+
+        if self.plan == "team":
+            return team_features
+        elif self.plan == "pro":
+            return pro_features
+        return free_features
+
+    def has_feature(self, feature: str) -> bool:
+        """Check if a feature is available in the current plan."""
+        return feature in (self.features or [])
+
+
+# Global license state
+_current_license: LicenseInfo | None = None
+
+
+def get_license() -> LicenseInfo:
+    """
+    Get the current license information.
+
+    Returns:
+        LicenseInfo with current license status.
+    """
+    global _current_license
+
+    if _current_license is None:
+        # For MVP, return free license
+        # TODO: Implement LemonSqueezy validation
+        _current_license = LicenseInfo(valid=True, plan="free")
+
+    return _current_license
+
+
+def validate_license(license_key: str) -> LicenseInfo:
+    """
+    Validate a license key with LemonSqueezy.
+
+    Args:
+        license_key: The license key to validate.
+
+    Returns:
+        LicenseInfo with validation result.
+
+    Note:
+        This is a stub for MVP. Full implementation will call LemonSqueezy API.
+    """
+    global _current_license
+
+    # TODO: Implement actual LemonSqueezy API validation
+    # API endpoint: https://api.lemonsqueezy.com/v1/licenses/validate
+    #
+    # For now, accept any key that looks valid (for testing)
+    if license_key and len(license_key) >= 10:
+        _current_license = LicenseInfo(
+            valid=True,
+            license_key=license_key,
+            plan="pro",
+        )
+    else:
+        _current_license = LicenseInfo(valid=False, plan="free")
+
+    return _current_license
+
+
+def require_feature(feature: str) -> bool:
+    """
+    Check if a feature is available, showing upgrade message if not.
+
+    Args:
+        feature: The feature to check.
+
+    Returns:
+        True if feature is available, False otherwise.
+    """
+    license_info = get_license()
+
+    if license_info.has_feature(feature):
+        return True
+
+    # Feature not available
+    return False
+
+
+def get_upgrade_message(feature: str) -> str:
+    """Get the upgrade message for a feature."""
+    feature_names = {
+        "pr": "PR description generation",
+        "changelog": "Changelog generation",
+        "release": "Release notes generation",
+        "custom-prompts": "Custom prompt templates",
+        "team-config": "Team configuration",
+    }
+
+    feature_name = feature_names.get(feature, feature)
+
+    return (
+        f"\n{feature_name} is a Pro feature.\n\n"
+        f"Upgrade to GitWit Pro: https://gitwit.dev/pricing\n"
+        f"Use code LAUNCH50 for 50% off!\n"
+    )

gitwit/prompts.py

@@ -0,0 +1,161 @@
+"""Prompt templates for AI operations."""
+
+COMMIT_SYSTEM_PROMPT = """You are a git commit message generator. Generate a single conventional commit message following the format:
+
+type(scope): description
+
+Rules:
+- Types: feat, fix, docs, style, refactor, test, chore, perf, ci, build
+- Scope is optional but encouraged (e.g., auth, api, ui, config)
+- Description must be lowercase, imperative mood, no period at end
+- Keep the first line under 72 characters
+- Only output the commit message, nothing else
+- No quotes around the message
+- No explanation or preamble
+
+Examples of good commit messages:
+- feat(auth): add OAuth2 login support
+- fix(api): handle null response from user endpoint
+- docs: update installation instructions
+- refactor(utils): simplify date parsing logic
+- chore: update dependencies to latest versions"""
+
+COMMIT_USER_PROMPT = """Generate a commit message for the following git diff:
+
+{diff}"""
+
+COMMIT_SUMMARY_USER_PROMPT = """Generate a commit message for the following changes:
+
+Files changed: {files}
+Total additions: +{additions}
+Total deletions: -{deletions}
+
+Partial diff (truncated due to size):
+{diff}"""
+
+
+def get_commit_prompt(diff: str, max_diff_chars: int = 4000) -> tuple[str, str]:
+    """
+    Get the system and user prompts for commit message generation.
+
+    If the diff is too long, summarizes it to fit within token limits.
+
+    Args:
+        diff: The git diff string
+        max_diff_chars: Maximum characters for the diff in the prompt
+
+    Returns:
+        Tuple of (system_prompt, user_prompt)
+    """
+    if len(diff) <= max_diff_chars:
+        return COMMIT_SYSTEM_PROMPT, COMMIT_USER_PROMPT.format(diff=diff)
+
+    # Parse diff to get summary info
+    lines = diff.split('\n')
+    files = []
+    additions = 0
+    deletions = 0
+
+    for line in lines:
+        if line.startswith('diff --git'):
+            # Extract filename from "diff --git a/path b/path"
+            parts = line.split(' ')
+            if len(parts) >= 4:
+                files.append(parts[2][2:])  # Remove 'a/' prefix
+        elif line.startswith('+') and not line.startswith('+++'):
+            additions += 1
+        elif line.startswith('-') and not line.startswith('---'):
+            deletions += 1
+
+    # Truncate diff to fit
+    truncated_diff = diff[:max_diff_chars - 500]  # Leave room for summary
+
+    user_prompt = COMMIT_SUMMARY_USER_PROMPT.format(
+        files=', '.join(files[:10]) + ('...' if len(files) > 10 else ''),
+        additions=additions,
+        deletions=deletions,
+        diff=truncated_diff
+    )
+
+    return COMMIT_SYSTEM_PROMPT, user_prompt
+
+
+# PR (Pull Request) prompts
+
+PR_SYSTEM_PROMPT = """You are a pull request description generator. Generate a PR title and description in the following exact format:
+
+Title: <short descriptive title under 72 characters>
+
+## Summary
+<1-3 sentences describing what this PR does and why>
+
+## Changes
+<bullet list of key changes>
+
+Rules:
+- Title should be concise and descriptive, no type prefix needed
+- Summary should explain the purpose and impact
+- Changes should list the main modifications made
+- Keep it professional and clear
+- No code blocks unless absolutely necessary
+- Output ONLY the formatted content, no extra text"""
+
+PR_USER_PROMPT = """Generate a PR title and description for the following branch diff:
+
+Branch: {branch_name}
+
+{diff}"""
+
+PR_SUMMARY_USER_PROMPT = """Generate a PR title and description for the following changes:
+
+Branch: {branch_name}
+Files changed: {files}
+Total additions: +{additions}
+Total deletions: -{deletions}
+
+Partial diff (truncated due to size):
+{diff}"""
+
+
+def get_pr_prompt(diff: str, branch_name: str, max_diff_chars: int = 6000) -> tuple[str, str]:
+    """
+    Get the system and user prompts for PR description generation.
+
+    Args:
+        diff: The git diff string
+        branch_name: The current branch name
+        max_diff_chars: Maximum characters for the diff in the prompt
+
+    Returns:
+        Tuple of (system_prompt, user_prompt)
+    """
+    if len(diff) <= max_diff_chars:
+        return PR_SYSTEM_PROMPT, PR_USER_PROMPT.format(branch_name=branch_name, diff=diff)
+
+    # Parse diff to get summary info
+    lines = diff.split('\n')
+    files = []
+    additions = 0
+    deletions = 0
+
+    for line in lines:
+        if line.startswith('diff --git'):
+            parts = line.split(' ')
+            if len(parts) >= 4:
+                files.append(parts[2][2:])
+        elif line.startswith('+') and not line.startswith('+++'):
+            additions += 1
+        elif line.startswith('-') and not line.startswith('---'):
+            deletions += 1
+
+    truncated_diff = diff[:max_diff_chars - 500]
+
+    user_prompt = PR_SUMMARY_USER_PROMPT.format(
+        branch_name=branch_name,
+        files=', '.join(files[:10]) + ('...' if len(files) > 10 else ''),
+        additions=additions,
+        deletions=deletions,
+        diff=truncated_diff
+    )
+
+    return PR_SYSTEM_PROMPT, user_prompt