# bash completion for strace                               -*- shell-script -*-

proc _strace
{
    var cur = '', prev = '', words = '', cword = ''
    _init_completion -n = || return

    # check if we're still completing strace
    var offset = '0', i = ''
    for (( i=1; i <= cword; i++ )); do
        case ${words[$i]} in
            -o|-e|-p)
                i=$((i+1))
                continue
                ;;
            -*)
                continue
                ;;
        esac
        offset=$i
        break
    done

    if [[ $offset -gt 0 ]] {
        _command_offset $offset
    } else {

        match $prev {
            with -e
                if [[ "$cur" == *=* ]] {
                    set prev = $(cur/=*/)
                    set cur = $(cur/*=/)

                    match $prev {
                        with trace
                            # Import arch-specific syscalls
                            #+ -- not foolproof IMHO --David Paleino
                            var define = '', syscall = '', rest = ''
                            var -A syscalls = ''
                            while read -r define syscall rest {
                                [[ $define == "#define" && \
                                    $syscall =~ ^__NR_(.+) ]] && \
                                    syscalls[$(BASH_REMATCH[1])]=1
                            } 2>/dev/null < /usr/include/asm/unistd.h
                            if [[ ! $syscalls ]] {
                                var unistd = '', arch = $[ command uname -m]
                                if [[ "$arch" == *86 ]] {
                                    set unistd = '/usr/include/asm/unistd_32.h'
                                } else {
                                    set unistd = '/usr/include/asm/unistd_64.h'
                                }
                                while read -r define syscall rest {
                                    [[ $define == "#define" && \
                                        $syscall =~ ^__NR_(.+) ]] && \
                                        syscalls[$(BASH_REMATCH[1])]=1
                                } 2>/dev/null < $unistd
                            }

                            setglobal COMPREPLY = '( '$( compgen -W '${!syscalls[@]} file
                                process network signal ipc desc all none' \
                                    -- "$cur" ) )
                            return 0
                            
                    }
                } else {
                    compopt -o nospace
                    setglobal COMPREPLY = '( '$( compgen -S"=" -W 'trace abbrev verbose raw
                        signal read write' -- "$cur" ) )
                }
                return 0
                
            with -o
                _filedir
                return 0
                
            with -p
                _pids
                return 0
                
            with -S
                setglobal COMPREPLY = '( '$( compgen -W 'time calls name nothing' \
                    -- "$cur" ) )
                return 0
                
            with -u
                _allowed_users
                return 0
                
        }

        if [[ "$cur" == -* ]] {
            setglobal COMPREPLY = '( '$( compgen -W '-c -d -f -ff -F --help -i -q -r -t -tt
                -ttt -T -v -V -x -xx -a -e -o -O -p -s -S -u -E' -- "$cur" ) )
        } else {
            setglobal COMPREPLY = '( '$( compgen -c -- "$cur" ) )
        }
    }
} &&
complete -F _strace -o default strace

# ex: ts=4 sw=4 et filetype=sh