#compdef ufw
# ------------------------------------------------------------------------------
# Copyright (c) 2016 Github zsh-users - https://github.com/zsh-users
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#     * Redistributions of source code must retain the above copyright
#       notice, this list of conditions and the following disclaimer.
#     * Redistributions in binary form must reproduce the above copyright
#       notice, this list of conditions and the following disclaimer in the
#       documentation and/or other materials provided with the distribution.
#     * Neither the name of the zsh-users nor the
#       names of its contributors may be used to endorse or promote products
#       derived from this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
# DISCLAIMED. IN NO EVENT SHALL ZSH-USERS BE LIABLE FOR ANY
# DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
# ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
# ------------------------------------------------------------------------------
# Description
# -----------
#
#  Completion script for The Uncomplicated Firewall (ufw) v0.36.2. (https://launchpad.net/ufw).
#
# ------------------------------------------------------------------------------
# Authors
# -------
#
#  * Jindřich Pilař (https://github.com/JindrichPilar)
#
# ------------------------------------------------------------------------------

_ufw_logging() {
  local params additional second
  second=$words[2]

  if [[ ! -z "$second" ]]; then
    return
  fi

  params=("on" "off")
  additional=("low" "medium" "high" "full")

  _describe -t params 'on/off' params
  _describe -t additional 'level' additional
}

_ufw_delete() {
  local rules complrules second

  second=$words[2]

  if [[ ! -z "$second" ]]; then
    return
  fi

  complrules=()
  rules=("${(f)$(ufw status | tr -s ' ' | tail -n +5 | tr -s '\n')}")

  for ((i=1; i<=${#rules[@]}; i++)); do
    complrules+=("$i:$rules[i]");
  done

  _describe -t complrules 'Rules' complrules
}

_ufw_app() {
  local ret=1

  local -a subcmds=(
    "list:list application profiles"
    "info:show information on profile"
    "update:update profile"
    "default:set default application profile"
  )

  _arguments \
    "1: :{_describe 'command' subcmds}" \
    && ret=0

  return ret
}

_ufw() {
  local curcontext="$curcontext" ret=1
  local -a state line commands

  commands=(
    "enable:enable the firewall"
    "disable:disable the firewall"
    "default:set default policy"
    "logging:set logging level"
    "allow:add allow rule"
    "deny:add deny rule"
    "reject:add reject rule"
    "limit:add limit rule"
    "delete:delete rule"
    "insert:insert rule at position"
    "route:add route rule"
    "reload:reload firewall"
    "reset:reset firewall"
    "status:show firewall status"
    "show:show firewall report"
    "version:display version information"
    "prepend:add rule before all of the same type"
    "app:application profile command"
  )

  _arguments -C -s -S -n \
    '(- 1 *)'--version"[display version information]: :->full" \
    '(- 1 *)'{-h,--help}'[display usage information]: :->full' \
    '(- 1 *)'--dry-run"[don't modify anything, just show the changes]: :->cmds" \
    '1:cmd:->cmds' \
    '*:: :->args' && ret=0

  case "$state" in
    (cmds)
      _describe -t commands 'commands' commands
      ;;
    (args)
      local cmd
      cmd=$words[1]
      case "$cmd" in
        (logging)
          _ufw_logging && ret=0
          ;;
        (delete)
          _ufw_delete && ret=0
          ;;
        (app)
          _ufw_app && ret=0
          ;;
        (*)
          _default && ret=0
          ;;
      esac
      ;;
    (*)
      ;;
  esac

  return ret
}

_ufw "$@"

# Local Variables:
# mode: Shell-Script
# sh-indentation: 2
# indent-tabs-mode: nil
# sh-basic-offset: 2
# End:
# vim: ft=zsh sw=2 ts=2 et
