# shellcheck shell=bash
# ==========================================================================
#     _   _ _ ____            ____          _____
#    | | | (_)  _ \ ___ _ __ / ___|___  _ _|_   _| __ __ _  ___ ___ _ __
#    | |_| | | |_) / _ \ '__| |   / _ \| '_ \| || '__/ _` |/ __/ _ \ '__|
#    |  _  | |  __/  __/ |  | |__| (_) | | | | || | | (_| | (_|  __/ |
#    |_| |_|_|_|   \___|_|   \____\___/|_| |_|_||_|  \__,_|\___\___|_|
#
#       ---  High-Performance Connectivity Tracer (HiPerConTracer)  ---
#                 https://www.nntb.no/~dreibh/hipercontracer/
# ==========================================================================
#
# High-Performance Connectivity Tracer (HiPerConTracer)
# Copyright (C) 2015-2026 by Thomas Dreibholz
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
# Contact: dreibh@simula.no


# ###### Bash completion for hpct-query #####################################
_hpct_query()
{
   # Based on: https://www.benningtons.net/index.php/bash-completion/
   local cur prev words cword
   if type -t _comp_initialize >/dev/null; then
      _comp_initialize || return
   elif type -t _init_completion >/dev/null; then
      _init_completion || return
   else
      # Manual initialization for older bash completion versions:
      COMPREPLY=()
      # shellcheck disable=SC2034
      cur="${COMP_WORDS[COMP_CWORD]}"
      # shellcheck disable=SC2034
      prev="${COMP_WORDS[COMP_CWORD-1]}"
      # shellcheck disable=SC2034,SC2124
      words="${COMP_WORDS[@]}"
      # shellcheck disable=SC2034
      cword="${COMP_CWORD}"
   fi

   # ====== Database configuration file =====================================
   if [ "${cword}" -eq 1 ] ; then
      _filedir '@(conf)'
      return

   # ====== Measurement type ================================================
   elif [ "${cword}" -eq 2 ] ; then
      # shellcheck disable=SC2207
      COMPREPLY=( $(compgen -W "ping traceroute" --  "${cur}") )
      return

   # ====== Options =========================================================
   else
      case "${prev}" in
         #  ====== Generic value ============================================
         -L | --loglevel       | \
         -T | --table          | \
         -V | --table-version  | \
         --from-time           | \
         --to-time             | \
         --from-measurement-id | \
         --to-measurement-id)
            return
            ;;
         # ====== Special case: file ========================================
         -o | --output)
            # HiPerConTracer results file names:
            _filedir '@(hpct\.xz|hpct\.bz2|hpct\.gz|hpct|results\.xz|results\.bz2|results\.gz|results)'
            return
            ;;
         # ====== Special case: file ========================================
         -M | --address-to-measurementid)
            # Arbitrary file names:
            _filedir
            return
            ;;
         # ====== Special case: log file ====================================
         -O | --logfile)
            # Files with extension .log:
            _filedir '@(log)'
            return
            ;;
         # ====== Special case: on/off ======================================
         -Z | --logcolor)
            # shellcheck disable=SC2207
            COMPREPLY=( $(compgen -W "on off" --  "${cur}") )
            return
            ;;
      esac
   fi

   # ====== All options =====================================================
   local opts="
-L
--loglevel
-Z
--logcolor
-O
--logfile
-q
--quiet
-v
--verbose
-o
--output
-T
--table
-V
--table-version
-M
--address-to-measurementid
--deduplication
--from-time
--to-time
--from-measurement-id
--to-measurement-id
-h
--help
--version
"
   # shellcheck disable=SC2207
   COMPREPLY=( $( compgen -W "${opts}" -- "${cur}" ) )
   return 0
}

complete -F _hpct_query hpct-query
