#!/bin/bash
# info: list iptables rules
# options: [FORMAT]
#
# example: v-list-firewall json
#
# This function of obtaining the list of all iptables rules.

#----------------------------------------------------------#
#                Variables & Functions                     #
#----------------------------------------------------------#

# Argument definition
format=${1-shell}

# Includes
# shellcheck source=/etc/hestiacp/hestia.conf
source /etc/hestiacp/hestia.conf
# shellcheck source=/usr/local/hestia/func/main.sh
source $HESTIA/func/main.sh
# load config file
source_conf "$HESTIA/conf/hestia.conf"

# JSON list function
json_list() {
	IFS=$'\n'
	i=1
	objects=$(grep RULE $HESTIA/data/firewall/rules.conf | wc -l)
	echo "{"
	while read str; do
		[[ -z "$str" ]] && continue
		parse_object_kv_list "$str"
		echo -n '    "'$RULE'": {
        "ACTION": "'$ACTION'",
        "PROTOCOL": "'$PROTOCOL'",
        "PORT": "'$PORT'",
        "IP": "'$IP'",
        "COMMENT": "'$COMMENT'",
        "SUSPENDED": "'$SUSPENDED'",
        "TIME": "'$TIME'",
        "DATE": "'$DATE'"
    }'
		if [ "$i" -lt "$objects" ]; then
			echo ','
		else
			echo
		fi
		((i++))
	done < <(cat $HESTIA/data/firewall/rules.conf)
	echo '}'
}

# SHELL list function
shell_list() {
	IFS=$'\n'
	echo "RULE^ACTION^PROTO^PORT^IP^SPND^DATE"
	echo "----^------^-----^----^--^----^----"
	while read str; do
		[[ -z "$str" ]] && continue
		parse_object_kv_list "$str"
		echo "$RULE^$ACTION^$PROTOCOL^$PORT^$IP^$SUSPENDED^$DATE"
	done < <(cat $HESTIA/data/firewall/rules.conf)
}

# PLAIN list function
plain_list() {
	IFS=$'\n'
	while read str; do
		[[ -z "$str" ]] && continue
		parse_object_kv_list "$str"
		echo -ne "$RULE\t$ACTION\t$PROTOCOL\t$PORT\t$IP\t$COMMENT\t"
		echo -e "$SUSPENDED\t$TIME\t$DATE"
	done < <(cat $HESTIA/data/firewall/rules.conf)
}

# CSV list function
csv_list() {
	IFS=$'\n'
	echo "RULE,ACTION,PROTOCOL,PORT,IP,COMMENT,SUSPENDED,TIME,DATE"
	while read str; do
		[[ -z "$str" ]] && continue
		parse_object_kv_list "$str"
		echo -n "$RULE,$ACTION,$PROTOCOL,$PORT,$IP,\"$COMMENT\","
		echo "$SUSPENDED,$TIME,$DATE"
	done < <(cat $HESTIA/data/firewall/rules.conf)
}

#----------------------------------------------------------#
#                       Action                             #
#----------------------------------------------------------#

# Listing data
case $format in
	json) json_list ;;
	plain) plain_list ;;
	csv) csv_list ;;
	shell) shell_list | column -t -s '^' ;;
esac

#----------------------------------------------------------#
#                       Hestia                             #
#----------------------------------------------------------#

exit