commit-msg

#!/usr/bin/env python3
"""
Pre-commit hook for commit message validation.
Uses local LLM to check and fix commit message format.
"""

import subprocess
import sys
import json
import os


# Your commit message standards
COMMIT_STANDARDS = """
Required format:
- Start with JIRA ticket ID in brackets: [PROJ-123]
- Followed by brief description in present tense
- Max 72 characters for subject line
- No trailing periods
- Examples:
  ✓ [INFRA-456] Add retry logic to API client
  ✓ [SEC-789] Fix SQL injection in user query
  ✗ [PROJ-123] Added new feature. (past tense, period)
  ✗ Fixed bug (missing ticket ID)
"""


def extract_commit_message():
    """Get the commit message from git."""
    try:
        msg = subprocess.check_output(
            ['git', 'log', '-1', '--pretty=%B', 'HEAD'],
            stderr=subprocess.DEVNULL
        ).decode('utf-8').strip()
        return msg
    except:
        # If no commits yet, read from COMMIT_EDITMSG
        commit_msg_file = '.git/COMMIT_EDITMSG'
        if os.path.exists(commit_msg_file):
            with open(commit_msg_file, 'r') as f:
                return f.read().strip()
    return None


def validate_with_llm(message: str) -> dict:
    """
    Send commit message to local LLM for validation.
    Replace this with your actual LLM integration.
    """
    # TODO: Replace with your LLM API call
    # Example for local Claude/GPT setup:
    
    prompt = f"""{COMMIT_STANDARDS}

Current commit message:
{message}

Respond in JSON format:
{{
  "valid": true/false,
  "issues": ["list of problems"],
  "suggested_fix": "corrected message if invalid"
}}
"""
    
    # Placeholder - replace with actual LLM call
    # For now, do basic regex validation
    import re
    
    ticket_pattern = r'^\[([A-Z]+-\d+)\]\s+'
    match = re.match(ticket_pattern, message)
    
    issues = []
    if not match:
        issues.append("Missing JIRA ticket ID in [PROJ-123] format")
    
    if message.endswith('.'):
        issues.append("Remove trailing period")
    
    if len(message.split('\n')[0]) > 72:
        issues.append("Subject line exceeds 72 characters")
    
    # Check for past tense verbs
    past_tense = ['added', 'fixed', 'updated', 'removed', 'changed']
    message_lower = message.lower()
    if any(word in message_lower for word in past_tense):
        issues.append("Use present tense (add, fix, update, not added, fixed, updated)")
    
    valid = len(issues) == 0
    
    # Generate suggested fix if invalid
    suggested = message
    if issues:
        # Basic auto-fix attempts
        suggested = suggested.rstrip('.')
        suggested = suggested.replace('Added', 'Add').replace('added', 'add')
        suggested = suggested.replace('Fixed', 'Fix').replace('fixed', 'fix')
        suggested = suggested.replace('Updated', 'Update').replace('updated', 'update')
        
        # If missing ticket, try to extract from branch name
        if not match:
            branch = subprocess.check_output(
                ['git', 'rev-parse', '--abbrev-ref', 'HEAD']
            ).decode('utf-8').strip()
            
            ticket_from_branch = re.search(r'([A-Z]+-\d+)', branch)
            if ticket_from_branch:
                suggested = f"[{ticket_from_branch.group(1)}] {suggested}"
    
    return {
        'valid': valid,
        'issues': issues,
        'suggested_fix': suggested if not valid else message
    }


def main():
    """Main validation logic."""
    message = extract_commit_message()
    
    if not message:
        print("No commit message found")
        sys.exit(1)
    
    # Skip validation for merge commits
    if message.startswith('Merge'):
        sys.exit(0)
    
    result = validate_with_llm(message)
    
    if result['valid']:
        print("✓ Commit message is valid")
        sys.exit(0)
    
    print("\n❌ Commit message validation failed:\n")
    for issue in result['issues']:
        print(f"  • {issue}")
    
    print(f"\nCurrent message:\n  {message}")
    print(f"\nSuggested fix:\n  {result['suggested_fix']}")
    
    # Ask user if they want to apply the fix
    response = input("\nApply suggested fix? (y/n): ").lower()
    
    if response == 'y':
        # Amend the commit with fixed message
        subprocess.run(['git', 'commit', '--amend', '-m', result['suggested_fix']])
        print("✓ Commit message updated")
        sys.exit(0)
    else:
        print("Commit aborted. Please fix the message manually.")
        sys.exit(1)


if __name__ == '__main__':
    main()