wcp/bin/v-add-remote-dns-domain at 74c376178564af6db5ec747ec66e2f74af8457aa · 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
#!/bin/bash
# info: add remote dns domain
# options: USER DOMAIN
#
# The function synchronize dns domain with the remote server.


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

# Argument defenition
user=$1
domain=$2

# Includes
source $VESTA/conf/vesta.conf
source $VESTA/func/main.sh
source $VESTA/func/remote.sh


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

check_args '2' "$#" 'USER DOMAIN'
validate_format 'user' 'domain'
is_system_enabled "$DNS_CLUSTER" 'DNS_CLUSTER'
is_object_valid 'user' 'USER' "$user"
is_object_valid 'dns' 'DOMAIN' "$domain"

if [ ! -e "$VESTA/conf/dns-cluster.conf" ]; then
    echo "Error: dns-cluster.conf doesn't exist"
    log_event "$E_NOTEXIST $EVENT"
    exit $E_NOTEXIST
fi

number_of_proc=$(ps auxf | grep -v grep | grep $VESTA/bin/$SCRIPT | wc -l)
if [ "$number_of_proc" -gt 2 ]; then
    echo "Error: another sync process already exists"
    log_event "$E_EXISTS $EVENT"
    exit $E_EXISTS
fi


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

old_ifs="$IFS"
IFS=$'\n'

# Starting cluster loop
for cluster_str in $(cat $VESTA/conf/dns-cluster.conf); do

    # Get host values
    eval $cluster_str

    # Check connection type
    if [ -z "TYPE" ]; then
        TYPE='api'
    fi

    # Switch on connection type
    case $TYPE in
        ssh) send_cmd="send_ssh_cmd" ;;
        *)  send_cmd="send_api_cmd" ;;
    esac

    # Check host connection
    $send_cmd v-list-sys-config
    if [ $? -ne 0 ]; then
        echo "Error: $TYPE connection to $HOST failed"
        log_event "$E_CONNECT $EVENT"
        exit $E_CONNECT
    fi

    # Check recipient dns user
    if [ -z "$DNS_USER" ]; then
        DNS_USER='dns-cluster'
    fi
    $send_cmd v-list-user $DNS_USER
    if [ $? -ne 0 ]; then
        echo "Error: dns user $DNS_USER doesn't exist"
        log_event "$E_NOTEXIST $EVENT"
        exit $E_NOTEXIST
    fi

    # Check dns exceptions
    if [ -z "$DNS_CLUSTER_IGNORE" ]; then
        DNS_CLUSTER_IGNORE='dns-cluster'
    fi

    # Sync domain
    str=$(grep "DOMAIN='$domain'" $USER_DATA/dns.conf)
    eval $str
    $send_cmd v-insert-dns-domain $DNS_USER "$str" $HOSTNAME 'flush' 'no'

    # Sync records
    if [ "$TYPE" = 'ssh' ]; then
        tmp=$(mktemp -u)
        scp_cmd $USER_DATA/dns/$DOMAIN.conf $tmp
        $send_cmd v-insert-dns-records $DNS_USER $DOMAIN $tmp 'no'
    else
        for str in $(cat $USER_DATA/dns/$DOMAIN.conf); do
            str=$(echo "$str" | sed 's/"/\\"/g')
            $send_cmd v-insert-dns-record $DNS_USER $DOMAIN "$str"
        done
    fi

    # Rebuild dns zone
    $send_cmd v-rebuild-dns-domain $DNS_USER $domain 'scheduled'
    if [ $? -ne 0 ]; then
        echo "Error: $TYPE connection to $HOST failed (rebuild)"
        log_event "$E_CONNECT $EVENT"
        exit $E_CONNECT
    fi

done

# Update pipe
pipe="$VESTA/data/queue/dns-cluster.pipe"
str=$(grep -n "$SCRIPT $1 $2$" $pipe | cut -f1 -d: | head -n1)
if [ ! -z "$str" ]; then
    sed -i "$str d"  $pipe
fi


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

exit