# bash completion for smartctl(8)                          -*- shell-script -*-

_smartctl_quietmode()
{
    COMPREPLY=( $(compgen -W 'errorsonly silent noserial' -- "$cur") )
}
_smartctl_device()
{
    case $cur in
        areca*|3ware*|megaraid*|cciss*)
            COMPREPLY+=( ${cur%%,*},{0..31} )
            COMPREPLY=( $(compgen -W '"${COMPREPLY[@]}"' -- "$cur") )
            ;;
        hpt*)
            COMPREPLY+=( hpt,{1..4}/{1..8} hpt,{1..4}/{1..8}/{1..5} )
            COMPREPLY=( $(compgen -W '"${COMPREPLY[@]}"' -- "$cur") )
            ;;
        *)
            COMPREPLY=( $(compgen -W "ata scsi sat usbcypress usbjmicron
                usbsunplus marvell areca 3ware hpt megaraid cciss auto test" \
                    -- "$cur") )
            case "${COMPREPLY[@]}" in
                areca|3ware|hpt|megaraid|cciss)
                    compopt -o nospace
                    ;;
            esac
            ;;
    esac
}
_smartctl_tolerance()
{
    COMPREPLY=( $(compgen -W 'normal conservative permissive verypermissive' \
        -- "$cur") )
}
_smartctl_badsum()
{
    COMPREPLY=( $(compgen -W 'warn exit ignore' -- "$cur") )
}
_smartctl_report()
{
    COMPREPLY=( $(compgen -W 'ioctl ataioctl scsiioctl' -- "$cur") )
}
_smartctl_powermode()
{
    COMPREPLY=( $(compgen -W 'never sleep standby idle' -- "$cur") )
}
_smartctl_feature()
{
    COMPREPLY=( $(compgen -W 'on off' -- "$cur") )
}
_smartctl_log()
{
    COMPREPLY=( $(compgen -W 'error selftest selective directory background
        sasphy sasphy,reset sataphy sataphy,reset scttemp scttempsts
        scttemphist scterc gplog smartlog xerror xselftest' -- "$cur") )
}
_smartctl_vendorattribute()
{
    COMPREPLY=( $(compgen -W 'help 9,minutes 9,seconds 9,halfminutes 9,temp
        192,emergencyretractcyclect 193,loadunload 194,10xCelsius 194,unknown
        198,offlinescanuncsectorct 200,writeerrorcount 201,detectedtacount
        220,temp' -- "$cur") )
}
_smartctl_firmwarebug()
{
    COMPREPLY=( $(compgen -W 'none samsung samsung2 samsung3 swapid' \
        -- "$cur") )
}
_smartctl_presets()
{
    COMPREPLY=( $(compgen -W 'use ignore show showall' -- "$cur") )
}
_smartctl_test()
{
    [[ $cur == @(pending|scttempint|vendor), ]] && return
    COMPREPLY=( $(compgen -W 'offline short long conveyance select,
        select,redo select,next afterselect,on afterselect,off pending,
        scttempint, vendor,' -- "$cur") )
    [[ $COMPREPLY == *, ]] && compopt -o nospace
}
_smartctl_drivedb()
{
    local prefix=
    if [[ $cur == +* ]]; then
        prefix=+
        cur="${cur#+}"
    fi
    _filedir h
    [[ -n $prefix ]] && COMPREPLY=( "${COMPREPLY[@]/#/$prefix}" )
}

_smartctl()
{
    local cur prev words cword split
    _init_completion -s || return

    case $prev in
        --quietmode|-!(-*)q)
            _smartctl_quietmode
            ;;
        --device|-!(-*)d)
            _smartctl_device
            return
            ;;
        --tolerance|-!(-*)T)
            _smartctl_tolerance
            return
            ;;
        --badsum|-!(-*)b)
            _smartctl_badsum
            return
            ;;
        --report|-!(-*)r)
            _smartctl_report
            return
            ;;
        --nocheck|-!(-*)n)
            _smartctl_powermode
            return
            ;;
        --smart|--offlineauto|--saveauto|-!(-*)[soS])
            _smartctl_feature
            return
            ;;
        --log|-!(-*)l)
            _smartctl_log
            return
            ;;
        --vendorattribute|-!(-*)v)
            _smartctl_vendorattribute
            return
            ;;
        --firmwarebug|-!(-*)F)
            _smartctl_firmwarebug
            return
            ;;
        --presets|-!(-*)P)
            _smartctl_presets
            return
            ;;
        --drivedb|-!(-*)B)
            _smartctl_drivedb
            return
            ;;
        --test|-!(-*)t)
            _smartctl_test
            return
            ;;
    esac

    $split && return

    if [[ "$cur" == -* ]]; then
        COMPREPLY=( $(compgen -W '$(_parse_help "$1")' -- "$cur") )
        [[ $COMPREPLY == *= ]] && compopt -o nospace
    else
        cur=${cur:=/dev/}
        _filedir
    fi
} &&
complete -F _smartctl smartctl

# ex: filetype=sh