#!/bin/bash
# ident    "@(#)fwcmd.bash  v1.0  Last Mod: 120918" 
#
########################################################################
# fwcmd.bash
#
# Christopher B. Lee
# for LeeVault
#
########################################################################
#
# Last Modification: 120918
# Purpose: @(#)    Script to run selected firewall-cmd commands from a  
#    @(#) menu-like interface
# Scope:    Runs firewall-cmd with either stsed options on the command line,
#        or in the absence of command line options, presents a menu of 
#        frequently used options.
#
# Modification History
#
########################################################################
#
# DECLARED VARIABLES
FWCMD="firewall-cmd --"
OPT1="state"
OPT2="reload"
OPT3="panic-on"
OPT4="panic-off"
OPT5="query-panic"
OPT6="list-all"
OPT7="add-service="
OPT8="add-port="
OPT9="help"

########################################################################

if [ -n "$1" ]  # Test whether command-line argument is present (non-empty).
then 
    $FWCMD$1
    exit 0
else
    echo ""
    echo "FIREWALL COMMANDER"
    echo ""
    echo "Select From the floowing options:"
    echo "     (1) Show STATE of the Firewall."
    echo "     (2) Reload the Firewall configuration."
    echo "     (3) Set Panic Mode ON."
    echo "     (4) Set Panic Mode OFF."
    echo "     (5) Show the Panic Mode Status."
    echo "     (6) Show the list of allowed services."
    echo "     (7) Add a Service."
    echo "     (8) Add a Port."
    echo "     (9) Print out the HELP information."
    echo "Selection (1-7)"
fi

read -s PICK
#echo "$PICK"     #for debugging
if [[ -n "$PICK" ]]
    if [[ $PICK = "1" ]]
        then
        OPTION="$OPT1"
    elif [[ $PICK = "2" ]]
        then
        OPTION="$OPT2"
    elif [[ $PICK = "3" ]]
        then
        OPTION="$OPT3"
    elif [[ $PICK = "4" ]]
        then
        OPTION="$OPT4"
    elif [[ $PICK = "5" ]]
        then
        OPTION="$OPT5"
    elif [[ $PICK = "6" ]]
        then
        OPTION="$OPT6"
    elif [[ $PICK = "7" ]]
        then
        echo "Please type in the <ServiceName> to add."
        read SERVICE
        OPTION="$OPT8$SERVICE --permanent"
        OPTION="$OPT7"
    elif [[ $PICK = "8" ]]
        then
        echo "Please type in the <PortName/Type> TCP or UDP, to add."
        read PORT
        OPTION="$OPT8$PORT --permanent"
    elif [[ $PICK = "9" ]]
        then
        OPTION="$OPT9"    
    elif [[ $PICK != {1,2,3,4,5,6,7,8,9} ]]
        then
        echo "No valid selection was made."
        exit 0
    fi
then
    $FWCMD${OPTION}
    echo "Script Complete."
    exit 0
fi

EOF