wcp/bin/v_list_user_ips at 558f4b719fffbc8914440287c796db4301110e9d · ghzhost/wcp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
#!/bin/bash
# info: list user ips
# options user [format]
#
# The function for obtainig the list of available ip addresses.


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

# Argument defenition
user=$1
format=${2-shell}

# Importing variables
source $VESTA/conf/vars.conf
source $V_FUNC/shared.func

# Json function
json_list_user_ips() {
    # Print top bracket
    echo '{'

    owned_ips=$(grep -l "OWNER='$user'" $V_IPS/*)
    shared_ips=$(grep -A5 "OWNER='vesta'" $V_IPS/* |\
                    grep "STATUS='shared'"|\
                    cut -f 1 -d - )

    ip_list="$owned_ips $shared_ips"
    fileds_count=$(echo "$fields" | wc -w)

    # Starting main loop
    for IP in $ip_list; do
        IP=$(basename $IP)
        ip_data=$(cat $V_IPS/$IP)

        # Assing key=value
        eval $ip_data

        # Closing bracket if there already was output
        if [ -n "$data" ]; then
            echo -e '        },'
        fi
        i=1
        for field in $fields; do
            eval value=$field

            if [ $i -eq 1 ]; then
                # Printing parrent
                (( ++i))
                echo -e "\t\"$value\": {"
            else
                # Printing child
                if [ $i -lt $fileds_count ]; then
                    (( ++i))
                    echo -e "\t\t\"${field//$/}\": \"${value//,/, }\","
                else
                    echo -e "\t\t\"${field//$/}\": \"${value//,/, }\""
                    data=1
                fi
            fi
        done
    done

    # Closing bracket if there was output
    if [ -n "$data" ]; then
        echo -e '        }'
    fi

    # Printing bottom bracket
    echo -e '}'
}

# Shell function
shell_list_user_ips() {
    owned_ips=$(grep -l "OWNER='$user'" $V_IPS/*)
    shared_ips=$(grep -A5 "OWNER='vesta'" $V_IPS/* |\
                    grep "STATUS='shared'"|\
                    cut -f 1 -d - )

    ip_list="$owned_ips $shared_ips"
    if [ -z "$nohead" ]; then
        # Print brief info
        echo "${fields//$/}"
        for a in $fields; do
            echo -e "--------- \c"
        done
        echo
    fi

    # Starting main loop
    for IP in $ip_list; do
        IP=$(basename $IP)
        ip_data=$(cat $V_IPS/$IP)

        # Assign key/value config
        eval $ip_data

        # Print result line
        eval echo "$fields"
    done
}


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

# Checking arg number
check_args '1' "$#" 'user [format]'

# Checking argument format
format_validation 'user'

# Checking user
is_user_valid


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

# Defining fileds to select
fields='$IP $OWNER $STATUS $NAME'

# Listing ips
case $format in
    json)   json_list_user_ips ;;
    plain)  nohead=1; shell_list_user_ips ;;
    shell)  shell_list_user_ips | column -t ;;
    *)      check_args '1' '0' 'user [format]' ;;
esac


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

exit