From a5ff23de7478f619dd8d124367bb942af1be0408 Mon Sep 17 00:00:00 2001
From: Serghey Rodin <serghey.rodin@gmail.com>
Date: Wed, 3 Jun 2015 18:39:54 +0300
Subject: [PATCH] Restore traffic pipe

---
 bin/v-restore-user | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/bin/v-restore-user b/bin/v-restore-user
index e24713e2..efdb21e2 100755
--- a/bin/v-restore-user
+++ b/bin/v-restore-user
@@ -89,7 +89,7 @@ fi
 disk_usage=$(df $BACKUP |tail -n1 |tr ' ' '\n' |grep % |cut -f 1 -d %)
 if [ "$disk_usage" -ge "$BACKUP_DISK_LIMIT" ]; then
     rm -rf $tmpdir
-    echo "Not enough disk space to run backup" | $send_mail -s "$subj" $email
+    echo "Not enough disk space to run restore" | $send_mail -s "$subj" $email
     echo "Error: Not enough disk space"
     sed -i "/ $user /d" $VESTA/data/queue/backup.pipe
     log_event "$E_DISK" "$EVENT"
@@ -399,15 +399,23 @@ if [ "$web" != 'no' ]; then
 
     done
 
+    # Add user to traff queue
+    sed -i "/ $user /d" $VESTA/data/queue/traffic.pipe
+    echo "$BIN/v-update-web-domains-traff $user" >>\
+        $VESTA/data/queue/traffic.pipe
+
     # Restart WEB
     $BIN/v-restart-web
     if [ $? -ne 0 ]; then
         exit $E_RESTART
     fi
 
-    $BIN/v-restart-proxy
-    if [ $? -ne 0 ]; then
-        exit $E_RESTART
+    # Restart Proxy
+    if [ ! -z "$PROXY_SYSTEM" ]; then
+        $BIN/v-restart-proxy
+        if [ $? -ne 0 ]; then
+            exit $E_RESTART
+        fi
     fi
 
     echo