#!/bin/bash

function commit-with-llm() {

    mkdir -p tmp
    PROMPT_FILE="tmp/llm_update"
    echo -e "\nWrite prompt to $PROMPT_FILE."
    cat << EOF > "$PROMPT_FILE"
Look at the git log and the git diff below and reply with a git commit message matching the changes and style of the log.
The commit message must comply with the conventional commits specification. Reply with the commit message only.

<<<  git log -n 5 --pretty=format:"%ci: %B" >>>
$(git log -n 5 --pretty=format:"%ci: %B")

<<< git diff --staged >>>
$(git diff --staged)
EOF

    # Run the llm command
    echo -e "Send prompt with staged git diff and wait for the response of the $(llm models | sed -n 's/^Default: //p') LLM."
    RESULT=$(cat "$PROMPT_FILE" | llm)

    # Check if result is empty
    if [ -z "$RESULT" ]; then
        echo "No response from the model. Exiting."
        exit 1
    fi

    # Save the result to a file
    RESULT_FILE="tmp/llm_update_result"
    echo "$RESULT" > "$RESULT_FILE"
    echo -e "Saved response to $RESULT_FILE.\n"

    # Show a preview of result file
    less "$RESULT_FILE"

    # Ask for confirmation before updating files
    read -p "Do you want to commit or edit before commit the message? (y/e/n): " CONFIRM
    if [[ ! "$CONFIRM" =~ ^[YyEe]$ ]]; then
        exit 0
    fi

    git commit -m "$RESULT"
    if [[ "$CONFIRM" =~ ^[Ee]$ ]]; then
        git commit --amend
    fi
}
