#!/bin/bash

# Include this script in your shell PATH to enable argument completion for IREE
# tools.

_iree_tools_autocomplete() {
    local tool="$1"
    local commands
    local cur

    # Get the list of commands from the appropriate tool's `--help-list`
    commands=$($tool --help-list | awk '{print $1}' | grep -v '^ *=' | sed 's/[=<].*//')

    cur=${COMP_WORDS[COMP_CWORD]}

    case "$cur" in
      -*)
        COMPREPLY=($(compgen -W "$commands" -- "${cur}"))
        ;;
      *)
        COMPREPLY=($(compgen -f -- "${cur}"))
        ;;
    esac

    return 0
}

complete -F _iree_tools_autocomplete iree-compile
complete -F _iree_tools_autocomplete iree-link
complete -F _iree_tools_autocomplete iree-opt