gcommit_app.py

#!/usr/bin/env python3
"""
GCommit - Main application class for AI-powered git commit message generation
"""

import subprocess
import sys
from typing import List, Tuple
from rich.console import Console
from rich.theme import Theme
from rich.table import Table
from rich.panel import Panel
from rich.progress import track
from rich.prompt import Prompt, Confirm
from rich.text import Text
from git_helper import GitHelper
from ollama_client import OllamaClient


class GCommit:
 """Main application class"""
 
 def __init__(self, ollama_url: str = "http://localhost:11434", model: str = "llama3", hint: str = ""):
 self.git = GitHelper()
 self.ollama = OllamaClient(ollama_url, model)
 self.hint = hint
 
 # Set up Rich console with custom theme
 custom_theme = Theme({
 "info": "dim cyan",
 "warning": "yellow",
 "danger": "bold red",
 "success": "bold green",
 "header": "bold magenta",
 "filename": "cyan",
 "highlight": "bold yellow"
 })
 self.console = Console(theme=custom_theme)
 
 def check_untracked_files(self) -> None:
 """Display warning for untracked files"""
 has_untracked, untracked_files = self.git.has_untracked_files()
 if has_untracked:
 self.console.print(Panel(
 "[warning]⚠️ Warning: Untracked files detected[/warning]",
 title="Git Status",
 border_style="yellow"
 ))
 
 table = Table(show_header=True, header_style="warning")
 table.add_column("Untracked Files", style="filename")
 for file in untracked_files:
 table.add_row(file)
 
 self.console.print(table)
 self.console.print("[info]These files will not be included in the commit.[/info]")
 self.console.print("[info]Use [highlight]git add <files>[/highlight] to track them.[/info]\n")
 
 def run(self) -> int:
 """Main application entry point"""
 # Display welcome header
 self.console.print(Panel(
 "[header]🤖 gcommit[/header] - AI-powered Git commit message generator",
 title="Welcome",
 border_style="magenta"
 ))
 
 # Check if we're in a git repository
 try:
 subprocess.run(['git', 'rev-parse', '--git-dir'], 
 capture_output=True, check=True)
 except subprocess.CalledProcessError:
 self.console.print("[danger]Error: Not in a git repository[/danger]")
 return 1
 
 # Check for untracked files
 self.check_untracked_files()
 
 # Get staged files
 staged_files = self.git.get_staged_files()
 if not staged_files:
 self.console.print(Panel(
 "[warning]No staged changes to commit[/warning] 😥",
 title="Git Status",
 border_style="yellow"
 ))
 self.console.print("[info]Use [highlight]git add <files>[/highlight] to stage your changes first.[/info]")
 return 0
 
 # Check Ollama availability
 if not self.ollama.is_available():
 self.console.print("[danger]Error: Ollama is not running or not accessible.[/danger]")
 self.console.print("[info]Please start Ollama with: [highlight]ollama serve[/highlight][/info]")
 return 1
 
 # Process each file
 self.console.rule("[header]Analyzing Staged Files[/header]")
 file_summaries = []
 
 for filepath in track(staged_files, description="Processing files..."):
 diff = self.git.get_file_diff(filepath)
 if diff:
 with self.console.status(f"[info]Analyzing[/info] [filename]{filepath}[/filename]...") as status:
 summary = self.ollama.summarize_file_changes(filepath, diff, self.hint)
 if summary:
 file_summaries.append((filepath, summary))
 
 if not file_summaries:
 self.console.print("[danger]Error: Could not summarize any file changes[/danger]")
 return 1
 
 # Display file summaries table
 self.console.rule("[header]File Analysis Results[/header]")
 summary_table = Table(show_header=True, header_style="header")
 summary_table.add_column("File", style="filename", width=30)
 summary_table.add_column("Summary", style="white")
 
 for filepath, summary in file_summaries:
 summary_table.add_row(filepath, summary)
 
 self.console.print(summary_table)
 
 # Generate commit message from summaries
 self.console.rule("[header]Generating Commit Message[/header]")
 with self.console.status("[info]Creating commit message...[/info]") as status:
 commit_message = self.ollama.generate_commit_message(file_summaries, self.hint)
 
 if not commit_message:
 self.console.print("[danger]Error: Failed to generate commit message[/danger]")
 return 1
 
 # Display and get confirmation
 self.console.rule("[header]Commit Message Preview[/header]")
 self.console.print(Panel(
 commit_message,
 title="Generated Commit Message",
 border_style="green",
 expand=False
 ))
 
 choices = ["Accept and commit", "Reject and cancel", "Edit message"]
 choice = Prompt.ask(
 "\nWhat would you like to do?",
 choices=["accept", "reject", "edit"],
 default="accept"
 )
 
 if choice == "accept":
 final_message = commit_message
 elif choice == "reject":
 self.console.print("[warning]Commit cancelled.[/warning]")
 return 0
 elif choice == "edit":
 final_message = Prompt.ask("Enter your commit message", default=commit_message)
 
 # Commit changes
 self.console.rule("[header]Committing Changes[/header]")
 with self.console.status("[info]Committing...[/info]") as status:
 if self.git.commit_changes(final_message):
 self.console.print("[success]✅ Changes committed successfully![/success]")
 return 0
 else:
 return 1