#!/bin/bash

set -euo pipefail

# Source shared library

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"

source "$SCRIPT_DIR/git-ai-lib.sh"

# Check for llm installation

if ! command -v llm >/dev/null 2>&1; then

echo "Error: llm command not found. Install from https://llm.datasette.io/en/stable/setup.html" >&2

exit 1

fi

usage() {

cat << EOF

Usage: $(basename "$0") [options]

Generate and commit changes using AI messages.

Commits both staged and unstaged changes to tracked files (like git commit -a).

By default, also stages and commits untracked files (skip with --no-verify).

Options:

-m, --model MODEL Specify the LLM model to use (default: deepseek-coder or o1-mini)

Examples: gpt-4, claude-3-opus, mistral-7b, ...

-d, --dry-run Show changes without committing

-n, --no-verify Skip pre-commit hooks, skip adding untracked files, and pass --no-verify to git commit

--author=<author> Set commit author

-s, --signoff, --no-signoff

Add or remove signoff

--cleanup=<mode> Set commit cleanup mode

-e, --edit, --no-edit

Edit or skip editing commit message

-l, --list List available models

-h, --help Show this help

EOF

exit 1

}

# Default values

model=$(select_default_model)

script_dir="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"

# Capture original arguments before parsing

original_args=("$@")

dry_run=false

no_verify=false

extra_commit_args=()

# Parse arguments

while [[ $# -gt 0 ]]; do

case "$1" in

-m|--model)

model="$2"

shift 2

;;

-d|--dry-run)

dry_run=true

shift

;;

-n|--no-verify)

no_verify=true

extra_commit_args+=("--no-verify")

shift

;;

-l|--list)

echo "Available models:"

llm models

exit 0

;;

-h|--help)

usage

;;

--author=*|--cleanup=*|-e|--edit|--no-edit|-s|--signoff|--no-signoff)

extra_commit_args+=("$1")

shift

;;

*)

echo "Error: Unknown option: $1" >&2

usage

;;

esac

done

# Check if we're in a git repository

if ! git rev-parse --is-inside-work-tree >/dev/null 2>&1; then

echo "Error: Not in a git repository" >&2

exit 1

fi

# Run pre-commit hook unless --no-verify was specified

if [ "$no_verify" = false ]; then

git_dir=$(git rev-parse --git-dir)

pre_commit_hook="$git_dir/hooks/pre-commit"

if [ -x "$pre_commit_hook" ]; then

echo "Running pre-commit hook..."

"$pre_commit_hook"

fi

fi

# Check for any changes (staged, unstaged, or untracked)

if [ -z "$(git status --porcelain)" ]; then

echo "Error: No changes to commit" >&2

exit 1

fi

if [ "$dry_run" = true ]; then

echo "=== Changes to be committed ==="

git status --short

fi

# Load additional instructions from config files

repo_root=$(git rev-parse --show-toplevel)

additional_instructions=$(load_additional_instructions "$repo_root")

# Combine additional instructions with custom prompt (git-ai-commit doesn't have custom_prompt variable, so just use additional_instructions)

combined_prompt="$additional_instructions"

# Generate commit message with progress indicator

model_display=$(resolve_model_display "$model")

echo -e "\n== Commit Message (via ${model_display}) =="

echo -n "Generating commit message... " >&2

# Generate the diff and pipe through the message generator

if [ "$no_verify" = false ]; then

# Include untracked files

commit_msg=$({ git diff HEAD; git ls-files --others --exclude-standard | xargs -I{} echo "A {}"; } | generate_commit_message "$model" "$combined_prompt" "")

else

# Only tracked files

commit_msg=$(git diff HEAD | generate_commit_message "$model" "$combined_prompt" "")

fi

printf "\r\033[K" >&2 # Clear progress message

# Display the message

echo "$commit_msg"

if [ "$dry_run" = true ]; then

# Build suggestion command safely

suggestion="$(basename "$0")"

filtered_args=()

for arg in "${original_args[@]}"; do

[[ "$arg" != "-d" && "$arg" != "--dry-run" ]] && filtered_args+=("$arg")

done

if [ ${#filtered_args[@]} -gt 0 ]; then

suggestion+=" ${filtered_args[*]}"

fi

echo -e "\n== Next Steps ==\nTo use a message like this, run:\n $suggestion"

else

# Stage and commit changes

echo -e "\n== Committing changes... ==" >&2

if [ "$no_verify" = false ]; then

git add -A # Add all changes including untracked files

else

git add -u # Only add changes to tracked files

fi

git commit -m "$commit_msg" ${extra_commit_args[@]:-} >/dev/null && echo "Changes committed successfully!"

fi