#!/bin/bash
# info: list user nameservers
# options: USER [FORMAT]
#
# example: v-list-user-ns admin
#
# Function for obtaining the list of user's DNS servers.

#----------------------------------------------------------#
#                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() {
	i=1
	objects=$(echo -e "${ns//,/\\n}" | wc -l)
	echo '['
	for nameserver in ${ns//,/ }; do
		if [ "$i" -ne "$objects" ]; then
			echo -e "\t\"$nameserver\","
		else
			echo -e "\t\"$nameserver\""
		fi
		((++i))
	done
	echo ']'
}

# SHELL list function
shell_list() {
	echo "NS"
	echo "--"
	for nameserver in ${ns//,/ }; do
		echo "$nameserver"
	done
}

# PLAIN list function
plain_list() {
	for nameserver in ${ns//,/ }; do
		echo "$nameserver"
	done
}

# CSV list function
csv_list() {
	echo "NAMESERVER"
	for nameserver in ${ns//,/ }; do
		echo "$nameserver"
	done
}

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

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

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

# Parsing name servers
ns=$(grep "^NS='" "$USER_DATA/user.conf" | cut -f 2 -d \')

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

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

exit