#!/bin/bash
# info: list user IPs
# options: USER [FORMAT]
#
# example: v-list-user-ips admin
#
# This function for obtaining the list of available IP addresses.

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

# Argument definition
user="$1"
format="${2-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() {
	echo '{'
	ip_count="$(echo "$ips" | wc -l)"
	i=1
	for IP in $ips; do
		source_conf "$HESTIA/data/ips/$IP"
		echo -n '    "'$IP'": {
        "OWNER": "'$OWNER'",
        "STATUS": "'$STATUS'",
        "NAME": "'$NAME'",
        "NAT": "'$NAT'"
        }'
		if [ "$i" -lt "$ip_count" ]; then
			echo ','
		else
			echo
		fi
		((i++))
	done
	echo '}'
}

# SHELL list function
shell_list() {
	echo "IP  NAT  OWNER   STATUS   NAME"
	echo "--  ---  -----   ------   ---"
	for IP in $ips; do
		source_conf "$HESTIA/data/ips/$IP"
		if [ -z "$NAT" ]; then
			NAT='no'
		fi
		if [ -z "$NAME" ]; then
			NAME='no'
		fi
		echo "$IP $NAT $OWNER $STATUS $NAME"
	done
}

# PLAIN list function
plain_list() {
	for IP in $ips; do
		source_conf "$HESTIA/data/ips/$IP"
		echo -e "$IP\t$OWNER\t$STATUS\t$NAME\t$NAT"
	done
}

# CSV list function
csv_list() {
	echo "IP,OWNER,STATUS,NAME,NAT"
	for IP in $ips; do
		source_conf "$HESTIA/data/ips/$IP"
		echo "$IP,$OWNER,$STATUS,$NAME,$NAT"
	done
}

#----------------------------------------------------------#
#                    Verifications                         #
#----------------------------------------------------------#

check_args '1' "$#" 'USER [FORMAT]'
is_format_valid 'user'
is_object_valid 'user' 'USER' "$user"

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

# Defining fileds to select
owner='admin'
owner_ips="$(grep -A 1 -H "OWNER='$owner'" $HESTIA/data/ips/*)"
owner_ips="$(echo "$owner_ips" | grep "STATUS='shared'")"
owner_ips="$(echo "$owner_ips" | cut -f 7 -d / | cut -f 1 -d -)"
user_ips="$(grep -H "OWNER='$user'" $HESTIA/data/ips/*)"
user_ips="$(echo "$user_ips" | cut -f 7 -d / | cut -f 1 -d :)"
ips="$(echo -e "$user_ips\n$owner_ips" | sort -u | sed "/^$/d")"
fields='$IP $OWNER $STATUS $NAME $NAT'

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

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

exit