#!/usr/bin/env bash
set -e

help-table() {
    local -a rows
    local max_cmd=7 max_opt=6 max_desc=11

    rows+=("Command|Option|Description")
    rows+=("===Task===| | ")
    rows+=("clone-taskfile||Clone the taskfile.build repo.")
    rows+=("init-project||Setup a taskfile project in the current directory.")
    rows+=("init-dotenv-dir||Setup task dotenv dir.")
    rows+=("template-dotenv||Generate .env from .env.template.")
    rows+=("open-with-os||Open file or url with default application.")
    rows+=("open-url-with-delay|[url][delay]|Open url after n seconds.")
    rows+=("show-dotenv-setup||Show info about the dotenv setup.")
    rows+=("setup-completion|[shell]|Show shell completion instructions.")
    rows+=("list-taskfile-aliases||List all library command aliases.")
    rows+=("===Files===| | ")
    rows+=("ensure-directory|[path]|Ensure directory path exists.")
    rows+=("create-symlink|[target][source]|Remove existing file/directory and create symlink.")
    rows+=("create-symlink-files|[target][source]|Create symlink for files in source pattern.")
    rows+=("===Python===| | ")
    rows+=("activate-venv||Activate or initialize Python virtual env.")
    rows+=("remove-venv||Remove the Python virtual env.")
    rows+=("===Environment===| | ")
    rows+=("create-dotenv|[env][type]|Create .env file of type nextcloud, odoo, upgrade.")
    rows+=("list-dotenv||List dotenv files.")
    rows+=("show-dotenv|[env]|Output content of the .env file.")
    rows+=("copy-dotenv|[env][env]|Copy .env file.")
    rows+=("edit-dotenv|[env]|Open .env file in default editor.")
    rows+=("load-dotenv|[env]|Load and export .env file.")
    rows+=("rename-dotenv|[env][env]|Rename .env file.")
    rows+=("remove-dotenv|[env]|Remove .env file.")
    rows+=("backup-dotenv-files|[path]|Archive and copy .env files to target location.")
    rows+=("restore-dotenv-files|[path]|Extract and copy .env files from backup file.")
    rows+=("===LLM===| | ")
    rows+=("commit-with-llm||Commit with llm generated commit message.")
    rows+=("update-with-llm|[glob][prompt]|Feed files matching glob to LLM and apply changes.")
    rows+=("update-changelog-with-llm||Update CHANGELOG.md with changes from staged git diff using LLM.")
    rows+=("create-prompt||Create a new prompt file from template.")
    rows+=("list-prompts|[path]|List all prompt files.")
    rows+=("===Logging===| | ")
    rows+=("log-error||Colored output for errors.")
    rows+=("log-success||Colored output for success.")
    rows+=("log-warning||Colored output for warnings.")
    rows+=("===Git===| | ")
    rows+=("add-git-folder|[url][branch][path]|Add git folder to .gitmodules.")
    rows+=("add-git-merge|[remote][branch][path]|Add git merge ref to gitmodules.")
    rows+=("add-git-remote|[remote][url][path]|Add git remote to git folder.")
    rows+=("aggregate-git-folders|[path]|Run gitaggregate for the repos.yaml.")
    rows+=("commit-git-folder|[message][path]|Commit all changes in git folders in path.")
    rows+=("clean-git-folder|[path]|Clean git folder.")
    rows+=("checkout-git-folder|[branch][path]|Checkout git commit from .gitmodules files.")
    rows+=("clone-git-folder|[path]|Clone git folder from .gitmodules file.")
    rows+=("download-git-folder|[path]|Download git folder from .gitmodules file.")
    rows+=("find-git-folder|[path]|Find all .git folders in path.")
    rows+=("generate-ssh-key||Generate ssh key pair.")
    rows+=("list-git-folder|[path]|List path and url of git folders from .gitmodules file.")
    rows+=("ls-git-folder|[path]|Space separated list of git folders in path.")
    rows+=("merge-git-remote|[path]|Merge all remotes and refs defined in .gitmodules.")
    rows+=("pull-git-folder|[branch][path]|Pull all git folders from .gitmodules file.")
    rows+=("switch-git-folder|[branch][path]|Switch branch for all git folders from .gitmodules file.")
    rows+=("sync-git-folder|[branch][path]|Switch, stash and pull all git folders from .gitmodules file.")
    rows+=("remove-git-folder|[path]|Remove git folders in path from .gitmodules file.")
    rows+=("push-git-folder|[path]|Push all git folders in path.")
    rows+=("reset-git-folder|[path]|Abort, rebase and reset git folder.")
    rows+=("run-pre-commit|[path]|Run pre-commit in path.")
    rows+=("save-git-folder||Save commit refs to .gitmodules.")
    rows+=("stage-git-folder|[path]|Stage all files in git folders in path.")
    rows+=("status-git-folder|[path]|Show status of git folders in path.")
    rows+=("update-submodule||Update git submodules.")
    rows+=("===Pass===| | ")
    rows+=("save-dotenv-to-pass|[file]|Store content of .env in pass entry.")
    rows+=("load-dotenv-from-pass|[file]|Restore content to .env from pass entry.")
    rows+=("===KeePass===| | ")
    rows+=("save-pass-to-keepass|[pass][entry]|Store password to KeePass entry.")
    rows+=("get-pass-from-keepass|[entry]|Get password from KeePass entry.")
    rows+=("save-dotenv-to-keepass|[file]|Store content of .env in KeePass entry.")
    rows+=("load-dotenv-from-keepass|[file]|Restore content to .env from KeePass entry.")
    rows+=("===Kubernetes===| | ")
    rows+=("save-kubeconfig-to-pass|[ctx]|Save kubeconfig file in pass.")
    rows+=("save-kubeconfig-to-keepass|[ctx]|Save kubeconfig file in KeePass.")
    rows+=("load-kubeconfig-from-pass|[ctx]|Load kubeconfig file from pass.")
    rows+=("load-kubeconfig-from-keepass|[ctx]|Load kubeconfig file from KeePass.")
    rows+=("===Docker===| | ")
    rows+=("login-docker|[user][token]|Run Docker login and create docker config.")
    rows+=("build-image|[output]|Build Docker image with buildx.")
    rows+=("===Ansible===| | ")
    rows+=("install-ansible-build-scripts|[role]|Install scripts of the specified Ansbile role.")

    for row in "${rows[@]}"; do
        IFS='|' read -r cmd opt desc <<< "$row"
        (( ${#cmd} > max_cmd )) && max_cmd=${#cmd}
        (( ${#opt} > max_opt )) && max_opt=${#opt}
        (( ${#desc} > max_desc )) && max_desc=${#desc}
    done

    printf '| %-*s | %-*s | %-*s |\n' \
        "$max_cmd" "Command" \
        "$max_opt" "Option" \
        "$max_desc" "Description"

    printf '|-%*s-|-%*s-|-%*s-|\n' \
        "$max_cmd" "$(printf '%*s' "$max_cmd" '' | tr ' ' '-')" \
        "$max_opt" "$(printf '%*s' "$max_opt" '' | tr ' ' '-')" \
        "$max_desc" "$(printf '%*s' "$max_desc" '' | tr ' ' '-')"

    local first_row=true
    for row in "${rows[@]:1}"; do
        IFS='|' read -r cmd opt desc <<< "$row"
        if [[ "$cmd" =~ ^===.*===$ ]]; then
            group_name="${cmd/===/}"
            group_name="${group_name/===/}"
            
            if [ "$first_row" = true ]; then
                first_row=false
            else
                printf '|-%*s-|-%*s-|-%*s-|\n' \
                    "$max_cmd" "$(printf '%*s' "$max_cmd" '' | tr ' ' '-')" \
                    "$max_opt" "$(printf '%*s' "$max_opt" '' | tr ' ' '-')" \
                    "$max_desc" "$(printf '%*s' "$max_desc" '' | tr ' ' '-')"
            fi
            
            local left_pad=$(( (max_cmd - ${#group_name}) / 2 ))
            local right_pad=$(( max_cmd - left_pad - ${#group_name} ))
            printf '| %*s%s%*s | %*s | %*s |\n' \
                "$left_pad" "" \
                "$group_name" \
                "$right_pad" "" \
                "$max_opt" "" \
                "$max_desc" ""
            
            printf '|-%*s-|-%*s-|-%*s-|\n' \
                "$max_cmd" "$(printf '%*s' "$max_cmd" '' | tr ' ' '-')" \
                "$max_opt" "$(printf '%*s' "$max_opt" '' | tr ' ' '-')" \
                "$max_desc" "$(printf '%*s' "$max_desc" '' | tr ' ' '-')"
        else
            first_row=false
            printf '| %-*s | %-*s | %-*s |\n' \
                "$max_cmd" "$cmd" \
                "$max_opt" "$opt" \
                "$max_desc" "$desc"
        fi
    done
}

help() {
    local grep_pattern
    echo
    if [[ -n "$1" ]]; then
        grep_pattern="$1"
        help-table | grep -i "$grep_pattern" | column -t -s'|'
    else
        echo "task <command> [options]"
        echo
        echo "commands:"
        echo
        help-table
    fi
    echo
}

if [[ -d "$HOME/taskfile.build/bin" ]]; then
    for file in "$HOME/taskfile.build/bin/"*; do
        if [[ -f "$file" ]]; then
            # shellcheck disable=SC1090
            source "$file"
        fi
    done
fi

if [[ "$1" == *","* ]]; then
    # Split comma-separated commands
    IFS=',' read -ra COMMANDS === "$1"
    for cmd in "${COMMANDS[@]}"; do
        # Trim whitespace
        cmd=$(echo "$cmd" | xargs)
        if declare -f "$cmd" > /dev/null; then
            "$cmd"
        else
            echo "Unknown command: $cmd"
            help "$@"
            exit 1
        fi
    done
elif declare -f "$1" > /dev/null; then
    "$1" "${@:2}"
else
    case "$1" in
        *)
            echo "Unknown command: $1"
            exit 1
            ;;
    esac
fi