# bash completion for tlp, tlp-stat, bluetooth, nfc, wwan, wifi
# Copyright (c) 2025 Thomas Koch <linrunner at gmx.net> and others.
# SPDX-License-Identifier: GPL-2.0-or-later

_batteries() {
    # show list of batteries
    local bats b

    bats=$( {
        for b in /sys/class/power_supply/*; do
            if echo "$b" | grep -E -v -q "hid" \
                    && [ "$(cat $b/present 2> /dev/null)" = "1" ] \
                    && [ "$(cat $b/type)" = "Battery" ]; then
                echo "${b##/*/} "
            fi
        done } )

    if [ -n "$bats" ]; then
        COMPREPLY=( $(compgen -W "${bats}" -- ${cur}) )
    fi
}

_target_level() {
    local thresh

    if thresh="$(cat /sys/class/power_supply/${COMP_WORDS[2]}/charge_control_end_threshold 2> /dev/null)" \
        || thresh="$(cat /sys/devices/platform/smapi/${COMP_WORDS[2]}/stop_charge_thresh 2> /dev/null)"; then
        COMPREPLY=( $(compgen -W "$thresh" -- ${cur}) )
    fi
}

_tlp() {
    local cur prev words cword opts bats
    _init_completion || return

    opts="start performance ac balanced bat power-saver usb setcharge fullcharge chargeonce discharge recalibrate bayoff diskid --version"

    case $cword in
        1)
            # subcmds only
            COMPREPLY=( $(compgen -W "${opts}" -- ${cur}) )
            return 0
            ;;

        2)
            case "${prev}" in
                fullcharge|chargeonce|discharge|recalibrate)
                    _batteries
                    return 0
                    ;;
            esac
            ;;

        3)
            if [ "${COMP_WORDS[1]}" = "discharge" ]; then
                _target_level
                return 0
            fi
            ;;

        4)
            if [ "${COMP_WORDS[1]}" = "setcharge" ]; then
                _batteries
                return 0
            fi
            ;;
    esac
} &&
    complete -F _tlp tlp

_tlp_rf() {
    local cur prev words cword opts
    _init_completion || return

    opts="on off toggle cycle --version"

    if [ $cword -eq 1 ]; then
        COMPREPLY=( $(compgen -W "${opts}" -- ${cur}) )
        return 0
    fi
} &&
    complete -F _tlp_rf bluetooth nfc wwan wifi

_tlp_stat() {
    local cur prev words cword opts
    _init_completion || return

    opts="--battery --cdiff --config --disk --graphics --mode --pcie
        --pd-diag --pev --processor --psup --quiet --rfkill --system
        --temp --trace --trace-nm --udev --usb --verbose --warn"

    if [ $cword -eq 1 ]; then
        COMPREPLY=( $(compgen -W "${opts}" -- ${cur}) )
        return 0
    fi
} &&
    complete -F _tlp_stat tlp-stat

complete -F _command run-on-ac run-on-bat
