install.py

"""
Installation script for commit message generator git hook.
"""

import os
import sys
import shutil
import subprocess
from pathlib import Path

def find_git_root():
    """Find the git root directory starting from current directory."""
    current = Path.cwd()
    while current != current.parent:
        if (current / ".git").exists():
            return current
        current = current.parent
    return None

def install_hook():
    """Install the commit message generator as a git hook."""
    git_root = find_git_root()
    if not git_root:
        print("Error: Not in a git repository. Please run this from within a git repo.", file=sys.stderr)
        return False
    
    hooks_dir = git_root / ".git" / "hooks"
    script_dir = Path(__file__).parent.absolute()
    
    print(f"Installing hook in: {hooks_dir}")
    print(f"Script location: {script_dir}")
    
    hooks_dir.mkdir(exist_ok=True)
    
    target_dir = hooks_dir / "commit-msg-gen"
    if target_dir.exists():
        print("Removing existing installation...")
        shutil.rmtree(target_dir)
    
    print("Copying files...")
    shutil.copytree(script_dir, target_dir)
    
    hook_file = hooks_dir / "prepare-commit-msg"
    
    if os.name == 'nt':  # Windows
        hook_content = f'''@echo off
"%~dp0commit-msg-gen\\venv\\Scripts\\python.exe" "%~dp0commit-msg-gen\\main.py" %*
'''
    else:  # Unix-like
        # Create a shell script that calls our wrapper
        hook_content = f'''#!/bin/bash
SCRIPT_DIR="$(cd "$(dirname "${{BASH_SOURCE[0]}}")" && pwd)"
"$SCRIPT_DIR/commit-msg-gen/venv/bin/python" "$SCRIPT_DIR/commit-msg-gen/main.py" "$@"
'''
    
    with open(hook_file, 'w') as f:
        f.write(hook_content)
    
    if os.name != 'nt':
        os.chmod(hook_file, 0o755)
    
    print("Installation complete!")
    print("\nMake sure you have:")
    print("1. Set up your OpenAI API key in commit-msg-gen/.env")
    print("2. The virtual environment is properly set up (run setup.py if needed)")
    
    return True

def main():
    """Main installation function."""
    if len(sys.argv) > 1 and sys.argv[1] in ['--help', '-h']:
        print("Usage: python install.py")
        print("Installs the commit message generator as a git hook in the current repository.")
        return
    
    if install_hook():
        print("\n Hook installed successfully!")
    else:
        print("\n Installation failed!")
        sys.exit(1)

if __name__ == "__main__":
    main()