#!/bin/bash
# info: list user ips
# options: USER [FORMAT] [VERSION]
#
# The function for obtaining the list of available ip addresses.


#----------------------------------------------------------#
#                    Variable&Function                     #
#----------------------------------------------------------#

# Argument definition
user=$1
format=${2-shell}
version=${3-all}

# Includes
source $VESTA/func/main.sh

# JSON list function
json_list() {
    echo '{'
    ip_count=$(echo "$ips" |wc -l)
    i=1
    for IP in $ips; do
        source $VESTA/data/ips/$IP
        echo -n '    "'$IP'": {
        "OWNER": "'$OWNER'",
        "STATUS": "'$STATUS'",
        "NAME": "'$NAME'",
        "VERSION": "'$VERSION'",
        "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   VERSION"
    echo "--  ---  -----   ------   ----   -------"
    for IP in $ips; do
        source $VESTA/data/ips/$IP
        if [ -z "$NAT" ]; then
            NAT='no'
        fi
        if [ -z "$NAME" ]; then
            NAME='no'
        fi
        echo "$IP $NAT $OWNER $STATUS $NAME $VERSION"
    done
}

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

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


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

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


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

# Defining fileds to select
if [ "$version" == "6" ]; then
    owner='admin'
    owner_ips=$(grep -A 10 -H  "OWNER='$owner'" $VESTA/data/ips/*)
    owner_ips=$(echo "$owner_ips" |grep -A 10 "STATUS='shared'" |grep "VERSION='6'")
    owner_ips=$(echo "$owner_ips" |cut -f 7 -d / |cut -f 1 -d -)
    user_ips=$(grep -A 10 -H "OWNER='$user'" $VESTA/data/ips/* |grep "VERSION='6'")
    user_ips=$(echo "$user_ips" |cut -f 1 -d '-' |cut -f 7 -d / |awk -F ':OWNER' '{print $1}')
    ips=$(echo -e "$user_ips\n$owner_ips" |sort -u |sed "/^$/d")
    fields='$IP $OWNER $STATUS $NAME $NAT'
elif [ "$version" == "4" ]; then
    owner='admin'
    owner_ips=$(grep -A 10 -H  "OWNER='$owner'" $VESTA/data/ips/*)
    owner_ips=$(echo "$owner_ips" |grep -A 10 "STATUS='shared'" |grep "VERSION='4'")
    owner_ips=$(echo "$owner_ips" |cut -f 7 -d / |cut -f 1 -d -)
    user_ips=$(grep -A 10 -H "OWNER='$user'" $VESTA/data/ips/* |grep "VERSION='4'")
    user_ips=$(echo "$user_ips" |cut -f 1 -d '-' |cut -f 7 -d / |awk -F ':OWNER' '{print $1}')
    ips=$(echo -e "$user_ips\n$owner_ips" |sort -u |sed "/^$/d")
    fields='$IP $OWNER $STATUS $NAME $NAT'
else
    owner='admin'
    owner_ips=$(grep -A 1 -H  "OWNER='$owner'" $VESTA/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'" $VESTA/data/ips/*)
    user_ips=$(echo "$user_ips" |cut -f 1 -d '-' |cut -f 7 -d / |awk -F ':OWNER' '{print $1}')
    ips=$(echo -e "$user_ips\n$owner_ips" |sort -u |sed "/^$/d")
    fields='$IP $OWNER $STATUS $NAME $NAT'
fi

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


#----------------------------------------------------------#
#                       Vesta                              #
#----------------------------------------------------------#

exit