#!/bin/bash

__swift-section_cursor_index_in_current_word() {
    local remaining="${COMP_LINE}"

    local word
    for word in "${COMP_WORDS[@]::COMP_CWORD}"; do
        remaining="${remaining##*([[:space:]])"${word}"*([[:space:]])}"
    done

    local -ir index="$((COMP_POINT - ${#COMP_LINE} + ${#remaining}))"
    if [[ "${index}" -le 0 ]]; then
        printf 0
    else
        printf %s "${index}"
    fi
}

# positional arguments:
#
# - 1: the current (sub)command's count of positional arguments
#
# required variables:
#
# - repeating_flags: the repeating flags that the current (sub)command can accept
# - non_repeating_flags: the non-repeating flags that the current (sub)command can accept
# - repeating_options: the repeating options that the current (sub)command can accept
# - non_repeating_options: the non-repeating options that the current (sub)command can accept
# - positional_number: value ignored
# - unparsed_words: unparsed words from the current command line
#
# modified variables:
#
# - non_repeating_flags: remove flags for this (sub)command that are already on the command line
# - non_repeating_options: remove options for this (sub)command that are already on the command line
# - positional_number: set to the current positional number
# - unparsed_words: remove all flags, options, and option values for this (sub)command
__swift-section_offer_flags_options() {
    local -ir positional_count="${1}"
    positional_number=0

    local was_flag_option_terminator_seen=false
    local is_parsing_option_value=false

    local -ar unparsed_word_indices=("${!unparsed_words[@]}")
    local -i word_index
    for word_index in "${unparsed_word_indices[@]}"; do
        if "${is_parsing_option_value}"; then
            # This word is an option value:
            # Reset marker for next word iff not currently the last word
            [[ "${word_index}" -ne "${unparsed_word_indices[${#unparsed_word_indices[@]} - 1]}" ]] && is_parsing_option_value=false
            unset "unparsed_words[${word_index}]"
            # Do not process this word as a flag or an option
            continue
        fi

        local word="${unparsed_words["${word_index}"]}"
        if ! "${was_flag_option_terminator_seen}"; then
            case "${word}" in
            --)
                unset "unparsed_words[${word_index}]"
                # by itself -- is a flag/option terminator, but if it is the last word, it is the start of a completion
                if [[ "${word_index}" -ne "${unparsed_word_indices[${#unparsed_word_indices[@]} - 1]}" ]]; then
                    was_flag_option_terminator_seen=true
                fi
                continue
                ;;
            -*)
                # ${word} is a flag or an option
                # If ${word} is an option, mark that the next word to be parsed is an option value
                local option
                for option in "${repeating_options[@]}" "${non_repeating_options[@]}"; do
                    [[ "${word}" = "${option}" ]] && is_parsing_option_value=true && break
                done

                # Remove ${word} from ${non_repeating_flags} or ${non_repeating_options} so it isn't offered again
                local not_found=true
                local -i index
                for index in "${!non_repeating_flags[@]}"; do
                    if [[ "${non_repeating_flags[${index}]}" = "${word}" ]]; then
                        unset "non_repeating_flags[${index}]"
                        non_repeating_flags=("${non_repeating_flags[@]}")
                        not_found=false
                        break
                    fi
                done
                if "${not_found}"; then
                    for index in "${!non_repeating_flags[@]}"; do
                        if [[ "${non_repeating_flags[${index}]}" = "${word}" ]]; then
                            unset "non_repeating_flags[${index}]"
                            non_repeating_flags=("${non_repeating_flags[@]}")
                            break
                        fi
                    done
                fi
                unset "unparsed_words[${word_index}]"
                continue
                ;;
            esac
        fi

        # ${word} is neither a flag, nor an option, nor an option value
        if [[ "${positional_number}" -lt "${positional_count}" || "${positional_count}" -lt 0 ]]; then
            # ${word} is a positional
            ((positional_number++))
            unset "unparsed_words[${word_index}]"
        else
            if [[ -z "${word}" ]]; then
                # Could be completing a flag, option, or subcommand
                positional_number=-1
            else
                # ${word} is a subcommand or invalid, so stop processing this (sub)command
                positional_number=-2
            fi
            break
        fi
    done

    unparsed_words=("${unparsed_words[@]}")

    if\
        ! "${was_flag_option_terminator_seen}"\
        && ! "${is_parsing_option_value}"\
        && [[ ("${cur}" = -* && "${positional_number}" -ge 0) || "${positional_number}" -eq -1 ]]
    then
        COMPREPLY+=($(compgen -W "${repeating_flags[*]} ${non_repeating_flags[*]} ${repeating_options[*]} ${non_repeating_options[*]}" -- "${cur}"))
    fi
}

__swift-section_add_completions() {
    local completion
    while IFS='' read -r completion; do
        COMPREPLY+=("${completion}")
    done < <(IFS=$'\n' compgen "${@}" -- "${cur}")
}

__swift-section_custom_complete() {
    if [[ -n "${cur}" || -z ${COMP_WORDS[${COMP_CWORD}]} || "${COMP_LINE:${COMP_POINT}:1}" != ' ' ]]; then
        local -ar words=("${COMP_WORDS[@]}")
    else
        local -ar words=("${COMP_WORDS[@]::${COMP_CWORD}}" '' "${COMP_WORDS[@]:${COMP_CWORD}}")
    fi

    "${COMP_WORDS[0]}" "${@}" "${words[@]}"
}

_swift-section() {
    local state
    state="$(shopt -p;shopt -po)"
    trap "${state//$'\n'/;}" RETURN
    shopt -s extglob
    set +o history +o posix

    local -xr SAP_SHELL=bash
    local -x SAP_SHELL_VERSION
    SAP_SHELL_VERSION="$(IFS='.';printf %s "${BASH_VERSINFO[*]}")"
    local -r SAP_SHELL_VERSION

    local -r cur="${2}"
    local -r prev="${3}"

    local -i positional_number
    local -a unparsed_words=("${COMP_WORDS[@]:1:${COMP_CWORD}}")

    local -a repeating_flags=()
    local -a non_repeating_flags=(--version -h --help)
    local -a repeating_options=()
    local -a non_repeating_options=()
    __swift-section_offer_flags_options 0

    # Offer subcommand / subcommand argument completions
    local -r subcommand="${unparsed_words[0]}"
    unset 'unparsed_words[0]'
    unparsed_words=("${unparsed_words[@]}")
    case "${subcommand}" in
    dump|interface|help)
        # Offer subcommand argument completions
        "_swift-section_${subcommand}"
        ;;
    *)
        # Offer subcommand completions
        COMPREPLY+=($(compgen -W 'dump interface help' -- "${cur}"))
        ;;
    esac
}

_swift-section_dump() {
    repeating_flags=()
    non_repeating_flags=(--dyld-shared-cache --uses-system-dyld-shared-cache --enable-synthesize-sugar-on-types --disable-synthesize-sugar-on-types --enable-display-debugger-generated-module --disable-display-debugger-generated-module --enable-qualify-entities --disable-qualify-entities --enable-display-extension-contexts --disable-display-extension-contexts --enable-display-unmangled-suffix --disable-display-unmangled-suffix --enable-display-module-names --disable-display-module-names --enable-display-generic-specializations --disable-display-generic-specializations --enable-display-protocol-conformances --disable-display-protocol-conformances --enable-display-where-clauses --disable-display-where-clauses --enable-display-entity-types --disable-display-entity-types --enable-shorten-partial-apply --disable-shorten-partial-apply --enable-shorten-thunk --disable-shorten-thunk --enable-shorten-value-witness --disable-shorten-value-witness --enable-shorten-archetype --disable-shorten-archetype --enable-show-private-discriminators --disable-show-private-discriminators --enable-show-function-argument-types --disable-show-function-argument-types --enable-show-async-resume-partial --disable-show-async-resume-partial --enable-display-stdlib-module --disable-display-stdlib-module --enable-display-objc-module --disable-display-objc-module --enable-print-for-type-name --disable-print-for-type-name --enable-show-closure-signature --disable-show-closure-signature --enable-show-module-in-dependent-member-type --disable-show-module-in-dependent-member-type --emit-offset-comments --emit-member-addresses --emit-vtable-offsets --emit-expanded-field-offsets --emit-pwt-addresses --preferred-binary-order --version -h --help)
    repeating_options=(--sections -s)
    non_repeating_options=(--cache-image-path -p --cache-image-name -n --architecture -a --demangle-options --output-path -o --color-scheme -c)
    __swift-section_offer_flags_options 1

    # Offer option value completions
    case "${prev}" in
    '--cache-image-path'|'-p')
        return
        ;;
    '--cache-image-name'|'-n')
        return
        ;;
    '--architecture'|'-a')
        __swift-section_add_completions -W 'x86_64'$'\n''arm64'$'\n''arm64e'
        return
        ;;
    '--demangle-options')
        __swift-section_add_completions -W 'default'$'\n''simplified'$'\n''interface'
        return
        ;;
    '--output-path'|'-o')
        __swift-section_add_completions -f
        return
        ;;
    '--sections'|'-s')
        __swift-section_add_completions -W 'types'$'\n''protocols'$'\n''protocolConformances'$'\n''associatedTypes'
        return
        ;;
    '--color-scheme'|'-c')
        __swift-section_add_completions -W 'none'$'\n''light'$'\n''dark'
        return
        ;;
    esac

    # Offer positional completions
    case "${positional_number}" in
    1)
        __swift-section_add_completions -f
        return
        ;;
    esac
}

_swift-section_interface() {
    repeating_flags=()
    non_repeating_flags=(--dyld-shared-cache --uses-system-dyld-shared-cache --show-c-imported-types --parse-opaque-return-type --emit-offset-comments --emit-member-addresses --emit-vtable-offsets --emit-expanded-field-offsets --sort-members-by-offset --version -h --help)
    repeating_options=()
    non_repeating_options=(--cache-image-path -p --cache-image-name -n --architecture -a --output-path -o --color-scheme -c)
    __swift-section_offer_flags_options 1

    # Offer option value completions
    case "${prev}" in
    '--cache-image-path'|'-p')
        return
        ;;
    '--cache-image-name'|'-n')
        return
        ;;
    '--architecture'|'-a')
        __swift-section_add_completions -W 'x86_64'$'\n''arm64'$'\n''arm64e'
        return
        ;;
    '--output-path'|'-o')
        __swift-section_add_completions -f
        return
        ;;
    '--color-scheme'|'-c')
        __swift-section_add_completions -W 'none'$'\n''light'$'\n''dark'
        return
        ;;
    esac

    # Offer positional completions
    case "${positional_number}" in
    1)
        __swift-section_add_completions -f
        return
        ;;
    esac
}

_swift-section_help() {
    repeating_flags=()
    non_repeating_flags=(--version)
    repeating_options=()
    non_repeating_options=()
    __swift-section_offer_flags_options -1
}

complete -o filenames -F _swift-section swift-section
